finish redirect middlewares

example-project/cmd/web-app/handlers/routes.go

@@ -14,10 +14,20 @@ import (
 const baseLayoutTmpl = "base.tmpl"
 
 // API returns a handler for a set of routes.
-func APP(shutdown chan os.Signal, log *log.Logger, staticDir, templateDir string, masterDB *sqlx.DB, authenticator *auth.Authenticator, renderer web.Renderer) http.Handler {
+func APP(shutdown chan os.Signal, log *log.Logger, staticDir, templateDir string, masterDB *sqlx.DB, authenticator *auth.Authenticator, renderer web.Renderer, globalMids ...web.Middleware) http.Handler {
+
+	// Define base middlewares applied to all requests.
+	middlewares := []web.Middleware{
+		mid.Trace(), mid.Logger(log), mid.Errors(log), mid.Metrics(), mid.Panics(),
+	}
+
+	// Append any global middlewares if they were included.
+	if len(globalMids) > 0 {
+		middlewares = append(middlewares, globalMids...)
+	}
 
 	// Construct the web.App which holds all routes as well as common Middleware.
-	app := web.NewApp(shutdown, log, mid.Trace(), mid.Logger(log), mid.Errors(log), mid.Metrics(), mid.Panics())
+	app := web.NewApp(shutdown, log, middlewares...)
 
 	// Register health check endpoint. This route is not authenticated.
 	check := Check{

example-project/cmd/web-app/main.go

@@ -5,9 +5,11 @@ import (
 	"encoding/json"
 	"expvar"
 	"fmt"
+	"geeks-accelerator/oss/saas-starter-kit/example-project/internal/mid"
 	"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/tracer"
 	"html/template"
 	"log"
+	"net"
 	"net/http"
 	_ "net/http/pprof"
 	"net/url"
@@ -285,6 +287,29 @@ func main() {
 		return
 	}
 
+	// =========================================================================
+	// Init redirect middleware to ensure all requests go to the primary domain.
+
+	baseSiteUrl, err := url.Parse(cfg.App.BaseUrl)
+	if err != nil {
+		log.Fatalf("main : Parse App Base URL : %s : %v", cfg.App.BaseUrl, err)
+	}
+
+	var primaryDomain string
+	if strings.Contains(baseSiteUrl.Host, ":") {
+		primaryDomain, _, err = net.SplitHostPort(baseSiteUrl.Host)
+		if err != nil {
+			log.Fatalf("main : SplitHostPort : %s : %v", baseSiteUrl.Host, err)
+		}
+	} else {
+		primaryDomain = baseSiteUrl.Host
+	}
+
+	redirect := mid.DomainNameRedirect(mid.DomainNameRedirectConfig{
+		DomainName: primaryDomain,
+		HTTPSEnabled: (cfg.HTTPS.Host != ""),
+	})
+
 	// =========================================================================
 	// URL Formatter
 	// s3UrlFormatter is a help function used by to convert an s3 key to
@@ -530,7 +555,7 @@ func main() {
 
 	app := http.Server{
 		Addr:           cfg.HTTP.Host,
-		Handler:        handlers.APP(shutdown, log, cfg.App.StaticDir, cfg.App.TemplateDir, masterDb, nil, renderer),
+		Handler:        handlers.APP(shutdown, log, cfg.App.StaticDir, cfg.App.TemplateDir, masterDb, nil, renderer, redirect),
 		ReadTimeout:    cfg.HTTP.ReadTimeout,
 		WriteTimeout:   cfg.HTTP.WriteTimeout,
 		MaxHeaderBytes: 1 << 20,