diff options
Diffstat (limited to 'plugin.go')
-rw-r--r-- | plugin.go | 151 |
1 files changed, 0 insertions, 151 deletions
@@ -1,13 +1,9 @@ package koushin import ( - "fmt" "html/template" - "path/filepath" "github.com/labstack/echo/v4" - "github.com/yuin/gopher-lua" - "layeh.com/gopher-luar" ) type Plugin interface { @@ -17,150 +13,3 @@ type Plugin interface { Render(name string, data interface{}) 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 -} |