✨ Add repo and tag template funcs

clevyr · May 6, 2022 · f8b374e · f8b374e
1 parent 65f9402
commit f8b374e
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/pkg/template/funcs.go b/pkg/template/funcs.go
@@ -0,0 +1,13 @@
+package template
+
+import "strings"
+
+func DockerRepo(image string) string {
+	split := strings.SplitN(image, ":", 2)
+	return split[0]
+}
+
+func DockerTag(image string) string {
+	split := strings.SplitN(image, ":", 2)
+	return split[1]
+}
diff --git a/pkg/template/line_comment.go b/pkg/template/line_comment.go
@@ -8,6 +8,13 @@ import (
 	"text/template"
 )
 
+var funcMap = sprig.TxtFuncMap()
+
+func init() {
+	funcMap["repo"] = DockerRepo
+	funcMap["tag"] = DockerTag
+}
+
 func RecurseNode(conf config.Config, node *yaml.Node) error {
 	if len(node.Content) == 0 {
 		if err := TemplateLineComment(conf, node); err != nil {
@@ -26,7 +33,7 @@ func RecurseNode(conf config.Config, node *yaml.Node) error {
 func TemplateLineComment(conf config.Config, node *yaml.Node) error {
 	if node.LineComment != "" && strings.HasPrefix(node.LineComment, conf.Prefix) {
 		tmpl, err := template.New("").
-			Funcs(sprig.TxtFuncMap()).
+			Funcs(funcMap).
 			Delims(conf.LeftDelim, conf.RightDelim).
 			Option("missingkey=error").
 			Parse(strings.TrimSpace(node.LineComment[len(conf.Prefix):]))