package user import ( "strconv" "git.huangwc.com/pig/pig-farm-controller/internal/app/controller" "git.huangwc.com/pig/pig-farm-controller/internal/app/dto" "git.huangwc.com/pig/pig-farm-controller/internal/app/service" domain_notify "git.huangwc.com/pig/pig-farm-controller/internal/domain/notify" "git.huangwc.com/pig/pig-farm-controller/internal/domain/token" "git.huangwc.com/pig/pig-farm-controller/internal/infra/logs" "git.huangwc.com/pig/pig-farm-controller/internal/infra/models" "git.huangwc.com/pig/pig-farm-controller/internal/infra/repository" "github.com/labstack/echo/v4" "gorm.io/gorm" ) // Controller 用户控制器 type Controller struct { userRepo repository.UserRepository monitorService service.MonitorService tokenService token.Service notifyService domain_notify.Service logger *logs.Logger } // NewController 创建用户控制器实例 func NewController( userRepo repository.UserRepository, monitorService service.MonitorService, logger *logs.Logger, tokenService token.Service, notifyService domain_notify.Service, ) *Controller { return &Controller{ userRepo: userRepo, monitorService: monitorService, tokenService: tokenService, notifyService: notifyService, logger: logger, } } // --- Controller Methods --- // CreateUser godoc // @Summary 创建新用户 // @Description 根据用户名和密码创建一个新的系统用户。 // @Tags 用户管理 // @Accept json // @Produce json // @Param user body dto.CreateUserRequest true "用户信息" // @Success 200 {object} controller.Response{data=dto.CreateUserResponse} "业务码为201代表创建成功" // @Router /api/v1/users [post] func (c *Controller) CreateUser(ctx echo.Context) error { var req dto.CreateUserRequest if err := ctx.Bind(&req); err != nil { c.logger.Errorf("创建用户: 参数绑定失败: %v", err) return controller.SendErrorResponse(ctx, controller.CodeBadRequest, err.Error()) } user := &models.User{ Username: req.Username, Password: req.Password, // 密码会在 BeforeSave 钩子中哈希 } if err := c.userRepo.Create(user); err != nil { c.logger.Errorf("创建用户: 创建用户失败: %v", err) // 尝试查询用户,以判断是否是用户名重复导致的错误 _, findErr := c.userRepo.FindByUsername(req.Username) if findErr == nil { // 如果能找到用户,说明是用户名重复 return controller.SendErrorResponse(ctx, controller.CodeConflict, "用户名已存在") } // 其他创建失败的情况 return controller.SendErrorResponse(ctx, controller.CodeInternalError, "创建用户失败") } return controller.SendResponse(ctx, controller.CodeCreated, "用户创建成功", dto.CreateUserResponse{ Username: user.Username, ID: user.ID, }) } // Login godoc // @Summary 用户登录 // @Description 用户可以使用用户名、邮箱、手机号、微信号或飞书账号进行登录,成功后返回 JWT 令牌。 // @Tags 用户管理 // @Accept json // @Produce json // @Param credentials body dto.LoginRequest true "登录凭证" // @Success 200 {object} controller.Response{data=dto.LoginResponse} "业务码为200代表登录成功" // @Router /api/v1/users/login [post] func (c *Controller) Login(ctx echo.Context) error { var req dto.LoginRequest if err := ctx.Bind(&req); err != nil { c.logger.Errorf("登录: 参数绑定失败: %v", err) return controller.SendErrorResponse(ctx, controller.CodeBadRequest, err.Error()) } // 使用新的方法,通过唯一标识符(用户名、邮箱等)查找用户 user, err := c.userRepo.FindUserForLogin(req.Identifier) if err != nil { if err == gorm.ErrRecordNotFound { return controller.SendErrorResponse(ctx, controller.CodeUnauthorized, "登录凭证不正确") } c.logger.Errorf("登录: 查询用户失败: %v", err) return controller.SendErrorResponse(ctx, controller.CodeInternalError, "登录失败") } if !user.CheckPassword(req.Password) { return controller.SendErrorResponse(ctx, controller.CodeUnauthorized, "登录凭证不正确") } // 登录成功,生成 JWT token tokenString, err := c.tokenService.GenerateToken(user.ID) if err != nil { c.logger.Errorf("登录: 生成令牌失败: %v", err) return controller.SendErrorResponse(ctx, controller.CodeInternalError, "登录失败,无法生成认证信息") } return controller.SendResponse(ctx, controller.CodeSuccess, "登录成功", dto.LoginResponse{ Username: user.Username, ID: user.ID, Token: tokenString, }) } // SendTestNotification godoc // @Summary 发送测试通知 // @Description 为指定用户发送一条特定渠道的测试消息,以验证其配置是否正确。 // @Tags 用户管理 // @Security BearerAuth // @Accept json // @Produce json // @Param id path int true "用户ID" // @Param body body dto.SendTestNotificationRequest true "请求体" // @Success 200 {object} controller.Response{data=string} "成功响应" // @Router /api/v1/users/{id}/notifications/test [post] func (c *Controller) SendTestNotification(ctx echo.Context) error { const actionType = "发送测试通知" // 1. 从 URL 中获取用户 ID userID, err := strconv.ParseUint(ctx.Param("id"), 10, 32) if err != nil { c.logger.Errorf("%s: 无效的用户ID格式: %v", actionType, err) return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的用户ID格式", actionType, "无效的用户ID格式", ctx.Param("id")) } // 2. 从请求体 (JSON Body) 中获取要测试的通知类型 var req dto.SendTestNotificationRequest if err := ctx.Bind(&req); err != nil { c.logger.Errorf("%s: 参数绑定失败: %v", actionType, err) return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "请求体格式错误或缺少 'type' 字段: "+err.Error(), actionType, "请求体绑定失败", req) } // 3. 调用领域服务 err = c.notifyService.SendTestMessage(uint(userID), req.Type) if err != nil { c.logger.Errorf("%s: 服务层调用失败: %v", actionType, err) return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "发送测试消息失败: "+err.Error(), actionType, "服务层调用失败", map[string]interface{}{"userID": userID, "type": req.Type}) } // 4. 返回成功响应 c.logger.Infof("%s: 成功为用户 %d 发送类型为 %s 的测试消息", actionType, userID, req.Type) return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "测试消息已发送,请检查您的接收端。", nil, actionType, "测试消息发送成功", map[string]interface{}{"userID": userID, "type": req.Type}) }