-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
63 lines (49 loc) · 1.22 KB
/
main.go
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
package main
import (
"bytes"
"errors"
"flag"
"fmt"
"github.com/ethanfrogers/starlark-spinnaker/pkg/cache"
"github.com/ethanfrogers/starlark-spinnaker/pkg/writer"
"go.starlark.net/starlark"
"go.starlark.net/starlarkstruct"
)
func includedModules() starlark.StringDict {
return starlark.StringDict{
"struct": starlark.NewBuiltin("struct", starlarkstruct.Make),
}
}
func predeclaredModules() (starlark.StringDict) {
return starlark.StringDict{
"struct": starlark.NewBuiltin("struct", starlarkstruct.Make),
}
}
func main() {
mainConfig := flag.String("config", "main.sky", "entrypoint filename")
flag.Parse()
starlarkLoader := cache.NewLoader(
cache.WithModuleLoaders(&cache.LocalModuleLoader{}),
cache.WithPredeclaredModules(predeclaredModules()))
thread := starlark.Thread{
Name: "spinlark",
Load: starlarkLoader.Load,
}
globals, err := starlarkLoader.Load(&thread, *mainConfig)
if err != nil {
panic(err)
}
mainFunc, ok := globals["main"]
if !ok {
panic(errors.New("no main function found"))
}
v, err := starlark.Call(&thread, mainFunc, nil , nil)
if err != nil {
panic(err)
}
buf := new(bytes.Buffer)
if err := writer.ToJson(buf, v); err != nil {
panic(err)
}
fmt.Println(buf.String())
}