86 lines
2.3 KiB
Go
86 lines
2.3 KiB
Go
package alidrive
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/Xhofe/alist/conf"
|
|
log "github.com/sirupsen/logrus"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// convert body to json
|
|
func BodyToJson(url string, req interface{}, resp RespHandle, drive *conf.Drive) error {
|
|
if body, err := DoPost(url, req, drive.AccessToken); err != nil {
|
|
log.Errorf("doPost出错:%s", err.Error())
|
|
return err
|
|
} else {
|
|
if err = json.Unmarshal(body, &resp); err != nil {
|
|
log.Errorf("解析json[%s]出错:%s", string(body), err.Error())
|
|
return err
|
|
}
|
|
}
|
|
if resp.IsAvailable() {
|
|
return nil
|
|
}
|
|
if resp.GetCode() == conf.AccessTokenInvalid {
|
|
resp.SetCode("")
|
|
if RefreshToken(drive) {
|
|
return BodyToJson(url, req, resp, drive)
|
|
}
|
|
}
|
|
return fmt.Errorf(resp.GetMessage())
|
|
}
|
|
|
|
// do post request
|
|
func DoPost(url string, request interface{}, auth string) (body []byte, err error) {
|
|
var (
|
|
resp *http.Response
|
|
)
|
|
requestBody := new(bytes.Buffer)
|
|
err = json.NewEncoder(requestBody).Encode(request)
|
|
if err != nil {
|
|
log.Errorf("创建requestBody出错:%s", err.Error())
|
|
}
|
|
req, err := http.NewRequest("POST", url, requestBody)
|
|
log.Debugf("do_post_req:%+v", req)
|
|
if err != nil {
|
|
log.Errorf("创建request出错:%s", err.Error())
|
|
return
|
|
}
|
|
if auth != "" {
|
|
req.Header.Set("authorization", conf.Bearer+auth)
|
|
}
|
|
req.Header.Add("content-type", "application/json")
|
|
req.Header.Add("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")
|
|
req.Header.Add("origin", "https://aliyundrive.com")
|
|
req.Header.Add("accept", "*/*")
|
|
req.Header.Add("Accept-Language", "zh-CN,zh;q=0.8,en-US;q=0.5,en;q=0.3")
|
|
req.Header.Add("Connection", "keep-alive")
|
|
|
|
for retryCount := 3; retryCount >= 0; retryCount-- {
|
|
resp, err = conf.Client.Do(req)
|
|
if err != nil && strings.Contains(err.Error(), "timeout") {
|
|
<-time.After(time.Second)
|
|
} else {
|
|
if body, err = ioutil.ReadAll(resp.Body); err != nil {
|
|
log.Errorf("读取api返回内容失败")
|
|
}
|
|
if string(body) != "" {
|
|
break
|
|
}
|
|
log.Errorf("返回为空,1s后重新请求")
|
|
<-time.After(time.Second)
|
|
}
|
|
}
|
|
if err != nil {
|
|
log.Errorf("请求阿里云盘api时出错:%s", err.Error())
|
|
return
|
|
}
|
|
log.Debugf("请求返回信息:%s", string(body))
|
|
return
|
|
}
|