package shell import ( "fmt" "io" "strings" "github.com/seaweedfs/seaweedfs/weed/pb/filer_pb" "github.com/seaweedfs/seaweedfs/weed/util" ) func init() { Commands = append(Commands, &commandFsTree{}) } type commandFsTree struct { } func (c *commandFsTree) Name() string { return "fs.tree" } func (c *commandFsTree) Help() string { return `recursively list all files under a directory fs.tree /some/dir ` } func (c *commandFsTree) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) { path, err := commandEnv.parseUrl(findInputDirectory(args)) if err != nil { return err } dir, name := util.FullPath(path).DirAndName() dirCount, fCount, terr := treeTraverseDirectory(writer, commandEnv, util.FullPath(dir), name, newPrefix(), -1) if terr == nil { fmt.Fprintf(writer, "%d directories, %d files\n", dirCount, fCount) } return terr } func treeTraverseDirectory(writer io.Writer, filerClient filer_pb.FilerClient, dir util.FullPath, name string, prefix *Prefix, level int) (directoryCount, fileCount int64, err error) { prefix.addMarker(level) err = filer_pb.ReadDirAllEntries(filerClient, dir, name, func(entry *filer_pb.Entry, isLast bool) error { if level < 0 && name != "" { if entry.Name != name { return nil } } fmt.Fprintf(writer, "%s%s\n", prefix.getPrefix(level, isLast), entry.Name) if entry.IsDirectory { directoryCount++ subDir := dir.Child(entry.Name) dirCount, fCount, terr := treeTraverseDirectory(writer, filerClient, subDir, "", prefix, level+1) directoryCount += dirCount fileCount += fCount err = terr } else { fileCount++ } return nil }) return } type Prefix struct { markers map[int]bool } func newPrefix() *Prefix { return &Prefix{ markers: make(map[int]bool), } } func (p *Prefix) addMarker(marker int) { p.markers[marker] = true } func (p *Prefix) removeMarker(marker int) { delete(p.markers, marker) } func (p *Prefix) getPrefix(level int, isLastChild bool) string { var sb strings.Builder if level < 0 { return "" } for i := 0; i < level; i++ { if _, ok := p.markers[i]; ok { sb.WriteString("│") } else { sb.WriteString(" ") } sb.WriteString(" ") } if isLastChild { sb.WriteString("└──") p.removeMarker(level) } else { sb.WriteString("├──") } return sb.String() }