1 files changed, 14 insertions, 2 deletions
diff --git a/main.go b/main.go
index b8d2436..dd735a1 100644
--- a/main.go
+++ b/main.go
@@ -11,6 +11,9 @@ import (
 
 	"gosten/model"
 	"gosten/templ"
+
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
 )
 
 // flags
@@ -35,10 +38,19 @@ func main() {
 	queries := model.New(db)
 	_, err = queries.GetUsers(context.Background())
 
-	http.HandleFunc("/", func(w http.ResponseWriter, _ *http.Request) {
+	r := chi.NewRouter()
+
+	// A good base middleware stack
+	r.Use(middleware.RequestID)
+	r.Use(middleware.RealIP)
+	r.Use(middleware.CleanPath)
+	r.Use(middleware.Logger)
+	r.Use(middleware.Recoverer)
+
+	r.Get("/", func(w http.ResponseWriter, _ *http.Request) {
 		templ.Lookup("index").Execute(w, nil)
 	})
 
 	address := net.JoinHostPort(host, strconv.FormatUint(port, 10))
-	log.Fatal(http.ListenAndServe(address, nil))
+	log.Fatal(http.ListenAndServe(address, r))
 }