80 lines
1.6 KiB
Go
80 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"gitea.arsenm.dev/Arsen6331/scpt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
)
|
|
|
|
func main() {
|
|
useStdin := flag.Bool("stdin", false, "Parse STDIN")
|
|
dumpAST := flag.Bool("dump-ast", false, "Dump the AST as JSON to STDOUT and quit")
|
|
loadAST := flag.String("load-ast", "", "Load JSON AST from specified file and execute it")
|
|
flag.Parse()
|
|
|
|
var ast *scpt.AST
|
|
|
|
if *loadAST != "" {
|
|
data, err := ioutil.ReadFile(*loadAST)
|
|
if err != nil {
|
|
log.Fatalln("Error opening specified file:", err)
|
|
}
|
|
ast, err = scpt.LoadAST(data)
|
|
if err != nil {
|
|
log.Fatalln("Error loading AST:", err)
|
|
}
|
|
} else if *useStdin {
|
|
var err error
|
|
ast, err = scpt.Parse(os.Stdin)
|
|
if err != nil {
|
|
log.Fatalln("Error parsing STDIN:", err)
|
|
}
|
|
} else {
|
|
if flag.NArg() < 1 {
|
|
log.Fatalln("Filepath or --stdin required")
|
|
}
|
|
filePath := flag.Args()[0]
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
log.Fatalln("Error opening specified file:", err)
|
|
}
|
|
ast, err = scpt.Parse(file)
|
|
if err != nil {
|
|
log.Fatalln("Error parsing file:", err)
|
|
}
|
|
}
|
|
|
|
if *dumpAST {
|
|
data, err := ast.DumpPretty()
|
|
if err != nil {
|
|
log.Fatalln("Error dumping AST:", err)
|
|
}
|
|
fmt.Println(string(data))
|
|
os.Exit(0)
|
|
}
|
|
|
|
scpt.AddFuncs(scpt.FuncMap{
|
|
"print": scptPrint,
|
|
"display-dialog": displayDialog,
|
|
"do-shell-script": doShellScript,
|
|
})
|
|
|
|
err := ast.Execute()
|
|
if err != nil {
|
|
log.Fatalln("Error executing script:", err)
|
|
}
|
|
}
|
|
|
|
func scptPrint(args map[string]interface{}) (interface{}, error) {
|
|
val, ok := args[""]
|
|
if !ok {
|
|
return nil, errors.New("print requires an unnamed argument")
|
|
}
|
|
fmt.Println(val)
|
|
return nil, nil
|
|
}
|