53 lines
947 B
Go
53 lines
947 B
Go
package data
|
|
|
|
import (
|
|
ui "github.com/gizak/termui/v3"
|
|
"github.com/sqshq/sampler/config"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
)
|
|
|
|
type Item struct {
|
|
Label string
|
|
SampleScript string
|
|
InitScript *string
|
|
TransformScript *string
|
|
Color *ui.Color
|
|
}
|
|
|
|
func NewItems(cfgs []config.Item) []Item {
|
|
|
|
items := make([]Item, 0)
|
|
|
|
for _, i := range cfgs {
|
|
item := Item{
|
|
Label: *i.Label,
|
|
SampleScript: *i.SampleScript,
|
|
InitScript: i.InitScript,
|
|
TransformScript: i.TransformScript,
|
|
Color: i.Color}
|
|
items = append(items, item)
|
|
}
|
|
|
|
return items
|
|
}
|
|
|
|
func (i *Item) nextValue(variables []string) (value string, err error) {
|
|
|
|
cmd := exec.Command("sh", "-c", i.SampleScript)
|
|
cmd.Env = os.Environ()
|
|
|
|
for _, variable := range variables {
|
|
cmd.Env = append(cmd.Env, variable)
|
|
}
|
|
|
|
output, err := cmd.Output()
|
|
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return strings.TrimSpace(string(output)), nil
|
|
}
|