101 lines
2.6 KiB
Go
101 lines
2.6 KiB
Go
|
package controller
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"ai_xhs/common"
|
|||
|
|
"ai_xhs/service"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type AuthController struct {
|
|||
|
|
authService *service.AuthService
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewAuthController() *AuthController {
|
|||
|
|
return &AuthController{
|
|||
|
|
authService: service.NewAuthService(),
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// WechatLogin 微信小程序登录
|
|||
|
|
func (ctrl *AuthController) WechatLogin(c *gin.Context) {
|
|||
|
|
var req struct {
|
|||
|
|
Code string `json:"code" binding:"required"`
|
|||
|
|
Phone string `json:"phone"` // 可选，员工手机号（直接传明文）
|
|||
|
|
PhoneCode string `json:"phone_code"` // 可选，微信手机号加密code
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|||
|
|
common.Error(c, common.CodeInvalidParams, "参数错误: "+err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 调用登录服务
|
|||
|
|
token, employee, err := ctrl.authService.WechatLogin(req.Code, req.Phone, req.PhoneCode)
|
|||
|
|
if err != nil {
|
|||
|
|
common.Error(c, common.CodeServerError, err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取用户显示名称（优先使用真实姓名，其次用户名）
|
|||
|
|
displayName := employee.RealName
|
|||
|
|
if displayName == "" {
|
|||
|
|
displayName = employee.Username
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
common.SuccessWithMessage(c, "登录成功", gin.H{
|
|||
|
|
"token": token,
|
|||
|
|
"employee": gin.H{
|
|||
|
|
"id": employee.ID,
|
|||
|
|
"name": displayName,
|
|||
|
|
"username": employee.Username,
|
|||
|
|
"real_name": employee.RealName,
|
|||
|
|
"phone": employee.Phone,
|
|||
|
|
"role": employee.Role,
|
|||
|
|
"enterprise_id": employee.EnterpriseID,
|
|||
|
|
"enterprise_name": employee.EnterpriseName,
|
|||
|
|
"is_bound_xhs": employee.IsBoundXHS,
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// PhoneLogin 手机号登录（用于测试）
|
|||
|
|
func (ctrl *AuthController) PhoneLogin(c *gin.Context) {
|
|||
|
|
var req struct {
|
|||
|
|
Phone string `json:"phone" binding:"required"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|||
|
|
common.Error(c, common.CodeInvalidParams, "参数错误: "+err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 调用手机号登录服务
|
|||
|
|
token, employee, err := ctrl.authService.PhoneLogin(req.Phone)
|
|||
|
|
if err != nil {
|
|||
|
|
common.Error(c, common.CodeServerError, err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取用户显示名称（优先使用真实姓名，其次用户名）
|
|||
|
|
displayName := employee.RealName
|
|||
|
|
if displayName == "" {
|
|||
|
|
displayName = employee.Username
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
common.SuccessWithMessage(c, "登录成功", gin.H{
|
|||
|
|
"token": token,
|
|||
|
|
"employee": gin.H{
|
|||
|
|
"id": employee.ID,
|
|||
|
|
"name": displayName,
|
|||
|
|
"username": employee.Username,
|
|||
|
|
"real_name": employee.RealName,
|
|||
|
|
"phone": employee.Phone,
|
|||
|
|
"role": employee.Role,
|
|||
|
|
"enterprise_id": employee.EnterpriseID,
|
|||
|
|
"enterprise_name": employee.EnterpriseName,
|
|||
|
|
"is_bound_xhs": employee.IsBoundXHS,
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
}
|