package controllers

import (
	"opsxloki/dto"
	"opsxloki/models"
	"opsxloki/pkg/jwt"
	"opsxloki/services"
	"opsxloki/utils"

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

type AuthController struct {
	service *services.AuthService
}

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

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

func (ctrl *AuthController) RefreshToken(c *gin.Context) {
	var req dto.RefreshTokenRequest
	if !utils.BindAndValidate(c, &req) {
		return
	}
	claims, err := jwt.ParseRefreshToken(req.RefreshToken)
	if err != nil {
		utils.Unauthorized(c, "invalid or expired refresh token")
		return
	}
	// 对照数据库校验用户存在性与会话版本：改密/重置密码后旧 refresh token 立即失效
	user, err := ctrl.service.GetUserByID(claims.UserID)
	if err != nil {
		utils.Unauthorized(c, "user no longer exists")
		return
	}
	if claims.TokenVersion != user.TokenVersion {
		utils.Unauthorized(c, "session revoked, please login again")
		return
	}
	// 以用户当前 role 与版本签发，角色变更在刷新时即时生效
	tokenPair, err := jwt.GenerateTokenPair(user.ID, user.Username, user.Role, user.TokenVersion)
	if err != nil {
		utils.Error(c, 500, err.Error())
		return
	}
	utils.Success(c, gin.H{
		"accessToken":  tokenPair.AccessToken,
		"refreshToken": tokenPair.RefreshToken,
		"expiresAt":    tokenPair.ExpiresAt.Unix(),
	})
}

func (ctrl *AuthController) GetUser(c *gin.Context) {
	claims, ok := jwt.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 dto.ChangePasswordRequest
	if !utils.BindAndValidate(c, &req) {
		return
	}
	claims, ok := jwt.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"})
}