make hostname-based redirection configurable

1 files changed, 10 insertions, 7 deletions

diff --git a/internal/server/server.go b/internal/server/server.go
index 9edebe3..b9db417 100644
--- a/internal/server/server.go
+++ b/internal/server/server.go
@@ -29,6 +29,7 @@ var (
 type Config struct {
 	Development   bool   `conf:"default:false,flag:dev"`
 	Root          string `conf:"default:website"`
+	Redirect      bool   `conf:"default:false"`
 	ListenAddress string `conf:"default:localhost"`
 	Port          string `conf:"default:3000,short:p"`
 	TLS           bool   `conf:"default:false"`
@@ -132,14 +133,16 @@ func New(runtimeConfig *Config) (*Server, error) {
 		return nil, errors.Wrap(err, "could not create website mux")
 	}
 	log.Debug("binding main handler to", "host", listenAddress)
-	hostname := config.BaseURL.Hostname()
 
-	loggingMux.Handle(hostname+"/", mux)
-
-	loggingMux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
-		newURL := config.BaseURL.JoinPath(r.URL.String())
-		http.Redirect(w, r, newURL.String(), 301)
-	})
+	if runtimeConfig.Development || runtimeConfig.Redirect {
+		loggingMux.Handle(config.BaseURL.Hostname()+"/", mux)
+		loggingMux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+			newURL := config.BaseURL.JoinPath(r.URL.String())
+			http.Redirect(w, r, newURL.String(), 301)
+		})
+	} else {
+		loggingMux.Handle("/", mux)
+	}
 
 	top.Handle("/",
 		serverHeaderHandler(