69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
package main
|
|
|
|
//go:generate go generate github.com/containous/yaegi/interp
|
|
//go:generate go generate github.com/containous/yaegi/cmd/goexports
|
|
//go:generate go generate github.com/containous/yaegi/stdlib
|
|
//go:generate go generate github.com/containous/yaegi/stdlib/syscall
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/containous/yaegi/interp"
|
|
"github.com/containous/yaegi/stdlib"
|
|
)
|
|
|
|
func main() {
|
|
var interactive, astDot, cfgDot, noRun bool
|
|
flag.BoolVar(&astDot, "a", false, "display AST graph")
|
|
flag.BoolVar(&cfgDot, "c", false, "display CFG graph")
|
|
flag.BoolVar(&interactive, "i", false, "start an interactive REPL")
|
|
flag.BoolVar(&noRun, "n", false, "do not run")
|
|
flag.Usage = func() {
|
|
fmt.Println("Usage:", os.Args[0], "[options] [script] [args]")
|
|
fmt.Println("Options:")
|
|
flag.PrintDefaults()
|
|
}
|
|
flag.Parse()
|
|
args := flag.Args()
|
|
log.SetFlags(log.Lshortfile)
|
|
|
|
i := interp.New()
|
|
i.Use(stdlib.Value)
|
|
i.Use(interp.ExportValue)
|
|
if astDot {
|
|
interp.AstDot(i)
|
|
}
|
|
if cfgDot {
|
|
interp.CfgDot(i)
|
|
}
|
|
if noRun {
|
|
interp.NoRun(i)
|
|
}
|
|
|
|
if len(args) > 0 {
|
|
b, err := ioutil.ReadFile(args[0])
|
|
if err != nil {
|
|
log.Fatal("Could not read file: ", args[0])
|
|
}
|
|
s := string(b)
|
|
if s[:2] == "#!" {
|
|
// Allow executable go scripts, but fix them prior to parse
|
|
s = strings.Replace(s, "#!", "//", 1)
|
|
}
|
|
i.Name = args[0]
|
|
if _, err := i.Eval(s); err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
if interactive {
|
|
i.Repl(os.Stdin, os.Stdout)
|
|
}
|
|
} else {
|
|
i.Repl(os.Stdin, os.Stdout)
|
|
}
|
|
}
|