alps/plugin_go.go
2019-12-17 10:58:31 +01:00

110 lines
2.3 KiB
Go

package koushin
import (
"html/template"
"net/http"
"path/filepath"
"github.com/labstack/echo/v4"
)
type goPlugin struct {
p *GoPlugin
}
func (p *goPlugin) Name() string {
return p.p.Name
}
func (p *goPlugin) LoadTemplate(t *template.Template) error {
t.Funcs(p.p.templateFuncs)
paths, err := filepath.Glob(pluginDir + "/" + p.p.Name + "/public/*.html")
if err != nil {
return err
}
if len(paths) > 0 {
if _, err := t.ParseFiles(paths...); err != nil {
return err
}
}
return nil
}
func (p *goPlugin) SetRoutes(group *echo.Group) {
for _, r := range p.p.routes {
group.Add(r.Method, r.Path, r.Handler)
}
group.Static("/plugins/"+p.p.Name+"/assets", pluginDir+"/"+p.p.Name+"/public/assets")
}
func (p *goPlugin) Inject(name string, data interface{}) error {
return nil
}
func (p *goPlugin) Close() error {
return nil
}
type goPluginRoute struct {
Method string
Path string
Handler echo.HandlerFunc
}
// GoPlugin is a helper to create Go plugins.
//
// Use this struct to define your plugin, then call RegisterPlugin:
//
// p := GoPlugin{Name: "my-plugin"}
// // Define routes, template functions, etc
// koushin.RegisterPlugin(p.Plugin())
type GoPlugin struct {
Name string
routes []goPluginRoute
templateFuncs template.FuncMap
}
// AddRoute registers a new HTTP route.
//
// The echo.Context passed to the HTTP handler can be type-asserted to
// *koushin.Context.
func (p *GoPlugin) AddRoute(method, path string, handler echo.HandlerFunc) {
p.routes = append(p.routes, goPluginRoute{method, path, handler})
}
func (p *GoPlugin) DELETE(path string, handler echo.HandlerFunc) {
p.AddRoute(http.MethodDelete, path, handler)
}
func (p *GoPlugin) GET(path string, handler echo.HandlerFunc) {
p.AddRoute(http.MethodGet, path, handler)
}
func (p *GoPlugin) POST(path string, handler echo.HandlerFunc) {
p.AddRoute(http.MethodPost, path, handler)
}
func (p *GoPlugin) PUT(path string, handler echo.HandlerFunc) {
p.AddRoute(http.MethodPut, path, handler)
}
// TemplateFuncs registers new template functions.
func (p *GoPlugin) TemplateFuncs(funcs template.FuncMap) {
if p.templateFuncs == nil {
p.templateFuncs = make(template.FuncMap, len(funcs))
}
for k, f := range funcs {
p.templateFuncs[k] = f
}
}
// Plugin returns an object implementing Plugin.
func (p *GoPlugin) Plugin() Plugin {
return &goPlugin{p}
}