59 lines
992 B
Go
59 lines
992 B
Go
|
package structure
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type PathTree struct {
|
||
|
|
Name string
|
||
|
|
IsDir bool
|
||
|
|
Children []*PathTree
|
||
|
|
}
|
||
|
|
|
||
|
|
func BuildTree(path string) (*PathTree, error) {
|
||
|
|
fileInfo, err := os.Stat(path)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
node := &PathTree{
|
||
|
|
Name: filepath.Base(path),
|
||
|
|
IsDir: fileInfo.IsDir(),
|
||
|
|
}
|
||
|
|
|
||
|
|
if !node.IsDir {
|
||
|
|
return node, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
files, err := os.ReadDir(path)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, file := range files {
|
||
|
|
if strings.HasPrefix(file.Name(), ".") {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
childPath := filepath.Join(path, file.Name())
|
||
|
|
childNode, err := BuildTree(childPath)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
node.Children = append(node.Children, childNode)
|
||
|
|
}
|
||
|
|
|
||
|
|
return node, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func PrintTree(node *PathTree, prefix string) {
|
||
|
|
fmt.Printf("%v* %s\n", prefix, node.Name)
|
||
|
|
newPrefix := prefix + " "
|
||
|
|
if len(node.Children) > 0 {
|
||
|
|
for _, child := range node.Children {
|
||
|
|
PrintTree(child, newPrefix+" ")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|