aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat
-rw-r--r--plugin.go151
-rw-r--r--plugin_lua.go158
2 files changed, 158 insertions, 151 deletions
diff --git a/plugin.go b/plugin.go
index a25f34b..6a5b7e7 100644
--- a/plugin.go
+++ b/plugin.go
@@ -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
-}
diff --git a/plugin_lua.go b/plugin_lua.go
new file mode 100644
index 0000000..c1a8ebe
--- /dev/null
+++ b/plugin_lua.go
@@ -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
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-22 13:27:36 +0000