package handler

import (
	"net/http"

	"github.com/gin-gonic/gin"
	"kefu-sys/server/internal/middleware"
	"kefu-sys/server/internal/model"
)

type KnowledgeHandler struct{}

func NewKnowledgeHandler() *KnowledgeHandler { return &KnowledgeHandler{} }

func (h *KnowledgeHandler) ListCategories(c *gin.Context) {
	tenantID := middleware.GetTenantID(c)

	var categories []model.Category
	model.DB.Where("tenant_id = ?", tenantID).Find(&categories)

	middleware.JSON(c, categories)
}

func (h *KnowledgeHandler) CreateCategory(c *gin.Context) {
	var category model.Category
	if err := c.ShouldBindJSON(&category); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "参数错误"})
		return
	}
	category.TenantID = middleware.GetTenantID(c)

	if err := model.DB.Create(&category).Error; err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"code": 500, "message": "创建失败"})
		return
	}

	middleware.JSON(c, category)
}

func (h *KnowledgeHandler) ListEntries(c *gin.Context) {
	tenantID := middleware.GetTenantID(c)
	page, pageSize := middleware.GetPageParams(c)
	categoryID := c.Query("category_id")
	search := c.Query("search")
	status := c.Query("status")

	var entries []model.KnowledgeEntry
	var total int64

	query := model.DB.Where("tenant_id = ?", tenantID)
	if categoryID != "" {
		query = query.Where("category_id = ?", categoryID)
	}
	if search != "" {
		query = query.Where("title LIKE ? OR content LIKE ?", "%"+search+"%", "%"+search+"%")
	}
	if status != "" {
		query = query.Where("status = ?", status)
	}

	query.Model(&model.KnowledgeEntry{}).Count(&total)
	query.Order("updated_at desc").Offset((page - 1) * pageSize).Limit(pageSize).Find(&entries)

	middleware.JSONList(c, entries, total, page, pageSize)
}

func (h *KnowledgeHandler) CreateEntry(c *gin.Context) {
	var entry model.KnowledgeEntry
	if err := c.ShouldBindJSON(&entry); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "参数错误"})
		return
	}
	entry.TenantID = middleware.GetTenantID(c)

	if err := model.DB.Create(&entry).Error; err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"code": 500, "message": "创建失败"})
		return
	}

	middleware.JSON(c, entry)
}

func (h *KnowledgeHandler) UpdateEntry(c *gin.Context) {
	tenantID := middleware.GetTenantID(c)
	id := c.Param("id")

	var entry model.KnowledgeEntry
	if err := model.DB.Where("id = ? AND tenant_id = ?", id, tenantID).First(&entry).Error; err != nil {
		c.JSON(http.StatusNotFound, gin.H{"code": 404, "message": "条目不存在"})
		return
	}

	var updates map[string]interface{}
	if err := c.ShouldBindJSON(&updates); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "参数错误"})
		return
	}

	delete(updates, "tenant_id")
	delete(updates, "id")

	model.DB.Model(&entry).Updates(updates)
	middleware.JSON(c, entry)
}

func (h *KnowledgeHandler) DeleteEntry(c *gin.Context) {
	tenantID := middleware.GetTenantID(c)
	id := c.Param("id")

	result := model.DB.Where("id = ? AND tenant_id = ?", id, tenantID).Delete(&model.KnowledgeEntry{})
	if result.RowsAffected == 0 {
		c.JSON(http.StatusNotFound, gin.H{"code": 404, "message": "条目不存在"})
		return
	}

	middleware.JSON(c, gin.H{"message": "已删除"})
}