✨ 修复cookie问题

internal/application/cpdaily/login/common.go

@@ -32,7 +32,7 @@ func ActivateSession(ctx context.Context, host string, client *resty.Client) {
 
 		defer sessionResponse.Body.Close()
 
-		if !sessionResponse.IsSuccess() {
+		if !sessionResponse.IsSuccess() && sessionResponse.StatusCode() != 502 {
 			utils.OkOrPanic(errors.NewBusinessError(503, "Session激活失败"),
 				func(err error) error {
 					return oops.
@@ -87,10 +87,7 @@ func ActivateSession(ctx context.Context, host string, client *resty.Client) {
 }
 
 func CookieEnabled(ctx context.Context, host string, client *resty.Client) bool {
-	client.SetRedirectPolicy(resty.FlexibleRedirectPolicy(20))
-	defer func() {
-		client.SetRedirectPolicy(resty.NoRedirectPolicy())
-	}()
+	client.SetRedirectPolicy(resty.NoRedirectPolicy())
 	host = strings.TrimSuffix(host, "/")
 	sessionResponse, err := client.R().
 		WithContext(ctx).
@@ -98,13 +95,7 @@ func CookieEnabled(ctx context.Context, host string, client *resty.Client) bool
 		Post(fmt.Sprintf("%s/wec-counselor-attendance-apps/student/attendance/getStuAttendacesInOneDay", host))
 
 	utils.OkOrPanic(err)
-	defer sessionResponse.Body.Close()
-
-	if !sessionResponse.IsSuccess() {
-		return sessionResponse.StatusCode() == 405
-	}
-	responseContentType := sessionResponse.Header().Get("Content-Type")
-	return strings.Contains(responseContentType, "json")
+	return sessionResponse.StatusCode() != 302
 }
 
 func GetClient(ctx context.Context, cookie string) *resty.Client {

internal/application/cpdaily/school/school_fetcher.go

@@ -65,7 +65,7 @@ func GetSchoolDetail(ctx context.Context, tenantId string) SchoolDetailInfo {
 		var ret SchoolDetailInfo
 		err := sonic.Unmarshal([]byte(redisResult.Val()), &ret)
 		utils.OkOrPanic(err)
-		return ret
+		return parseSpecialSchool(ret)
 	}
 	response, err := httpclient.Cpdaily.Clone(ctx).R().
 		SetHeaders(map[string]string{
@@ -105,5 +105,13 @@ func GetSchoolDetail(ctx context.Context, tenantId string) SchoolDetailInfo {
 			Wrap(errors.NewBusinessError(503, "学校ID不存在").Decorate("empty school detail")))
 	}
 	client.Set(ctx, cst2.RedisSchoolInfo+tenantId, ret[0], time.Duration(cst2.RedisSchoolInfoExpire)*time.Second)
-	return ret[0]
+	return parseSpecialSchool(ret[0])
+}
+
+func parseSpecialSchool(s SchoolDetailInfo) SchoolDetailInfo {
+	// 甘肃农业大学
+	if s.Id == "gsau" {
+		s.WecloudUrl = "https://fdm.gsau.edu.cn"
+	}
+	return s
 }

internal/framework/schedule/cpdaily_handler.go

@@ -94,14 +94,13 @@ func (c *CpdailyTaskHandler) ProcessTask(ctx context.Context, task *asynq.Task)
 	schoolDetail := school.GetSchoolDetail(ctx, targetTask.Edges.CpUser.SchoolID)
 	cookie := targetTask.Edges.CpUser.Cookie
 	restyClient := login.GetClient(ctx, cookie)
-	//if !login.CookieEnabled(ctx, schoolDetail.WecloudUrl, restyClient) {
-	// 定时任务强制登录并更新cookie
-	loginExecutor := login.GetCpdailyLoginBySchoolInfo(ctx, &schoolDetail)
-	loginExecutor.Login(ctx, &schoolDetail, targetTask.Edges.CpUser.Username, targetTask.Edges.CpUser.Password)
-	cookie = loginExecutor.GetCookieString()
-	db.Global.CpUser.UpdateOneID(targetTask.Edges.CpUser.ID).SetCookie(cookie).ExecX(ctx)
-	restyClient = login.GetClient(ctx, cookie)
-	//}
+	if !login.CookieEnabled(ctx, schoolDetail.WecloudUrl, restyClient) {
+		loginExecutor := login.GetCpdailyLoginBySchoolInfo(ctx, &schoolDetail)
+		loginExecutor.Login(ctx, &schoolDetail, targetTask.Edges.CpUser.Username, targetTask.Edges.CpUser.Password)
+		cookie = loginExecutor.GetCookieString()
+		db.Global.CpUser.UpdateOneID(targetTask.Edges.CpUser.ID).SetCookie(cookie).ExecX(ctx)
+		restyClient = login.GetClient(ctx, cookie)
+	}
 	// 获取任务执行handler
 	taskExecutor := task2.GetCpdailyTaskService(ctx, targetTask.TaskType, restyClient, schoolDetail)