fix(quark_uc_tv): auto re-scan when refresh token is invalid (#557)

drivers/quark_uc_tv/driver.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"net/http"
 	"strconv"
+	"strings"
 	"time"
 
 	"github.com/OpenListTeam/OpenList/v4/internal/op"
@@ -62,20 +63,22 @@ func (d *QuarkUCTV) Init(ctx context.Context) error {
 			// 通过query token获取code -> refresh token
 			code, err := d.getCode(ctx1)
 			if err != nil {
-				return err
+				// 获取授权码失败，重置状态并重新获取二维码
+				return d.reInit(ctx)
 			}
 			// 通过code获取refresh token
 			err = d.getRefreshTokenByTV(ctx1, code, false)
 			if err != nil {
-				return err
+				// 换取凭证失败，重置状态并重试扫码流程
+				return d.reInit(ctx)
 			}
 		}
 	}
 	// 通过refresh token获取access token
 	if d.QuarkUCTVCommon.AccessToken == "" {
 		err := d.getRefreshTokenByTV(ctx1, d.Addition.RefreshToken, true)
 		if err != nil {
-			return err
+			return d.handleInvalidRefreshToken(ctx, err)
 		}
 	}
 
@@ -87,6 +90,22 @@ func (d *QuarkUCTV) Init(ctx context.Context) error {
 	return nil
 }
 
+// reInit 重置授权状态，用于在登录凭证失效或过期时触发重新扫码流程
+func (d *QuarkUCTV) reInit(ctx context.Context) error {
+	d.Addition.QueryToken = ""
+	op.MustSaveDriverStorage(d)
+	return d.Init(ctx)
+}
+
+// handleInvalidRefreshToken 检查是否为 Refresh Token 无效错误，若是则触发重新扫码
+func (d *QuarkUCTV) handleInvalidRefreshToken(ctx context.Context, err error) error {
+	if strings.Contains(err.Error(), "Refresh Token无效") {
+		d.Addition.RefreshToken = ""
+		return d.reInit(ctx)
+	}
+	return err
+}
+
 func (d *QuarkUCTV) Drop(ctx context.Context) error {
 	return nil
 }

drivers/quark_uc_tv/util.go

@@ -83,7 +83,7 @@ func (d *QuarkUCTV) request(ctx context.Context, pathname string, method string,
 		// token 过期 / 无效
 		err = d.getRefreshTokenByTV(ctx, d.Addition.RefreshToken, true)
 		if err != nil {
-			return nil, err
+			return nil, d.handleInvalidRefreshToken(ctx, err)
 		}
 		ctx1, cancelFunc := context.WithTimeout(ctx, 10*time.Second)
 		defer cancelFunc()
@@ -187,6 +187,9 @@ func (d *QuarkUCTV) getRefreshTokenByTV(ctx context.Context, code string, isRefr
 		op.MustSaveDriverStorage(d)
 		d.QuarkUCTVCommon.AccessToken = resp.Data.AccessToken
 	} else {
+		if resp.Data.ErrorInfo != "" {
+			return errors.New(resp.Data.ErrorInfo)
+		}
 		return errors.New("refresh token is empty")
 	}
 	return nil