core/kg/api/rest/router.go

package rest

import (
	"net/http"

	"github.com/giolekva/pcloud/core/kg/common"
	"github.com/giolekva/pcloud/core/kg/log"
	"github.com/gorilla/mux"
)

const APIURLSuffix = "/api/v1"

type Router struct {
	App    common.AppIface
	Logger common.LoggerIface

	Root    *mux.Router // ''
	APIRoot *mux.Router // 'api/v1'
	Users   *mux.Router // 'api/v1/users'
	User    *mux.Router // 'api/v1/users/{user_id:[A-Za-z0-9]+}'
}

func NewRouter(root *mux.Router, app common.AppIface, logger common.LoggerIface) *Router {
	apiRoot := root.PathPrefix(APIURLSuffix).Subrouter()
	users := apiRoot.PathPrefix("/users").Subrouter()
	user := apiRoot.PathPrefix("/users/{user_id:[A-Za-z0-9]+}").Subrouter()

	routers := &Router{
		App:    app,
		Logger: logger,

		Root:    root,
		APIRoot: apiRoot,
		Users:   users,
		User:    user,
	}

	root.Handle("/api/v1/{anything:.*}", http.HandlerFunc(http.NotFound))
	routers.initUsers()
	root.Use(routers.loggerMiddleware)
	return routers
}

func (router *Router) ServeHTTP(w http.ResponseWriter, req *http.Request) {
	router.Root.ServeHTTP(w, req)
}

func (router *Router) loggerMiddleware(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		router.Logger.Debug(r.Method, log.String("url", r.URL.String()))
		next.ServeHTTP(w, r)
	})
}