From 3e1b87f5f275d5466f3ba877a807a691b525a1be Mon Sep 17 00:00:00 2001
From: xiaoyaofenfen <1254525673@qq.com>
Date: Fri, 19 Aug 2022 17:27:54 +0800
Subject: [PATCH] add tree command

---
 internal/command/tree.go | 114 +++++++++++++++++++++++++++++++++++++++
 main.go                  |   5 +-
 2 files changed, 118 insertions(+), 1 deletion(-)
 create mode 100644 internal/command/tree.go

diff --git a/internal/command/tree.go b/internal/command/tree.go
new file mode 100644
index 0000000..a127684
--- /dev/null
+++ b/internal/command/tree.go
@@ -0,0 +1,114 @@
+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"
+	"path"
+	"strings"
+)
+
+func CmdTree() cli.Command {
+	return cli.Command{
+		Name:      "tree",
+		Usage:     "列出目录的树形图",
+		UsageText: cmder.App().Name + " tree <目录>",
+		Description: `
+	列出指定目录内的文件和目录, 并以树形图的方式呈现
+
+	示例:
+
+	列出 当前工作目录 内的文件和目录的树形图
+	aliyunpan tree
+
+	列出 /我的资源 内的文件和目录的树形图
+	aliyunpan tree /我的资源
+
+`,
+		Category: "阿里云盘",
+		Before:   cmder.ReloadConfigFunc,
+		Action: func(c *cli.Context) error {
+			if config.Config.ActiveUser() == nil {
+				fmt.Println("未登录账号")
+				return nil
+			}
+			RunTree(parseDriveId(c), c.Args().Get(0))
+			return nil
+		},
+		Flags: []cli.Flag{
+			cli.StringFlag{
+				Name:  "driveId",
+				Usage: "网盘ID",
+				Value: "",
+			},
+		},
+	}
+}
+
+const (
+	indentPrefix   = "│   "
+	pathPrefix     = "├──"
+	lastFilePrefix = "└──"
+)
+
+func getTree(driveId, pathStr string, depth int) {
+	activeUser := config.Config.ActiveUser()
+	pathStr = activeUser.PathJoin(driveId, pathStr)
+	pathStr = path.Clean(pathStr)
+
+	targetPathInfo, err1 := activeUser.PanClient().FileInfoByPath(driveId, pathStr)
+	if err1 != nil {
+		fmt.Println(err1)
+		return
+	}
+	fileList := aliyunpan.FileList{}
+	fileListParam := &aliyunpan.FileListParam{}
+	fileListParam.ParentFileId = targetPathInfo.FileId
+	fileListParam.DriveId = driveId
+	fileListParam.OrderBy = aliyunpan.FileOrderByName
+	fileListParam.OrderDirection = aliyunpan.FileOrderDirectionAsc
+	if targetPathInfo.IsFolder() {
+		fileResult, err := activeUser.PanClient().FileListGetAll(fileListParam, 0)
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+		fileList = append(fileList, fileResult...)
+	} else {
+		fileList = append(fileList, targetPathInfo)
+	}
+
+	var (
+		prefix          = pathPrefix
+		fN              = len(fileList)
+		indentPrefixStr = strings.Repeat(indentPrefix, depth)
+	)
+	for i, file := range fileList {
+		if file.IsFolder() {
+			fmt.Printf("%v%v %v/\n", indentPrefixStr, pathPrefix, file.FileName)
+			getTree(driveId, pathStr+"/"+file.Path, depth+1)
+			continue
+		}
+
+		if i+1 == fN {
+			prefix = lastFilePrefix
+		}
+
+		fmt.Printf("%v%v %v\n", indentPrefixStr, prefix, file.FileName)
+	}
+
+	return
+}
+
+// RunTree 列出树形图
+func RunTree(driveId, pathStr string) {
+	activeUser := config.Config.ActiveUser()
+	activeUser.PanClient().ClearCache()
+	activeUser.PanClient().EnableCache()
+	defer activeUser.PanClient().DisableCache()
+	pathStr = activeUser.PathJoin(driveId, pathStr)
+	fmt.Printf("%s\n", pathStr)
+	getTree(driveId, pathStr, 0)
+}
diff --git a/main.go b/main.go
index 956bae8..4710831 100644
--- a/main.go
+++ b/main.go
@@ -173,7 +173,7 @@ func main() {
 				acceptCompleteFileCommands = []string{
 					"cd", "cp", "xcp", "download", "ls", "mkdir", "mv", "pwd", "rename", "rm", "share", "upload", "login", "loglist", "logout",
 					"clear", "quit", "exit", "quota", "who", "sign", "update", "who", "su", "config",
-					"drive", "export", "import", "sync",
+					"drive", "export", "import", "sync", "tree",
 				}
 				closed = strings.LastIndex(line, " ") == len(line)-1
 			)
@@ -416,6 +416,9 @@ func main() {
 		// 列出目录 ls
 		command.CmdLs(),
 
+		// 显示树形目录 tree
+		command.CmdTree(),
+
 		// 创建目录 mkdir
 		command.CmdMkdir(),