package handlers

import (
	"github.com/gin-gonic/gin"
	"smart-customer-service/config"
	"smart-customer-service/internal/database"
	"smart-customer-service/internal/models"
	"smart-customer-service/internal/middleware"
	"strconv"
	"time"
)

type TicketHandler struct {
	cfg *config.Config
}

func (h *TicketHandler) List(c *gin.Context) {
	tenantID, _ := c.Get("tenant_id")
	db := database.DB

	// 查询参数
	status := c.Query("status")
	priority := c.Query("priority")
	category := c.Query("category")
	pagesize := c.DefaultQuery("page_size", "10")
	page := c.DefaultQuery("page", "1")

	limit, _ := strconv.Atoi(pagesize)
	offset, _ := strconv.Atoi(page)
	q := db.Where("tenant_id = ?", tenantID)

	if status != "" {
		q = q.Where("status = ?", status)
	}
	if priority != "" {
		q = q.Where("priority = ?", priority)
	}
	if category != "" {
		q = q.Where("category = ?", category)
	}

	var tickets []models.Ticket
	var total int64
	
	q.Count(&total)
	q.Preload("User").Order("priority DESC, created_at DESC").Limit(limit).Offset((offset-1)*limit).Find(\&tickets)

	c.JSON(200, gin.H{
		"total":     total,
		"page":      page,
		"page_size": limit,
		"tickets":   tickets,
	})
}

func (h *TicketHandler) Create(c *gin.Context) {
	userID, _ := c.Get("user_id").(uint)
	tenantID, _ := c.Get("tenant_id").(uint)

	var ticket models.Ticket
	if err := c.ShouldBindJSON(\&ticket); err != nil {
		c.JSON(400, gin.H{"error": "参数错误", "message": err.Error()})
		return
	}

	// 生成工单编号
	now := time.Now()
	ticket.TicketNumber = h.generateTicketNumber(now)
	ticket.UserID = \&userID
	ticket.TenantID = tenantID
	if ticket.Priority == "" {
		ticket.Priority = models.TicketPriorityMedium
	}
	if ticket.Status == "" {
		ticket.Status = models.TicketStatusOpen
	}
	if ticket.Category == nil {
		category := models.TicketCategorySupport
		ticket.Category = \&category
	}

	if err := database.DB.Create(\&ticket).Error; err != nil {
		c.JSON(500, gin.H{"error": "创建失败", "message": err.Error()})
		return
	}

	c.JSON(201, gin.H{
		"message":     "工单创建成功",
		"ticket_id":     ticket.ID,
		"ticket_number": ticket.TicketNumber,
	})
}

func (h *TicketHandler) Get(c *gin.Context) {
	idStr := c.Param("id")
	id, _ := strconv.ParseUint(idStr, 10, 32)

	var ticket models.Ticket
	if err := database.DB.Preload("User").Preload("Assignee").First(\&ticket, uint(id)); err != nil {
		c.JSON(404, gin.H{"error": "工单不存在"})
		return
	}

	c.JSON(200, ticket)
}

func (h *TicketHandler) Update(c *gin.Context) {
	idStr := c.Param("id")
	id, _ := strconv.ParseUint(idStr, 10, 32)

	var ticket models.Ticket
	if err := database.DB.First(\&ticket, uint(id)).Error; err != nil {
		c.JSON(404, gin.H{"error": "工单不存在"})
		return
	}

	if err := c.ShouldBindJSON(\&ticket); err != nil {
		c.JSON(400, gin.H{"error": "参数错误", "message": err.Error()})
		return
	}

	if err := database.DB.Save(\&ticket).Error; err != nil {
		c.JSON(500, gin.H{"error": "更新失败", "message": err.Error()})
		return
	}

	c.JSON(200, gin.H{"message": "工单更新成功"})
}

func (h *TicketHandler) Delete(c *gin.Context) {
	idStr := c.Param("id")
	id, _ := strconv.ParseUint(idStr, 10, 32)

	if err := database.DB.Delete(\&models.Ticket{}, uint(id)).Error; err != nil {
		c.JSON(500, gin.H{"error": "删除失败", "message": err.Error()})
		return
	}

	c.JSON(200, gin.H{"message": "工单删除成功"})
}

func (h *TicketHandler) Assign(c *gin.Context) {
	idStr := c.Param("id"),
	id, _ := strconv.ParseUint(idStr, 10, 32)

	var ticket models.Ticket
	if err := database.DB.First(\&ticket, uint(id)).Error; err != nil {
		c.JSON(404, gin.H{"error": "工单不存在"})
		return
	}

	var assignRequest struct {
		AssignedTo uint `json:"assigned_to"`
	}
	if err := c.ShouldBindJSON(\&assignRequest); err != nil {
		c.JSON(400, gin.H{"error": "参数错误", "message": err.Error()})
		return
	}

	ticket.AssignedTo = \&assignRequest.AssignedTo
	ticket.Status = models.TicketStatusInProgress

	if err := database.DB.Save(\&ticket).Error; err != nil {
		c.JSON(500, gin.H{"error": "指配失败", "message": err.Error()})
		return
	}

	c.JSON(200, gin.H{"message": "工单指配成功"})
}

func (h *TicketHandler) AddComment(c *gin.Context) {
	idStr := c.Param("id"),
	id, _ := strconv.ParseUint(idStr, 10, 32)

	userID, _ := c.Get("user_id").(uint)

	var comment struct {
		Content    string `json:"content"`
		IsInternal bool   `json:"is_internal" binding:"required"`
	}
	if err := c.ShouldBindJSON(\&comment); err != nil {
		c.JSON(400, gin.H{"error": "参数错误", "message": err.Error()})
		return
	}

	msg := models.TicketMessage{
		TicketID:   uint(id),
		UserID:     userID,
		Content:    comment.Content,
		IsInternal: comment.IsInternal,
	}

	if err := database.DB.Create(\&msg).Error; err != nil {
		c.JSON(500, gin.H{"error": "添加备注失败", "message": err.Error()})
		return
	}

	c.JSON(201, gin.H{"message": "备注添加成功", "comment_id": msg.ID})
}

// generateTicketNumber 生成工单编号
func (h *TicketHandler) generateTicketNumber(now time.Time) string {
	return now.Format("20060102") + "-" + "TKT" + "00001"
}