[core] Facilities for pushing runtime variables back up the stack

Author: teo

core/workflow/iteratorrole.go

@@ -254,3 +254,17 @@ func (i *iteratorRole) GetUserVars() gera.StringMap {
 	}
 	return i.template.GetUserVars()
 }
+
+func (i *iteratorRole) SetRuntimeVar(key string, value string) {
+	if i == nil {
+		return
+	}
+	i.template.SetRuntimeVar(key, value)
+}
+
+func (i *iteratorRole) SetRuntimeVars(kv map[string]string) {
+	if i == nil {
+		return
+	}
+	i.template.SetRuntimeVars(kv)
+}

core/workflow/role.go

@@ -55,6 +55,8 @@ type Role interface {
 	setParent(role Updatable)
 	ProcessTemplates(workflowRepo *repos.Repo) error
 	GlobFilter(g glob.Glob) []Role
+	SetRuntimeVar(key string, value string)
+	SetRuntimeVars(kv map[string]string)
 }
 
 type Updatable interface {

core/workflow/rolebase.go

@@ -62,6 +62,16 @@ type roleBase struct {
 	Locals     map[string]string        `yaml:"-"` // only used for passing iterator from template to new role
 }
 
+func (r *roleBase) SetRuntimeVar(key string, value string) {
+	r.UserVars.Set(key, value)
+}
+
+func (r *roleBase) SetRuntimeVars(kv map[string]string) {
+	for k, v := range kv {
+		r.UserVars.Set(k, v)
+	}
+}
+
 func (r *roleBase) ConsolidatedVarStack() (varStack map[string]string, err error) {
 	// This function is used in task.go to get the parent role's varStack
 	var defaults, vars, userVars map[string]string