func (ls *LState) Load(reader io.Reader, name string) (*LFunction, error) {
chunk, err := parse.Parse(reader, name)
if err != nil {
return nil, newApiErrorS(ApiErrorSyntax, err.Error())
}
proto, err := Compile(chunk, name)
if err != nil {
return nil, newApiErrorS(ApiErrorSyntax, err.Error())
}
return newLFunctionL(proto, ls.currentEnv(), 0), nil
}
func testScriptCompile(t *testing.T, script string) {
file, err := os.Open(script)
if err != nil {
t.Fatal(err)
return
}
chunk, err2 := parse.Parse(file, script)
if err2 != nil {
t.Fatal(err2)
return
}
parse.Dump(chunk)
proto, err3 := Compile(chunk, script)
if err3 != nil {
t.Fatal(err3)
return
}
proto.String()
}
func main() {
var opt_e, opt_l string
var opt_i, opt_v, opt_dt, opt_dc bool
var opt_m int
flag.StringVar(&opt_e, "e", "", "")
flag.StringVar(&opt_l, "l", "", "")
flag.IntVar(&opt_m, "mx", 0, "")
flag.BoolVar(&opt_i, "i", false, "")
flag.BoolVar(&opt_v, "v", false, "")
flag.BoolVar(&opt_dt, "dt", false, "")
flag.BoolVar(&opt_dc, "dc", false, "")
flag.Usage = func() {
fmt.Println(`usage: glua.exe [options] [script [args]].
Available options are:
-e stat execute string 'stat'
-l name require library 'name'
-mx MB memory limit(default: unlimited)
-dt dump AST trees
-dc dump VM codes
-i enter interactive mode after executing 'script'
-v show version information
`)
}
flag.Parse()
if len(opt_e) == 0 && !opt_i && !opt_v && flag.NArg() == 0 {
opt_i = true
}
status := 0
L := lua.NewState()
defer L.Close()
if opt_m > 0 {
L.SetMx(opt_m)
}
if opt_v || opt_i {
fmt.Println(lua.PackageCopyRight)
}
if len(opt_l) > 0 {
if err := L.DoFile(opt_l); err != nil {
fmt.Println(err.Error())
}
}
if nargs := flag.NArg(); nargs > 0 {
script := flag.Arg(0)
argtb := L.NewTable()
for i := 1; i < nargs; i++ {
L.RawSet(argtb, lua.LNumber(i), lua.LString(flag.Arg(i)))
}
L.SetGlobal("arg", argtb)
if opt_dt || opt_dc {
file, err := os.Open(script)
if err != nil {
fmt.Println(err.Error())
os.Exit(1)
}
chunk, err2 := parse.Parse(file, script)
if err2 != nil {
fmt.Println(err2.Error())
os.Exit(1)
}
if opt_dt {
fmt.Println(parse.Dump(chunk))
}
if opt_dc {
proto, err3 := lua.Compile(chunk, script)
if err3 != nil {
fmt.Println(err3.Error())
os.Exit(1)
}
fmt.Println(proto.String())
}
}
if err := L.DoFile(script); err != nil {
fmt.Println(err.Error())
status = 1
}
}
if len(opt_e) > 0 {
if err := L.DoString(opt_e); err != nil {
fmt.Println(err.Error())
status = 1
}
}
if opt_i {
reader := bufio.NewReader(os.Stdin)
for {
fmt.Print("> ")
if buf, err := reader.ReadString('\n'); err == io.EOF {
break
} else if err != nil {
} else {
if err := L.DoString(buf); err != nil {
fmt.Println(err.Error())
}
}
}
}
os.Exit(status)
}