lua parser

server/go.mod

@@ -9,5 +9,6 @@ require (
 	github.com/patrickmn/go-cache v2.1.0+incompatible
 	github.com/sirupsen/logrus v1.3.0
 	github.com/stretchr/testify v1.2.2
+	github.com/yuin/gopher-lua v0.0.0-20190206043414-8bfc7677f583
 	golang.org/x/image v0.0.0-20190118043309-183bebdce1b2 // indirect
 )

server/go.sum

@@ -1,3 +1,6 @@
+github.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=
+github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=
+github.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1/go.mod h1:Q3SI9o4m/ZMnBNeIyt5eFwwo7qiLfzFZmjNmxjkiQlU=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/disintegration/imaging v1.5.0 h1:uYqUhwNmLU4K1FN44vhqS4TZJRAA4RhBINgbQlKyGi0=
@@ -21,9 +24,12 @@ github.com/sirupsen/logrus v1.3.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPx
 github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/testify v1.2.2 h1:bSDNvY7ZPG5RlJ8otE/7V6gMiyenm9RtJ7IUVIAoJ1w=
 github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
+github.com/yuin/gopher-lua v0.0.0-20190206043414-8bfc7677f583 h1:SZPG5w7Qxq7bMcMVl6e3Ht2X7f+AAGQdzjkbyOnNNZ8=
+github.com/yuin/gopher-lua v0.0.0-20190206043414-8bfc7677f583/go.mod h1:gqRgreBUhTSL0GeU64rtZ3Uq3wtjOa/TB2YfrtkCbVQ=
 golang.org/x/crypto v0.0.0-20180904163835-0709b304e793 h1:u+LnwYTOOW7Ukr/fppxEb1Nwz0AtPflrblfvUudpo+I=
 golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
 golang.org/x/image v0.0.0-20190118043309-183bebdce1b2 h1:FNSSV4jv1PrPsiM2iKGpqLPPgYACqh9Muav7Pollk1k=
 golang.org/x/image v0.0.0-20190118043309-183bebdce1b2/go.mod h1:ux5Hcp/YLpHSI86hEcLt0YII63i6oz57MZXIpbrjZUs=
 golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33 h1:I6FyU15t786LL7oL/hn43zqTuEGr4PN7F4XJ1p4E3Y8=
 golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20190204203706-41f3e6584952/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=

server/luaparser/luaparser.go

@@ -0,0 +1,58 @@
+package luaparser
+
+import (
+  "github.com/yuin/gopher-lua"
+  "errors"
+  "strconv"
+)
+
+func New() *LuaParser {
+  p := LuaParser{
+    state: lua.NewState(lua.Options{SkipOpenLibs: true}),
+  }
+
+  return &p
+}
+
+type LuaParser struct {
+  state *lua.LState
+}
+
+func (this *LuaParser) ParseMap(expr string) (map[string]interface{}, error) {
+  result := make(map[string]interface{})
+
+  err := this.state.DoString(expr)
+  if err != nil {
+    return result, err
+  }
+
+  lv := this.state.Get(-1)
+
+  tbl, ok := lv.(*lua.LTable)
+  if !ok {
+    return result, errors.New("parsing failed")
+  }
+
+  tbl.ForEach(func(k, v lua.LValue){
+    key, ok := k.(lua.LString)
+
+    if !ok {
+      return
+    }
+
+    boolValue, ok := v.(lua.LBool)
+    if ok {
+      result[key.String()] = boolValue == lua.LTrue
+    }
+    intValue, ok := v.(lua.LNumber)
+    if ok {
+      result[key.String()], _ = strconv.Atoi(intValue.String())
+    }
+    strValue, ok := v.(lua.LString)
+    if ok {
+      result[key.String()] = strValue.String()
+    }
+  })
+
+  return result, nil
+}

server/luaparser/luaparser_test.go

@@ -0,0 +1,30 @@
+package luaparser
+
+import (
+  "testing"
+  "fmt"
+)
+
+func TestParse(t *testing.T){
+  p := New()
+  m, err := p.ParseMap(`return {a=1, b=true, c="abc"}`)
+
+  if err != nil {
+    panic(err)
+  }
+
+  fmt.Println(m)
+
+  if m["a"].(int) != 1 {
+    t.Fatal("parsing error")
+  }
+
+  if !m["b"].(bool) {
+    t.Fatal("parsing error")
+  }
+
+  if m["c"].(string) != "abc" {
+    t.Fatal("parsing error")
+  }
+
+}