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)
+}