schedule_converter/client/http.go

package client

import (
	"fmt"
	"log"
	"net/http"
	"net/http/cookiejar"
	"net/url"
	"strings"
	"time"

	"jwts/config"
)

var Client *http.Client

var RecognizerClient *http.Client

func InitClient() {
	jar, err := cookiejar.New(nil)
	if err != nil {
		log.Fatalf("Failed to create cookie jar: %v", err)
	}
	Client = &http.Client{
		Jar: jar,
	}

	RecognizerClient = &http.Client{
		Timeout: 120 * time.Second,
	}
}

// 设置请求头
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"},
	}
}

// 获取带Content-Type的请求头
func GetHeadersWithContentType(contentType string) http.Header {
	headers := GetHeaders()
	headers["Content-Type"] = []string{contentType}
	return headers
}

// 步骤1: 获取初始Cookie
func GetInitialCookies() error {
	fmt.Println("[1] 访问登录页面获取初始Cookie...")

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

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

	fmt.Printf("    登录页面状态码: %d\n", resp.StatusCode)
	PrintCookies()

	return nil
}

// 打印Cookies
func PrintCookies() {
	fmt.Printf("    当前Cookies: ")
	for _, cookie := range Client.Jar.Cookies(parseURL(config.BaseURL)) {
		fmt.Printf("%s=%s; ", cookie.Name, cookie.Value)
	}
	fmt.Println()
}

// 解析URL
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, "验证码")
}