Added before hooks for plugins

Author: romanoff

api.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"github.com/gophergala2016/dbserver/plugins"
+	"net/http"
 	"os"
 )
 
@@ -11,6 +12,7 @@ type Api struct {
 	MinVersion         int
 	Routes             []*Route
 	Plugins            map[string]Plugin
+	PluginsList        []string
 }
 
 func (self *Api) IsDeprecated(version int) bool {
@@ -28,13 +30,19 @@ func (self *Api) RegisterPlugin(name string, plugin Plugin) {
 	}
 	plugin.ParseConfig("plugins/" + name + ".toml")
 	self.Plugins[name] = plugin
+	self.PluginsList = append(self.PluginsList, name)
 }
 
 func (self *Api) GetPlugin(name string) Plugin {
 	return self.Plugins[name]
 }
 
+func (self *Api) GetPlugins() []string {
+	return self.PluginsList
+}
+
 type Plugin interface {
 	ParseConfig(path string) error
 	Process(data map[string]interface{}, arg map[string]interface{}) *plugins.Response
+	ProcessBeforeHook(data map[string]interface{}, r *http.Request)
 }

main.go

@@ -75,7 +75,15 @@ func handler(api *Api, route *Route, version int) func(http.ResponseWriter, *htt
 			urlParams[urlParam.Key] = urlParam.Value
 		}
 		params, err := getRequestParams(r, urlParams)
-		sql, err := route.Sql(params, apiVersion)
+		if err != nil {
+			w.WriteHeader(http.StatusBadRequest)
+			return
+		}
+		data := make(map[string]interface{})
+		data["params"] = params
+
+		runBeforeHooks(api, data, r)
+		sql, err := route.Sql(data, apiVersion)
 		if err != nil && sql != "" {
 			w.WriteHeader(http.StatusBadRequest)
 			fmt.Fprint(w, sql)
@@ -209,3 +217,11 @@ func goThroughPipelines(api *Api,
 	}
 	return nil
 }
+
+func runBeforeHooks(api *Api, data map[string]interface{}, r *http.Request) {
+	plugins := api.GetPlugins()
+	for _, name := range plugins {
+		plugin := api.GetPlugin(name)
+		plugin.ProcessBeforeHook(data, r)
+	}
+}

plugins/jwt/jwt.go

@@ -8,6 +8,8 @@ import (
 	"github.com/SermoDigital/jose/jws"
 	"github.com/gophergala2016/dbserver/plugins"
 	"io/ioutil"
+	"net/http"
+	"strings"
 	"time"
 )
 
@@ -83,7 +85,19 @@ func (self *JWT) GenerateToken(payload map[string]interface{}) ([]byte, error) {
 	return serializedToken, nil
 }
 
-// app.Register(&JWT{}, "jwt") || app.Register("jwt", JWT)
-
-// Hooks: 1. Before request
-//        2. Process - when called in pipeline
+func (self *JWT) ProcessBeforeHook(data map[string]interface{}, r *http.Request) {
+	headerValue := r.Header.Get("Authorization")
+	if headerValue == "" {
+		return
+	}
+	if !strings.HasPrefix(headerValue, "Bearer ") {
+		return
+	}
+	headerValue = strings.Replace(headerValue, "Bearer ", "", 1)
+	//TODO: Verify secret
+	token, err := jws.ParseJWT([]byte(headerValue))
+	if err != nil {
+		return
+	}
+	data["jwt"] = token.Claims()
+}

route.go

@@ -32,7 +32,7 @@ type RouteVersion struct {
 	SqlTemplate *template.Template
 }
 
-func (self *Route) validate(params map[string]interface{}, version int) (string, error) {
+func (self *Route) validate(params interface{}, version int) (string, error) {
 	route := self.Versions[version]
 	if route == nil {
 		return "", fmt.Errorf("Route version %v missing from %v route", version, self.Name)
@@ -59,14 +59,14 @@ func (self *Route) validate(params map[string]interface{}, version int) (string,
 	return "", nil
 }
 
-func (self *Route) Sql(params map[string]interface{}, version int) (string, error) {
+func (self *Route) Sql(data map[string]interface{}, version int) (string, error) {
 	version = self.GetAvailableVersion(version)
 	route := self.Versions[version]
 	if route == nil {
 		return "", fmt.Errorf("Route version %v missing from %v route", version, self.Name)
 	}
 	var out bytes.Buffer
-	response, err := self.validate(params, version)
+	response, err := self.validate(data["params"], version)
 	if err != nil {
 		return "", err
 	}
@@ -76,7 +76,7 @@ func (self *Route) Sql(params map[string]interface{}, version int) (string, erro
 	if !self.Custom {
 		out.Write([]byte("with response_table as ("))
 	}
-	err = route.SqlTemplate.Execute(&out, params)
+	err = route.SqlTemplate.Execute(&out, data)
 	if err != nil {
 		return "", err
 	}
@@ -124,5 +124,4 @@ func makeTemplate(t string) (*template.Template, error) {
 		"quote": quoteString,
 	}
 	return template.New("").Funcs(funcMap).Parse(t)
-
 }