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

package config

import (
	"fmt"
	"io"
	"os"
	"strings"

	"gopkg.in/yaml.v3"
)

// Config is a tmpl config
type Config struct {
	Prompts []Prompt `yaml:"prompts"`
}

// PromptType is a type of prompt
type PromptType string

const (
	PromptTypeInput     PromptType = "input"
	PromptTypeMultiline PromptType = "multi"
	PromptTypeEditor    PromptType = "editor"
	PromptTypeConfirm   PromptType = "confirm"
	PromptTypeSelect    PromptType = "select"
)

// Prompt is a tmpl prompt
type Prompt struct {
	ID      string     `yaml:"id"`
	Label   string     `yaml:"label"`
	Help    string     `yaml:"help"`
	Default string     `yaml:"default"`
	Options []string   `yaml:"options"`
	Type    PromptType `yaml:"type"`
}

// Load loads a tmpl config
func Load(r io.Reader) (*Config, error) {
	configBytes, err := io.ReadAll(r)
	if err != nil {
		return nil, err
	}

	configContents := os.Expand(string(configBytes), func(s string) string {
		if strings.HasPrefix(s, "TMPL_PROMPT") {
			return fmt.Sprintf("${%s}", s)
		}
		return os.Getenv(s)
	})

	var c Config
	return &c, yaml.Unmarshal([]byte(configContents), &c)
}