saasitone/controllers/router.go

package controllers

import (
	"net/http"

	"goweb/middleware"

	"github.com/gorilla/sessions"
	"github.com/labstack/echo-contrib/session"

	"github.com/labstack/echo/v4"
	echomw "github.com/labstack/echo/v4/middleware"

	"goweb/container"
)

const StaticDir = "static"

func BuildRouter(c *container.Container) {
	// Static files with proper cache control
	// funcmap.File() should be used in templates to append a cache key to the URL in order to break cache
	// after each server restart
	c.Web.Group("", middleware.CacheControl(c.Config.Cache.MaxAge.StaticFile)).
		Static("/", StaticDir)

	// Middleware
	c.Web.Use(echomw.RemoveTrailingSlashWithConfig(echomw.TrailingSlashConfig{
		RedirectCode: http.StatusMovedPermanently,
	}))
	c.Web.Use(echomw.RequestID())
	c.Web.Use(echomw.Recover())
	c.Web.Use(echomw.Gzip())
	c.Web.Use(echomw.Logger())
	c.Web.Use(echomw.TimeoutWithConfig(echomw.TimeoutConfig{
		Timeout: c.Config.App.Timeout,
	}))
	c.Web.Use(middleware.PageCache(c.Cache))
	c.Web.Use(session.Middleware(sessions.NewCookieStore([]byte(c.Config.App.EncryptionKey))))
	c.Web.Use(echomw.CSRFWithConfig(echomw.CSRFConfig{
		TokenLookup: "form:csrf",
	}))

	// Base controller
	ctr := NewController(c)

	// Error handler
	err := Error{Controller: ctr}
	c.Web.HTTPErrorHandler = err.Get

	// Routes
	navRoutes(c.Web, ctr)
	userRoutes(c.Web, ctr)
}

func navRoutes(e *echo.Echo, ctr Controller) {
	home := Home{Controller: ctr}
	e.GET("/", home.Get).Name = "home"

	about := About{Controller: ctr}
	e.GET("/about", about.Get).Name = "about"

	contact := Contact{Controller: ctr}
	e.GET("/contact", contact.Get).Name = "contact"
	e.POST("/contact", contact.Post).Name = "contact.post"
}

func userRoutes(e *echo.Echo, ctr Controller) {
	login := Login{Controller: ctr}
	e.GET("/user/login", login.Get).Name = "login"
	e.POST("/user/login", login.Post).Name = "login.post"

	register := Register{Controller: ctr}
	e.GET("/user/register", register.Get).Name = "register"
	e.POST("/user/register", register.Post).Name = "register.post"
}