feat: add match function for templating

TektonOps · Jul 8, 2021 · 2f22ba6 · 2f22ba6
1 parent f8609dc
commit 2f22ba6
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/cmd/dir/dir.go b/cmd/dir/dir.go
@@ -47,7 +47,7 @@ func Render() *cli.Command {
 
 			for _, v := range paths {
 				funcMap := sprig.TxtFuncMap()
-				t := template.Must(template.New(filepath.Base(v)).Funcs(funcMap).ParseFiles(v))
+				t := template.Must(template.New(filepath.Base(v)).Funcs(funcMap).Funcs(helpers.MatchFunc()).ParseFiles(v))
 
 				if len(helpers.GlobalOpts.Prefix) != 0 {
 					helpers.EnvVariables = helpers.MatchPrefix(helpers.GlobalOpts.Prefix)

diff --git a/cmd/helpers/helpers.go b/cmd/helpers/helpers.go
@@ -33,7 +33,7 @@ func GetVars() (enVars map[string]string) {
 // ParseString will parse any input provided as string
 func ParseString(str string) (*template.Template, error) {
 	funcMap := sprig.TxtFuncMap()
-	return template.Must(template.New("").Funcs(funcMap).Parse(str)), nil
+	return template.Must(template.New("").Funcs(funcMap).Funcs(MatchFunc()).Parse(str)), nil
 }
 
 // MatchPrefix will match a given prefix pattern of all env variables and render only those.
@@ -77,3 +77,11 @@ func CreateDirIfNotExist(path string) error {
 	}
 	return nil
 }
+
+// MatchFunc returns a custom functions map
+func MatchFunc() template.FuncMap {
+	var functionMap = map[string]interface{}{
+		"match": MatchPrefix,
+	}
+	return functionMap
+}