package handler

import (
	"net/http"
	"strconv"
	"time"

	"github.com/gin-gonic/gin"
	"gpt-load/internal/db"
	"gpt-load/internal/models"
	"gpt-load/internal/response"
)

// GetLogs godoc
// @Summary Get request logs
// @Description Get request logs with pagination and filtering
// @Tags Logs
// @Accept  json
// @Produce  json
// @Param page query int false "Page number"
// @Param size query int false "Page size"
// @Param group_id query int false "Group ID"
// @Param start_time query string false "Start time (RFC3339)"
// @Param end_time query string false "End time (RFC3339)"
// @Param status_code query int false "Status code"
// @Success 200 {array} models.RequestLog
// @Router /api/logs [get]
func GetLogs(c *gin.Context) {
	page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
	size, _ := strconv.Atoi(c.DefaultQuery("size", "10"))
	offset := (page - 1) * size

	query := db.DB.Model(&models.RequestLog{})

	if groupIDStr := c.Query("group_id"); groupIDStr != "" {
		groupID, err := strconv.Atoi(groupIDStr)
		if err == nil {
			query = query.Where("group_id = ?", groupID)
		}
	}

	if startTimeStr := c.Query("start_time"); startTimeStr != "" {
		startTime, err := time.Parse(time.RFC3339, startTimeStr)
		if err == nil {
			query = query.Where("timestamp >= ?", startTime)
		}
	}

	if endTimeStr := c.Query("end_time"); endTimeStr != "" {
		endTime, err := time.Parse(time.RFC3339, endTimeStr)
		if err == nil {
			query = query.Where("timestamp <= ?", endTime)
		}
	}

	if statusCodeStr := c.Query("status_code"); statusCodeStr != "" {
		statusCode, err := strconv.Atoi(statusCodeStr)
		if err == nil {
			query = query.Where("status_code = ?", statusCode)
		}
	}

	var logs []models.RequestLog
	var total int64

	query.Count(&total)
	err := query.Order("timestamp desc").Offset(offset).Limit(size).Find(&logs).Error
	if err != nil {
		response.Error(c, http.StatusInternalServerError, "Failed to get logs")
		return
	}

	c.JSON(http.StatusOK, gin.H{
		"total": total,
		"page":  page,
		"size":  size,
		"data":  logs,
	})
}