74438c1f7f
- 重命名模块 jwts -> schedule_converter
- 新增 service 包，统一业务逻辑层
- 新增 pkg/logger 包，使用 slog 结构化日志
- 新增 pkg/errors 包，统一错误处理
- 拆分 captcha 包，图片识别移至 service/recognizer
- 清理 models 包，只保留数据结构定义
- 消除 grpc/handler.go 中的重复代码
- 简化 main.go，使用 slog 和 context
- 配置支持单例模式和结构化配置
- 使用 any 替代 interface{}
- 使用 sync.Map 替代 map + RWMutex
101 lines
2.2 KiB
Go
101 lines
2.2 KiB
Go
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, "验证码")
|
|
}
|