router.go

package baseapp

import (
	"fmt"
	"os"

	sdk "github.com/pokt-network/pocket-core/types"
)

type router struct {
	routes map[string]sdk.Handler
}

var _ sdk.Router = NewRouter()

// NewRouter returns a reference to a new router.
//
// TODO: Either make the function private or make return type (router) public.
func NewRouter() *router { // nolint: golint
	return &router{
		routes: make(map[string]sdk.Handler),
	}
}

// AddRoute adds a route path to the router with a given handler. The route must
// be alphanumeric.
func (rtr *router) AddRoute(path string, h sdk.Handler) sdk.Router {
	if !isAlphaNumeric(path) {
		fmt.Println("route expressions can only contain alphanumeric characters")
		os.Exit(1)
	}
	if rtr.routes[path] != nil {
		fmt.Println(fmt.Errorf("route %s has already been initialized", path))
		os.Exit(1)
	}

	rtr.routes[path] = h
	return rtr
}

// Route returns a handler for a given route path.
//
// TODO: Handle expressive matches.
func (rtr *router) Route(path string) sdk.Handler {
	return rtr.routes[path]
}