1
0
mirror of https://github.com/go-task/task.git synced 2025-01-22 05:10:17 +02:00

feat: add ability to call task with variable by reference

This commit is contained in:
Pete Davison 2023-12-29 03:49:12 +00:00
parent c655d90ab3
commit 1f3fca50b3
5 changed files with 37 additions and 20 deletions

View File

@ -67,6 +67,7 @@ func (c *Compiler) getVariables(t *ast.Task, call *ast.Call, evaluateShVars bool
newVar.Value = value
}
newVar.Sh = tr.Replace(v.Sh)
newVar.Ref = v.Ref
newVar.Json = tr.Replace(v.Json)
newVar.Yaml = tr.Replace(v.Yaml)
newVar.Dir = v.Dir

View File

@ -113,6 +113,9 @@ func (r *Templater) replaceVars(vars *ast.Vars, extra map[string]any) *ast.Vars
}
newVar.Live = v.Live
newVar.Sh = r.ReplaceWithExtra(v.Sh, extra)
newVar.Ref = v.Ref
newVar.Json = r.ReplaceWithExtra(v.Json, extra)
newVar.Yaml = r.ReplaceWithExtra(v.Yaml, extra)
newVars.Set(k, newVar)
return nil
})

View File

@ -76,6 +76,7 @@ type Var struct {
Value any
Live any
Sh string
Ref string
Json string
Yaml string
Dir string
@ -107,9 +108,10 @@ func (v *Var) UnmarshalYAML(node *yaml.Node) error {
case yaml.MappingNode:
key := node.Content[0].Value
switch key {
case "sh", "map", "json", "yaml":
case "sh", "ref", "map", "json", "yaml":
var m struct {
Sh string
Ref string
Map any
Json string
Yaml string
@ -118,12 +120,13 @@ func (v *Var) UnmarshalYAML(node *yaml.Node) error {
return err
}
v.Sh = m.Sh
v.Ref = m.Ref
v.Value = m.Map
v.Json = m.Json
v.Yaml = m.Yaml
return nil
default:
return fmt.Errorf(`yaml: line %d: %q is not a valid variable type. Try "sh", "map", "json", "yaml" or using a scalar value`, node.Line, key)
return fmt.Errorf(`yaml: line %d: %q is not a valid variable type. Try "sh", "ref", "map", "json", "yaml" or using a scalar value`, node.Line, key)
}
default:
var value any

View File

@ -11,14 +11,10 @@ tasks:
MAP:
map: {"name":"Alice","age":30,"children":[{"name":"Bob","age":5},{"name":"Charlie","age":3},{"name":"Diane","age":1}]}
cmds:
- >-
echo "{{.MAP.name}} has {{len .MAP.children}} children called
{{- $children := .MAP.children -}}
{{- range $i, $child := $children -}}
{{- if lt $i (sub (len $children) 1)}} {{$child.name -}},
{{- else}} and {{$child.name -}}
{{- end -}}
{{- end -}}"
- task: print-var
vars:
VAR:
ref: MAP
json:
vars:
@ -27,14 +23,10 @@ tasks:
JSON:
json: "{{.JSON_STRING}}"
cmds:
- >-
echo "{{.JSON.name}} has {{len .JSON.children}} children called
{{- $children := .JSON.children -}}
{{- range $i, $child := $children -}}
{{- if lt $i (sub (len $children) 1)}} {{$child.name -}},
{{- else}} and {{$child.name -}}
{{- end -}}
{{- end -}}"
- task: print-var
vars:
VAR:
ref: JSON
yaml:
vars:
@ -42,10 +34,17 @@ tasks:
sh: cat example.yaml
YAML:
yaml: "{{.YAML_STRING}}"
cmds:
- task: print-var
vars:
VAR:
ref: YAML
print-var:
cmds:
- >-
echo "{{.YAML.name}} has {{len .YAML.children}} children called
{{- $children := .YAML.children -}}
echo "{{.VAR.name}} has {{len .VAR.children}} children called
{{- $children := .VAR.children -}}
{{- range $i, $child := $children -}}
{{- if lt $i (sub (len $children) 1)}} {{$child.name -}},
{{- else}} and {{$child.name -}}

View File

@ -211,6 +211,17 @@ func (e *Executor) compiledTask(call ast.Call, evaluateShVars bool) (*ast.Task,
newCmd.Cmd = r.Replace(cmd.Cmd)
newCmd.Task = r.Replace(cmd.Task)
newCmd.Vars = r.ReplaceVars(cmd.Vars)
// Loop over the command's variables and resolve any references to other variables
err := cmd.Vars.Range(func(k string, v ast.Var) error {
if v.Ref != "" {
refVal := vars.Get(v.Ref)
newCmd.Vars.Set(k, refVal)
}
return nil
})
if err != nil {
return nil, err
}
new.Cmds = append(new.Cmds, newCmd)
}
}