90 lines
1.9 KiB
Go
90 lines
1.9 KiB
Go
|
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, "验证码")
|
||
|
|
}
|