refactor(GH-33): pass the whole config.Config instead of just func

cmd/travelgrunt/main.go

@@ -95,7 +95,7 @@ func main() {
 		log.Fatalf("failed to load travelgrunt config: %s", err.Error())
 	}
 
-	entries, paths, err := directory.Collect(rootPath, cfg.IncludeFn)
+	entries, paths, err := directory.Collect(rootPath, cfg)
 
 	if err != nil {
 		log.Fatalf("failed to collect Terragrunt project directories: %s", err.Error())

pkg/directory/directory.go

@@ -4,14 +4,16 @@ import (
 	"os"
 	"path/filepath"
 	"strings"
+
+	"github.com/ivanilves/travelgrunt/pkg/config"
 )
 
 func isHidden(d os.DirEntry) bool {
 	return d.IsDir() && string(d.Name()[0]) == "."
 }
 
 // Collect gets a list of directory path entries containing file "terragrunt.hcl"
-func Collect(rootPath string, includeFn func(os.DirEntry) bool) (entries map[string]string, paths []string, err error) {
+func Collect(rootPath string, cfg config.Config) (entries map[string]string, paths []string, err error) {
 	entries = make(map[string]string, 0)
 	paths = make([]string, 0)
 
@@ -25,7 +27,7 @@ func Collect(rootPath string, includeFn func(os.DirEntry) bool) (entries map[str
 				return filepath.SkipDir
 			}
 
-			if includeFn(d) {
+			if cfg.IncludeFn(d) {
 				abs := filepath.Dir(path)
 				rel := strings.TrimPrefix(abs, rootPath+"/")
 

pkg/directory/directory_test.go

@@ -24,7 +24,7 @@ func TestCollect(t *testing.T) {
 	}
 
 	for path, expectedSuccess := range testCases {
-		entries, paths, err := Collect(path, cfg.IncludeFn)
+		entries, paths, err := Collect(path, cfg)
 
 		if expectedSuccess {
 			assert.Greater(len(entries), 0)

pkg/directory/tree/tree_test.go

@@ -20,7 +20,7 @@ var (
 func mockTree() Tree {
 	cfg := config.DefaultConfig()
 
-	_, paths, _ := directory.Collect(fixturePath, cfg.IncludeFn)
+	_, paths, _ := directory.Collect(fixturePath, cfg)
 
 	return NewTree(paths)
 }