package main

import (
	"encoding/json"
	"fmt"
	"io"
	"log"
	"net/http"

	"kfz-goods-pricing/internal/config"
	"kfz-goods-pricing/internal/database"
	"kfz-goods-pricing/internal/handler"
	"kfz-goods-pricing/internal/repository"
	"kfz-goods-pricing/internal/service"
)

func main() {
	// 日志同时输出到GUI窗口和本地文件
	log.SetOutput(io.MultiWriter(&guiLogWriter{}, initFileLog()))
	log.SetFlags(log.Ltime)

	log.Printf("孔网商品定价 %s 启动中...", version)

	// 加载配置
	cfg, err := config.Load("./config/config.yaml")
	if err != nil {
		fatalExit("加载配置文件失败: %v", err)
	}
	config.SetGlobal(cfg)
	log.Printf("配置加载成功: port=%s, timer=%ds, rate_limit=%ds", cfg.Port, cfg.TimerInterval, cfg.APIRateLimit)

	global := config.GetGlobal()
	marshal, _ := json.Marshal(global)
	log.Printf("config: %s", string(marshal))

	// 初始化数据库
	if err := database.InitDB("./data/goods_pricing.db"); err != nil {
		fatalExit("初始化数据库失败: %v", err)
	}
	defer database.CloseDB()
	log.Println("数据库初始化成功")

	// 初始化服务
	proxy := ""
	tokenRepo := repository.NewTokenRepository()
	goodsService := service.NewGoodsService(proxy, cfg.APIRateLimit, cfg.CallbackURL, tokenRepo)
	goodsHandler := handler.NewGoodsHandler(goodsService)

	// Token相关
	tokenHandler := handler.NewTokenHandler(tokenRepo)

	// Kfz登录
	kfzHandler := handler.NewKfzHandler()

	// 配置相关
	configHandler := handler.NewConfigHandler("./config/config.yaml")

	// 注册路由（带 CORS 中间件）
	http.HandleFunc("/api/goods/query", corsMiddleware(goodsHandler.QueryGoods))

	// Token路由
	http.HandleFunc("/api/token/add", corsMiddleware(tokenHandler.BatchAddTokens))
	http.HandleFunc("/api/token/list", corsMiddleware(tokenHandler.GetAllTokens))
	http.HandleFunc("/api/token/delete", corsMiddleware(tokenHandler.DeleteToken))
	http.HandleFunc("/api/token/update", corsMiddleware(tokenHandler.UpdateToken))
	http.HandleFunc("/api/token/enabled", corsMiddleware(tokenHandler.GetEnabledTokens))

	// Kfz登录路由
	http.HandleFunc("/api/kfz/login", corsMiddleware(kfzHandler.KfzLogin))

	// 配置路由
	http.HandleFunc("/api/config/price/get", corsMiddleware(configHandler.GetConfigPrice))
	http.HandleFunc("/api/config/price/set", corsMiddleware(configHandler.SetConfigPrice))

	// 启动定时器
	goodsService.StartTimerScheduler(cfg.TimerInterval)
	log.Printf("定时器已启动，%d秒后开始首次同步", cfg.TimerInterval)

	// 启动HTTP服务（后台）
	go func() {
		log.Printf("服务器正在启动 %s", cfg.Port)
		if err := http.ListenAndServe(fmt.Sprintf(":%s", cfg.Port), nil); err != nil {
			fatalExit("启动服务器失败: %v", err)
		}
	}()

	// 启动GUI窗口（阻塞直到窗口关闭）
	runGUI()
}

// corsMiddleware CORS 跨域中间件
func corsMiddleware(next http.HandlerFunc) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		// 允许所有来源
		w.Header().Set("Access-Control-Allow-Origin", "*")
		// 允许的方法
		w.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
		// 允许的请求头
		w.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization, X-Requested-With")

		// 预检请求直接返回
		if r.Method == http.MethodOptions {
			w.WriteHeader(http.StatusOK)
			return
		}

		// 调用实际处理函数
		next(w, r)
	}
}