Add a way to extend the handlers and add custom stuff

crt-fork · Sep 22, 2017 · 98735a5 · 98735a5
1 parent ff905a0
commit 98735a5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/api.go b/api.go
@@ -42,7 +42,7 @@ func main() {
 	sch.Start()
 
 	handlers.Bootstrap(core, e)
-	handlers.Register()
+	handlers.Register(nil)
 }
 
 func initStorage() storage.StorageApi {

diff --git a/handlers/bootstrap.go b/handlers/bootstrap.go
@@ -25,12 +25,14 @@ var core pwd.PWDApi
 var e event.EventApi
 var ws *socketio.Server
 
+type HandlerExtender func(h *mux.Router)
+
 func Bootstrap(c pwd.PWDApi, ev event.EventApi) {
 	core = c
 	e = ev
 }
 
-func Register() {
+func Register(extend HandlerExtender) {
 
 	bypassCaptcha := len(os.Getenv("GOOGLE_RECAPTCHA_DISABLED")) > 0
 
@@ -91,6 +93,10 @@ func Register() {
 
 	corsRouter.HandleFunc("/", NewSession).Methods("POST")
 
+	if extend != nil {
+		extend(corsRouter)
+	}
+
 	n := negroni.Classic()
 	r.PathPrefix("/").Handler(negroni.New(negroni.Wrap(corsHandler(corsRouter))))
 	n.UseHandler(r)