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
69 lines
1.6 KiB
Go
69 lines
1.6 KiB
Go
package auth
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"schedule_converter/client"
|
|
"schedule_converter/config"
|
|
)
|
|
|
|
// 登录
|
|
func Login(captchaCode string) (bool, error) {
|
|
return LoginWithClient(client.Client, captchaCode, config.Username, config.Password)
|
|
}
|
|
|
|
// LoginWithClient 使用指定的 HTTP 客户端和凭据登录
|
|
func LoginWithClient(httpClient *http.Client, captchaCode, username, password string) (bool, error) {
|
|
fmt.Println("[4] 执行登录...")
|
|
|
|
data := fmt.Sprintf("usercode=%s&password=%s&code=%s", username, password, captchaCode)
|
|
req, err := http.NewRequest("POST", config.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
|
|
}
|
|
|
|
if client.CheckLoginSuccess(string(body)) {
|
|
fmt.Println(" ✓ 登录成功！")
|
|
|
|
// 提取用户名
|
|
nameMatch := regexp.MustCompile(`(\w+)同学`).FindStringSubmatch(string(body))
|
|
if len(nameMatch) > 1 {
|
|
fmt.Printf(" 欢迎, %s!\n", nameMatch[1])
|
|
}
|
|
|
|
// 保存登录后的页面
|
|
if err := os.WriteFile("login_result.html", body, 0644); err != nil {
|
|
fmt.Printf(" 警告: 无法保存登录页面: %v\n", err)
|
|
} else {
|
|
fmt.Println(" 登录后页面已保存到: login_result.html")
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
fmt.Println(" ✗ 登录失败")
|
|
if client.IsCaptchaError(string(body)) {
|
|
fmt.Println(" 错误: 验证码错误")
|
|
}
|
|
|
|
return false, nil
|
|
}
|