Extract Lua plugin logic into separate file
This commit is contained in:
Simon Ser
parent
a9a607b9b2
commit
95da47f7c4
2 changed files with 158 additions and 151 deletions
151
plugin.go
151
plugin.go
|
|
@ -1,13 +1,9 @@
|
||||||
package koushin
|
package koushin
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"html/template"
|
"html/template"
|
||||||
"path/filepath"
|
|
||||||
|
|
||||||
"github.com/labstack/echo/v4"
|
"github.com/labstack/echo/v4"
|
||||||
"github.com/yuin/gopher-lua"
|
|
||||||
"layeh.com/gopher-luar"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type Plugin interface {
|
type Plugin interface {
|
||||||
|
|
@ -17,150 +13,3 @@ type Plugin interface {
|
||||||
Render(name string, data interface{}) error
|
Render(name string, data interface{}) error
|
||||||
Close() error
|
Close() error
|
||||||
}
|
}
|
||||||
|
|
||||||
type luaRoute struct {
|
|
||||||
method string
|
|
||||||
path string
|
|
||||||
f *lua.LFunction
|
|
||||||
}
|
|
||||||
|
|
||||||
type luaPlugin struct {
|
|
||||||
filename string
|
|
||||||
state *lua.LState
|
|
||||||
renderCallbacks map[string]*lua.LFunction
|
|
||||||
filters template.FuncMap
|
|
||||||
routes []luaRoute
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *luaPlugin) Name() string {
|
|
||||||
return p.filename
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *luaPlugin) onRender(l *lua.LState) int {
|
|
||||||
name := l.CheckString(1)
|
|
||||||
f := l.CheckFunction(2)
|
|
||||||
p.renderCallbacks[name] = f
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *luaPlugin) setFilter(l *lua.LState) int {
|
|
||||||
name := l.CheckString(1)
|
|
||||||
f := l.CheckFunction(2)
|
|
||||||
p.filters[name] = func(args ...interface{}) string {
|
|
||||||
luaArgs := make([]lua.LValue, len(args))
|
|
||||||
for i, v := range args {
|
|
||||||
luaArgs[i] = luar.New(l, v)
|
|
||||||
}
|
|
||||||
|
|
||||||
err := l.CallByParam(lua.P{
|
|
||||||
Fn: f,
|
|
||||||
NRet: 1,
|
|
||||||
Protect: true,
|
|
||||||
}, luaArgs...)
|
|
||||||
if err != nil {
|
|
||||||
panic(err) // TODO: better error handling?
|
|
||||||
}
|
|
||||||
|
|
||||||
ret := l.CheckString(-1)
|
|
||||||
l.Pop(1)
|
|
||||||
return ret
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *luaPlugin) setRoute(l *lua.LState) int {
|
|
||||||
method := l.CheckString(1)
|
|
||||||
path := l.CheckString(2)
|
|
||||||
f := l.CheckFunction(3)
|
|
||||||
p.routes = append(p.routes, luaRoute{method, path, f})
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *luaPlugin) Render(name string, data interface{}) error {
|
|
||||||
f, ok := p.renderCallbacks[name]
|
|
||||||
if !ok {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
err := p.state.CallByParam(lua.P{
|
|
||||||
Fn: f,
|
|
||||||
NRet: 0,
|
|
||||||
Protect: true,
|
|
||||||
}, luar.New(p.state, data))
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *luaPlugin) Filters() template.FuncMap {
|
|
||||||
return p.filters
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *luaPlugin) SetRoutes(group *echo.Group) {
|
|
||||||
for _, r := range p.routes {
|
|
||||||
group.Match([]string{r.method}, r.path, func(ctx echo.Context) error {
|
|
||||||
err := p.state.CallByParam(lua.P{
|
|
||||||
Fn: r.f,
|
|
||||||
NRet: 0,
|
|
||||||
Protect: true,
|
|
||||||
}, luar.New(p.state, ctx))
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("Lua plugin error: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *luaPlugin) Close() error {
|
|
||||||
p.state.Close()
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func loadLuaPlugin(filename string) (*luaPlugin, error) {
|
|
||||||
l := lua.NewState()
|
|
||||||
p := &luaPlugin{
|
|
||||||
filename: filename,
|
|
||||||
state: l,
|
|
||||||
renderCallbacks: make(map[string]*lua.LFunction),
|
|
||||||
filters: make(template.FuncMap),
|
|
||||||
}
|
|
||||||
|
|
||||||
mt := l.NewTypeMetatable("koushin")
|
|
||||||
l.SetGlobal("koushin", mt)
|
|
||||||
l.SetField(mt, "on_render", l.NewFunction(p.onRender))
|
|
||||||
l.SetField(mt, "set_filter", l.NewFunction(p.setFilter))
|
|
||||||
l.SetField(mt, "set_route", l.NewFunction(p.setRoute))
|
|
||||||
|
|
||||||
if err := l.DoFile(filename); err != nil {
|
|
||||||
l.Close()
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return p, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func loadAllLuaPlugins(log echo.Logger) ([]Plugin, error) {
|
|
||||||
filenames, err := filepath.Glob("plugins/*.lua")
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("filepath.Glob failed: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
plugins := make([]Plugin, 0, len(filenames))
|
|
||||||
for _, filename := range filenames {
|
|
||||||
log.Printf("Loading Lua plugin '%v'", filename)
|
|
||||||
p, err := loadLuaPlugin(filename)
|
|
||||||
if err != nil {
|
|
||||||
for _, p := range plugins {
|
|
||||||
p.Close()
|
|
||||||
}
|
|
||||||
return nil, fmt.Errorf("failed to load Lua plugin '%v': %v", filename, err)
|
|
||||||
}
|
|
||||||
plugins = append(plugins, p)
|
|
||||||
}
|
|
||||||
|
|
||||||
return plugins, nil
|
|
||||||
}
|
|
||||||
|
|
|
||||||
158
plugin_lua.go
Normal file
158
plugin_lua.go
Normal file
|
|
@ -0,0 +1,158 @@
|
||||||
|
package koushin
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"html/template"
|
||||||
|
"path/filepath"
|
||||||
|
|
||||||
|
"github.com/labstack/echo/v4"
|
||||||
|
"github.com/yuin/gopher-lua"
|
||||||
|
"layeh.com/gopher-luar"
|
||||||
|
)
|
||||||
|
|
||||||
|
type luaRoute struct {
|
||||||
|
method string
|
||||||
|
path string
|
||||||
|
f *lua.LFunction
|
||||||
|
}
|
||||||
|
|
||||||
|
type luaPlugin struct {
|
||||||
|
filename string
|
||||||
|
state *lua.LState
|
||||||
|
renderCallbacks map[string]*lua.LFunction
|
||||||
|
filters template.FuncMap
|
||||||
|
routes []luaRoute
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *luaPlugin) Name() string {
|
||||||
|
return p.filename
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *luaPlugin) onRender(l *lua.LState) int {
|
||||||
|
name := l.CheckString(1)
|
||||||
|
f := l.CheckFunction(2)
|
||||||
|
p.renderCallbacks[name] = f
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *luaPlugin) setFilter(l *lua.LState) int {
|
||||||
|
name := l.CheckString(1)
|
||||||
|
f := l.CheckFunction(2)
|
||||||
|
p.filters[name] = func(args ...interface{}) string {
|
||||||
|
luaArgs := make([]lua.LValue, len(args))
|
||||||
|
for i, v := range args {
|
||||||
|
luaArgs[i] = luar.New(l, v)
|
||||||
|
}
|
||||||
|
|
||||||
|
err := l.CallByParam(lua.P{
|
||||||
|
Fn: f,
|
||||||
|
NRet: 1,
|
||||||
|
Protect: true,
|
||||||
|
}, luaArgs...)
|
||||||
|
if err != nil {
|
||||||
|
panic(err) // TODO: better error handling?
|
||||||
|
}
|
||||||
|
|
||||||
|
ret := l.CheckString(-1)
|
||||||
|
l.Pop(1)
|
||||||
|
return ret
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *luaPlugin) setRoute(l *lua.LState) int {
|
||||||
|
method := l.CheckString(1)
|
||||||
|
path := l.CheckString(2)
|
||||||
|
f := l.CheckFunction(3)
|
||||||
|
p.routes = append(p.routes, luaRoute{method, path, f})
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *luaPlugin) Render(name string, data interface{}) error {
|
||||||
|
f, ok := p.renderCallbacks[name]
|
||||||
|
if !ok {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
err := p.state.CallByParam(lua.P{
|
||||||
|
Fn: f,
|
||||||
|
NRet: 0,
|
||||||
|
Protect: true,
|
||||||
|
}, luar.New(p.state, data))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *luaPlugin) Filters() template.FuncMap {
|
||||||
|
return p.filters
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *luaPlugin) SetRoutes(group *echo.Group) {
|
||||||
|
for _, r := range p.routes {
|
||||||
|
group.Match([]string{r.method}, r.path, func(ctx echo.Context) error {
|
||||||
|
err := p.state.CallByParam(lua.P{
|
||||||
|
Fn: r.f,
|
||||||
|
NRet: 0,
|
||||||
|
Protect: true,
|
||||||
|
}, luar.New(p.state, ctx))
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("Lua plugin error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *luaPlugin) Close() error {
|
||||||
|
p.state.Close()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func loadLuaPlugin(filename string) (*luaPlugin, error) {
|
||||||
|
l := lua.NewState()
|
||||||
|
p := &luaPlugin{
|
||||||
|
filename: filename,
|
||||||
|
state: l,
|
||||||
|
renderCallbacks: make(map[string]*lua.LFunction),
|
||||||
|
filters: make(template.FuncMap),
|
||||||
|
}
|
||||||
|
|
||||||
|
mt := l.NewTypeMetatable("koushin")
|
||||||
|
l.SetGlobal("koushin", mt)
|
||||||
|
l.SetField(mt, "on_render", l.NewFunction(p.onRender))
|
||||||
|
l.SetField(mt, "set_filter", l.NewFunction(p.setFilter))
|
||||||
|
l.SetField(mt, "set_route", l.NewFunction(p.setRoute))
|
||||||
|
|
||||||
|
if err := l.DoFile(filename); err != nil {
|
||||||
|
l.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return p, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func loadAllLuaPlugins(log echo.Logger) ([]Plugin, error) {
|
||||||
|
filenames, err := filepath.Glob("plugins/*.lua")
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("filepath.Glob failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
plugins := make([]Plugin, 0, len(filenames))
|
||||||
|
for _, filename := range filenames {
|
||||||
|
log.Printf("Loading Lua plugin '%v'", filename)
|
||||||
|
p, err := loadLuaPlugin(filename)
|
||||||
|
if err != nil {
|
||||||
|
for _, p := range plugins {
|
||||||
|
p.Close()
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("failed to load Lua plugin '%v': %v", filename, err)
|
||||||
|
}
|
||||||
|
plugins = append(plugins, p)
|
||||||
|
}
|
||||||
|
|
||||||
|
return plugins, nil
|
||||||
|
}
|
||||||
Loading…
Reference in a new issue