termfx/execute.go at main · jji05/termfx · GitHub

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
package termfx

import (
	"fmt"
	"io"
	"strings"

	"github.com/valyala/fasttemplate"
)

//ExecuteString will parse and run a termfx file/input and return the result as a string
func (r *Registry) ExecuteString(input string) (string, error) {

	t, err := fasttemplate.NewTemplate(input, r.split[0], r.split[1])
	if err != nil {
		return "", err
	}

	return t.ExecuteFuncStringWithErr(func(w io.Writer, tag string) (int, error) {

		var cmdArgs string
		el := strings.Split(tag, "(")
		for _, element := range el {
			cmdArgs = strings.Split(element, ")")[0]
		}

		if len(el) > 1 {
			el[0] += "()"
		}

		command, ok := r.commands[el[0]]
		if ok == false {
			return w.Write([]byte(fmt.Sprintf("[#Unknown tag %q#]", el[0])))
		}

		return command(w, cmdArgs)
	})

}

//Execute will parse and run a termfx file/input
func (r *Registry) Execute(input string, writer io.Writer) error {

	t, err := fasttemplate.NewTemplate(input, r.split[0], r.split[1])
	if err != nil {
		return err
	}

	_, err = t.ExecuteFunc(writer, func(w io.Writer, tag string) (int, error) {

		var cmdArgs string
		el := strings.Split(tag, "(")
		for _, element := range el {
			cmdArgs = strings.Split(element, ")")[0]
		}

		if len(el) > 1 {
			el[0] += "()"
		}

		command, ok := r.commands[el[0]]
		if ok == false {
			return w.Write([]byte(fmt.Sprintf("[#Unknown tag %q#]", el[0])))
		}

		return command(w, cmdArgs)
	})

	return err

}