From 32c477da5f2725adc53035cc32ace6031bb7a511 Mon Sep 17 00:00:00 2001
From: fabriziopandini <fpandini@vmware.com>
Date: Mon, 8 Apr 2024 14:54:51 +0200
Subject: [PATCH] Add more templating func to prowjob-gen

---
 hack/tools/prowjob-gen/generator.go | 42 ++++++++++++++++++++++++++++-
 1 file changed, 41 insertions(+), 1 deletion(-)

diff --git a/hack/tools/prowjob-gen/generator.go b/hack/tools/prowjob-gen/generator.go
index b597c20570a0..f923b903fede 100644
--- a/hack/tools/prowjob-gen/generator.go
+++ b/hack/tools/prowjob-gen/generator.go
@@ -175,7 +175,11 @@ func (g *generator) templateFunctions() template.FuncMap {
 	funcs["TrimPrefix"] = strings.TrimPrefix
 	funcs["TrimSuffix"] = strings.TrimSuffix
 	funcs["ReplaceAll"] = strings.ReplaceAll
+	// function cloned from sprig
 	funcs["last"] = last
+	funcs["list"] = list
+	funcs["has"] = has
+	funcs["trim"] = strings.TrimSpace
 	return funcs
 }
 
@@ -192,6 +196,42 @@ func last(list any) any {
 
 		return l2.Index(l - 1).Interface()
 	default:
-		panic(fmt.Sprintf("Cannot find last on type %s", tp))
+		panic(fmt.Sprintf("cannot find last on type %s", tp))
+	}
+}
+
+func list(v ...interface{}) []interface{} {
+	return v
+}
+
+func has(needle interface{}, haystack interface{}) bool {
+	l, err := mustHas(needle, haystack)
+	if err != nil {
+		panic(err)
+	}
+
+	return l
+}
+
+func mustHas(needle interface{}, haystack interface{}) (bool, error) {
+	if haystack == nil {
+		return false, nil
+	}
+	tp := reflect.TypeOf(haystack).Kind()
+	switch tp {
+	case reflect.Slice, reflect.Array:
+		l2 := reflect.ValueOf(haystack)
+		var item interface{}
+		l := l2.Len()
+		for i := 0; i < l; i++ {
+			item = l2.Index(i).Interface()
+			if reflect.DeepEqual(needle, item) {
+				return true, nil
+			}
+		}
+
+		return false, nil
+	default:
+		return false, fmt.Errorf("cannot find has on type %s", tp)
 	}
 }