Wrapping typed context in gorilla/mux

wrapper.go

package wrapper

import (
	"net/http"

	"github.com/gorilla/mux"
)

// wrapper
type ContextWraper struct {
	context *Context
}

func New(context Context) *ContextWraper {
	return &ContextWraper{
		context: &context,
	}
}

func (cw ContextWraper) WrapFunc(handlerFunc HandlerFunc) http.Handler {

	return &handler{
		context:     cw.context,
		handlerFunc: handlerFunc,
	}
}

func (cw ContextWraper) WrapMiddlewareFunc(middlewareFunc MiddlewareFunc) mux.MiddlewareFunc {
	return func(h http.Handler) http.Handler {
		return middlewareFunc(cw.context, h)
	}
}

// handler
var _ http.Handler = (*handler)(nil)

type handler struct {
	context     *Context
	handlerFunc HandlerFunc
}

func (h handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	h.handlerFunc(h.context, w, r)
}

// types
type Context struct {
	Foo string
	Bar int
}

type HandlerFunc func(*Context, http.ResponseWriter, *http.Request)
type MiddlewareFunc func(*Context, http.Handler) http.Handler