Add Template Functions (#45)

fmenezes · Sep 1, 2020 · 71cef28 · 71cef28
1 parent 0e08516
commit 71cef28
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/cmd/codeownerslint/linter.go b/cmd/codeownerslint/linter.go
@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"io"
 	"path/filepath"
+	"strings"
 	"text/template"
 
 	"github.com/fmenezes/codeowners"
@@ -38,7 +39,10 @@ func run(wr io.Writer, opt options) exitCode {
 		format = opt.format
 	}
 	format = fmt.Sprintf("%s\n", format)
-	tpl, err := template.New("main").Parse(format)
+	tpl, err := template.New("main").Funcs(template.FuncMap{
+		"ToLower": strings.ToLower,
+		"ToUpper": strings.ToUpper,
+	}).Parse(format)
 	if err != nil {
 		fmt.Fprintf(wr, "Unexpected error when parsing format: %v", err)
 		return unexpectedErrorCode

diff --git a/cmd/codeownerslint/linter_test.go b/cmd/codeownerslint/linter_test.go
@@ -50,6 +50,14 @@ func TestCustomFormat(t *testing.T) {
 `)
 }
 
+func TestCustomFormatFunc(t *testing.T) {
+	assert(t, options{
+		directory: "../../test/data/noowners",
+		format:    "{{ToLower .Position.FilePath}}",
+	}, errorCode, `codeowners
+`)
+}
+
 func TestInvalidFormatParse(t *testing.T) {
 	assertCode(t, options{
 		directory: "../../test/data/noowners",