310 lines
9.1 KiB
Go
310 lines
9.1 KiB
Go
package service
|
|
|
|
import (
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/flipped-aurora/gin-vue-admin/server/global"
|
|
"github.com/flipped-aurora/gin-vue-admin/server/plugin/wechat-integration/model"
|
|
"github.com/silenceper/wechat/v2"
|
|
"github.com/silenceper/wechat/v2/cache"
|
|
"github.com/silenceper/wechat/v2/miniprogram"
|
|
miniConfig "github.com/silenceper/wechat/v2/miniprogram/config"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type WechatMiniService struct{}
|
|
|
|
// GetWechatMiniProgram 获取微信小程序实例
|
|
func (w *WechatMiniService) GetWechatMiniProgram() (*miniprogram.MiniProgram, error) {
|
|
if global.GVA_CONFIG.Wechat.MiniAppID == "" || global.GVA_CONFIG.Wechat.MiniAppSecret == "" {
|
|
return nil, errors.New("微信小程序配置不完整")
|
|
}
|
|
|
|
wc := wechat.NewWechat()
|
|
memory := cache.NewMemory()
|
|
cfg := &miniConfig.Config{
|
|
AppID: global.GVA_CONFIG.Wechat.MiniAppID,
|
|
AppSecret: global.GVA_CONFIG.Wechat.MiniAppSecret,
|
|
Cache: memory,
|
|
}
|
|
|
|
return wc.GetMiniProgram(cfg), nil
|
|
}
|
|
|
|
// Code2Session 小程序登录凭证校验
|
|
func (w *WechatMiniService) Code2Session(code string) (*model.WechatMiniUser, error) {
|
|
mini, err := w.GetWechatMiniProgram()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 调用微信接口获取 session
|
|
session, err := mini.GetAuth().Code2Session(code)
|
|
if err != nil {
|
|
global.GVA_LOG.Error("微信小程序 Code2Session 失败: " + err.Error())
|
|
return nil, err
|
|
}
|
|
|
|
if session.OpenID == "" {
|
|
return nil, errors.New("获取用户 OpenID 失败")
|
|
}
|
|
|
|
// 查找或创建用户
|
|
var user model.WechatMiniUser
|
|
err = global.GVA_DB.Where("openid = ?", session.OpenID).First(&user).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
// 创建新用户
|
|
user = model.WechatMiniUser{
|
|
OpenID: session.OpenID,
|
|
SessionKey: &session.SessionKey,
|
|
}
|
|
if session.UnionID != "" {
|
|
user.UnionID = &session.UnionID
|
|
}
|
|
|
|
now := time.Now()
|
|
user.LastLoginTime = &now
|
|
|
|
// 尝试关联系统用户
|
|
err = w.linkSystemUser(&user)
|
|
if err != nil {
|
|
global.GVA_LOG.Error("关联系统用户失败: " + err.Error())
|
|
}
|
|
|
|
err = global.GVA_DB.Create(&user).Error
|
|
if err != nil {
|
|
global.GVA_LOG.Error("创建微信小程序用户失败: " + err.Error())
|
|
return nil, err
|
|
}
|
|
} else {
|
|
global.GVA_LOG.Error("查询微信小程序用户失败: " + err.Error())
|
|
return nil, err
|
|
}
|
|
} else {
|
|
// 更新用户信息
|
|
now := time.Now()
|
|
user.SessionKey = &session.SessionKey
|
|
user.LastLoginTime = &now
|
|
if session.UnionID != "" {
|
|
user.UnionID = &session.UnionID
|
|
}
|
|
|
|
// 如果还没有关联系统用户，尝试关联
|
|
if user.UserID == nil {
|
|
err = w.linkSystemUser(&user)
|
|
if err != nil {
|
|
global.GVA_LOG.Error("关联系统用户失败: " + err.Error())
|
|
}
|
|
}
|
|
|
|
err = global.GVA_DB.Save(&user).Error
|
|
if err != nil {
|
|
global.GVA_LOG.Error("更新微信小程序用户失败: " + err.Error())
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
// GetUserInfo 获取用户信息
|
|
func (w *WechatMiniService) GetUserInfo(openid string) (*model.WechatMiniUser, error) {
|
|
var user model.WechatMiniUser
|
|
err := global.GVA_DB.Where("openid = ?", openid).First(&user).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, errors.New("用户不存在")
|
|
}
|
|
return nil, err
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
// UpdateUserInfo 更新用户信息
|
|
func (w *WechatMiniService) UpdateUserInfo(openid string, userInfo map[string]interface{}) error {
|
|
updates := make(map[string]interface{})
|
|
|
|
if nickname, ok := userInfo["nickname"]; ok {
|
|
if str, ok := nickname.(string); ok && str != "" {
|
|
updates["nickname"] = str
|
|
}
|
|
}
|
|
if avatarUrl, ok := userInfo["avatarUrl"]; ok {
|
|
if str, ok := avatarUrl.(string); ok && str != "" {
|
|
updates["avatar_url"] = str
|
|
}
|
|
}
|
|
if gender, ok := userInfo["gender"]; ok {
|
|
if num, ok := gender.(float64); ok {
|
|
updates["gender"] = int(num)
|
|
}
|
|
}
|
|
if country, ok := userInfo["country"]; ok {
|
|
if str, ok := country.(string); ok && str != "" {
|
|
updates["country"] = str
|
|
}
|
|
}
|
|
if province, ok := userInfo["province"]; ok {
|
|
if str, ok := province.(string); ok && str != "" {
|
|
updates["province"] = str
|
|
}
|
|
}
|
|
if city, ok := userInfo["city"]; ok {
|
|
if str, ok := city.(string); ok && str != "" {
|
|
updates["city"] = str
|
|
}
|
|
}
|
|
if language, ok := userInfo["language"]; ok {
|
|
if str, ok := language.(string); ok && str != "" {
|
|
updates["language"] = str
|
|
}
|
|
}
|
|
|
|
if len(updates) == 0 {
|
|
return errors.New("没有需要更新的信息")
|
|
}
|
|
|
|
err := global.GVA_DB.Model(&model.WechatMiniUser{}).Where("openid = ?", openid).Updates(updates).Error
|
|
if err != nil {
|
|
global.GVA_LOG.Error("更新微信小程序用户信息失败: " + err.Error())
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetUserList 获取用户列表
|
|
func (w *WechatMiniService) GetUserList(page, pageSize int) ([]model.WechatMiniUser, int64, error) {
|
|
var users []model.WechatMiniUser
|
|
var total int64
|
|
|
|
db := global.GVA_DB.Model(&model.WechatMiniUser{})
|
|
|
|
// 获取总数
|
|
err := db.Count(&total).Error
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
// 分页查询
|
|
offset := (page - 1) * pageSize
|
|
err = db.Offset(offset).Limit(pageSize).Order("created_at desc").Find(&users).Error
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
return users, total, nil
|
|
}
|
|
|
|
// linkSystemUser 关联系统用户 (通过 unionid 关联，为未来APP扩展预留)
|
|
func (w *WechatMiniService) linkSystemUser(user *model.WechatMiniUser) error {
|
|
if user.UnionID == nil || *user.UnionID == "" {
|
|
// 没有 unionid，无法跨平台关联，但小程序仍可正常使用
|
|
global.GVA_LOG.Warn("小程序用户没有 UnionID，无法跨平台关联")
|
|
return nil
|
|
}
|
|
|
|
// 查找是否有其他小程序用户（或未来的APP用户）已经关联了系统用户
|
|
var existingMiniUser model.WechatMiniUser
|
|
err := global.GVA_DB.Where("unionid = ? AND user_id IS NOT NULL AND openid != ?", *user.UnionID, user.OpenID).First(&existingMiniUser).Error
|
|
if err == nil {
|
|
// 找到了已关联系统用户的用户，使用相同的 user_id
|
|
user.UserID = existingMiniUser.UserID
|
|
global.GVA_LOG.Info("通过 UnionID 关联到已存在的系统用户: " + *user.UnionID)
|
|
return nil
|
|
}
|
|
|
|
// 如果没有找到，暂时不自动创建系统用户
|
|
// 等用户输入手机号后，通过业务逻辑关联或创建系统用户
|
|
global.GVA_LOG.Info("新用户，等待手机号验证后关联系统用户，unionid: " + *user.UnionID)
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetUserByUnionID 根据 UnionID 获取用户信息
|
|
func (w *WechatMiniService) GetUserByUnionID(unionid string) (*model.WechatMiniUser, error) {
|
|
var user model.WechatMiniUser
|
|
err := global.GVA_DB.Where("unionid = ?", unionid).First(&user).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, errors.New("用户不存在")
|
|
}
|
|
return nil, err
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
// LinkToSystemUser 手动关联系统用户
|
|
func (w *WechatMiniService) LinkToSystemUser(openid string, userID uint) error {
|
|
err := global.GVA_DB.Model(&model.WechatMiniUser{}).Where("openid = ?", openid).Update("user_id", userID).Error
|
|
if err != nil {
|
|
global.GVA_LOG.Error("关联系统用户失败: " + err.Error())
|
|
return err
|
|
}
|
|
|
|
global.GVA_LOG.Info("成功关联系统用户")
|
|
return nil
|
|
}
|
|
|
|
// BindPhoneAndLinkUser 绑定手机号并关联系统用户
|
|
func (w *WechatMiniService) BindPhoneAndLinkUser(openid, phone string) (*model.WechatMiniUser, error) {
|
|
// 1. 更新小程序用户的手机号
|
|
err := global.GVA_DB.Model(&model.WechatMiniUser{}).Where("openid = ?", openid).Update("phone", phone).Error
|
|
if err != nil {
|
|
global.GVA_LOG.Error("更新手机号失败: " + err.Error())
|
|
return nil, err
|
|
}
|
|
|
|
// 2. 获取更新后的用户信息
|
|
var user model.WechatMiniUser
|
|
err = global.GVA_DB.Where("openid = ?", openid).First(&user).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 3. 根据手机号查找或创建系统用户
|
|
// 这里需要根据实际的系统用户表结构来实现
|
|
// 暂时先记录日志，实际项目中需要调用系统用户服务
|
|
global.GVA_LOG.Info("用户绑定手机号: " + phone + ", openid: " + openid)
|
|
|
|
// 4. 如果有 UnionID，同时更新所有相关的微信账号
|
|
if user.UnionID != nil && *user.UnionID != "" {
|
|
// 未来如果有APP用户，也会通过相同的 UnionID 关联到同一个系统用户
|
|
global.GVA_LOG.Info("用户有 UnionID，支持跨平台关联: " + *user.UnionID)
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
// GetUserByPhone 根据手机号获取用户
|
|
func (w *WechatMiniService) GetUserByPhone(phone string) (*model.WechatMiniUser, error) {
|
|
var user model.WechatMiniUser
|
|
err := global.GVA_DB.Where("phone = ?", phone).First(&user).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, errors.New("用户不存在")
|
|
}
|
|
return nil, err
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
// CheckUnionIDExists 检查 UnionID 是否已存在（为APP登录预留）
|
|
func (w *WechatMiniService) CheckUnionIDExists(unionid string) (*model.WechatMiniUser, error) {
|
|
if unionid == "" {
|
|
return nil, errors.New("unionid 不能为空")
|
|
}
|
|
|
|
var user model.WechatMiniUser
|
|
err := global.GVA_DB.Where("unionid = ?", unionid).First(&user).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, nil // 不存在，返回 nil 但不报错
|
|
}
|
|
return nil, err
|
|
}
|
|
return &user, nil
|
|
}
|