Add profiling endpoints

server/server.go

@@ -1,13 +1,13 @@
 package server
 
 import (
-	"fmt"
 	"io/fs"
 	"net/http"
 
 	"github.com/mstarongithub/passkey"
 	"github.com/rs/zerolog/log"
 	"gitlab.com/mstarongitlab/goutils/other"
+	"gitlab.com/mstarongitlab/linstrom/config"
 	"gitlab.com/mstarongitlab/linstrom/storage"
 )
 
@@ -32,7 +32,18 @@ func buildRootHandler(pkey *passkey.Passkey, reactiveFS, staticFS fs.FS) http.Ha
 	pkey.MountRoutes(mux, "/webauthn/")
 	mux.Handle("/", setupFrontendRouter(reactiveFS, staticFS))
 	mux.Handle("/pk/", http.StripPrefix("/pk", http.FileServer(http.Dir("pk-auth"))))
-	mux.HandleFunc("/health", func(w http.ResponseWriter, r *http.Request) { fmt.Fprint(w, true) })
+	mux.HandleFunc("/alive", isAliveHandler)
+
+	profilingHandler := setupProfilingHandler()
+	mux.Handle("/profiling/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		// Only allow access to profiling if a password is provided and it matches the one in the config
+		if r.FormValue("password") != config.GlobalConfig.Admin.ProfilingPassword {
+			// Specifically reply with a plain 404
+			http.Error(w, "", 404)
+			return
+		}
+		profilingHandler.ServeHTTP(w, r)
+	}))
 	mux.Handle(
 		"/authonly/",
 		pkey.Auth(