增加baidu_youth秒存功能

Author: morecup

backend/baidu_youth/api/api.go

@@ -317,3 +317,25 @@ func (b *BaiduApi) Create(path string, preCreateFileData *PreCreateFileData, upl
 	}
 	return opts, nil
 }
+
+func (b *BaiduApi) Rapid(path string, contentMd5 string, sliceMd5 string, size int64) (opts *rest.Opts, err error) {
+	data := url.Values{}
+	data.Add("path", FixToBaiduPath(path))
+	data.Add("content-length", strconv.FormatInt(size, 10))
+	data.Add("content-md5", contentMd5)
+	data.Add("slice-md5", sliceMd5)
+	data.Add("rtype", "1")
+	data.Add("ondup", "newcopy")
+
+	opts = &rest.Opts{
+		Method:  "POST",
+		RootURL: "https://d.pcs.baidu.com/rest/2.0/pcs/file",
+		Parameters: url.Values{
+			"method": []string{"rapidupload"},
+			//"app_id":	[]string{"25179614"},
+		},
+		ContentType: "application/x-www-form-urlencoded",
+		Body:        strings.NewReader(data.Encode()),
+	}
+	return opts, nil
+}

backend/baidu_youth/api/types.go

@@ -138,6 +138,21 @@ type CreateVO struct {
 	BaseItem
 }
 
+type PcsBaseBaiduResponse struct {
+	ErrnoCode int   `json:"error_code"`
+	ErrorMsg  int   `json:"error_msg"`
+	RequestId int64 `json:"request_id"`
+}
+
+func (b PcsBaseBaiduResponse) GetErrno() int {
+	return b.ErrnoCode
+}
+
+type RapidVO struct {
+	BaseBaiduResponse
+	BaseItem
+}
+
 type QuotaInfoResponse struct {
 	BaseBaiduResponse
 	Total   int64  `json:"total"`

backend/baidu_youth/compat_fs.go

@@ -644,3 +644,24 @@ func (f *Fs) GetQuotaInfo(ctx context.Context) (*api.QuotaInfoResponse, error) {
 	}
 	return info, nil
 }
+
+// Rapid fs.ErrorFileRapidUpload
+func (f *Fs) Rapid(ctx context.Context, path string, contentMd5 string, sliceMd5 string, size int64) (baseItem *api.BaseItem, success bool, err error) {
+	info := new(api.RapidVO)
+	err = f.pacer.Call(func() (bool, error) {
+		opts, err := f.api.Rapid(path, contentMd5, sliceMd5, size)
+		if err != nil {
+			return false, err
+		}
+		resp, err := f.srv.CallJSON(ctx, opts, nil, info)
+		return shouldRetry(ctx, resp, err)
+	})
+	if err != nil {
+		return nil, false, err
+	}
+	if info != nil {
+		return &info.BaseItem, true, nil
+	} else {
+		return nil, false, fs.ErrorFileRapidUpload
+	}
+}