pig-farm-controller/internal/app/controller/user/user_controller.go

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})
}