201 lines
6.5 KiB
Go
201 lines
6.5 KiB
Go
package controller
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
|
||
"github.com/gogf/gf/v2/crypto/gmd5"
|
||
"github.com/gogf/gf/v2/database/gdb"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/os/gctx"
|
||
"github.com/gogf/gf/v2/util/grand"
|
||
"github.com/medivhzhan/weapp/v3"
|
||
"github.com/medivhzhan/weapp/v3/auth"
|
||
"github.com/tiger1103/gfast/v3/api/v1/common/coryCommon"
|
||
"github.com/tiger1103/gfast/v3/api/wxApplet/wxApplet"
|
||
systemController "github.com/tiger1103/gfast/v3/internal/app/system/controller"
|
||
"github.com/tiger1103/gfast/v3/internal/app/system/dao"
|
||
"github.com/tiger1103/gfast/v3/internal/app/system/model/do"
|
||
"github.com/tiger1103/gfast/v3/internal/app/system/model/entity"
|
||
se "github.com/tiger1103/gfast/v3/internal/app/system/service"
|
||
wxDao "github.com/tiger1103/gfast/v3/internal/app/wxApplet/dao"
|
||
wxModel "github.com/tiger1103/gfast/v3/internal/app/wxApplet/model/do"
|
||
"github.com/tiger1103/gfast/v3/internal/app/wxApplet/service"
|
||
"github.com/tiger1103/gfast/v3/library/libUtils"
|
||
"github.com/tiger1103/gfast/v3/library/liberr"
|
||
)
|
||
|
||
type WXLoginController struct {
|
||
systemController.BaseController
|
||
}
|
||
|
||
var WXLogin = new(WXLoginController)
|
||
|
||
func (e *WXLoginController) BusLogin(ctx context.Context, req *wxApplet.LoginReq) (res *wxApplet.LoginRes, err error) {
|
||
res = &wxApplet.LoginRes{}
|
||
err = g.DB().Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||
err = g.Try(ctx, func(ctx context.Context) {
|
||
appId, err := g.Cfg().Get(gctx.New(), "wx.appId")
|
||
if err != nil {
|
||
return
|
||
}
|
||
appSecret, err := g.Cfg().Get(gctx.New(), "wx.appSecret")
|
||
if err != nil {
|
||
return
|
||
}
|
||
sdk := weapp.NewClient(appId.String(), appSecret.String())
|
||
cli := sdk.NewAuth()
|
||
//// 登录凭证校验
|
||
rsp, err := cli.Code2Session(&auth.Code2SessionRequest{
|
||
Appid: appId.String(),
|
||
Secret: appSecret.String(),
|
||
JsCode: req.Code,
|
||
GrantType: "authorization_code",
|
||
})
|
||
if err != nil {
|
||
return
|
||
}
|
||
if rsp.ErrCode != 0 {
|
||
err = rsp.GetResponseError()
|
||
return
|
||
}
|
||
user := wxApplet.BusConstructionUserAddReq{}
|
||
user.Openid = rsp.Openid
|
||
res.Openid = user.Openid
|
||
value, err := dao.BusConstructionUser.Ctx(ctx).Where("openid=?", user.Openid).Fields("head_icon").Value()
|
||
if err != nil {
|
||
return
|
||
}
|
||
result := 0
|
||
if value != nil {
|
||
result = 1
|
||
}
|
||
// 上传微信头像
|
||
str, err := coryCommon.UploadFileTwo(ctx, req.HeadIcon, coryCommon.Helmet)
|
||
if err != nil {
|
||
liberr.ErrIsNil(ctx, err)
|
||
return
|
||
}
|
||
replace := strings.Replace(str, "resource/public", "/wxfile", 1)
|
||
// 删除以前的头像
|
||
if strings.Contains(value.String(), "/wxfile/") {
|
||
s := strings.Replace(value.String(), "/wxfile/", coryCommon.GetCWD()+"/resource/public/", 1)
|
||
os.Remove(filepath.ToSlash(s))
|
||
}
|
||
if result == 0 {
|
||
// user.HeadIcon = req.HeadIcon
|
||
//user.NickName = req.NickName
|
||
//user.HeadIcon = replace
|
||
_, err := service.BusConstructionUser().Add(ctx, &user)
|
||
if err != nil {
|
||
return
|
||
}
|
||
} else {
|
||
_, err = dao.BusConstructionUser.Ctx(ctx).Data(g.Map{
|
||
//"head_icon": req.HeadIcon,
|
||
"nick_name": req.NickName,
|
||
"head_icon": replace,
|
||
}).Where("openid=?", user.Openid).Update()
|
||
if err != nil {
|
||
return
|
||
}
|
||
}
|
||
res.Token = gettoken(ctx, user.Openid, user)
|
||
return
|
||
})
|
||
return err
|
||
})
|
||
return
|
||
}
|
||
|
||
//func gettoken(ctx context.Context, openid string, user wxApplet.BusConstructionUserAddReq) string {
|
||
// key := gmd5.MustEncryptString(openid)
|
||
// token, err := se.GfToken().GenerateToken(ctx, key+key, user)
|
||
// fmt.Println("创建的token", token)
|
||
// fmt.Println(err)
|
||
// token = "Bearer " + token
|
||
// fmt.Println("更改的-----", token)
|
||
// return token
|
||
//}
|
||
|
||
func gettoken(ctx context.Context, openid string, user wxApplet.BusConstructionUserAddReq) string {
|
||
encryptedKey := gmd5.MustEncryptString(openid)
|
||
|
||
key := openid + "-" + encryptedKey
|
||
authToken, _ := se.GfToken().GenerateToken(ctx, key, user)
|
||
|
||
return "Bearer " + authToken
|
||
}
|
||
|
||
// IsBindApp 是否绑定 App
|
||
func (e *WXLoginController) IsBindApp(ctx context.Context, req *wxApplet.IsBindAppReq) (*wxApplet.IsBindAppRes, error) {
|
||
exist, err := dao.BusConstructionUser.Ctx(ctx).As("bcu").
|
||
InnerJoin("sys_user AS su", `bcu.user_name COLLATE utf8mb4_general_ci = su.user_name COLLATE utf8mb4_general_ci
|
||
AND bcu.phone COLLATE utf8mb4_general_ci = su.mobile COLLATE utf8mb4_general_ci`).
|
||
Where("bcu.openid", req.Openid).
|
||
Count()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 如果已绑定 App 则返回 1，否则返回 0
|
||
if exist > 0 {
|
||
return &wxApplet.IsBindAppRes{IsBind: 1}, nil
|
||
}
|
||
|
||
return &wxApplet.IsBindAppRes{IsBind: 0}, nil
|
||
}
|
||
|
||
// AppletBindApp 小程序绑定 App
|
||
func (e *WXLoginController) AppletBindApp(ctx context.Context, req *wxApplet.AppletBindAppReq) (*wxApplet.AppletBindAppRes, error) {
|
||
// 获取小程序用户信息
|
||
busConstructUser := entity.BusConstructionUser{}
|
||
if err := dao.BusConstructionUser.Ctx(ctx).Where(dao.BusConstructionUser.Columns().Openid, req.Openid).Scan(&busConstructUser); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 判断当前用户的手机号和用户名是否为空
|
||
if busConstructUser.Phone == "" || busConstructUser.UserName == "" {
|
||
return nil, fmt.Errorf("当前用户的手机号和用户名为空")
|
||
}
|
||
|
||
exist, err := dao.SysUser.Ctx(ctx).Where(dao.SysUser.Columns().UserNickname, busConstructUser.UserName).
|
||
Where(dao.SysUser.Columns().Mobile, busConstructUser.Phone).Count()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 判断是否已绑定 App
|
||
if exist > 0 {
|
||
return nil, fmt.Errorf("当前用户已经绑定了 App")
|
||
}
|
||
|
||
passwordSalt := grand.S(10) // 密码盐值
|
||
encryptedPassword := libUtils.EncryptPassword(req.Password, passwordSalt) // 加密密码
|
||
|
||
err = g.Try(ctx, func(ctx context.Context) {
|
||
// 创建用户
|
||
userId, err := dao.SysUser.Ctx(ctx).InsertAndGetId(do.SysUser{
|
||
UserName: busConstructUser.UserName, // 用户名
|
||
UserNickname: busConstructUser.NickName, // 用户昵称
|
||
Mobile: busConstructUser.Phone, // 手机号
|
||
UserPassword: encryptedPassword, // 密码
|
||
UserSalt: passwordSalt, // 密码盐
|
||
Sex: busConstructUser.Sex, // 性别
|
||
})
|
||
liberr.ErrIsNil(ctx, err, "创建用户失败")
|
||
|
||
// 关联后台项目
|
||
_, err = wxDao.SysUserProjectRelevancy.Ctx(ctx).Insert(wxModel.SysUserProjectRelevancy{
|
||
UserId: userId, // 用户 ID
|
||
ProjectId: busConstructUser.ProjectId, // 项目 ID
|
||
})
|
||
liberr.ErrIsNil(ctx, err, "关联后台管理项目失败")
|
||
})
|
||
|
||
return nil, nil
|
||
}
|