converted dir tree

server/sys/fs.go

@@ -2,7 +2,9 @@ package sys
 
 import (
 	"os"
+	"path/filepath"
 
+	"github.com/marcopeocchi/yt-dlp-web-ui/server/internal"
 	"golang.org/x/sys/unix"
 )
 
@@ -18,3 +20,42 @@ func FreeSpace() (uint64, error) {
 	unix.Statfs(wd+"/downloads", &stat)
 	return (stat.Bavail * uint64(stat.Bsize)), nil
 }
+
+func DirectoryTree(rootPath string) (*[]string, error) {
+	type Node struct {
+		path     string
+		children []Node
+	}
+
+	stack := internal.Stack[Node]{
+		Nodes: make([]*internal.Node[Node], 5),
+	}
+	flattened := make([]string, 0)
+
+	root := Node{path: rootPath}
+	stack.Push(&internal.Node[Node]{
+		Value: root,
+	})
+	flattened = append(flattened, rootPath)
+
+	for stack.IsNotEmpty() {
+		current := stack.Pop().Value
+		children, err := os.ReadDir(current.path)
+		if err != nil {
+			return nil, err
+		}
+		for _, entry := range children {
+			childPath := filepath.Join(current.path, entry.Name())
+			childNode := Node{path: childPath}
+
+			if entry.IsDir() {
+				current.children = append(current.children, childNode)
+				stack.Push(&internal.Node[Node]{
+					Value: childNode,
+				})
+				flattened = append(flattened, childNode.path)
+			}
+		}
+	}
+	return &flattened, nil
+}