main.go

package main

import (
  "fmt"
  "os"

  "github.com/fnordpipe/gorillua/modules/bindings/luabase64"
  "github.com/fnordpipe/gorillua/modules/bindings/luacron"
  "github.com/fnordpipe/gorillua/modules/bindings/luahttp"
  "github.com/fnordpipe/gorillua/modules/bindings/luajson"
  "github.com/fnordpipe/gorillua/modules/bindings/lualogger"
  "github.com/fnordpipe/gorillua/modules/bindings/luamariadb"
  "github.com/fnordpipe/gorillua/modules/bindings/luarequest"
  "github.com/fnordpipe/gorillua/modules/bindings/luasocket"
  "github.com/fnordpipe/gorillua/modules/bindings/luasrp"
  "github.com/fnordpipe/gorillua/modules/bindings/luauuid"
  "github.com/fnordpipe/gorillua/modules/logger"
  "github.com/yuin/gopher-lua"
)

var L *lua.LState

var _LUA_PATH string

func main() {
  if len(os.Args) < 2 {
    logger.Info(fmt.Sprintf("USAGE: %s <lua> [...]", os.Args[0]))
    os.Exit(1)
  }

  lua.LuaPathDefault = _LUA_PATH

  L := lua.NewState()
  defer L.Close()

  t := L.CreateTable(0, len(os.Args))
  for _, v := range os.Args {
    t.Append(lua.LString(v))
  }
  L.SetGlobal("arg", t)

  L.PreloadModule("base64", luabase64.Loader)
  L.PreloadModule("cron", luacron.Loader)
  L.PreloadModule("http", luahttp.Loader)
  L.PreloadModule("json", luajson.Loader)
  L.PreloadModule("logger", lualogger.Loader)
  L.PreloadModule("mariadb", luamariadb.Loader)
  L.PreloadModule("request", luarequest.Loader)
  L.PreloadModule("socket", luasocket.Loader)
  L.PreloadModule("srp", luasrp.Loader)
  L.PreloadModule("uuid", luauuid.Loader)

  if err := L.DoFile(os.Args[1]); err != nil {
    logger.Error("Cannot parse lua script")
    logger.Error(err.Error())
    os.Exit(2)
  }
}