306 lines
7.3 KiB
Go
306 lines
7.3 KiB
Go
|
package grpc
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"log"
|
|||
|
|
"net/http"
|
|||
|
|
"net/http/cookiejar"
|
|||
|
|
"net/url"
|
|||
|
|
"strings"
|
|||
|
|
"sync"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
pb "jwts/proto/runner"
|
|||
|
|
|
|||
|
|
"jwts/auth"
|
|||
|
|
"jwts/captcha"
|
|||
|
|
"jwts/config"
|
|||
|
|
"jwts/models"
|
|||
|
|
"jwts/parser"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// TaskHandler 任务处理器
|
|||
|
|
type TaskHandler struct {
|
|||
|
|
cancelledTasks map[string]bool
|
|||
|
|
mu sync.RWMutex
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewTaskHandler 创建新的任务处理器
|
|||
|
|
func NewTaskHandler() *TaskHandler {
|
|||
|
|
return &TaskHandler{
|
|||
|
|
cancelledTasks: make(map[string]bool),
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Execute 执行任务
|
|||
|
|
func (h *TaskHandler) Execute(task *pb.Task) *pb.TaskResult {
|
|||
|
|
result := &pb.TaskResult{
|
|||
|
|
TaskId: task.TaskId,
|
|||
|
|
CompletedAt: time.Now().UnixMilli(),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查任务是否已取消
|
|||
|
|
if h.IsCancelled(task.TaskId) {
|
|||
|
|
result.Status = pb.TaskStatus_TASK_STATUS_CANCELLED
|
|||
|
|
result.ErrorMessage = "任务已取消"
|
|||
|
|
return result
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
switch task.Type {
|
|||
|
|
case pb.TaskType_TASK_TYPE_GET_SCHEDULE:
|
|||
|
|
data, err := h.getSchedule(task.Payload)
|
|||
|
|
if err != nil {
|
|||
|
|
result.Status = pb.TaskStatus_TASK_STATUS_FAILED
|
|||
|
|
result.ErrorMessage = err.Error()
|
|||
|
|
} else {
|
|||
|
|
result.Status = pb.TaskStatus_TASK_STATUS_SUCCESS
|
|||
|
|
result.Data = data
|
|||
|
|
}
|
|||
|
|
case pb.TaskType_TASK_TYPE_LOGIN:
|
|||
|
|
data, err := h.login(task.Payload)
|
|||
|
|
if err != nil {
|
|||
|
|
result.Status = pb.TaskStatus_TASK_STATUS_FAILED
|
|||
|
|
result.ErrorMessage = err.Error()
|
|||
|
|
} else {
|
|||
|
|
result.Status = pb.TaskStatus_TASK_STATUS_SUCCESS
|
|||
|
|
result.Data = data
|
|||
|
|
}
|
|||
|
|
default:
|
|||
|
|
result.Status = pb.TaskStatus_TASK_STATUS_FAILED
|
|||
|
|
result.ErrorMessage = fmt.Sprintf("未知的任务类型: %v", task.Type)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return result
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CancelTask 取消任务
|
|||
|
|
func (h *TaskHandler) CancelTask(taskID string) {
|
|||
|
|
h.mu.Lock()
|
|||
|
|
defer h.mu.Unlock()
|
|||
|
|
h.cancelledTasks[taskID] = true
|
|||
|
|
log.Printf("任务已标记为取消: %s", taskID)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IsCancelled 检查任务是否已取消
|
|||
|
|
func (h *TaskHandler) IsCancelled(taskID string) bool {
|
|||
|
|
h.mu.RLock()
|
|||
|
|
defer h.mu.RUnlock()
|
|||
|
|
return h.cancelledTasks[taskID]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getSchedule 获取课表
|
|||
|
|
func (h *TaskHandler) getSchedule(payload []byte) ([]byte, error) {
|
|||
|
|
var req pb.GetSchedulePayload
|
|||
|
|
if err := json.Unmarshal(payload, &req); err != nil {
|
|||
|
|
return nil, fmt.Errorf("解析任务载荷失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
semester := strings.TrimSpace(req.Semester)
|
|||
|
|
if semester == "" {
|
|||
|
|
semester = config.Semester
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
log.Printf("获取课表: 用户=%s, 学期=%s", req.Username, semester)
|
|||
|
|
|
|||
|
|
// 创建新的 HTTP 客户端
|
|||
|
|
jar, err := cookiejar.New(nil)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("创建 cookie jar 失败: %w", err)
|
|||
|
|
}
|
|||
|
|
httpClient := &http.Client{
|
|||
|
|
Jar: jar,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置 TWFID cookie (使用默认配置)
|
|||
|
|
setTWFIDCookie(httpClient, config.BaseURL, "")
|
|||
|
|
|
|||
|
|
// 步骤1: 获取初始 Cookie
|
|||
|
|
if err := getInitialCookies(httpClient); err != nil {
|
|||
|
|
return nil, fmt.Errorf("获取初始 Cookie 失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 重试登录
|
|||
|
|
maxRetries := 3
|
|||
|
|
var loginSuccess bool
|
|||
|
|
for attempt := 1; attempt <= maxRetries; attempt++ {
|
|||
|
|
log.Printf("尝试登录 (第%d次)...", attempt)
|
|||
|
|
|
|||
|
|
// 获取并识别验证码
|
|||
|
|
captchaCode, err := captcha.GetAndRecognize(httpClient)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf(" 验证码获取失败: %v", err)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 登录
|
|||
|
|
success, err := loginWithCredentials(httpClient, captchaCode, req.Username, req.Password)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf(" 登录失败: %v", err)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if success {
|
|||
|
|
loginSuccess = true
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
log.Println(" 登录失败，3秒后重试...")
|
|||
|
|
time.Sleep(3 * time.Second)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if !loginSuccess {
|
|||
|
|
return nil, fmt.Errorf("登录失败，已重试 %d 次", maxRetries)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取课表
|
|||
|
|
courses, err := parser.FetchScheduleWithClient(httpClient, semester)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("获取课表失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 构建结果
|
|||
|
|
result := &pb.ScheduleResultData{
|
|||
|
|
StudentId: req.Username,
|
|||
|
|
Semester: semester,
|
|||
|
|
Courses: convertCourses(courses),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return json.Marshal(result)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// login 登录验证
|
|||
|
|
func (h *TaskHandler) login(payload []byte) ([]byte, error) {
|
|||
|
|
var req pb.LoginPayload
|
|||
|
|
if err := json.Unmarshal(payload, &req); err != nil {
|
|||
|
|
return nil, fmt.Errorf("解析任务载荷失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
log.Printf("登录验证: 用户=%s", req.Username)
|
|||
|
|
|
|||
|
|
// 创建新的 HTTP 客户端
|
|||
|
|
jar, err := cookiejar.New(nil)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("创建 cookie jar 失败: %w", err)
|
|||
|
|
}
|
|||
|
|
httpClient := &http.Client{
|
|||
|
|
Jar: jar,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置 TWFID cookie
|
|||
|
|
setTWFIDCookie(httpClient, config.BaseURL, req.Twfid)
|
|||
|
|
|
|||
|
|
// 步骤1: 获取初始 Cookie
|
|||
|
|
if err := getInitialCookies(httpClient); err != nil {
|
|||
|
|
return nil, fmt.Errorf("获取初始 Cookie 失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 重试登录
|
|||
|
|
maxRetries := 3
|
|||
|
|
for attempt := 1; attempt <= maxRetries; attempt++ {
|
|||
|
|
log.Printf("尝试登录 (第%d次)...", attempt)
|
|||
|
|
|
|||
|
|
// 获取并识别验证码
|
|||
|
|
captchaCode, err := captcha.GetAndRecognize(httpClient)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf(" 验证码获取失败: %v", err)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 登录
|
|||
|
|
success, err := loginWithCredentials(httpClient, captchaCode, req.Username, req.Password)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf(" 登录失败: %v", err)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if success {
|
|||
|
|
return json.Marshal(map[string]interface{}{
|
|||
|
|
"success": true,
|
|||
|
|
"message": "登录成功",
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
log.Println(" 登录失败，3秒后重试...")
|
|||
|
|
time.Sleep(3 * time.Second)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return nil, fmt.Errorf("登录失败，已重试 %d 次", maxRetries)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getInitialCookies 获取初始 Cookie
|
|||
|
|
func getInitialCookies(httpClient *http.Client) error {
|
|||
|
|
log.Println("[1] 访问登录页面获取初始 Cookie...")
|
|||
|
|
|
|||
|
|
req, err := http.NewRequest("GET", config.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()
|
|||
|
|
|
|||
|
|
log.Printf(" 登录页面状态码: %d", resp.StatusCode)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// loginWithCredentials 使用凭据登录
|
|||
|
|
func loginWithCredentials(httpClient *http.Client, captchaCode, username, password string) (bool, error) {
|
|||
|
|
return auth.LoginWithClient(httpClient, captchaCode, username, password)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// setTWFIDCookie 设置 TWFID Cookie
|
|||
|
|
func setTWFIDCookie(httpClient *http.Client, baseURL, twfid string) {
|
|||
|
|
if twfid == "" {
|
|||
|
|
twfid = config.TWFID
|
|||
|
|
}
|
|||
|
|
u, _ := url.Parse(baseURL)
|
|||
|
|
httpClient.Jar.SetCookies(u, []*http.Cookie{
|
|||
|
|
{Name: "TWFID", Value: twfid},
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getHeaders 获取请求头
|
|||
|
|
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"},
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// convertCourses 转换课程格式
|
|||
|
|
func convertCourses(courses []models.Course) []*pb.Course {
|
|||
|
|
var result []*pb.Course
|
|||
|
|
for _, c := range courses {
|
|||
|
|
var schedule []*pb.ScheduleTime
|
|||
|
|
for _, s := range c.Schedule {
|
|||
|
|
schedule = append(schedule, &pb.ScheduleTime{
|
|||
|
|
Day: s.Day,
|
|||
|
|
Sections: convertToInt32(s.Section),
|
|||
|
|
Weeks: convertToInt32(s.Weeks),
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
result = append(result, &pb.Course{
|
|||
|
|
Name: c.Name,
|
|||
|
|
Teacher: c.Teacher,
|
|||
|
|
Room: c.Room,
|
|||
|
|
Schedule: schedule,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
return result
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// convertToInt32 将 []int 转换为 []int32
|
|||
|
|
func convertToInt32(arr []int) []int32 {
|
|||
|
|
result := make([]int32, len(arr))
|
|||
|
|
for i, v := range arr {
|
|||
|
|
result[i] = int32(v)
|
|||
|
|
}
|
|||
|
|
return result
|
|||
|
|
}
|