Fix broken path prefix

pkg/api/api.go

@@ -3,7 +3,6 @@ package api
 import (
 	"encoding/base64"
 	"fmt"
-	"github.com/sosedoff/pgweb/static"
 	"net/http"
 	neturl "net/url"
 	"regexp"
@@ -18,6 +17,7 @@ import (
 	"github.com/sosedoff/pgweb/pkg/command"
 	"github.com/sosedoff/pgweb/pkg/connection"
 	"github.com/sosedoff/pgweb/pkg/shared"
+	"github.com/sosedoff/pgweb/static"
 )
 
 var (
@@ -58,12 +58,20 @@ func setClient(c *gin.Context, newClient *client.Client) error {
 }
 
 // GetHome renderes the home page
-func GetHome() http.Handler {
+func GetHome(prefix string) http.Handler {
-	return http.FileServer(http.FS(static.Static))
+	if prefix != "" {
+		prefix = "/" + prefix
+	}
+	return http.StripPrefix(prefix, http.FileServer(http.FS(static.Static)))
 }
 
-func GetAssets() http.Handler {
+func GetAssets(prefix string) http.Handler {
-	return http.StripPrefix("/static/", http.FileServer(http.FS(static.Static)))
+	if prefix != "" {
+		prefix = "/" + prefix + "static/"
+	} else {
+		prefix = "/static/"
+	}
+	return http.StripPrefix(prefix, http.FileServer(http.FS(static.Static)))
 }
 
 // GetSessions renders the number of active sessions

pkg/api/routes.go

@@ -2,6 +2,7 @@ package api
 
 import (
 	"github.com/gin-gonic/gin"
+
 	"github.com/sosedoff/pgweb/pkg/command"
 )
 
@@ -20,8 +21,8 @@ func SetupMiddlewares(group *gin.RouterGroup) {
 func SetupRoutes(router *gin.Engine) {
 	root := router.Group(command.Opts.Prefix)
 
-	root.GET("/", gin.WrapH(GetHome()))
+	root.GET("/", gin.WrapH(GetHome(command.Opts.Prefix)))
-	root.GET("/static/*path", gin.WrapH(GetAssets()))
+	root.GET("/static/*path", gin.WrapH(GetAssets(command.Opts.Prefix)))
 	root.GET("/connect/:resource", ConnectWithBackend)
 
 	api := root.Group("/api")