lan
accc6567f2
All checks were successful
Build / build (push) Successful in 5m28s
Refactor the gRPC client and task handling logic to improve modularity, error handling, and observability. - Split `grpc/handler.go` into specialized handler files (login, classroom, exams, grades, schedule) to reduce complexity. - Replace standard `log` with `log/slog` throughout the project for structured logging. - Refactor `grpc/client.go` to use a thread-safe connection management pattern with `sync.RWMutex`. - Remove the legacy `service` package and HTTP server implementation, transitioning the application to a pure gRPC runner mode. - Clean up `config` and `pkg/errors` to remove unused utilities and simplify the API. - Improve error reporting in parsers and HTTP clients by checking response status codes.
57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package auth
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"log/slog"
|
|
"net/http"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"schedule_converter/client"
|
|
"schedule_converter/config"
|
|
)
|
|
|
|
var nameRegexp = regexp.MustCompile(`(\w+)同学`)
|
|
|
|
// LoginWithClient 使用指定的 HTTP 客户端和凭据登录
|
|
func LoginWithClient(httpClient *http.Client, captchaCode, username, password string) (bool, error) {
|
|
log := slog.With("username", username)
|
|
log.Debug("执行登录")
|
|
|
|
data := fmt.Sprintf("usercode=%s&password=%s&code=%s", username, password, captchaCode)
|
|
req, err := http.NewRequest("POST", config.Get().BaseURL+"/login", strings.NewReader(data))
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
req.Header = client.GetHeadersWithContentType("application/x-www-form-urlencoded")
|
|
|
|
resp, err := httpClient.Do(req)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
bodyStr := string(body)
|
|
if client.CheckLoginSuccess(bodyStr) {
|
|
if match := nameRegexp.FindStringSubmatch(bodyStr); len(match) > 1 {
|
|
log.Info("登录成功", "name", match[1])
|
|
} else {
|
|
log.Info("登录成功")
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
if client.IsCaptchaError(bodyStr) {
|
|
log.Warn("登录失败：验证码错误")
|
|
} else {
|
|
log.Warn("登录失败")
|
|
}
|
|
return false, nil
|
|
} |