summaryrefslogtreecommitdiff
path: root/runner/python/indexer.go
blob: 3a61f7ebd6497b33ba1bfcc5af686084b468ece4 (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
package python

import (
	"github.com/pelletier/go-toml/v2"
	"io/fs"
	"os"
	"path/filepath"
	"pik/model"
)

type pyproj struct {
	Project struct {
		Scripts map[string]string
	}
}

func (p python) Index(path string, f fs.FS, runners []model.Runner) ([]model.Target, error) {
	for _, pt := range VenvPaths {
		if stat, err := fs.Stat(f, filepath.Join(pt)); err == nil {
			if stat.IsDir() {
				p.Venvs[path] = filepath.Join(path, pt)
			}
		}
	}
	content, err := fs.ReadFile(f, "pyproject.toml")
	if os.IsNotExist(err) {
		return nil, nil
	} else if err != nil {
		return nil, err
	}

	pp := &pyproj{}

	err = toml.Unmarshal(content, pp)
	if err != nil {
		return nil, err
	}

	var targets = make([]model.Target, 0, len(pp.Project.Scripts))
	for n, s := range pp.Project.Scripts {
		targets = append(targets, Python.CreateProjTarget(n, s))
	}
	return targets, nil
}