update

server/api/v1/course/course_users.go

@@ -332,37 +332,76 @@ func (courseUsersApi *CourseUsersApi) MetaMaskLogin(c *gin.Context) {
 		}
 	}
 
-	_,err = courseUsersService.GetCourseUsersByAddr(loginUser.Invitor)
-	if err != nil && err == gorm.ErrRecordNotFound{
-		response.FailWithMessage("邀请者不存在", c)
-		return
-	} else if err != nil && err != gorm.ErrRecordNotFound {
-		response.FailWithMessage(err.Error(), c)
-		return
-	}
-
-
-	_,err = courseUsersService.GetCourseUsersByAddr(loginUser.Addr)
+	user,err := courseUsersService.GetCourseUsersByAddr(loginUser.Addr)
 	if err != nil && err != gorm.ErrRecordNotFound {
 		response.FailWithMessage(err.Error(), c)
 		return
-	}
+	} else if err != nil && err == gorm.ErrRecordNotFound {
+		if loginUser.Invitor != "" {
+			err = address.CheckAddress(loginUser.Invitor,-1)
+			if err != nil {
+				response.FailWithMessage(err.Error(), c)
+				return
+			}
+		} else {
+			response.FailWithMessage("邀请者不存在", c)
+			return
+		}
+		_,err = courseUsersService.GetCourseUsersByAddr(loginUser.Invitor)
+		if err != nil && err == gorm.ErrRecordNotFound{
+			response.FailWithMessage("邀请者不存在", c)
+			return
+		} else if err != nil && err != gorm.ErrRecordNotFound {
+			response.FailWithMessage(err.Error(), c)
+			return
+		}
 
-	if err == nil {
+		var courseUser course.CourseUsers
+		courseUser.Invitor = loginUser.Invitor
+		courseUser.Address = loginUser.Addr
+		err = courseUsersService.CreateCourseUsers(courseUser)
+		if err != nil {
+			global.GVA_LOG.Error("生成用户信息失败!", zap.Error(err))
+			response.FailWithMessage("创建用户信息失败", c)
+			return
+		}
 		response.OkWithData(token, c)
-		return
 	}
 
-	var courseUser course.CourseUsers
-	courseUser.Invitor = loginUser.Invitor
-	courseUser.Address = loginUser.Addr
-	err = courseUsersService.CreateCourseUsers(courseUser)
-	if err != nil {
-		global.GVA_LOG.Error("生成用户信息失败!", zap.Error(err))
-		response.FailWithMessage("创建用户信息失败", c)
+	// 用户已存在 有邀请人  直接返回token
+	if err == nil && user.Invitor != "" {
+		response.OkWithData(token, c)
 		return
+	}  else if err == nil && user.Invitor == "" {  // 不存在邀请人 则更新邀请关系
+		if loginUser.Invitor != "" {
+			err = address.CheckAddress(loginUser.Invitor,-1)
+			if err != nil {
+				response.FailWithMessage(err.Error(), c)
+				return
+			}
+		} else {
+			response.FailWithMessage("邀请者不存在", c)
+			return
+		}
+		iuser,err := courseUsersService.GetCourseUsersByAddr(loginUser.Invitor)
+		if err != nil && err == gorm.ErrRecordNotFound{
+			response.FailWithMessage("邀请者不存在", c)
+			return
+		} else if err != nil && err != gorm.ErrRecordNotFound {
+			response.FailWithMessage(err.Error(), c)
+			return
+		}
+		iuser.Invitor = loginUser.Invitor
+		err = courseUsersService.UpdateCourseUsers(iuser)
+		if err != nil {
+			global.GVA_LOG.Error("更新用户信息失败!", zap.Error(err))
+			response.FailWithMessage("更新用户信息失败", c)
+			return
+		}
+		response.OkWithData(token, c)
 	}
-	response.OkWithData(token, c)
+
+
 }
 
 // GetDeviceCode 获取设备码

web/src/view/courseSignup/courseSignup.vue

@@ -313,5 +313,14 @@ const enterDialog = async () => {
 }
 </script>
 
-<style>
+<style lang="scss">
+.header-img-box {
+  width: 200px;
+  height: 200px;
+  border: 1px dashed #ccc;
+  border-radius: 20px;
+  text-align: center;
+  line-height: 200px;
+  cursor: pointer;
+}
 </style>