🎇 多盘支持

alidrive/auth.go

@@ -2,77 +2,44 @@ package alidrive
 
 import (
 	"encoding/json"
-	"fmt"
 	"github.com/Xhofe/alist/conf"
 	"github.com/Xhofe/alist/utils"
 	log "github.com/sirupsen/logrus"
 )
 
-// use token login
-func TokenLogin() (*TokenLoginResp, error) {
-	log.Infof("尝试使用token登录...")
-	url := "https://auth.aliyundrive.com/v2/oauth/token_login"
-	req := TokenLoginReq{Token: conf.Conf.AliDrive.LoginToken}
-	log.Debugf("token_login_req:%+v", req)
-	var tokenLogin TokenLoginResp
-	if body, err := DoPost(url, req, false); err != nil {
-		log.Errorf("tokenLogin-doPost出错:%s", err.Error())
-		return nil, err
-	} else {
-		if err = json.Unmarshal(body, &tokenLogin); err != nil {
-			log.Errorf("解析json[%s]出错:%s", string(body), err.Error())
-			return nil, err
-		}
-	}
-	if tokenLogin.IsAvailable() {
-		return &tokenLogin, nil
-	}
-	return nil, fmt.Errorf("登录token失效,请更换:%s", tokenLogin.Message)
-}
-
-// get access token
-func GetToken(tokenLogin *TokenLoginResp) (*TokenResp, error) {
-	log.Infof("获取API token...")
-	url := "https://websv.aliyundrive.com/token/get"
-	code := utils.GetCode(tokenLogin.Goto)
-	if code == "" {
-		return nil, fmt.Errorf("获取code出错")
-	}
-	req := GetTokenReq{Code: code}
-	var token TokenResp
-	if body, err := DoPost(url, req, false); err != nil {
-		log.Errorf("tokenLogin-doPost出错:%s", err.Error())
-		return nil, err
-	} else {
-		if err = json.Unmarshal(body, &token); err != nil {
-			log.Errorf("解析json[%s]出错:%s", string(body), err.Error())
-			log.Errorf("此处json解析失败应该是code失效")
-			return nil, fmt.Errorf("code失效")
-		}
-	}
-	return &token, nil
-}
-
 // refresh access_token token by refresh_token
-func RefreshToken() bool {
-	log.Infof("刷新token...")
+func RefreshToken(drive *conf.Drive) bool {
+	log.Infof("刷新[%s]token...", drive.Name)
 	url := "https://websv.aliyundrive.com/token/refresh"
-	req := RefreshTokenReq{RefreshToken: conf.Conf.AliDrive.RefreshToken}
+	req := RefreshTokenReq{RefreshToken: drive.RefreshToken}
 	var token TokenResp
-	if body, err := DoPost(url, req, false); err != nil {
+	if body, err := DoPost(url, req, ""); err != nil {
 		log.Errorf("tokenLogin-doPost出错:%s", err.Error())
 		return false
 	} else {
 		if err = json.Unmarshal(body, &token); err != nil {
 			log.Errorf("解析json[%s]出错:%s", string(body), err.Error())
-			log.Errorf("此处json解析失败应该是refresh_token失效")
+			log.Errorf("此处json解析失败应该是[%s]refresh_token失效", drive.Name)
 			return false
 		}
 	}
-	//刷新成功 更新token并写入文件
-	conf.Conf.AliDrive.AccessToken = token.AccessToken
-	conf.Conf.AliDrive.RefreshToken = token.RefreshToken
-	conf.Authorization = token.TokenType + "\t" + token.AccessToken
-	utils.WriteToYml(conf.Con, conf.Conf)
+	//刷新成功 更新token
+	drive.AccessToken = token.AccessToken
+	drive.RefreshToken = token.RefreshToken
 	return true
 }
+
+func RefreshTokenAll() string {
+	log.Infof("刷新所有token...")
+	res := ""
+	for i, drive := range conf.Conf.AliDrive.Drives {
+		if !RefreshToken(&conf.Conf.AliDrive.Drives[i]) {
+			res = res + drive.Name + ","
+		}
+	}
+	utils.WriteToYml(conf.ConfigFile, conf.Conf)
+	if res != "" {
+		return res[:len(res)-1]
+	}
+	return ""
+}