sampler-fork/data/item.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
}