package controllers

import (
	"lokideploy/models"
	"lokideploy/services"
	"lokideploy/utils"

	"github.com/gin-gonic/gin"
)

type AuthController struct {
	service *services.AuthService
}

func NewAuthController() *AuthController {
	return &AuthController{service: services.NewAuthService()}
}

type LoginRequest struct {
	Username string `json:"username" binding:"required"`
	Password string `json:"password" binding:"required"`
}

type RegisterRequest struct {
	Username string `json:"username" binding:"required"`
	Password string `json:"password" binding:"required,min=6"`
	Nickname string `json:"nickname"`
	Role     string `json:"role"`
}

type ChangePasswordRequest struct {
	OldPassword string `json:"oldPassword" binding:"required"`
	NewPassword string `json:"newPassword" binding:"required,min=6"`
}

func (ctrl *AuthController) Login(c *gin.Context) {
	var req LoginRequest
	if !utils.BindAndValidate(c, &req) {
		return
	}
	user, token, err := ctrl.service.Login(req.Username, req.Password)
	if err != nil {
		utils.Unauthorized(c, "用户名或密码错误")
		return
	}
	utils.Success(c, gin.H{"token": token, "user": models.SanitizedJSON(user)})
}

func (ctrl *AuthController) Register(c *gin.Context) {
	var req RegisterRequest
	if !utils.BindAndValidate(c, &req) {
		return
	}
	if req.Role == "" {
		req.Role = "user"
	}
	user, err := ctrl.service.Register(req.Username, req.Password, req.Nickname, req.Role)
	if err != nil {
		utils.Error(c, 400, err.Error())
		return
	}
	utils.Success(c, models.SanitizedJSON(user))
}

func (ctrl *AuthController) GetUser(c *gin.Context) {
	claims, ok := utils.GetUserFromContext(c)
	if !ok {
		utils.Unauthorized(c, "")
		return
	}
	user, err := ctrl.service.GetUserByID(claims.UserID)
	if err != nil {
		utils.Error(c, 404, "用户不存在")
		return
	}
	utils.Success(c, models.SanitizedJSON(user))
}

func (ctrl *AuthController) ChangePassword(c *gin.Context) {
	var req ChangePasswordRequest
	if !utils.BindAndValidate(c, &req) {
		return
	}
	claims, ok := utils.GetUserFromContext(c)
	if !ok {
		utils.Unauthorized(c, "")
		return
	}
	if err := ctrl.service.ChangePassword(claims.UserID, req.OldPassword, req.NewPassword); err != nil {
		utils.Error(c, 400, err.Error())
		return
	}
	utils.Success(c, gin.H{"message": "password changed"})
}