-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils.go
118 lines (83 loc) · 2.23 KB
/
utils.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
package plumber
import (
"bytes"
"fmt"
"os"
"strings"
"text/template"
sprig "github.com/go-task/slim-sprig/v3"
"github.com/urfave/cli/v2"
"golang.org/x/exp/slices"
)
func TemplateFuncMap() template.FuncMap {
// functions can be found here: https://go-task.github.io/slim-sprig/
return sprig.FuncMap()
}
func EditCliFlag[Flag any](flags []cli.Flag, fn func(f Flag) bool, apply func(f Flag) Flag) []cli.Flag {
clone := slices.Clone(flags)
OverwriteCliFlag(clone, fn, apply)
return clone
}
func OverwriteCliFlag[Flag any](flags []cli.Flag, fn func(f Flag) bool, apply func(f Flag) Flag) {
index := slices.IndexFunc(flags, func(flag cli.Flag) bool {
converted, ok := flag.(Flag)
if !ok {
return false
}
return fn(converted)
})
if index < 0 {
panic(fmt.Errorf("Flag can not be found to modify."))
}
applied := apply(flags[index].(Flag))
cast := (interface{})(applied)
modified, ok := cast.(cli.Flag)
if !ok {
panic(fmt.Errorf("Can not cast the type of the given flag."))
}
flags[index] = modified
}
func ParseEnvironmentVariablesToMap() map[string]string {
vars := map[string]string{}
for _, v := range os.Environ() {
pair := strings.SplitN(v, "=", 2)
key := pair[0]
value := pair[1]
vars[key] = value
}
return vars
}
func InlineTemplate[Ctx any](tmpl string, ctx Ctx, funcs ...template.FuncMap) (string, error) {
if tmpl == "" {
return "", nil
}
parser := template.New("inline").Funcs(TemplateFuncMap())
for _, f := range funcs {
parser.Funcs(f)
}
tmp, err := parser.Parse(tmpl)
if err != nil {
return "", fmt.Errorf("Can not create inline template: %w", err)
}
var w bytes.Buffer
err = tmp.ExecuteTemplate(&w, "inline", ctx)
if err != nil {
return "", fmt.Errorf("Can not generate inline template: %w", err)
}
return w.String(), nil
}
func InlineTemplates[Ctx any](tmpls []string, ctx Ctx, funcs ...template.FuncMap) ([]string, error) {
results := []string{}
errors := []string{}
for _, tmpl := range tmpls {
tpl, err := InlineTemplate[Ctx](tmpl, ctx, funcs...)
if err != nil {
errors = append(errors, err.Error())
}
results = append(results, tpl)
}
if len(errors) > 0 {
return results, fmt.Errorf("%s", strings.Join(errors, "\n"))
}
return results, nil
}