first commit
This commit is contained in:
sjk
100
go_backend/controller/auth_controller.go
Normal file
100
go_backend/controller/auth_controller.go
Normal file
@@ -0,0 +1,100 @@
|
||||
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,
|
||||
},
|
||||
})
|
||||
}
|
||||
257
go_backend/controller/employee_controller.go
Normal file
257
go_backend/controller/employee_controller.go
Normal file
@@ -0,0 +1,257 @@
|
||||
package controller
|
||||
|
||||
import (
|
||||
"ai_xhs/common"
|
||||
"ai_xhs/service"
|
||||
"strconv"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
type EmployeeController struct {
|
||||
service *service.EmployeeService
|
||||
}
|
||||
|
||||
func NewEmployeeController() *EmployeeController {
|
||||
return &EmployeeController{
|
||||
service: &service.EmployeeService{},
|
||||
}
|
||||
}
|
||||
|
||||
// SendXHSCode 发送小红书验证码
|
||||
func (ctrl *EmployeeController) SendXHSCode(c *gin.Context) {
|
||||
var req struct {
|
||||
XHSPhone string `json:"xhs_phone" binding:"required"`
|
||||
}
|
||||
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
common.Error(c, common.CodeInvalidParams, "参数错误：手机号不能为空")
|
||||
return
|
||||
}
|
||||
|
||||
err := ctrl.service.SendXHSCode(req.XHSPhone)
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeInternalError, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
common.SuccessWithMessage(c, "验证码已发送，请在小红书APP中查看", nil)
|
||||
}
|
||||
|
||||
// GetProfile 获取员工个人信息
|
||||
func (ctrl *EmployeeController) GetProfile(c *gin.Context) {
|
||||
employeeID := c.GetInt("employee_id")
|
||||
|
||||
employee, err := ctrl.service.GetProfile(employeeID)
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeNotFound, "员工不存在")
|
||||
return
|
||||
}
|
||||
|
||||
// 获取用户显示名称（优先使用真实姓名，其次用户名）
|
||||
displayName := employee.RealName
|
||||
if displayName == "" {
|
||||
displayName = employee.Username
|
||||
}
|
||||
|
||||
data := map[string]interface{}{
|
||||
"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.Enterprise.Name,
|
||||
"is_bound_xhs": employee.IsBoundXHS,
|
||||
"xhs_account": employee.XHSAccount,
|
||||
"xhs_phone": employee.XHSPhone,
|
||||
"has_xhs_cookie": employee.XHSCookie != "", // 标识是否有Cookie，不返回完整Cookie
|
||||
}
|
||||
|
||||
if employee.BoundAt != nil {
|
||||
data["bound_at"] = employee.BoundAt.Format("2006-01-02 15:04:05")
|
||||
}
|
||||
|
||||
common.Success(c, data)
|
||||
}
|
||||
|
||||
// BindXHS 绑定小红书账号
|
||||
func (ctrl *EmployeeController) BindXHS(c *gin.Context) {
|
||||
employeeID := c.GetInt("employee_id")
|
||||
|
||||
var req struct {
|
||||
XHSPhone string `json:"xhs_phone" binding:"required"`
|
||||
Code string `json:"code" binding:"required"`
|
||||
}
|
||||
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
common.Error(c, common.CodeInvalidParams, "参数错误")
|
||||
return
|
||||
}
|
||||
|
||||
xhsAccount, err := ctrl.service.BindXHS(employeeID, req.XHSPhone, req.Code)
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeBindXHSFailed, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
common.SuccessWithMessage(c, "绑定成功", map[string]interface{}{
|
||||
"xhs_account": xhsAccount,
|
||||
})
|
||||
}
|
||||
|
||||
// UnbindXHS 解绑小红书账号
|
||||
func (ctrl *EmployeeController) UnbindXHS(c *gin.Context) {
|
||||
employeeID := c.GetInt("employee_id")
|
||||
|
||||
if err := ctrl.service.UnbindXHS(employeeID); err != nil {
|
||||
common.Error(c, common.CodeInternalError, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
common.SuccessWithMessage(c, "解绑成功", nil)
|
||||
}
|
||||
|
||||
// GetAvailableCopies 获取可领取文案列表
|
||||
func (ctrl *EmployeeController) GetAvailableCopies(c *gin.Context) {
|
||||
employeeID := c.GetInt("employee_id")
|
||||
productID, err := strconv.Atoi(c.Query("product_id"))
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeInvalidParams, "产品ID参数错误")
|
||||
return
|
||||
}
|
||||
|
||||
data, err := ctrl.service.GetAvailableCopies(employeeID, productID)
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeInternalError, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
common.Success(c, data)
|
||||
}
|
||||
|
||||
// ClaimCopy 领取文案
|
||||
func (ctrl *EmployeeController) ClaimCopy(c *gin.Context) {
|
||||
employeeID := c.GetInt("employee_id")
|
||||
|
||||
var req struct {
|
||||
CopyID int `json:"copy_id" binding:"required"`
|
||||
ProductID int `json:"product_id" binding:"required"`
|
||||
}
|
||||
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
common.Error(c, common.CodeInvalidParams, "参数错误")
|
||||
return
|
||||
}
|
||||
|
||||
data, err := ctrl.service.ClaimCopy(employeeID, req.CopyID, req.ProductID)
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeAlreadyClaimed, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
common.SuccessWithMessage(c, "领取成功", data)
|
||||
}
|
||||
|
||||
// ClaimRandomCopy 随机领取文案
|
||||
func (ctrl *EmployeeController) ClaimRandomCopy(c *gin.Context) {
|
||||
employeeID := c.GetInt("employee_id")
|
||||
|
||||
var req struct {
|
||||
ProductID int `json:"product_id" binding:"required"`
|
||||
}
|
||||
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
common.Error(c, common.CodeInvalidParams, "参数错误")
|
||||
return
|
||||
}
|
||||
|
||||
data, err := ctrl.service.ClaimRandomCopy(employeeID, req.ProductID)
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeCopyNotAvailable, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
common.SuccessWithMessage(c, "领取成功", data)
|
||||
}
|
||||
|
||||
// Publish 发布内容
|
||||
func (ctrl *EmployeeController) Publish(c *gin.Context) {
|
||||
employeeID := c.GetInt("employee_id")
|
||||
|
||||
var req service.PublishRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
common.Error(c, common.CodeInvalidParams, "参数错误")
|
||||
return
|
||||
}
|
||||
|
||||
recordID, err := ctrl.service.Publish(employeeID, req)
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeInternalError, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
common.SuccessWithMessage(c, "发布成功", map[string]interface{}{
|
||||
"record_id": recordID,
|
||||
})
|
||||
}
|
||||
|
||||
// GetMyPublishRecords 获取我的发布记录
|
||||
func (ctrl *EmployeeController) GetMyPublishRecords(c *gin.Context) {
|
||||
employeeID := c.GetInt("employee_id")
|
||||
|
||||
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
||||
pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "10"))
|
||||
|
||||
data, err := ctrl.service.GetMyPublishRecords(employeeID, page, pageSize)
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeInternalError, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
common.Success(c, data)
|
||||
}
|
||||
|
||||
// GetPublishRecordDetail 获取发布记录详情
|
||||
func (ctrl *EmployeeController) GetPublishRecordDetail(c *gin.Context) {
|
||||
employeeID := c.GetInt("employee_id")
|
||||
recordID, err := strconv.Atoi(c.Param("id"))
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeInvalidParams, "记录ID参数错误")
|
||||
return
|
||||
}
|
||||
|
||||
data, err := ctrl.service.GetPublishRecordDetail(employeeID, recordID)
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeNotFound, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
common.Success(c, data)
|
||||
}
|
||||
|
||||
// CheckXHSStatus 检查小红书绑定与Cookie状态
|
||||
func (ctrl *EmployeeController) CheckXHSStatus(c *gin.Context) {
|
||||
employeeID := c.GetInt("employee_id")
|
||||
|
||||
status, err := ctrl.service.CheckXHSStatus(employeeID)
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeInternalError, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
common.Success(c, status)
|
||||
}
|
||||
|
||||
// GetProducts 获取产品列表
|
||||
func (ctrl *EmployeeController) GetProducts(c *gin.Context) {
|
||||
data, err := ctrl.service.GetProducts()
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeInternalError, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
common.Success(c, map[string]interface{}{
|
||||
"list": data,
|
||||
})
|
||||
}
|
||||
75
go_backend/controller/xhs_controller.go
Normal file
75
go_backend/controller/xhs_controller.go
Normal file
@@ -0,0 +1,75 @@
|
||||
package controller
|
||||
|
||||
import (
|
||||
"ai_xhs/common"
|
||||
"ai_xhs/service"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
type XHSController struct {
|
||||
service *service.XHSService
|
||||
}
|
||||
|
||||
func NewXHSController() *XHSController {
|
||||
return &XHSController{
|
||||
service: &service.XHSService{},
|
||||
}
|
||||
}
|
||||
|
||||
// SendCode 发送小红书验证码
|
||||
func (ctrl *XHSController) SendCode(c *gin.Context) {
|
||||
var req struct {
|
||||
Phone string `json:"phone" binding:"required"`
|
||||
CountryCode string `json:"country_code"`
|
||||
}
|
||||
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
common.Error(c, common.CodeInvalidParams, "参数错误：手机号不能为空")
|
||||
return
|
||||
}
|
||||
|
||||
// 调用Service层发送验证码
|
||||
result, err := ctrl.service.SendVerificationCode(req.Phone, req.CountryCode)
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeInternalError, "调用Python服务失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
// 判断Python服务返回的结果
|
||||
if result.Code != 0 {
|
||||
common.Error(c, result.Code, result.Message)
|
||||
return
|
||||
}
|
||||
|
||||
common.SuccessWithMessage(c, result.Message, result.Data)
|
||||
}
|
||||
|
||||
// VerifyCode 验证小红书验证码并登录
|
||||
func (ctrl *XHSController) VerifyCode(c *gin.Context) {
|
||||
var req struct {
|
||||
Phone string `json:"phone" binding:"required"`
|
||||
Code string `json:"code" binding:"required"`
|
||||
CountryCode string `json:"country_code"`
|
||||
}
|
||||
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
common.Error(c, common.CodeInvalidParams, "参数错误：手机号和验证码不能为空")
|
||||
return
|
||||
}
|
||||
|
||||
// 调用Service层验证登录
|
||||
result, err := ctrl.service.VerifyLogin(req.Phone, req.Code, req.CountryCode)
|
||||
if err != nil {
|
||||
common.Error(c, common.CodeInternalError, "调用Python服务失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
// 判断Python服务返回的结果
|
||||
if result.Code != 0 {
|
||||
common.Error(c, result.Code, result.Message)
|
||||
return
|
||||
}
|
||||
|
||||
common.SuccessWithMessage(c, result.Message, result.Data)
|
||||
}
|
||||
Reference in New Issue
Block a user