fix:(baidu_photo): upload erron -6

AlistGo · Dec 31, 2024 · 2b5ac68 · 2b5ac68
1 parent 7fd4ac7
commit 2b5ac68
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/drivers/baidu_photo/driver.go b/drivers/baidu_photo/driver.go
@@ -28,8 +28,9 @@ type BaiduPhoto struct {
 	Addition
 
 	// AccessToken string
-	Uk   int64
-	root model.Obj
+	Uk       int64
+	bdstoken string
+	root     model.Obj
 
 	uploadThread int
 }
@@ -73,6 +74,10 @@ func (d *BaiduPhoto) Init(ctx context.Context) error {
 	if err != nil {
 		return err
 	}
+	d.bdstoken, err = d.getBDStoken()
+	if err != nil {
+		return err
+	}
 	d.Uk, err = strconv.ParseInt(info.YouaID, 10, 64)
 	return err
 }
@@ -296,6 +301,7 @@ func (d *BaiduPhoto) Put(ctx context.Context, dstDir model.Obj, stream model.Fil
 		_, err = d.Post(FILE_API_URL_V1+"/precreate", func(r *resty.Request) {
 			r.SetContext(ctx)
 			r.SetFormData(params)
+			r.SetQueryParam("bdstoken", d.bdstoken)
 		}, &precreateResp)
 		if err != nil {
 			return nil, err

diff --git a/drivers/baidu_photo/utils.go b/drivers/baidu_photo/utils.go
@@ -476,6 +476,21 @@ func (d *BaiduPhoto) uInfo() (*UInfo, error) {
 	return &info, nil
 }
 
+func (d *BaiduPhoto) getBDStoken() (string, error) {
+	var info struct {
+		Result struct {
+			Bdstoken string `json:"bdstoken"`
+			Token    string `json:"token"`
+			Uk       int64  `json:"uk"`
+		} `json:"result"`
+	}
+	_, err := d.Get("https://pan.baidu.com/api/gettemplatevariable?fields=[%22bdstoken%22,%22token%22,%22uk%22]", nil, &info)
+	if err != nil {
+		return "", err
+	}
+	return info.Result.Bdstoken, nil
+}
+
 func DecryptMd5(encryptMd5 string) string {
 	if _, err := hex.DecodeString(encryptMd5); err == nil {
 		return encryptMd5