小妖用户绑定微信

internal/controller/wx/wx_v1_we_chat_event.go

@@ -29,8 +29,7 @@ func (c *ControllerV1) WeChatEvent(ctx context.Context, req *v1.WeChatEventReq)
 		switch req.Event {
 		case "subscribe":
 			key := strings.TrimPrefix(req.EventKey, "qrscene_")
-			split := strings.Split(req.EventKey, "_")
-			out, err := service.User().Login(ctx, &model.UserLoginIn{OpenId: unionid, StoreCode: split[0]})
+			out, err := service.User().Login(ctx, &model.UserLoginIn{OpenId: unionid, SceneId: key})
 			if err != nil {
 				return nil, err
 			}
@@ -39,8 +38,7 @@ func (c *ControllerV1) WeChatEvent(ctx context.Context, req *v1.WeChatEventReq)
 			}
 			return nil, nil
 		case "SCAN":
-			split := strings.Split(req.EventKey, "_")
-			out, err := service.User().Login(ctx, &model.UserLoginIn{OpenId: unionid, StoreCode: split[0]})
+			out, err := service.User().Login(ctx, &model.UserLoginIn{OpenId: unionid, SceneId: req.EventKey})
 			if err != nil {
 				return nil, err
 			}

internal/logic/user/user.go

@@ -6,6 +6,7 @@ import (
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/os/glog"
 	"github.com/gogf/gf/v2/os/gtime"
+	"github.com/gogf/gf/v2/util/gconv"
 	"github.com/gogf/gf/v2/util/grand"
 	"server/internal/consts"
 	"server/internal/dao"
@@ -19,6 +20,7 @@ import (
 	"server/utility/gamelife"
 	"server/utility/jwt"
 	"server/utility/sms"
+	"strings"
 )
 
 type sUser struct{}
@@ -53,74 +55,114 @@ func New() service.IUser {
 }
 
 func (s *sUser) Login(ctx context.Context, in *model.UserLoginIn) (out *model.UserLoginOut, err error) {
+	// Fetch role information
 	value, err := dao.Roles.Ctx(ctx).Where(do.Roles{Code: consts.UserRoleCode}).Fields(dao.Roles.Columns().Code, dao.Roles.Columns().Id).One()
 	if err != nil {
 		return nil, ecode.Fail.Sub("查找角色失败")
 	}
-	// 根据 OpenId 查找用户
-	exist, err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId}).Exist()
-	if err != nil {
-		return nil, ecode.Fail.Sub("查找用户失败")
+
+	split := strings.Split(in.SceneId, "_")
+	if len(split) < 3 {
+		return nil, ecode.Fail.Sub("SceneId格式错误")
 	}
+	storeId := gconv.Int64(split[0])
+	var xyUserId string
 	var userId int64
-	v, err := dao.Stores.Ctx(ctx).Where(do.Stores{StoreCode: in.StoreCode}).Fields(dao.Stores.Columns().Id).Value()
-	if err != nil {
-		return nil, ecode.Fail.Sub("查找门店失败")
-	}
-	if !exist {
-		// 用户不存在，创建新用户
-		// 生成 username: qy_ + 8位随机字母数字
-		var username string
-		for {
-			randomStr := grand.Str("abcdefghijklmnopqrstuvwxyz0123456789", 8)
-			username = "qy_" + randomStr
-			// 检查 username 是否唯一
-			count, err := dao.Users.Ctx(ctx).Where(do.Users{Username: username}).Count()
+
+	if split[1] == "0" {
+		// Case 1: split[1] == "0", use in.OpenId to find or create user
+		exist, err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId}).Exist()
+		if err != nil {
+			return nil, ecode.Fail.Sub("查找用户失败")
+		}
+
+		if !exist {
+			// User does not exist, create new user
+			var username string
+			for {
+				randomStr := grand.Str("abcdefghijklmnopqrstuvwxyz0123456789", 8)
+				username = "qy_" + randomStr
+				count, err := dao.Users.Ctx(ctx).Where(do.Users{Username: username}).Count()
+				if err != nil {
+					return nil, ecode.Fail.Sub("检查用户名失败")
+				}
+				if count == 0 {
+					break // Username is unique
+				}
+			}
+			password, err := encrypt.EncryptPassword(consts.DefaultPassword)
 			if err != nil {
-				return nil, ecode.Fail.Sub("检查用户名失败")
+				return nil, ecode.Fail.Sub("加密密码失败")
 			}
-			if count == 0 {
-				break // username 唯一，退出循环
+			user := &entity.Users{
+				WxOpenId:         in.OpenId,
+				Username:         username,
+				Nickname:         username,
+				PasswordHash:     password,
+				Avatar:           consts.DefaultUserAvatar,
+				FirstVisitAt:     gtime.Now(),
+				LastLoginAt:      gtime.Now(),
+				WxPopenId:        utility.GenerateUserID("WX"),
+				QqPopenId:        utility.GenerateUserID("QQ"),
+				RoleId:           value[dao.Roles.Columns().Id].Int64(),
+				LastLoginStoreId: storeId,
+			}
+			result, err := dao.Users.Ctx(ctx).Insert(user)
+			if err != nil {
+				return nil, ecode.Fail.Sub("创建用户失败")
+			}
+			userId, err = result.LastInsertId()
+			if err != nil {
+				return nil, ecode.Fail.Sub("获取用户ID失败")
+			}
+		} else {
+			// User exists, update last login store ID and time
+			var user entity.Users
+			if err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId}).Scan(&user); err != nil {
+				return nil, ecode.Fail.Sub("查找用户失败")
+			}
+			userId = user.Id
+			if _, err := dao.Users.Ctx(ctx).Where(do.Users{Id: userId}).Update(do.Users{
+				LastLoginAt:      gtime.Now(),
+				LastLoginStoreId: storeId,
+			}); err != nil {
+				return nil, ecode.Fail.Sub("更新登录时间失败")
 			}
 		}
-		password, err := encrypt.EncryptPassword(consts.DefaultPassword)
-		if err != nil {
-			return nil, ecode.Fail.Sub("加密密码失败")
-		}
-		user := &entity.Users{
-			WxOpenId:         in.OpenId,
-			Username:         username,
-			Nickname:         username,
-			PasswordHash:     password,
-			Avatar:           consts.DefaultUserAvatar,
-			FirstVisitAt:     gtime.Now(),
-			LastLoginAt:      gtime.Now(),
-			WxPopenId:        utility.GenerateUserID("WX"),
-			QqPopenId:        utility.GenerateUserID("QQ"),
-			RoleId:           value[dao.Roles.Columns().Id].Int64(),
-			LastLoginStoreId: v.Int64(),
-		}
-		result, err := dao.Users.Ctx(ctx).Insert(user)
-		if err != nil {
-			return nil, ecode.Fail.Sub("创建用户失败")
-		}
-		userId, err = result.LastInsertId()
-		if err != nil {
-			return nil, ecode.Fail.Sub("获取用户ID失败")
-		}
-	} else {
-		// 用户存在，更新最后登录时间
+	} else if split[1] == "1" {
+		// Case 2: split[1] == "1", use xyUserId to find user and bind in.OpenId
+		xyUserId = split[2]
 		var user entity.Users
-		if err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId}).Scan(&user); err != nil {
+		exist, err := dao.Users.Ctx(ctx).Where(do.Users{XyUserId: xyUserId}).Exist()
+		if err != nil {
+			return nil, ecode.Fail.Sub("查找用户失败")
+		}
+		if !exist {
+			return nil, ecode.Fail.Sub("参数错误：用户不存在")
+		}
+
+		// User exists, check WxOpenId
+		if err := dao.Users.Ctx(ctx).Where(do.Users{XyUserId: xyUserId}).Scan(&user); err != nil {
 			return nil, ecode.Fail.Sub("查找用户失败")
 		}
 		userId = user.Id
-		if _, err := dao.Users.Ctx(ctx).Where(do.Users{Id: userId}).Update(do.Users{LastLoginAt: gtime.Now(), LastLoginStoreId: v.Int64()}); err != nil {
-			return nil, ecode.Fail.Sub("更新登录时间失败")
+		if user.WxOpenId != "" && user.WxOpenId != in.OpenId {
+			return nil, ecode.Fail.Sub("当前微信已绑定其他微信号")
 		}
+
+		// Bind in.OpenId and update last login store ID
+		if _, err := dao.Users.Ctx(ctx).Where(do.Users{Id: userId}).Update(do.Users{
+			WxOpenId:         in.OpenId,
+			LastLoginAt:      gtime.Now(),
+			LastLoginStoreId: storeId,
+		}); err != nil {
+			return nil, ecode.Fail.Sub("绑定OpenId失败")
+		}
+	} else {
+		return nil, ecode.Fail.Sub("无效的SceneId类型")
 	}
 
-	// 生成 token
+	// Generate token
 	token, err := jwt.GenerateToken(&jwt.TokenIn{
 		UserId: userId,
 		Role:   value[dao.Roles.Columns().Code].String(),
@@ -132,7 +174,7 @@ func (s *sUser) Login(ctx context.Context, in *model.UserLoginIn) (out *model.Us
 	out = &model.UserLoginOut{
 		Token: token,
 	}
-	return
+	return out, nil
 }
 func (s *sUser) WeChatLogin(ctx context.Context, in *model.WeChatLogin) (out *model.WeChatLoginOut, err error) {
 	return

internal/model/user.go

@@ -59,8 +59,8 @@ type LoginCache struct {
 }
 
 type UserLoginIn struct {
-	OpenId    string
-	StoreCode string // 门店编码，登录时根据门店编码查询，填写最近一次登录门店 id
+	OpenId  string
+	SceneId string
 }
 type WeChatLogin struct {
 	UUID string // 使用的是微信的唯一 id，而不是openid，保证微信服务号，小程序绑定的系统用户为同一个