mirror of
https://github.com/tickstep/aliyunpan.git
synced 2025-01-23 22:42:15 +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)
|
||
}
|