// Package user 提供用户相关功能的控制器
// 实现用户注册、登录等操作
package user

import (
	"net/http"

	"git.huangwc.com/pig/pig-farm-controller/internal/api/middleware"
	"git.huangwc.com/pig/pig-farm-controller/internal/logs"
	"git.huangwc.com/pig/pig-farm-controller/internal/storage/repository"
	"github.com/gin-gonic/gin"
	"golang.org/x/crypto/bcrypt"
)

// Controller 用户控制器
type Controller struct {
	userRepo repository.UserRepo
	logger   *logs.Logger
}

// NewController 创建用户控制器实例
func NewController(userRepo repository.UserRepo) *Controller {
	return &Controller{
		userRepo: userRepo,
		logger:   logs.NewLogger(),
	}
}

// RegisterRequest 注册请求结构体
type RegisterRequest struct {
	Username string `json:"username" binding:"required"`
	Password string `json:"password" binding:"required"`
}

// RegisterResponse 注册响应结构体
type RegisterResponse struct {
	ID        uint   `json:"id"`
	Username  string `json:"username"`
	CreatedAt string `json:"created_at"`
}

// Register 用户注册
func (c *Controller) Register(ctx *gin.Context) {
	var req RegisterRequest
	if err := ctx.ShouldBindJSON(&req); err != nil {
		ctx.JSON(http.StatusBadRequest, gin.H{"error": "请求参数错误"})
		return
	}

	user, err := c.userRepo.CreateUser(req.Username, req.Password)
	if err != nil {
		c.logger.Error("创建用户失败: " + err.Error())
		ctx.JSON(http.StatusInternalServerError, gin.H{"error": "创建用户失败"})
		return
	}

	response := RegisterResponse{
		ID:        user.ID,
		Username:  user.Username,
		CreatedAt: user.CreatedAt.Format("2006-01-02 15:04:05"),
	}

	ctx.JSON(http.StatusOK, gin.H{
		"message": "用户创建成功",
		"user":    response,
	})
}

// LoginRequest 登录请求结构体
type LoginRequest struct {
	Username string `json:"username" binding:"required"`
	Password string `json:"password" binding:"required"`
}

// LoginResponse 登录响应结构体
type LoginResponse struct {
	ID        uint   `json:"id"`
	Username  string `json:"username"`
	Token     string `json:"token"`
	CreatedAt string `json:"created_at"`
}

// Login 用户登录
func (c *Controller) Login(ctx *gin.Context) {
	var req LoginRequest
	if err := ctx.ShouldBindJSON(&req); err != nil {
		ctx.JSON(http.StatusBadRequest, gin.H{"error": "请求参数错误"})
		return
	}

	// 查找用户
	user, err := c.userRepo.FindByUsername(req.Username)
	if err != nil {
		c.logger.Error("查找用户失败: " + err.Error())
		ctx.JSON(http.StatusUnauthorized, gin.H{"error": "用户名或密码错误"})
		return
	}

	// 验证密码
	err = bcrypt.CompareHashAndPassword([]byte(user.PasswordHash), []byte(req.Password))
	if err != nil {
		ctx.JSON(http.StatusUnauthorized, gin.H{"error": "用户名或密码错误"})
		return
	}

	// 生成JWT访问令牌
	authMiddleware := middleware.NewAuthMiddleware(c.userRepo)
	token, err := authMiddleware.GenerateToken(user.ID, user.Username)
	if err != nil {
		c.logger.Error("生成令牌失败: " + err.Error())
		ctx.JSON(http.StatusInternalServerError, gin.H{"error": "登录失败"})
		return
	}

	response := LoginResponse{
		ID:        user.ID,
		Username:  user.Username,
		Token:     token,
		CreatedAt: user.CreatedAt.Format("2006-01-02 15:04:05"),
	}

	ctx.JSON(http.StatusOK, gin.H{
		"message": "登录成功",
		"user":    response,
	})
}