package handlers

import (
	"encoding/json"
	"fmt"
	"log"
	"net/http"
	"random-api-go/monitoring"
	"random-api-go/services"
	"random-api-go/stats"
	"random-api-go/utils"

	"strings"
	"time"
)

var statsManager *stats.StatsManager

// InitializeHandlers 初始化处理器
func InitializeHandlers(sm *stats.StatsManager) error {
	statsManager = sm
	return services.InitializeCSVService()
}

func HandleAPIRequest(w http.ResponseWriter, r *http.Request) {
	start := time.Now()
	realIP := utils.GetRealIP(r)

	path := strings.TrimPrefix(r.URL.Path, "/")
	pathSegments := strings.Split(path, "/")

	if len(pathSegments) < 2 {
		monitoring.LogRequest(monitoring.RequestLog{
			Time:       time.Now().Unix(),
			Path:       r.URL.Path,
			Method:     r.Method,
			StatusCode: http.StatusNotFound,
			Latency:    float64(time.Since(start).Microseconds()) / 1000,
			IP:         realIP,
			Referer:    r.Referer(),
		})
		http.NotFound(w, r)
		return
	}

	prefix := pathSegments[0]
	suffix := pathSegments[1]

	services.Mu.RLock()
	csvPath, ok := services.CSVPathsCache[prefix][suffix]
	services.Mu.RUnlock()

	if !ok {
		http.NotFound(w, r)
		return
	}

	selector, err := services.GetCSVContent(csvPath)
	if err != nil {
		http.Error(w, "Failed to fetch CSV content", http.StatusInternalServerError)
		log.Printf("Error fetching CSV content: %v", err)
		return
	}

	if len(selector.URLs) == 0 {
		http.Error(w, "No content available", http.StatusNotFound)
		return
	}

	randomURL := selector.GetRandomURL()

	// 记录统计
	endpoint := fmt.Sprintf("%s/%s", prefix, suffix)
	statsManager.IncrementCalls(endpoint)

	duration := time.Since(start)

	// 记录请求日志
	monitoring.LogRequest(monitoring.RequestLog{
		Time:       time.Now().Unix(),
		Path:       r.URL.Path,
		Method:     r.Method,
		StatusCode: http.StatusFound,
		Latency:    float64(duration.Microseconds()) / 1000, // 转换为毫秒
		IP:         realIP,
		Referer:    r.Referer(),
	})

	log.Printf(" %-12s | %-15s | %-6s | %-20s | %-20s | %-50s",
		duration,    // 持续时间
		realIP,      // 真实IP
		r.Method,    // HTTP方法
		r.URL.Path,  // 请求路径
		r.Referer(), // 来源信息
		randomURL,   // 重定向URL
	)

	http.Redirect(w, r, randomURL, http.StatusFound)
}

func HandleStats(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Content-Type", "application/json")
	stats := statsManager.GetStats()
	if err := json.NewEncoder(w).Encode(stats); err != nil {
		http.Error(w, "Error encoding stats", http.StatusInternalServerError)
		log.Printf("Error encoding stats: %v", err)
	}
}