mirror of https://github.com/Xhofe/alist
100 lines
2.6 KiB
Go
100 lines
2.6 KiB
Go
|
package aliyundrive_share
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/alist-org/alist/v3/drivers/base"
|
||
|
|
"github.com/alist-org/alist/v3/internal/op"
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (d *AliyundriveShare) refreshToken() error {
|
||
|
|
url := "https://auth.aliyundrive.com/v2/account/token"
|
||
|
|
var resp base.TokenResp
|
||
|
|
var e ErrorResp
|
||
|
|
_, err := base.RestyClient.R().
|
||
|
|
SetBody(base.Json{"refresh_token": d.RefreshToken, "grant_type": "refresh_token"}).
|
||
|
|
SetResult(&resp).
|
||
|
|
SetError(&e).
|
||
|
|
Post(url)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if e.Code != "" {
|
||
|
|
return fmt.Errorf("failed to refresh token: %s", e.Message)
|
||
|
|
}
|
||
|
|
d.RefreshToken, d.AccessToken = resp.RefreshToken, resp.AccessToken
|
||
|
|
op.MustSaveDriverStorage(d)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// do others that not defined in Driver interface
|
||
|
|
func (d *AliyundriveShare) getShareToken() error {
|
||
|
|
data := base.Json{
|
||
|
|
"share_id": d.ShareId,
|
||
|
|
}
|
||
|
|
if d.SharePwd != "" {
|
||
|
|
data["share_pwd"] = d.SharePwd
|
||
|
|
}
|
||
|
|
var e ErrorResp
|
||
|
|
var resp ShareTokenResp
|
||
|
|
_, err := base.RestyClient.R().
|
||
|
|
SetResult(&resp).SetError(&e).SetBody(data).
|
||
|
|
Post("https://api.aliyundrive.com/v2/share_link/get_share_token")
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if e.Code != "" {
|
||
|
|
return errors.New(e.Message)
|
||
|
|
}
|
||
|
|
d.ShareToken = resp.ShareToken
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *AliyundriveShare) getFiles(fileId string) ([]File, error) {
|
||
|
|
files := make([]File, 0)
|
||
|
|
data := base.Json{
|
||
|
|
"image_thumbnail_process": "image/resize,w_160/format,jpeg",
|
||
|
|
"image_url_process": "image/resize,w_1920/format,jpeg",
|
||
|
|
"limit": 100,
|
||
|
|
"order_by": d.OrderBy,
|
||
|
|
"order_direction": d.OrderDirection,
|
||
|
|
"parent_file_id": fileId,
|
||
|
|
"share_id": d.ShareId,
|
||
|
|
"video_thumbnail_process": "video/snapshot,t_1000,f_jpg,ar_auto,w_300",
|
||
|
|
"marker": "first",
|
||
|
|
}
|
||
|
|
for data["marker"] != "" {
|
||
|
|
if data["marker"] == "first" {
|
||
|
|
data["marker"] = ""
|
||
|
|
}
|
||
|
|
var e ErrorResp
|
||
|
|
var resp ListResp
|
||
|
|
res, err := base.RestyClient.R().
|
||
|
|
SetHeader("x-share-token", d.ShareToken).
|
||
|
|
SetResult(&resp).SetError(&e).SetBody(data).
|
||
|
|
Post("https://api.aliyundrive.com/adrive/v3/file/list")
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
log.Debugf("aliyundrive share get files: %s", res.String())
|
||
|
|
if e.Code != "" {
|
||
|
|
if e.Code == "AccessTokenInvalid" {
|
||
|
|
err = d.getShareToken()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return d.getFiles(fileId)
|
||
|
|
}
|
||
|
|
return nil, errors.New(e.Message)
|
||
|
|
}
|
||
|
|
data["marker"] = resp.NextMarker
|
||
|
|
files = append(files, resp.Items...)
|
||
|
|
}
|
||
|
|
if len(files) > 0 && d.DriveId == "" {
|
||
|
|
d.DriveId = files[0].DriveId
|
||
|
|
}
|
||
|
|
return files, nil
|
||
|
|
}
|