Merge pull request #55 from ivanilves/with

refactor: add WithNameEx() func to wrap arbitrary regex rule

.gitignore

@@ -18,3 +18,6 @@ vendor/
 /cmd/travelgrunt/*
 !/cmd/travelgrunt/*.go
 /dist/*
+
+# Local Makefile
+local.mk

Makefile

@@ -8,6 +8,8 @@ NEXT_VERSION  = v${API_VERSION}.${NEXT_PATCH}
 
 SHELL := /bin/bash
 
+-include local.mk
+
 default: dep build
 
 all: dep build test install

cmd/travelgrunt/main.go

@@ -6,7 +6,6 @@ import (
 	"os"
 
 	"github.com/ivanilves/travelgrunt/pkg/config"
-	"github.com/ivanilves/travelgrunt/pkg/config/rule"
 	"github.com/ivanilves/travelgrunt/pkg/directory"
 	"github.com/ivanilves/travelgrunt/pkg/directory/tree"
 	"github.com/ivanilves/travelgrunt/pkg/file"
@@ -117,9 +116,7 @@ func main() {
 	}
 
 	if expression != "" {
-		rules := []rule.Rule{{NameEx: expression}}
-
-		cfg.Rules = rules
+		cfg = cfg.WithNameEx(expression)
 	}
 
 	entries, paths, err := directory.Collect(rootPath, cfg)

pkg/config/with.go

@@ -0,0 +1,12 @@
+package config
+
+import (
+	"github.com/ivanilves/travelgrunt/pkg/config/rule"
+)
+
+// WithNameEx returns a copy of config with regex-based rule added
+func (cfg Config) WithNameEx(nameEx string) Config {
+	cfg.Rules = []rule.Rule{rule.Rule{NameEx: nameEx}}
+
+	return cfg
+}

pkg/config/with_test.go

@@ -0,0 +1,17 @@
+package config
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestWithNameEx(t *testing.T) {
+	assert := assert.New(t)
+
+	cfg := DefaultConfig()
+
+	cfg = cfg.WithNameEx("*.tf")
+
+	assert.Len(cfg.Rules, 1)
+}