Added --ui flag

Author: djthorpe

cmd/pgmanager/server.go

@@ -23,6 +23,7 @@ type ServerCommands struct {
 
 type RunServer struct {
 	URL string `arg:"" name:"url" help:"Database URL" default:""`
+	UI  bool   `name:"ui" help:"Enable frontend UI" default:"false"`
 
 	// Postgres options
 	PG struct {
@@ -76,7 +77,7 @@ func (cmd *RunServer) Run(ctx *Globals) error {
 	// Register HTTP handlers
 	router := http.NewServeMux()
 	httphandler.RegisterBackendHandlers(router, ctx.HTTP.Prefix, manager)
-	httphandler.RegisterFrontendHandler(router, "")
+	httphandler.RegisterFrontendHandler(router, "", cmd.UI)
 
 	// Create a TLS config
 	var tlsconfig *tls.Config

pkg/manager/httphandler/frontend_excluded.go

@@ -10,7 +10,7 @@ import (
 )
 
 // RegisterFrontendHandler registers a fallback handler when frontend is not included
-func RegisterFrontendHandler(router *http.ServeMux, prefix string) {
+func RegisterFrontendHandler(router *http.ServeMux, prefix string, enabled bool) {
 	// Catch all handler returns a "not found" error
 	router.HandleFunc(joinPath(prefix, "/"), func(w http.ResponseWriter, r *http.Request) {
 		_ = httpresponse.Error(w, httpresponse.ErrNotFound, r.URL.String())

pkg/manager/httphandler/frontend_included.go

@@ -8,13 +8,25 @@ import (
 	"embed"
 	"io/fs"
 	"net/http"
+
+	// Packages
+	"github.com/mutablelogic/go-server/pkg/httpresponse"
 )
 
 //go:embed frontend/*
 var frontendFS embed.FS
 
-// RegisterFrontendHandler registers the frontend static file handler
-func RegisterFrontendHandler(router *http.ServeMux, prefix string) {
+// RegisterFrontendHandler registers the frontend static file handler if enabled,
+// otherwise registers a fallback handler that returns a not found error.
+func RegisterFrontendHandler(router *http.ServeMux, prefix string, enabled bool) {
+	if !enabled {
+		// Fallback handler returns a "not found" error
+		router.HandleFunc(joinPath(prefix, "/"), func(w http.ResponseWriter, r *http.Request) {
+			_ = httpresponse.Error(w, httpresponse.ErrNotFound, r.URL.String())
+		})
+		return
+	}
+
 	// Get the subdirectory to strip the "frontend" prefix
 	subFS, err := fs.Sub(frontendFS, "frontend")
 	if err != nil {