mirror of
https://github.com/tickstep/aliyunpan.git
synced 2025-01-24 06:52:16 +08:00
100 lines
2.6 KiB
Go
100 lines
2.6 KiB
Go
|
// Copyright (c) 2020 tickstep.
|
|||
|
|
//
|
|||
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|||
|
|
// you may not use this file except in compliance with the License.
|
|||
|
|
// You may obtain a copy of the License at
|
|||
|
|
//
|
|||
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|||
|
|
//
|
|||
|
|
// Unless required by applicable law or agreed to in writing, software
|
|||
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|||
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|||
|
|
// See the License for the specific language governing permissions and
|
|||
|
|
// limitations under the License.
|
|||
|
|
package command
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/tickstep/aliyunpan-api/aliyunpan"
|
|||
|
|
"github.com/tickstep/aliyunpan/cmder"
|
|||
|
|
"github.com/tickstep/aliyunpan/internal/config"
|
|||
|
|
"github.com/urfave/cli"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func CmdToken() cli.Command {
|
|||
|
|
return cli.Command{
|
|||
|
|
Name: "token",
|
|||
|
|
Usage: "Token相关操作",
|
|||
|
|
UsageText: cmder.App().Name + " token",
|
|||
|
|
Category: "阿里云盘账号",
|
|||
|
|
Before: cmder.ReloadConfigFunc,
|
|||
|
|
Action: func(c *cli.Context) error {
|
|||
|
|
cli.ShowCommandHelp(c, c.Command.Name)
|
|||
|
|
return nil
|
|||
|
|
},
|
|||
|
|
Subcommands: []cli.Command{
|
|||
|
|
{
|
|||
|
|
Name: "update",
|
|||
|
|
Usage: "更新Token",
|
|||
|
|
UsageText: cmder.App().Name + " token update",
|
|||
|
|
Description: `
|
|||
|
|
示例:
|
|||
|
|
|
|||
|
|
更新当前登录用户的Token
|
|||
|
|
aliyunpan token update -mode 1
|
|||
|
|
|
|||
|
|
更新全部登录用户的Token
|
|||
|
|
aliyunpan token update -mode 2
|
|||
|
|
`,
|
|||
|
|
Action: func(c *cli.Context) error {
|
|||
|
|
if config.Config.ActiveUser() == nil {
|
|||
|
|
fmt.Println("未登录账号，无需刷新Token")
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
modeFlag := "1"
|
|||
|
|
if c.IsSet("mode") {
|
|||
|
|
modeFlag = c.String("mode")
|
|||
|
|
}
|
|||
|
|
RunTokenUpdate(modeFlag)
|
|||
|
|
return nil
|
|||
|
|
},
|
|||
|
|
Flags: []cli.Flag{
|
|||
|
|
cli.StringFlag{
|
|||
|
|
Name: "mode",
|
|||
|
|
Usage: "模式，1-登录用户，2-全部用户",
|
|||
|
|
Value: "1",
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RunTokenUpdate 执行Token更新
|
|||
|
|
func RunTokenUpdate(modeFlag string) {
|
|||
|
|
cmder.ReloadConfigFunc(nil)
|
|||
|
|
userList := config.Config.UserList
|
|||
|
|
if userList == nil || len(userList) == 0 {
|
|||
|
|
fmt.Printf("没有登录用户，无需刷新Token\n")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
for _,user := range userList {
|
|||
|
|
if modeFlag == "1" {
|
|||
|
|
if user.UserId != config.Config.ActiveUID {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
newToken,e := aliyunpan.GetAccessTokenFromRefreshToken(user.RefreshToken)
|
|||
|
|
if e != nil {
|
|||
|
|
fmt.Printf("无法为%s用户获取新的RefreshToken，可能需要重新登录\n", user.Nickname)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
if newToken != nil && newToken.RefreshToken != "" {
|
|||
|
|
user.RefreshToken = newToken.RefreshToken
|
|||
|
|
fmt.Printf("成功刷新%s用户的RefreshToken\n", user.Nickname)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
cmder.SaveConfigFunc(nil)
|
|||
|
|
}
|