mirror of https://github.com/Xhofe/alist
625 lines
18 KiB
Go
625 lines
18 KiB
Go
package _89
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/md5"
|
|
"encoding/base64"
|
|
"encoding/hex"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/Xhofe/alist/conf"
|
|
"github.com/Xhofe/alist/drivers/base"
|
|
"github.com/Xhofe/alist/model"
|
|
"github.com/Xhofe/alist/utils"
|
|
"github.com/go-resty/resty/v2"
|
|
jsoniter "github.com/json-iterator/go"
|
|
log "github.com/sirupsen/logrus"
|
|
"io"
|
|
"math"
|
|
"net/http"
|
|
"net/http/cookiejar"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
var client189Map map[string]*resty.Client
|
|
var infoMap = make(map[string]Rsa)
|
|
|
|
func (driver Cloud189) getClient(account *model.Account) (*resty.Client, error) {
|
|
client, ok := client189Map[account.Name]
|
|
if ok {
|
|
return client, nil
|
|
}
|
|
err := driver.Login(account)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
client, ok = client189Map[account.Name]
|
|
if !ok {
|
|
return nil, fmt.Errorf("can't find [%s] client", account.Name)
|
|
}
|
|
return client, nil
|
|
}
|
|
|
|
func (driver Cloud189) FormatFile(file *Cloud189File) *model.File {
|
|
f := &model.File{
|
|
Id: strconv.FormatInt(file.Id, 10),
|
|
Name: file.Name,
|
|
Size: file.Size,
|
|
Driver: driver.Config().Name,
|
|
UpdatedAt: nil,
|
|
Thumbnail: file.Icon.SmallUrl,
|
|
Url: file.Url,
|
|
}
|
|
loc, _ := time.LoadLocation("Local")
|
|
lastOpTime, err := time.ParseInLocation("2006-01-02 15:04:05", file.LastOpTime, loc)
|
|
if err == nil {
|
|
f.UpdatedAt = &lastOpTime
|
|
}
|
|
if file.Size == -1 {
|
|
f.Size = 0
|
|
}
|
|
f.Type = file.GetType()
|
|
return f
|
|
}
|
|
|
|
//func (c Cloud189) GetFile(path string, account *model.Account) (*Cloud189File, error) {
|
|
// dir, name := filepath.Split(path)
|
|
// dir = utils.ParsePath(dir)
|
|
// _, _, err := c.ParentPath(dir, account)
|
|
// if err != nil {
|
|
// return nil, err
|
|
// }
|
|
// parentFiles_, _ := conf.Cache.Get(conf.Ctx, fmt.Sprintf("%s%s", account.Name, dir))
|
|
// parentFiles, _ := parentFiles_.([]Cloud189File)
|
|
// for _, file := range parentFiles {
|
|
// if file.Name == name {
|
|
// if file.Size != -1 {
|
|
// return &file, err
|
|
// } else {
|
|
// return nil, ErrNotFile
|
|
// }
|
|
// }
|
|
// }
|
|
// return nil, ErrPathNotFound
|
|
//}
|
|
|
|
type LoginResp struct {
|
|
Msg string `json:"msg"`
|
|
Result int `json:"result"`
|
|
ToUrl string `json:"toUrl"`
|
|
}
|
|
|
|
// Login refer to PanIndex
|
|
func (driver Cloud189) Login(account *model.Account) error {
|
|
client, ok := client189Map[account.Name]
|
|
if !ok {
|
|
//cookieJar, _ := cookiejar.New(&cookiejar.Options{PublicSuffixList: publicsuffix.List})
|
|
client = resty.New()
|
|
//client.SetCookieJar(cookieJar)
|
|
client.SetTimeout(base.DefaultTimeout)
|
|
client.SetRetryCount(3)
|
|
client.SetHeader("Referer", "https://cloud.189.cn/")
|
|
}
|
|
// clear cookie
|
|
jar, _ := cookiejar.New(nil)
|
|
client.SetCookieJar(jar)
|
|
url := "https://cloud.189.cn/api/portal/loginUrl.action?redirectURL=https%3A%2F%2Fcloud.189.cn%2Fmain.action"
|
|
b := ""
|
|
lt := ""
|
|
ltText := regexp.MustCompile(`lt = "(.+?)"`)
|
|
var res *resty.Response
|
|
var err error
|
|
for i := 0; i < 3; i++ {
|
|
res, err = client.R().Get(url)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// 已经登陆
|
|
if res.RawResponse.Request.URL.String() == "https://cloud.189.cn/web/main" {
|
|
return nil
|
|
}
|
|
b = res.String()
|
|
ltTextArr := ltText.FindStringSubmatch(b)
|
|
if len(ltTextArr) > 0 {
|
|
lt = ltTextArr[1]
|
|
break
|
|
} else {
|
|
<-time.After(time.Second)
|
|
}
|
|
}
|
|
if lt == "" {
|
|
return fmt.Errorf("get page: %s \nstatus: %d \nrequest url: %s", b, res.StatusCode(), res.RawResponse.Request.URL.String())
|
|
}
|
|
captchaToken := regexp.MustCompile(`captchaToken' value='(.+?)'`).FindStringSubmatch(b)[1]
|
|
returnUrl := regexp.MustCompile(`returnUrl = '(.+?)'`).FindStringSubmatch(b)[1]
|
|
paramId := regexp.MustCompile(`paramId = "(.+?)"`).FindStringSubmatch(b)[1]
|
|
//reqId := regexp.MustCompile(`reqId = "(.+?)"`).FindStringSubmatch(b)[1]
|
|
jRsakey := regexp.MustCompile(`j_rsaKey" value="(\S+)"`).FindStringSubmatch(b)[1]
|
|
vCodeID := regexp.MustCompile(`picCaptcha\.do\?token\=([A-Za-z0-9\&\=]+)`).FindStringSubmatch(b)[1]
|
|
vCodeRS := ""
|
|
if vCodeID != "" {
|
|
// need ValidateCode
|
|
log.Debugf("try to identify verification codes")
|
|
timeStamp := strconv.FormatInt(time.Now().UnixNano()/1e6, 10)
|
|
u := "https://open.e.189.cn/api/logbox/oauth2/picCaptcha.do?token=" + vCodeID + timeStamp
|
|
imgRes, err := client.R().SetHeaders(map[string]string{
|
|
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:74.0) Gecko/20100101 Firefox/76.0",
|
|
"Referer": "https://open.e.189.cn/api/logbox/oauth2/unifyAccountLogin.do",
|
|
"Sec-Fetch-Dest": "image",
|
|
"Sec-Fetch-Mode": "no-cors",
|
|
"Sec-Fetch-Site": "same-origin",
|
|
}).Get(u)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
vRes, err := client.R().SetMultipartField(
|
|
"image", "validateCode.png", "image/png", bytes.NewReader(imgRes.Body())).
|
|
Post(conf.GetStr("ocr api"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if jsoniter.Get(vRes.Body(), "status").ToInt() != 200 {
|
|
return errors.New("ocr error:" + jsoniter.Get(vRes.Body(), "msg").ToString())
|
|
}
|
|
vCodeRS = jsoniter.Get(vRes.Body(), "result").ToString()
|
|
log.Debugln("code: ", vCodeRS)
|
|
}
|
|
userRsa := RsaEncode([]byte(account.Username), jRsakey, true)
|
|
passwordRsa := RsaEncode([]byte(account.Password), jRsakey, true)
|
|
url = "https://open.e.189.cn/api/logbox/oauth2/loginSubmit.do"
|
|
var loginResp LoginResp
|
|
res, err = client.R().
|
|
SetHeaders(map[string]string{
|
|
"lt": lt,
|
|
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36",
|
|
"Referer": "https://open.e.189.cn/",
|
|
"accept": "application/json;charset=UTF-8",
|
|
}).SetFormData(map[string]string{
|
|
"appKey": "cloud",
|
|
"accountType": "01",
|
|
"userName": "{RSA}" + userRsa,
|
|
"password": "{RSA}" + passwordRsa,
|
|
"validateCode": vCodeRS,
|
|
"captchaToken": captchaToken,
|
|
"returnUrl": returnUrl,
|
|
"mailSuffix": "@pan.cn",
|
|
"paramId": paramId,
|
|
"clientType": "10010",
|
|
"dynamicCheck": "FALSE",
|
|
"cb_SaveName": "1",
|
|
"isOauth2": "false",
|
|
}).Post(url)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = utils.Json.Unmarshal(res.Body(), &loginResp)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
return err
|
|
}
|
|
if loginResp.Result != 0 {
|
|
return fmt.Errorf(loginResp.Msg)
|
|
}
|
|
_, err = client.R().Get(loginResp.ToUrl)
|
|
if err != nil {
|
|
log.Errorf(err.Error())
|
|
return err
|
|
}
|
|
client189Map[account.Name] = client
|
|
return nil
|
|
}
|
|
|
|
func (driver Cloud189) isFamily(account *model.Account) bool {
|
|
return account.InternalType == "Family"
|
|
}
|
|
|
|
func (driver Cloud189) GetFiles(fileId string, account *model.Account) ([]Cloud189File, error) {
|
|
res := make([]Cloud189File, 0)
|
|
pageNum := 1
|
|
|
|
for {
|
|
var resp Cloud189Files
|
|
body, err := driver.Request("https://cloud.189.cn/api/open/file/listFiles.action", base.Get, map[string]string{
|
|
//"noCache": random(),
|
|
"pageSize": "60",
|
|
"pageNum": strconv.Itoa(pageNum),
|
|
"mediaType": "0",
|
|
"folderId": fileId,
|
|
"iconOption": "5",
|
|
"orderBy": "lastOpTime", //account.OrderBy
|
|
"descending": "true", //account.OrderDirection
|
|
}, nil, nil, account)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = utils.Json.Unmarshal(body, &resp)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp.ResCode != 0 {
|
|
return nil, fmt.Errorf(resp.ResMessage)
|
|
}
|
|
if resp.FileListAO.Count == 0 {
|
|
break
|
|
}
|
|
for _, folder := range resp.FileListAO.FolderList {
|
|
res = append(res, Cloud189File{
|
|
Id: folder.Id,
|
|
LastOpTime: folder.LastOpTime,
|
|
Name: folder.Name,
|
|
Size: -1,
|
|
})
|
|
}
|
|
res = append(res, resp.FileListAO.FileList...)
|
|
pageNum++
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func (driver Cloud189) Request(url string, method int, query, form map[string]string, headers map[string]string, account *model.Account) ([]byte, error) {
|
|
client, err := driver.getClient(account)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
//var resp base.Json
|
|
if driver.isFamily(account) {
|
|
url = strings.Replace(url, "/api/open", "/api/open/family", 1)
|
|
if query != nil {
|
|
query["familyId"] = account.SiteId
|
|
}
|
|
if form != nil {
|
|
form["familyId"] = account.SiteId
|
|
}
|
|
}
|
|
var e Cloud189Error
|
|
req := client.R().SetError(&e).
|
|
SetHeader("Accept", "application/json;charset=UTF-8").
|
|
SetQueryParams(map[string]string{
|
|
"noCache": random(),
|
|
})
|
|
if query != nil {
|
|
req = req.SetQueryParams(query)
|
|
}
|
|
if form != nil {
|
|
req = req.SetFormData(form)
|
|
}
|
|
if headers != nil {
|
|
req = req.SetHeaders(headers)
|
|
}
|
|
var res *resty.Response
|
|
switch method {
|
|
case base.Get:
|
|
res, err = req.Get(url)
|
|
case base.Post:
|
|
res, err = req.Post(url)
|
|
default:
|
|
return nil, base.ErrNotSupport
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
//log.Debug(res.String())
|
|
if e.ErrorCode != "" {
|
|
if e.ErrorCode == "InvalidSessionKey" {
|
|
err = driver.Login(account)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return driver.Request(url, method, query, form, nil, account)
|
|
}
|
|
}
|
|
if jsoniter.Get(res.Body(), "res_code").ToInt() != 0 {
|
|
err = errors.New(jsoniter.Get(res.Body(), "res_message").ToString())
|
|
}
|
|
return res.Body(), err
|
|
}
|
|
|
|
func (driver Cloud189) GetSessionKey(account *model.Account) (string, error) {
|
|
//info, ok := infoMap[account.Name]
|
|
//if !ok {
|
|
// info = Info{}
|
|
// infoMap[account.Name] = info
|
|
//} else {
|
|
// log.Debugf("hit")
|
|
//}
|
|
//if info.SessionKey != "" {
|
|
// return info.SessionKey, nil
|
|
//}
|
|
resp, err := driver.Request("https://cloud.189.cn/v2/getUserBriefInfo.action", base.Get, nil, nil, nil, account)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
sessionKey := jsoniter.Get(resp, "sessionKey").ToString()
|
|
//info.SessionKey = sessionKey
|
|
return sessionKey, nil
|
|
}
|
|
|
|
func (driver Cloud189) GetResKey(account *model.Account) (string, string, error) {
|
|
rsa, ok := infoMap[account.Name]
|
|
if !ok {
|
|
rsa = Rsa{}
|
|
infoMap[account.Name] = rsa
|
|
}
|
|
now := time.Now().UnixMilli()
|
|
if rsa.Expire > now {
|
|
return rsa.PubKey, rsa.PkId, nil
|
|
}
|
|
resp, err := driver.Request("https://cloud.189.cn/api/security/generateRsaKey.action", base.Get, nil, nil, nil, account)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
pubKey, pkId := jsoniter.Get(resp, "pubKey").ToString(), jsoniter.Get(resp, "pkId").ToString()
|
|
rsa.PubKey, rsa.PkId = pubKey, pkId
|
|
rsa.Expire = jsoniter.Get(resp, "expire").ToInt64()
|
|
return pubKey, pkId, nil
|
|
}
|
|
|
|
//func (driver Cloud189) UploadRequest1(uri string, form map[string]string, account *model.Account, resp interface{}) ([]byte, error) {
|
|
// //sessionKey, err := driver.GetSessionKey(account)
|
|
// //if err != nil {
|
|
// // return nil, err
|
|
// //}
|
|
// sessionKey := account.DriveId
|
|
// pubKey, pkId, err := driver.GetResKey(account)
|
|
// log.Debugln(sessionKey, pubKey, pkId)
|
|
// if err != nil {
|
|
// return nil, err
|
|
// }
|
|
// xRId := Random("xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx")
|
|
// pkey := Random("xxxxxxxxxxxx4xxxyxxxxxxxxxxxxxxx")[0 : 16+int(16*mathRand.Float32())]
|
|
// params := hex.EncodeToString(AesEncrypt([]byte(qs(form)), []byte(pkey[0:16])))
|
|
// date := strconv.FormatInt(time.Now().Unix(), 10)
|
|
// a := make(url.Values)
|
|
// a.Set("SessionKey", sessionKey)
|
|
// a.Set("Operate", http.MethodGet)
|
|
// a.Set("RequestURI", uri)
|
|
// a.Set("Date", date)
|
|
// a.Set("params", params)
|
|
// signature := hex.EncodeToString(SHA1(EncodeParam(a), pkey))
|
|
// encryptionText := RsaEncode([]byte(pkey), pubKey, false)
|
|
// headers := map[string]string{
|
|
// "signature": signature,
|
|
// "sessionKey": sessionKey,
|
|
// "encryptionText": encryptionText,
|
|
// "pkId": pkId,
|
|
// "x-request-id": xRId,
|
|
// "x-request-date": date,
|
|
// }
|
|
// req := base.RestyClient.R().SetHeaders(headers).SetQueryParam("params", params)
|
|
// if resp != nil {
|
|
// req.SetResult(resp)
|
|
// }
|
|
// res, err := req.Get("https://upload.cloud.189.cn" + uri)
|
|
// if err != nil {
|
|
// return nil, err
|
|
// }
|
|
// //log.Debug(res.String())
|
|
// data := res.Body()
|
|
// if jsoniter.Get(data, "code").ToString() != "SUCCESS" {
|
|
// return nil, errors.New(uri + "---" + jsoniter.Get(data, "msg").ToString())
|
|
// }
|
|
// return data, nil
|
|
//}
|
|
//
|
|
//func (driver Cloud189) UploadRequest2(uri string, form map[string]string, account *model.Account, resp interface{}) ([]byte, error) {
|
|
// c := strconv.FormatInt(time.Now().UnixMilli(), 10)
|
|
// r := Random("xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx")
|
|
// l := Random("xxxxxxxxxxxx4xxxyxxxxxxxxxxxxxxx")
|
|
// l = l[0 : 16+int(16*mathRand.Float32())]
|
|
//
|
|
// e := qs(form)
|
|
// data := AesEncrypt([]byte(e), []byte(l[0:16]))
|
|
// h := hex.EncodeToString(data)
|
|
//
|
|
// sessionKey := account.DriveId
|
|
// a := make(url.Values)
|
|
// a.Set("SessionKey", sessionKey)
|
|
// a.Set("Operate", http.MethodGet)
|
|
// a.Set("RequestURI", uri)
|
|
// a.Set("Date", c)
|
|
// a.Set("params", h)
|
|
// g := SHA1(EncodeParam(a), l)
|
|
//
|
|
// pubKey, pkId, err := driver.GetResKey(account)
|
|
// if err != nil {
|
|
// return nil, err
|
|
// }
|
|
// b := RsaEncode([]byte(l), pubKey, false)
|
|
// client, err := driver.getClient(account)
|
|
// if err != nil {
|
|
// return nil, err
|
|
// }
|
|
// req := client.R()
|
|
// req.Header.Set("accept", "application/json;charset=UTF-8")
|
|
// req.Header.Set("SessionKey", sessionKey)
|
|
// req.Header.Set("Signature", hex.EncodeToString(g))
|
|
// req.Header.Set("X-Request-Date", c)
|
|
// req.Header.Set("X-Request-ID", r)
|
|
// req.Header.Set("EncryptionText", b)
|
|
// req.Header.Set("PkId", pkId)
|
|
// if resp != nil {
|
|
// req.SetResult(resp)
|
|
// }
|
|
// res, err := req.Get("https://upload.cloud.189.cn" + uri + "?params=" + h)
|
|
// if err != nil {
|
|
// return nil, err
|
|
// }
|
|
// //log.Debug(res.String())
|
|
// data = res.Body()
|
|
// if jsoniter.Get(data, "code").ToString() != "SUCCESS" {
|
|
// return nil, errors.New(uri + "---" + jsoniter.Get(data, "msg").ToString())
|
|
// }
|
|
// return data, nil
|
|
//}
|
|
|
|
func (driver Cloud189) UploadRequest(uri string, form map[string]string, account *model.Account, resp interface{}) ([]byte, error) {
|
|
c := strconv.FormatInt(time.Now().UnixMilli(), 10)
|
|
r := Random("xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx")
|
|
l := Random("xxxxxxxxxxxx4xxxyxxxxxxxxxxxxxxx")
|
|
l = l[0 : 16+int(16*utils.Rand.Float32())]
|
|
|
|
e := qs(form)
|
|
data := AesEncrypt([]byte(e), []byte(l[0:16]))
|
|
h := hex.EncodeToString(data)
|
|
|
|
sessionKey := account.DriveId
|
|
signature := hmacSha1(fmt.Sprintf("SessionKey=%s&Operate=GET&RequestURI=%s&Date=%s¶ms=%s", sessionKey, uri, c, h), l)
|
|
|
|
pubKey, pkId, err := driver.GetResKey(account)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
b := RsaEncode([]byte(l), pubKey, false)
|
|
client, err := driver.getClient(account)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
req := client.R()
|
|
req.Header.Set("accept", "application/json;charset=UTF-8")
|
|
req.Header.Set("SessionKey", sessionKey)
|
|
req.Header.Set("Signature", signature)
|
|
req.Header.Set("X-Request-Date", c)
|
|
req.Header.Set("X-Request-ID", r)
|
|
req.Header.Set("EncryptionText", b)
|
|
req.Header.Set("PkId", pkId)
|
|
if resp != nil {
|
|
req.SetResult(resp)
|
|
}
|
|
res, err := req.Get("https://upload.cloud.189.cn" + uri + "?params=" + h)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
//log.Debug(res.String())
|
|
data = res.Body()
|
|
if jsoniter.Get(data, "code").ToString() != "SUCCESS" {
|
|
return nil, errors.New(uri + "---" + jsoniter.Get(data, "msg").ToString())
|
|
}
|
|
return data, nil
|
|
}
|
|
|
|
// NewUpload Error: signature check false
|
|
func (driver Cloud189) NewUpload(file *model.FileStream, account *model.Account) error {
|
|
sessionKey, err := driver.GetSessionKey(account)
|
|
if err != nil {
|
|
account.Status = err.Error()
|
|
} else {
|
|
account.Status = "work"
|
|
account.DriveId = sessionKey
|
|
}
|
|
_ = model.SaveAccount(account)
|
|
const DEFAULT uint64 = 10485760
|
|
var count = int64(math.Ceil(float64(file.GetSize()) / float64(DEFAULT)))
|
|
var finish uint64 = 0
|
|
parentFile, err := driver.File(file.ParentPath, account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !parentFile.IsDir() {
|
|
return base.ErrNotFolder
|
|
}
|
|
res, err := driver.UploadRequest("/person/initMultiUpload", map[string]string{
|
|
"parentFolderId": parentFile.Id,
|
|
"fileName": encode(file.Name),
|
|
"fileSize": strconv.FormatInt(int64(file.Size), 10),
|
|
"sliceSize": strconv.FormatInt(int64(DEFAULT), 10),
|
|
"lazyCheck": "1",
|
|
}, account, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
uploadFileId := jsoniter.Get(res, "data", "uploadFileId").ToString()
|
|
//_, err = driver.UploadRequest("/person/getUploadedPartsInfo", map[string]string{
|
|
// "uploadFileId": uploadFileId,
|
|
//}, account, nil)
|
|
var i int64
|
|
var byteSize uint64
|
|
md5s := make([]string, 0)
|
|
md5Sum := md5.New()
|
|
for i = 1; i <= count; i++ {
|
|
byteSize = file.GetSize() - finish
|
|
if DEFAULT < byteSize {
|
|
byteSize = DEFAULT
|
|
}
|
|
//log.Debugf("%d,%d", byteSize, finish)
|
|
byteData := make([]byte, byteSize)
|
|
n, err := io.ReadFull(file, byteData)
|
|
//log.Debug(err, n)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
finish += uint64(n)
|
|
md5Bytes := getMd5(byteData)
|
|
md5Hex := hex.EncodeToString(md5Bytes)
|
|
md5Base64 := base64.StdEncoding.EncodeToString(md5Bytes)
|
|
md5s = append(md5s, strings.ToUpper(md5Hex))
|
|
md5Sum.Write(byteData)
|
|
//log.Debugf("md5Bytes: %+v,md5Str:%s,md5Base64:%s", md5Bytes, md5Hex, md5Base64)
|
|
var resp UploadUrlsResp
|
|
res, err = driver.UploadRequest("/person/getMultiUploadUrls", map[string]string{
|
|
"partInfo": fmt.Sprintf("%s-%s", strconv.FormatInt(i, 10), md5Base64),
|
|
"uploadFileId": uploadFileId,
|
|
}, account, &resp)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
uploadData := resp.UploadUrls["partNumber_"+strconv.FormatInt(i, 10)]
|
|
log.Debugf("uploadData: %+v", uploadData)
|
|
requestURL := uploadData.RequestURL
|
|
uploadHeaders := strings.Split(decodeURIComponent(uploadData.RequestHeader), "&")
|
|
req, _ := http.NewRequest(http.MethodPut, requestURL, bytes.NewReader(byteData))
|
|
for _, v := range uploadHeaders {
|
|
i := strings.Index(v, "=")
|
|
req.Header.Set(v[0:i], v[i+1:])
|
|
}
|
|
|
|
r, err := base.HttpClient.Do(req)
|
|
log.Debugf("%+v %+v", r, r.Request.Header)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
fileMd5 := hex.EncodeToString(md5Sum.Sum(nil))
|
|
sliceMd5 := fileMd5
|
|
if file.GetSize() > DEFAULT {
|
|
sliceMd5 = utils.GetMD5Encode(strings.Join(md5s, "\n"))
|
|
}
|
|
res, err = driver.UploadRequest("/person/commitMultiUploadFile", map[string]string{
|
|
"uploadFileId": uploadFileId,
|
|
"fileMd5": fileMd5,
|
|
"sliceMd5": sliceMd5,
|
|
"lazyCheck": "1",
|
|
}, account, nil)
|
|
account.DriveId, _ = driver.GetSessionKey(account)
|
|
return err
|
|
}
|
|
|
|
func (driver Cloud189) OldUpload(file *model.FileStream, account *model.Account) error {
|
|
//return base.ErrNotImplement
|
|
client, err := driver.getClient(account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
parentFile, err := driver.File(file.ParentPath, account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// api refer to PanIndex
|
|
res, err := client.R().SetMultipartFormData(map[string]string{
|
|
"parentId": parentFile.Id,
|
|
"sessionKey": account.DriveId,
|
|
"opertype": "1",
|
|
"fname": file.GetFileName(),
|
|
}).SetMultipartField("Filedata", file.GetFileName(), file.GetMIMEType(), file).Post("https://hb02.upload.cloud.189.cn/v1/DCIWebUploadAction")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if jsoniter.Get(res.Body(), "MD5").ToString() != "" {
|
|
return nil
|
|
}
|
|
log.Debugf(res.String())
|
|
return errors.New(res.String())
|
|
}
|