From 538d8f4e8fb3d00612257fd62058a05835f995e6 Mon Sep 17 00:00:00 2001
From: xiaokai <xiaokai@kk.com>
Date: Mon, 2 Sep 2024 11:27:53 +0700
Subject: [PATCH] =?UTF-8?q?pikpak=20RefreshToken=20=E9=85=8D=E7=BD=AE?=
 =?UTF-8?q?=E8=BF=87=E6=9C=9F=E5=90=8E=E4=BC=9A=E5=B0=9D=E8=AF=95=E8=B5=B0?=
 =?UTF-8?q?=E4=B8=80=E6=AC=A1=E7=99=BB=E9=99=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 drivers/pikpak/driver.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/drivers/pikpak/driver.go b/drivers/pikpak/driver.go
index e2a2b82e..87f3bc7b 100644
--- a/drivers/pikpak/driver.go
+++ b/drivers/pikpak/driver.go
@@ -97,6 +97,17 @@ func (d *PikPak) Init(ctx context.Context) (err error) {
 				RefreshToken: d.Addition.RefreshToken,
 			}).Token()
 		}))
+		_, err := d.oauth2Token.Token()
+		if err != nil {
+			if err := d.login(); err != nil {
+				return err
+			}
+			d.oauth2Token = oauth2.ReuseTokenSource(nil, utils.TokenSource(func() (*oauth2.Token, error) {
+				return oauth2Config.TokenSource(ctx, &oauth2.Token{
+					RefreshToken: d.RefreshToken,
+				}).Token()
+			}))
+		}
 	} else {
 		// 如果没有填写RefreshToken，尝试登录 获取 refreshToken
 		if err := d.login(); err != nil {