103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
package grpc
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"log/slog"
|
|
"net/http"
|
|
"net/http/cookiejar"
|
|
"net/url"
|
|
"time"
|
|
|
|
"schedule_converter/auth"
|
|
"schedule_converter/captcha"
|
|
"schedule_converter/client"
|
|
"schedule_converter/config"
|
|
"schedule_converter/models"
|
|
"schedule_converter/parser"
|
|
apperr "schedule_converter/pkg/errors"
|
|
)
|
|
|
|
const (
|
|
maxLoginRetries = 3
|
|
loginRetryInterval = 3 * time.Second
|
|
)
|
|
|
|
func createHTTPClient() (*http.Client, error) {
|
|
jar, err := cookiejar.New(nil)
|
|
if err != nil {
|
|
return nil, apperr.New("create_jar", err, "鍒涘缓 cookie jar 澶辫触")
|
|
}
|
|
return &http.Client{Jar: jar}, nil
|
|
}
|
|
|
|
func setTWFIDCookie(httpClient *http.Client, baseURL, twfid string) {
|
|
cfg := config.Get()
|
|
if twfid == "" {
|
|
twfid = cfg.TWFID
|
|
}
|
|
u, err := url.Parse(baseURL)
|
|
if err != nil {
|
|
slog.Warn("瑙ｆ瀽 baseURL 澶辫触锛屾棤娉曡缃?TWFID cookie", "base_url", baseURL, "error", err)
|
|
return
|
|
}
|
|
httpClient.Jar.SetCookies(u, []*http.Cookie{
|
|
{Name: "TWFID", Value: twfid},
|
|
})
|
|
}
|
|
|
|
func executeLogin(ctx context.Context, httpClient *http.Client, username, password string) error {
|
|
log := slog.With("username", username)
|
|
|
|
if err := client.GetInitialCookiesWithClient(httpClient); err != nil {
|
|
return apperr.New("get_cookies", err, "鑾峰彇鍒濆 Cookie 澶辫触")
|
|
}
|
|
|
|
for attempt := 1; attempt <= maxLoginRetries; attempt++ {
|
|
select {
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
default:
|
|
}
|
|
|
|
log.Info("灏濊瘯鐧诲綍", "attempt", attempt, "max_retries", maxLoginRetries)
|
|
|
|
captchaCode := captcha.GetCaptchaCode()
|
|
success, err := auth.LoginWithClient(httpClient, captchaCode, username, password)
|
|
if err != nil {
|
|
log.Warn("鐧诲綍璇锋眰澶辫触", "error", err, "attempt", attempt)
|
|
time.Sleep(loginRetryInterval)
|
|
continue
|
|
}
|
|
|
|
if success {
|
|
return nil
|
|
}
|
|
|
|
log.Warn("鐧诲綍楠岃瘉澶辫触", "attempt", attempt)
|
|
time.Sleep(loginRetryInterval)
|
|
}
|
|
|
|
return apperr.ErrLoginFailed
|
|
}
|
|
|
|
func executeLoginAndFetchSchedule(ctx context.Context, httpClient *http.Client, username, password, semester string) ([]models.Course, error) {
|
|
if err := executeLogin(ctx, httpClient, username, password); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
cfg := config.Get()
|
|
if semester == "" {
|
|
semester = cfg.Semester
|
|
}
|
|
|
|
return parser.FetchScheduleWithClient(httpClient, semester)
|
|
}
|
|
|
|
func unmarshalPayload(payload []byte, v any) error {
|
|
if err := json.Unmarshal(payload, v); err != nil {
|
|
return apperr.New("parse_payload", err, "瑙ｆ瀽浠诲姟杞借嵎澶辫触")
|
|
}
|
|
return nil
|
|
}
|