134 lines
4.0 KiB
Go
134 lines
4.0 KiB
Go
package drivers
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/Xhofe/alist/conf"
|
|
"github.com/Xhofe/alist/model"
|
|
"github.com/go-resty/resty/v2"
|
|
"github.com/robfig/cron/v3"
|
|
log "github.com/sirupsen/logrus"
|
|
"time"
|
|
)
|
|
|
|
var aliClient = resty.New()
|
|
|
|
func init() {
|
|
aliClient.
|
|
SetRetryCount(3).
|
|
SetHeader("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").
|
|
SetHeader("content-type", "application/json").
|
|
SetHeader("origin", "https://aliyundrive.com")
|
|
}
|
|
|
|
type AliDrive struct {
|
|
}
|
|
|
|
type AliRespError struct {
|
|
Code string `json:"code"`
|
|
Message string `json:"message"`
|
|
}
|
|
|
|
type AliFile struct {
|
|
AliRespError
|
|
DriveId string `json:"drive_id"`
|
|
CreatedAt *time.Time `json:"created_at"`
|
|
DomainId string `json:"domain_id"`
|
|
EncryptMode string `json:"encrypt_mode"`
|
|
FileExtension string `json:"file_extension"`
|
|
FileId string `json:"file_id"`
|
|
Hidden bool `json:"hidden"`
|
|
Name string `json:"name"`
|
|
ParentFileId string `json:"parent_file_id"`
|
|
Starred bool `json:"starred"`
|
|
Status string `json:"status"`
|
|
Type string `json:"type"`
|
|
UpdatedAt *time.Time `json:"updated_at"`
|
|
Category string `json:"category"`
|
|
ContentHash string `json:"content_hash"`
|
|
ContentHashName string `json:"content_hash_name"`
|
|
ContentType string `json:"content_type"`
|
|
Crc64Hash string `json:"crc_64_hash"`
|
|
DownloadUrl string `json:"download_url"`
|
|
PunishFlag int64 `json:"punish_flag"`
|
|
Size int64 `json:"size"`
|
|
Thumbnail string `json:"thumbnail"`
|
|
Url string `json:"url"`
|
|
ImageMediaMetadata map[string]interface{} `json:"image_media_metadata"`
|
|
}
|
|
|
|
func (a AliDrive) Path(path string, account *model.Account) (*model.File, []*model.File, error) {
|
|
_, err := conf.Cache.Get(conf.Ctx, path)
|
|
if err == nil {
|
|
// return
|
|
}
|
|
|
|
panic("implement me")
|
|
}
|
|
|
|
func (a AliDrive) Link(path string, account *model.Account) (string, error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
type AliTokenResp struct {
|
|
AccessToken string `json:"access_token"`
|
|
RefreshToken string `json:"refresh_token"`
|
|
}
|
|
|
|
func AliRefreshToken(refresh string) (string, string, error) {
|
|
url := "https://auth.aliyundrive.com/v2/account/token"
|
|
var resp AliTokenResp
|
|
var e AliRespError
|
|
_, err := aliClient.R().
|
|
//ForceContentType("application/json").
|
|
SetBody(JsonStr(Json{"refresh_token": refresh, "grant_type": "refresh_token"})).
|
|
SetResult(&resp).
|
|
SetError(&e).
|
|
Post(url)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
log.Debugf("%+v,%+v", resp, e)
|
|
if e.Code != "" {
|
|
return "", "", fmt.Errorf("failed to refresh token: %s", e.Message)
|
|
}
|
|
return resp.RefreshToken, resp.AccessToken, nil
|
|
}
|
|
|
|
func (a AliDrive) Save(account *model.Account, old *model.Account) error {
|
|
if old != nil {
|
|
conf.Cron.Remove(cron.EntryID(old.CronId))
|
|
}
|
|
refresh, access, err := AliRefreshToken(account.RefreshToken)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
account.RefreshToken, account.AccessToken = refresh, access
|
|
cronId, err := conf.Cron.AddFunc("@every 2h", func() {
|
|
name := account.Name
|
|
newAccount, ok := model.GetAccount(name)
|
|
if !ok {
|
|
return
|
|
}
|
|
newAccount.RefreshToken, newAccount.AccessToken, err = AliRefreshToken(newAccount.RefreshToken)
|
|
if err != nil {
|
|
newAccount.Status = err.Error()
|
|
}
|
|
_ = model.SaveAccount(newAccount)
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
account.CronId = int(cronId)
|
|
err = model.SaveAccount(*account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
var _ Driver = (*AliDrive)(nil)
|
|
|
|
func init() {
|
|
RegisterDriver("AliDrive", &AliDrive{})
|
|
}
|