Initial commit

client/http.go

@@ -0,0 +1,89 @@
+package client
+
+import (
+	"fmt"
+	"log"
+	"net/http"
+	"net/http/cookiejar"
+	"net/url"
+	"strings"
+
+	"jwts/config"
+)
+
+// 全局HTTP客户端
+var Client *http.Client
+
+// 初始化HTTP客户端
+func InitClient() {
+	jar, err := cookiejar.New(nil)
+	if err != nil {
+		log.Fatalf("Failed to create cookie jar: %v", err)
+	}
+	Client = &http.Client{
+		Jar: jar,
+	}
+}
+
+// 设置请求头
+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, "验证码")
+}