add GetPaths method for node (#136)

JunNishimura · Jun 23, 2023 · a582d96 · a582d96
1 parent c263f29
commit a582d96
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/internal/object/tree.go b/internal/object/tree.go
@@ -17,6 +17,34 @@ type Node struct {
 	Children []*Node
 }
 
+func (n *Node) GetPaths() []string {
+	if len(n.Children) == 0 {
+		return []string{n.Name}
+	}
+
+	var paths []string
+	for _, child := range n.Children {
+		ps := child.getPaths(n.Name)
+		paths = append(paths, ps...)
+	}
+
+	return paths
+}
+
+func (n *Node) getPaths(parentDir string) []string {
+	if len(n.Children) == 0 {
+		return []string{fmt.Sprintf("%s/%s", parentDir, n.Name)}
+	}
+
+	var paths []string
+	for _, child := range n.Children {
+		ps := child.getPaths(fmt.Sprintf("%s/%s", parentDir, n.Name))
+		paths = append(paths, ps...)
+	}
+
+	return paths
+}
+
 type Tree struct {
 	object   *Object
 	Children []*Node