package controllers

import (
	"strconv"

	"lokistack/models"
	"lokistack/services"
	"lokistack/utils"

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

// AIController AIOps：模型配置 + 日志分析 + 审计历史
type AIController struct{}

func NewAIController() *AIController { return &AIController{} }

// GetConfig 返回模型配置（APIKey 掩码）+ 今日已用 token。
func (ctrl *AIController) GetConfig(c *gin.Context) {
	userID := c.GetUint("userId")
	cfg := services.MaskedAIConfig()
	utils.Success(c, gin.H{
		"config":         cfg,
		"tokenUsedToday": services.AITokenUsedToday(userID),
	})
}

// SaveConfig 保存模型配置。apiKey 传空或掩码 = 沿用旧值。
func (ctrl *AIController) SaveConfig(c *gin.Context) {
	var req struct {
		models.AIConfig
		APIKey string `json:"apiKey"`
	}
	if !utils.BindAndValidate(c, &req) {
		return
	}
	if err := services.SaveAIConfig(req.AIConfig, req.APIKey); err != nil {
		utils.Error(c, 400, err.Error())
		return
	}
	utils.Success(c, gin.H{"config": services.MaskedAIConfig()})
}

// TestConfig 连通性测试：用已保存（或本次提交）的配置发一句最小请求。
func (ctrl *AIController) TestConfig(c *gin.Context) {
	var req struct {
		APIKey  string `json:"apiKey"`
		BaseURL string `json:"baseUrl"`
		Model   string `json:"model"`
	}
	_ = c.ShouldBindJSON(&req)

	cfg := services.GetAIConfig()
	if req.BaseURL != "" {
		cfg.BaseURL = req.BaseURL
	}
	if req.Model != "" {
		cfg.Model = req.Model
	}
	// key 解析:传新值才覆盖;掩码/空一律沿用已保存的真实 key(GetAIConfig 返回的是解密后的真值)
	if req.APIKey != "" && !containsMask(req.APIKey) {
		cfg.APIKey = req.APIKey
	}

	result, err := services.TestAIConnection(cfg)
	if err != nil {
		utils.Error(c, 400, err.Error())
		return
	}
	utils.Success(c, result)
}

func containsMask(s string) bool {
	for i := 0; i+2 < len(s); i++ {
		if s[i] == '*' && s[i+1] == '*' && s[i+2] == '*' {
			return true
		}
	}
	return false
}

// ListRuns 审计历史（分页）。
func (ctrl *AIController) ListRuns(c *gin.Context) {
	page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
	pageSize, _ := strconv.Atoi(c.DefaultQuery("pageSize", "20"))
	runs, total := services.ListAIRuns(page, pageSize)
	utils.Success(c, gin.H{"list": runs, "total": total, "page": page, "pageSize": pageSize})
}

// GetRun 单条审计详情（含脱敏后的完整 payload）。
func (ctrl *AIController) GetRun(c *gin.Context) {
	id, err := utils.IntParam(c, "id")
	if err != nil {
		utils.BadRequest(c, "ID 无效")
		return
	}
	run := services.GetAIRun(uint(id))
	if run == nil {
		utils.Error(c, 404, "记录不存在")
		return
	}
	utils.Success(c, run)
}

// Chat 全站 AI 助手浮窗(多轮对话 + 页面上下文注入;Pro 闸门)
func (ctrl *AIController) Chat(c *gin.Context) {
	var req struct {
		History []services.ChatMessage `json:"history" binding:"required,min=1,max=40"`
		Page    string                 `json:"page"`
		Context string                 `json:"context"`
	}
	if !utils.BindAndValidate(c, &req) {
		return
	}
	cfg := services.GetAIConfig()
	reply, err := services.RunAIChat(c.Request.Context(), cfg, c.GetUint("userId"), req.History, req.Page, req.Context)
	if err != nil {
		utils.Error(c, 400, err.Error())
		return
	}
	utils.Success(c, gin.H{"reply": reply})
}