seaweedfs/weed/shell/command_fs_tree.go

114 lines
2.3 KiB
Go
Raw Normal View History

2019-03-28 07:05:04 +00:00
package shell
import (
"fmt"
"io"
"strings"
2019-12-13 08:22:37 +00:00
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
2020-03-23 07:01:34 +00:00
"github.com/chrislusf/seaweedfs/weed/util"
2019-03-28 07:05:04 +00:00
)
func init() {
Commands = append(Commands, &commandFsTree{})
2019-03-28 07:05:04 +00:00
}
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
2019-03-28 07:05:04 +00:00
`
}
func (c *commandFsTree) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) {
2019-03-28 07:05:04 +00:00
2020-03-24 04:26:15 +00:00
path, err := commandEnv.parseUrl(findInputDirectory(args))
2019-03-28 07:05:04 +00:00
if err != nil {
return err
}
2020-03-23 07:01:34 +00:00
dir, name := util.FullPath(path).DirAndName()
2019-03-28 07:05:04 +00:00
2020-03-24 04:26:15 +00:00
dirCount, fCount, terr := treeTraverseDirectory(writer, commandEnv, util.FullPath(dir), name, newPrefix(), -1)
2019-03-28 07:05:04 +00:00
2019-12-13 08:22:37 +00:00
if terr == nil {
fmt.Fprintf(writer, "%d directories, %d files\n", dirCount, fCount)
}
2019-03-28 07:05:04 +00:00
2019-12-13 08:22:37 +00:00
return terr
2019-03-28 07:05:04 +00:00
}
2020-03-23 07:01:34 +00:00
func treeTraverseDirectory(writer io.Writer, filerClient filer_pb.FilerClient, dir util.FullPath, name string, prefix *Prefix, level int) (directoryCount, fileCount int64, err error) {
2019-03-28 07:05:04 +00:00
2019-12-13 08:22:37 +00:00
prefix.addMarker(level)
2019-04-05 18:40:54 +00:00
err = filer_pb.ReadDirAllEntries(filerClient, dir, name, func(entry *filer_pb.Entry, isLast bool) error {
2019-12-13 08:22:37 +00:00
if level < 0 && name != "" {
if entry.Name != name {
return nil
2019-04-05 18:40:54 +00:00
}
2019-12-13 08:22:37 +00:00
}
2019-04-05 18:40:54 +00:00
2019-12-13 08:22:37 +00:00
fmt.Fprintf(writer, "%s%s\n", prefix.getPrefix(level, isLast), entry.Name)
2019-03-28 07:05:04 +00:00
2019-12-13 08:22:37 +00:00
if entry.IsDirectory {
directoryCount++
2020-01-20 07:59:46 +00:00
subDir := dir.Child(entry.Name)
2020-02-26 06:23:59 +00:00
dirCount, fCount, terr := treeTraverseDirectory(writer, filerClient, subDir, "", prefix, level+1)
2019-12-13 08:22:37 +00:00
directoryCount += dirCount
fileCount += fCount
err = terr
} else {
fileCount++
2019-03-28 07:05:04 +00:00
}
return nil
2019-12-13 08:22:37 +00:00
})
2019-03-28 07:05:04 +00:00
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
2019-04-05 18:40:54 +00:00
if level < 0 {
return ""
}
2019-03-28 07:05:04 +00:00
for i := 0; i < level; i++ {
if _, ok := p.markers[i]; ok {
sb.WriteString("│")
} else {
sb.WriteString(" ")
}
sb.WriteString(" ")
}
if isLastChild {
sb.WriteString("└──")
2019-03-31 06:09:16 +00:00
p.removeMarker(level)
2019-03-28 07:05:04 +00:00
} else {
sb.WriteString("├──")
}
return sb.String()
}