From e4439e66b9a03d139607584c2290ca2a99f7e184 Mon Sep 17 00:00:00 2001
From: foxxorcat <95907542+foxxorcat@users.noreply.github.com>
Date: Wed, 1 Jan 2025 21:13:34 +0800
Subject: [PATCH] fix:(baidu_photo): upload erron -6 (#7760 close #7744)

* fix:(baidu_photo): upload erron -6

* fix(baidu_photo):api add bdstoken
---
 drivers/baidu_photo/driver.go | 13 ++++++++++---
 drivers/baidu_photo/utils.go  | 15 +++++++++++++++
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/drivers/baidu_photo/driver.go b/drivers/baidu_photo/driver.go
index d0d69e82..b584c9a3 100644
--- 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
@@ -324,8 +330,8 @@ func (d *BaiduPhoto) Put(ctx context.Context, dstDir model.Obj, stream model.Fil
 					"path":     params["path"],
 					"partseq":  fmt.Sprint(partseq),
 					"uploadid": precreateResp.UploadID,
+					"app_id":   "16051585",
 				}
-
 				_, err = d.Post("https://c3.pcs.baidu.com/rest/2.0/pcs/superfile2", func(r *resty.Request) {
 					r.SetContext(ctx)
 					r.SetQueryParams(uploadParams)
@@ -352,6 +358,7 @@ func (d *BaiduPhoto) Put(ctx context.Context, dstDir model.Obj, stream model.Fil
 		_, err = d.Post(FILE_API_URL_V1+"/create", 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
index 0b960593..6061600e 100644
--- 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