mirror of
https://github.com/tickstep/aliyunpan.git
synced 2025-01-23 22:42:15 +08:00
275 lines
7.7 KiB
Go
275 lines
7.7 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/olekukonko/tablewriter"
|
||
"github.com/tickstep/aliyunpan-api/aliyunpan"
|
||
"github.com/tickstep/aliyunpan-api/aliyunpan_web"
|
||
"github.com/tickstep/aliyunpan/cmder"
|
||
"github.com/tickstep/aliyunpan/cmder/cmdtable"
|
||
"github.com/tickstep/aliyunpan/internal/config"
|
||
"github.com/tickstep/library-go/converter"
|
||
"github.com/tickstep/library-go/logger"
|
||
"github.com/urfave/cli"
|
||
"os"
|
||
"strconv"
|
||
"time"
|
||
)
|
||
|
||
func CmdRecycle() cli.Command {
|
||
return cli.Command{
|
||
Name: "recycle",
|
||
Usage: "回收站",
|
||
Description: `
|
||
回收站操作.
|
||
|
||
示例:
|
||
|
||
1. 从回收站还原两个文件, 其中的两个文件的 file_id 分别为 1013792297798440 和 643596340463870
|
||
aliyunpan recycle restore 1013792297798440 643596340463870
|
||
|
||
2. 从回收站删除两个文件, 其中的两个文件的 file_id 分别为 1013792297798440 和 643596340463870
|
||
aliyunpan recycle delete 1013792297798440 643596340463870
|
||
|
||
3. 清空回收站, 程序不会进行二次确认, 谨慎操作!!!
|
||
aliyunpan recycle delete -all
|
||
`,
|
||
Category: "阿里云盘",
|
||
Before: ReloadConfigFunc,
|
||
Action: func(c *cli.Context) error {
|
||
if c.NumFlags() <= 0 || c.NArg() <= 0 {
|
||
cli.ShowCommandHelp(c, c.Command.Name)
|
||
}
|
||
return nil
|
||
},
|
||
Subcommands: []cli.Command{
|
||
{
|
||
Name: "list",
|
||
Aliases: []string{"ls", "l"},
|
||
Usage: "列出回收站文件列表",
|
||
UsageText: cmder.App().Name + " recycle list",
|
||
Action: func(c *cli.Context) error {
|
||
if config.Config.ActiveUser() == nil {
|
||
fmt.Println("未登录账号")
|
||
return nil
|
||
}
|
||
if config.Config.ActiveUser().PanClient().WebapiPanClient() == nil {
|
||
fmt.Println("WEB客户端未登录，请登录后再使用该命令")
|
||
return nil
|
||
}
|
||
RunRecycleList(parseDriveId(c))
|
||
return nil
|
||
},
|
||
Flags: []cli.Flag{
|
||
cli.StringFlag{
|
||
Name: "driveId",
|
||
Usage: "网盘ID",
|
||
Value: "",
|
||
},
|
||
},
|
||
},
|
||
{
|
||
Name: "restore",
|
||
Aliases: []string{"r"},
|
||
Usage: "还原回收站文件或目录",
|
||
UsageText: cmder.App().Name + " recycle restore <file_id 1> <file_id 2> <file_id 3> ...",
|
||
Description: `根据文件/目录的 fs_id, 还原回收站指定的文件或目录`,
|
||
Action: func(c *cli.Context) error {
|
||
if config.Config.ActiveUser() == nil {
|
||
fmt.Println("未登录账号")
|
||
return nil
|
||
}
|
||
if config.Config.ActiveUser().PanClient().WebapiPanClient() == nil {
|
||
fmt.Println("WEB客户端未登录，请登录后再使用该命令")
|
||
return nil
|
||
}
|
||
if c.NArg() <= 0 {
|
||
cli.ShowCommandHelp(c, c.Command.Name)
|
||
return nil
|
||
}
|
||
RunRecycleRestore(parseDriveId(c), c.Args()...)
|
||
return nil
|
||
},
|
||
Flags: []cli.Flag{
|
||
cli.StringFlag{
|
||
Name: "driveId",
|
||
Usage: "网盘ID",
|
||
Value: "",
|
||
},
|
||
},
|
||
},
|
||
{
|
||
Name: "delete",
|
||
Aliases: []string{"d"},
|
||
Usage: "删除回收站文件或目录 / 清空回收站",
|
||
UsageText: cmder.App().Name + " recycle delete [-all] <file_id 1> <file_id 2> <file_id 3> ...",
|
||
Description: `根据文件/目录的 file_id 或 -all 参数, 删除回收站指定的文件或目录或清空回收站`,
|
||
Action: func(c *cli.Context) error {
|
||
if config.Config.ActiveUser() == nil {
|
||
fmt.Println("未登录账号")
|
||
return nil
|
||
}
|
||
if config.Config.ActiveUser().PanClient().WebapiPanClient() == nil {
|
||
fmt.Println("WEB客户端未登录，请登录后再使用该命令")
|
||
return nil
|
||
}
|
||
if c.Bool("all") {
|
||
// 清空回收站
|
||
RunRecycleClear(parseDriveId(c))
|
||
return nil
|
||
}
|
||
|
||
if c.NArg() <= 0 {
|
||
cli.ShowCommandHelp(c, c.Command.Name)
|
||
return nil
|
||
}
|
||
RunRecycleDelete(parseDriveId(c), c.Args()...)
|
||
return nil
|
||
},
|
||
Flags: []cli.Flag{
|
||
cli.BoolFlag{
|
||
Name: "all",
|
||
Usage: "清空回收站, 程序不会进行二次确认, 谨慎操作!!!",
|
||
},
|
||
cli.StringFlag{
|
||
Name: "driveId",
|
||
Usage: "网盘ID",
|
||
Value: "",
|
||
},
|
||
},
|
||
},
|
||
},
|
||
}
|
||
}
|
||
|
||
// RunRecycleList 执行列出回收站文件列表
|
||
func RunRecycleList(driveId string) {
|
||
panClient := GetActivePanClient()
|
||
fdl, err := panClient.WebapiPanClient().RecycleBinFileListGetAll(&aliyunpan_web.RecycleBinFileListParam{
|
||
DriveId: driveId,
|
||
Limit: 100,
|
||
})
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
return
|
||
}
|
||
|
||
tb := cmdtable.NewTable(os.Stdout)
|
||
tb.SetHeader([]string{"#", "file_id", "文件/目录名", "文件大小", "创建日期", "修改日期"})
|
||
tb.SetColumnAlignment([]int{tablewriter.ALIGN_DEFAULT, tablewriter.ALIGN_RIGHT, tablewriter.ALIGN_RIGHT, tablewriter.ALIGN_LEFT, tablewriter.ALIGN_LEFT})
|
||
for k, file := range fdl {
|
||
fn := file.FileName
|
||
fs := converter.ConvertFileSize(file.FileSize, 2)
|
||
if file.IsFolder() {
|
||
fn = fn + "/"
|
||
fs = "-"
|
||
}
|
||
tb.Append([]string{strconv.Itoa(k + 1), file.FileId, fn, fs, file.CreatedAt, file.UpdatedAt})
|
||
}
|
||
|
||
tb.Render()
|
||
}
|
||
|
||
// RunRecycleRestore 执行还原回收站文件或目录
|
||
func RunRecycleRestore(driveId string, fidStrList ...string) {
|
||
panClient := GetActivePanClient()
|
||
restoreFileList := []*aliyunpan.FileBatchActionParam{}
|
||
|
||
for _, fid := range fidStrList {
|
||
restoreFileList = append(restoreFileList, &aliyunpan.FileBatchActionParam{
|
||
DriveId: driveId,
|
||
FileId: fid,
|
||
})
|
||
}
|
||
|
||
if len(restoreFileList) == 0 {
|
||
fmt.Printf("没有需要还原的文件")
|
||
return
|
||
}
|
||
|
||
rbfr, err := panClient.WebapiPanClient().RecycleBinFileRestore(restoreFileList)
|
||
if rbfr != nil && len(rbfr) > 0 {
|
||
fmt.Printf("还原文件成功\n")
|
||
return
|
||
}
|
||
|
||
if len(rbfr) == 0 && err != nil {
|
||
fmt.Printf("还原文件失败：%s\n", err)
|
||
return
|
||
}
|
||
}
|
||
|
||
// RunRecycleDelete 执行删除回收站文件或目录
|
||
func RunRecycleDelete(driveId string, fidStrList ...string) {
|
||
panClient := GetActivePanClient()
|
||
deleteFileList := []*aliyunpan.FileBatchActionParam{}
|
||
|
||
for _, fid := range fidStrList {
|
||
deleteFileList = append(deleteFileList, &aliyunpan.FileBatchActionParam{
|
||
DriveId: driveId,
|
||
FileId: fid,
|
||
})
|
||
}
|
||
|
||
if len(deleteFileList) == 0 {
|
||
fmt.Printf("没有需要删除的文件")
|
||
return
|
||
}
|
||
|
||
rbfr, err := panClient.WebapiPanClient().RecycleBinFileDelete(deleteFileList)
|
||
if rbfr != nil && len(rbfr) > 0 {
|
||
fmt.Printf("彻底删除文件成功\n")
|
||
return
|
||
}
|
||
|
||
if len(rbfr) == 0 && err != nil {
|
||
fmt.Printf("彻底删除文件失败：%s\n", err)
|
||
return
|
||
}
|
||
}
|
||
|
||
// RunRecycleClear 清空回收站
|
||
func RunRecycleClear(driveId string) {
|
||
panClient := GetActivePanClient()
|
||
|
||
// 提交清空回收站异步任务
|
||
r, err := panClient.WebapiPanClient().RecycleBinFileClear(&aliyunpan_web.RecycleBinFileClearParam{
|
||
DriveId: driveId,
|
||
})
|
||
if err != nil {
|
||
logger.Verboseln(err)
|
||
fmt.Printf("当前无法清空回收站，请稍后重试\n")
|
||
}
|
||
|
||
for i := 0; i < 10; i++ {
|
||
ar, err1 := panClient.WebapiPanClient().AsyncTaskQueryStatus(&aliyunpan_web.AsyncTaskQueryStatusParam{
|
||
AsyncTaskId: r.AsyncTaskId,
|
||
})
|
||
if err1 != nil {
|
||
logger.Verboseln(err1)
|
||
time.Sleep(2 * time.Second)
|
||
continue
|
||
}
|
||
if ar.Status == "Succeed" {
|
||
fmt.Printf("清空回收站成功\n")
|
||
return
|
||
} else {
|
||
time.Sleep(1 * time.Second)
|
||
}
|
||
}
|
||
fmt.Printf("清空回收站失败，请稍后重试\n")
|
||
}
|