synapsecns/sanguine

package tree

import (
    "fmt"
    "os"
    "path/filepath"
    "strings"
)

// AddDirectoryPaths is a helper function that adds all paths in a directory to the tree
// you can optionally pass in a prefix to trim from the paths.
func AddDirectoryPaths(tree Tree, dirPath, trimPrefix string) error {
    err := filepath.Walk(dirPath, func(path string, _ os.FileInfo, err error) error {
        if err != nil {
            return fmt.Errorf("error walking path %s: %w", path, err)
        }

        if trimPrefix != "" {
            path = path[len(trimPrefix):]
        }

        // skip git files
        if strings.Contains(path, ".git") {
            return nil
        }

        tree.Add(path)
        return nil
    })
    if err != nil {
        return fmt.Errorf("error adding directory paths: %w", err)
    }
    return nil
}