-
Notifications
You must be signed in to change notification settings - Fork 467
/
rbwtemplatefuncs.go
74 lines (64 loc) · 1.8 KB
/
rbwtemplatefuncs.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
package cmd
import (
"encoding/json"
"os"
"os/exec"
"strings"
"github.com/coreos/go-semver/semver"
"github.com/twpayne/chezmoi/v2/internal/chezmoilog"
)
type rbwConfig struct {
Command string `json:"command" mapstructure:"command" yaml:"command"`
outputCache map[string][]byte
}
var rbwMinVersion = semver.Version{Major: 1, Minor: 7, Patch: 0}
func (c *Config) rbwFieldsTemplateFunc(name string, extraArgs ...string) map[string]any {
args := append([]string{"get", "--raw", name}, extraArgs...)
output, err := c.rbwOutput(args)
if err != nil {
panic(err)
}
var data struct {
Fields []map[string]any `json:"fields"`
}
if err := json.Unmarshal(output, &data); err != nil {
panic(newParseCmdOutputError(c.RBW.Command, args, output, err))
}
result := make(map[string]any)
for _, field := range data.Fields {
if name, ok := field["name"].(string); ok {
result[name] = field
}
}
return result
}
func (c *Config) rbwTemplateFunc(name string, extraArgs ...string) map[string]any {
args := append([]string{"get", "--raw", name}, extraArgs...)
output, err := c.rbwOutput(args)
if err != nil {
panic(err)
}
var data map[string]any
if err := json.Unmarshal(output, &data); err != nil {
panic(newParseCmdOutputError(c.RBW.Command, args, output, err))
}
return data
}
func (c *Config) rbwOutput(args []string) ([]byte, error) {
key := strings.Join(args, "\x00")
if data, ok := c.RBW.outputCache[key]; ok {
return data, nil
}
cmd := exec.Command(c.RBW.Command, args...) //nolint:gosec
cmd.Stdin = os.Stdin
cmd.Stderr = os.Stderr
output, err := chezmoilog.LogCmdOutput(c.logger, cmd)
if err != nil {
return nil, newCmdOutputError(cmd, output, err)
}
if c.RBW.outputCache == nil {
c.RBW.outputCache = make(map[string][]byte)
}
c.RBW.outputCache[key] = output
return output, nil
}