feat: Provide time.Parse and time.Now while evaluating trigger condition

Similar to argocd: - https://github.com/argoproj-labs/argocd-notifications/pull/22/files#diff-5b3e361e78e62137bc7f887d765e7c4bfbb85e01685b6640c41d74118d3e2f7dR11-R28 - https://argo-cd.readthedocs.io/en/stable/operator-manual/notifications/triggers/#functions Signed-off-by: mdsjip <[email protected]>
argoproj · Aug 21, 2022 · 585cead · 585cead
1 parent 69fd80c
commit 585cead
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/utils/record/record.go b/utils/record/record.go
@@ -230,7 +230,7 @@ func NewAPIFactorySettings() api.Settings {
 		ConfigMapName: NotificationConfigMap,
 		InitGetVars: func(cfg *api.Config, configMap *corev1.ConfigMap, secret *corev1.Secret) (api.GetVars, error) {
 			return func(obj map[string]interface{}, dest services.Destination) map[string]interface{} {
-				return map[string]interface{}{"rollout": obj}
+				return map[string]interface{}{"rollout": obj, "time": timeExprs}
 			}, nil
 		},
 	}
@@ -353,3 +353,20 @@ func translateReasonToTrigger(reason string) string {
 	trigger = matchAllCap.ReplaceAllString(trigger, "${1}-${2}")
 	return "on-" + strings.ToLower(trigger)
 }
+
+var timeExprs = map[string]interface{}{
+	"Parse": parse,
+	"Now":   now,
+}
+
+func parse(timestamp string) time.Time {
+	res, err := time.Parse(time.RFC3339, timestamp)
+	if err != nil {
+		panic(err)
+	}
+	return res
+}
+
+func now() time.Time {
+	return time.Now()
+}
diff --git a/utils/record/record_test.go b/utils/record/record_test.go
@@ -300,7 +300,7 @@ func TestNewAPIFactorySettings(t *testing.T) {
 	rollout := map[string]interface{}{"name": "hello"}
 	vars := getVars(rollout, services.Destination{})
 
-	assert.Equal(t, map[string]interface{}{"rollout": rollout}, vars)
+	assert.Equal(t, map[string]interface{}{"rollout": rollout, "time": timeExprs}, vars)
 }
 
 func TestWorkloadRefObjectMap(t *testing.T) {