summaryrefslogtreecommitdiff
path: root/runner/just/just.go
blob: e57453619b4c989972b1e9105e62bfbc69e23996 (plain)
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package just

import (
	"io/fs"
	"os/exec"
	"pik/identity"
	"pik/model"
	"pik/runner"
	"strings"
)

type just struct {
	path string
}

var Indexer = &just{}

func (j *just) Index(path string, f fs.FS, runners []model.Runner) ([]model.Target, error) {

	entries, err := fs.ReadDir(f, ".")
	if err != nil {
		return nil, err
	}
	hasJustfile := false
	for _, e := range entries {
		if !e.IsDir() && strings.ToLower(e.Name()) == "justfile" {
			hasJustfile = true
			break
		}
	}

	if !hasJustfile {
		return nil, nil
	}

	err = j.findJust()
	if err != nil {
		return nil, err
	}

	cmd := exec.Command(j.path, "--list")
	cmd.Dir = path
	out, err := cmd.CombinedOutput()
	if err != nil {
		return nil, err
	}
	return ParseOutput(string(out)), nil
}

func ParseOutput(input string) []model.Target {
	categories := make(map[string][]string)
	currentCategory := ""
	for _, line := range strings.Split(input, "\n") {
		// strip comment
		line = strings.SplitN(line, "#", 2)[0]
		line = strings.TrimSpace(line)

		if strings.HasPrefix(line, "[") && strings.HasSuffix(line, "]") {
			currentCategory = line[1 : len(line)-1]
			continue
		}

		tgt := strings.SplitN(line, " ", 2)[0]

		if tgt == "" {
			continue
		}

		categories[currentCategory] = append(categories[currentCategory], tgt)
	}

	var result []model.Target
	for c, targets := range categories {
		for _, t := range targets {
			result = append(result, &JustTarget{
				BaseTarget: runner.BaseTarget{
					Identity: identity.New(t),
				},
				Category: c,
			})
		}
	}
	return result
}

func (j *just) findJust() error {
	loc, err := exec.LookPath("just")
	if err != nil {
		return err
	}
	j.path = loc
	return nil
}