Add support for updating registered variables for the following tasks.

This update adds the function 'UpdateRegisteredVars' to update environment variables with values from a list of registered variables passed on by the caller. This ensures registered variables can be used across tasks are updated promptly.

cmd/spot/main.go

@@ -368,7 +368,8 @@ func runTaskForTarget(ctx context.Context, r *runner.Process, taskName, targetNa
 	}
 	log.Printf("[INFO] completed: hosts:%d, commands:%d in %v\n",
 		res.Hosts, res.Commands, time.Since(st).Truncate(100*time.Millisecond))
-	r.Playbook.UpdateTasksTargets(res.Vars) // for dynamic targets
+	r.Playbook.UpdateTasksTargets(res.Vars)         // for dynamic targets
+	r.Playbook.UpdateRegisteredVars(res.Registered) // for registered vars, cross-task
 	return nil
 }
 

pkg/config/command.go

@@ -11,6 +11,7 @@ import (
 	"strings"
 	"time"
 
+	"github.com/go-pkgz/stringutils"
 	"gopkg.in/yaml.v3"
 )
 
@@ -212,15 +213,21 @@ func (cmd *Cmd) scriptFile(inp string, register []string) (r io.Reader) {
 	}
 
 	// each exported variable is printed as a setvar command to be captured by the caller
+	exported := []string{} // collect all exported variables to avoid duplicates in setvar output
 	if len(exports) > 0 {
 		for _, v := range exports {
 			buf.WriteString(fmt.Sprintf("echo setvar %s=${%s}\n", v, v))
+			exported = append(exported, v)
 		}
 	}
 
 	// each register variable is printed as a setvar command to be captured by the caller
 	if len(register) > 0 {
 		for _, v := range register {
+			if stringutils.Contains(v, exported) {
+				// if already exported, we don't need to print it again
+				continue
+			}
 			buf.WriteString(fmt.Sprintf("echo setvar %s=${%s}\n", v, v))
 		}
 	}

pkg/config/playbook.go

@@ -437,6 +437,30 @@ func (p *PlayBook) UpdateTasksTargets(vars map[string]string) {
 	}
 }
 
+// UpdateRegisteredVars takes the list of registered vars from the caller handling task execution
+// and updates the environment variables of all commands in the playbook with the values from the list.
+func (p *PlayBook) UpdateRegisteredVars(vars map[string]string) {
+	if len(vars) == 0 {
+		return
+	}
+	log.Printf("[DEBUG] update registered vars %+v", vars)
+	for k, v := range vars {
+		for _, tsk := range p.Tasks {
+			for i, c := range tsk.Commands {
+				env := c.Environment
+				if env == nil {
+					env = make(map[string]string)
+				}
+				if _, ok := env[k]; ok { // don't allow override already set vars. TODO: not sure if this is correct
+					continue
+				}
+				env[k] = v
+				tsk.Commands[i].Environment = env
+			}
+		}
+	}
+}
+
 // loadInventory loads the inventory data from the specified location (file or URL) and returns it as an InventoryData struct.
 // The inventory data is parsed as either YAML or TOML, depending on the file extension.
 // The method also performs some additional processing on the inventory data:

pkg/runner/runner.go

@@ -53,6 +53,7 @@ type Playbook interface {
 	TargetHosts(name string) ([]config.Destination, error)
 	AllSecretValues() []string
 	UpdateTasksTargets(vars map[string]string)
+	UpdateRegisteredVars(vars map[string]string)
 }
 
 // ProcResp holds the information about processed commands and hosts.