package client

import (
	"log/slog"
	"net/http"
	"net/http/cookiejar"
	"net/url"
	"strings"
	"sync"

	"schedule_converter/config"
)

var (
	Client           *http.Client
	RecognizerClient *http.Client
	once             sync.Once
)

func InitClient() {
	once.Do(func() {
		jar, err := cookiejar.New(nil)
		if err != nil {
			slog.Error("创建 cookie jar 失败", "error", err)
			panic(err)
		}
		Client = &http.Client{Jar: jar}

		cfg := config.Get()
		RecognizerClient = &http.Client{
			Timeout: cfg.Recognizer.Timeout,
		}

		slog.Info("HTTP 客户端初始化完成")
	})
}

func GetHeaders() http.Header {
	return http.Header{
		"User-Agent":      []string{"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"},
		"Accept":          []string{"text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8"},
		"Accept-Language": []string{"zh-CN,zh;q=0.9,en;q=0.8"},
	}
}

func GetHeadersWithContentType(contentType string) http.Header {
	headers := GetHeaders()
	headers.Set("Content-Type", contentType)
	return headers
}

func GetInitialCookies() error {
	return GetInitialCookiesWithClient(Client)
}

func GetInitialCookiesWithClient(httpClient *http.Client) error {
	slog.Info("访问登录页面获取初始Cookie")

	cfg := config.Get()
	req, err := http.NewRequest("GET", cfg.BaseURL+"/loginNOCAS", nil)
	if err != nil {
		return err
	}
	req.Header = GetHeaders()

	resp, err := httpClient.Do(req)
	if err != nil {
		return err
	}
	defer resp.Body.Close()

	slog.Info("登录页面访问完成", "status_code", resp.StatusCode)
	logCookies(httpClient)

	return nil
}

func logCookies(httpClient *http.Client) {
	cfg := config.Get()
	cookies := httpClient.Jar.Cookies(parseURL(cfg.BaseURL))

	var cookieStrs []string
	for _, cookie := range cookies {
		cookieStrs = append(cookieStrs, cookie.Name+"="+cookie.Value)
	}
	slog.Debug("当前Cookies", "cookies", strings.Join(cookieStrs, "; "))
}

func parseURL(rawURL string) *url.URL {
	u, _ := url.Parse(rawURL)
	return u
}

func CheckLoginSuccess(body string) bool {
	return strings.Contains(body, "本科教学管理服务系统")
}

func IsCaptchaError(body string) bool {
	return strings.Contains(body, "验证码")
}