aliyunpan/internal/command/recycle.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")
}