mirror of
https://github.com/go-task/task.git
synced 2024-12-14 10:52:43 +02:00
130 lines
2.7 KiB
Go
130 lines
2.7 KiB
Go
package templater
|
|
|
|
import (
|
|
"bytes"
|
|
"strings"
|
|
"text/template"
|
|
|
|
"golang.org/x/exp/maps"
|
|
|
|
"github.com/go-task/task/v3/taskfile"
|
|
)
|
|
|
|
// Templater is a help struct that allow us to call "replaceX" funcs multiple
|
|
// times, without having to check for error each time. The first error that
|
|
// happen will be assigned to r.err, and consecutive calls to funcs will just
|
|
// return the zero value.
|
|
type Templater struct {
|
|
Vars *taskfile.Vars
|
|
RemoveNoValue bool
|
|
|
|
cacheMap map[string]any
|
|
err error
|
|
}
|
|
|
|
func (r *Templater) ResetCache() {
|
|
r.cacheMap = r.Vars.ToCacheMap()
|
|
}
|
|
|
|
func (r *Templater) Replace(str string) string {
|
|
return r.replace(str, nil)
|
|
}
|
|
|
|
func (r *Templater) ReplaceWithExtra(str string, extra map[string]any) string {
|
|
return r.replace(str, extra)
|
|
}
|
|
|
|
func (r *Templater) replace(str string, extra map[string]any) string {
|
|
if r.err != nil || str == "" {
|
|
return ""
|
|
}
|
|
|
|
templ, err := template.New("").Funcs(templateFuncs).Parse(str)
|
|
if err != nil {
|
|
r.err = err
|
|
return ""
|
|
}
|
|
|
|
if r.cacheMap == nil {
|
|
r.cacheMap = r.Vars.ToCacheMap()
|
|
}
|
|
|
|
var b bytes.Buffer
|
|
if extra == nil {
|
|
err = templ.Execute(&b, r.cacheMap)
|
|
} else {
|
|
// Copy the map to avoid modifying the cached map
|
|
m := maps.Clone(r.cacheMap)
|
|
maps.Copy(m, extra)
|
|
err = templ.Execute(&b, m)
|
|
}
|
|
if err != nil {
|
|
r.err = err
|
|
return ""
|
|
}
|
|
if r.RemoveNoValue {
|
|
return strings.ReplaceAll(b.String(), "<no value>", "")
|
|
}
|
|
return b.String()
|
|
}
|
|
|
|
func (r *Templater) ReplaceSlice(strs []string) []string {
|
|
if r.err != nil || len(strs) == 0 {
|
|
return nil
|
|
}
|
|
|
|
new := make([]string, len(strs))
|
|
for i, str := range strs {
|
|
new[i] = r.Replace(str)
|
|
}
|
|
return new
|
|
}
|
|
|
|
func (r *Templater) ReplaceGlobs(globs []*taskfile.Glob) []*taskfile.Glob {
|
|
if r.err != nil || len(globs) == 0 {
|
|
return nil
|
|
}
|
|
|
|
new := make([]*taskfile.Glob, len(globs))
|
|
for i, g := range globs {
|
|
new[i] = &taskfile.Glob{
|
|
Glob: r.Replace(g.Glob),
|
|
Negate: g.Negate,
|
|
}
|
|
}
|
|
return new
|
|
}
|
|
|
|
func (r *Templater) ReplaceVars(vars *taskfile.Vars) *taskfile.Vars {
|
|
return r.replaceVars(vars, nil)
|
|
}
|
|
|
|
func (r *Templater) ReplaceVarsWithExtra(vars *taskfile.Vars, extra map[string]any) *taskfile.Vars {
|
|
return r.replaceVars(vars, extra)
|
|
}
|
|
|
|
func (r *Templater) replaceVars(vars *taskfile.Vars, extra map[string]any) *taskfile.Vars {
|
|
if r.err != nil || vars.Len() == 0 {
|
|
return nil
|
|
}
|
|
|
|
var newVars taskfile.Vars
|
|
_ = vars.Range(func(k string, v taskfile.Var) error {
|
|
var newVar taskfile.Var
|
|
switch value := v.Value.(type) {
|
|
case string:
|
|
newVar.Value = r.ReplaceWithExtra(value, extra)
|
|
}
|
|
newVar.Live = v.Live
|
|
newVar.Sh = r.ReplaceWithExtra(v.Sh, extra)
|
|
newVars.Set(k, newVar)
|
|
return nil
|
|
})
|
|
|
|
return &newVars
|
|
}
|
|
|
|
func (r *Templater) Err() error {
|
|
return r.err
|
|
}
|