package handler

import (
	"dianshang/pkg/response"
	"net/http"
	"time"

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

// HealthHandler 健康检查处理器
type HealthHandler struct {
	db *gorm.DB
}

// NewHealthHandler 创建健康检查处理器
func NewHealthHandler(db *gorm.DB) *HealthHandler {
	return &HealthHandler{
		db: db,
	}
}

// HealthResponse 健康检查响应结构
type HealthResponse struct {
	Status    string    `json:"status"`
	Timestamp time.Time `json:"timestamp"`
	Version   string    `json:"version"`
	Database  string    `json:"database"`
	Uptime    string    `json:"uptime"`
}

var startTime = time.Now()

// Health 健康检查接口
// @Summary 健康检查
// @Description 检查服务是否正常运行，包括数据库连接状态
// @Tags 系统
// @Accept json
// @Produce json
// @Success 200 {object} response.Response{data=HealthResponse} "服务正常"
// @Failure 503 {object} response.Response "服务异常"
// @Router /health [get]
func (h *HealthHandler) Health(c *gin.Context) {
	// 检查数据库连接
	dbStatus := "connected"
	if h.db != nil {
		sqlDB, err := h.db.DB()
		if err != nil || sqlDB.Ping() != nil {
			dbStatus = "disconnected"
		}
	} else {
		dbStatus = "not_configured"
	}

	// 计算运行时间
	uptime := time.Since(startTime).String()

	healthData := HealthResponse{
		Status:    "healthy",
		Timestamp: time.Now(),
		Version:   "1.0.0",
		Database:  dbStatus,
		Uptime:    uptime,
	}

	// 如果数据库连接异常，返回503状态码
	if dbStatus != "connected" {
		healthData.Status = "unhealthy"
		c.JSON(http.StatusServiceUnavailable, response.Response{
			Code:    http.StatusServiceUnavailable,
			Message: "服务异常",
			Data:    healthData,
		})
		return
	}

	response.SuccessWithMessage(c, "服务正常", healthData)
}

// Ping 简单的ping接口
// @Summary Ping检查
// @Description 简单的服务可用性检查
// @Tags 系统
// @Accept json
// @Produce json
// @Success 200 {object} response.Response "pong"
// @Router /ping [get]
func (h *HealthHandler) Ping(c *gin.Context) {
	response.SuccessWithMessage(c, "pong", gin.H{
		"timestamp": time.Now(),
		"message":   "服务运行正常",
	})
}