Add ability to launch pgweb with url prefix

pkg/api/middleware.go

@@ -2,6 +2,7 @@ package api
 
 import (
 	"log"
+	"strings"
 
 	"github.com/gin-gonic/gin"
 
@@ -12,7 +13,9 @@ import (
 // Middleware function to check database connection status before running queries
 func dbCheckMiddleware() gin.HandlerFunc {
 	return func(c *gin.Context) {
-		if allowedPaths[c.Request.URL.Path] == true {
+		path := strings.Replace(c.Request.URL.Path, command.Opts.Prefix, "", -1)
+
+		if allowedPaths[path] == true {
 			c.Next()
 			return
 		}

pkg/api/routes.go

@@ -14,10 +14,12 @@ func SetupMiddlewares(group *gin.RouterGroup) {
 }
 
 func SetupRoutes(router *gin.Engine) {
-	router.GET("/", GetHome)
-	router.GET("/static/*path", GetAsset)
+	group := router.Group(command.Opts.Prefix)
 
-	api := router.Group("/api")
+	group.GET("/", GetHome)
+	group.GET("/static/*path", GetAsset)
+
+	api := group.Group("/api")
 	{
 		SetupMiddlewares(api)
 

pkg/command/options.go

@@ -2,6 +2,7 @@ package command
 
 import (
 	"os"
+	"strings"
 
 	"github.com/jessevdk/go-flags"
 )
@@ -22,6 +23,7 @@ type Options struct {
 	AuthPass string `long:"auth-pass" description:"HTTP basic auth password"`
 	SkipOpen bool   `short:"s" long:"skip-open" description:"Skip browser open on start"`
 	Sessions bool   `long:"sessions" description:"Enable multiple database sessions" default:"false"`
+	Prefix   string `long:"prefix" description:"Add a url prefix"`
 }
 
 var Opts Options
@@ -40,5 +42,9 @@ func ParseOptions() error {
 		Opts.Sessions = true
 	}
 
+	if Opts.Prefix != "" && !strings.Contains(Opts.Prefix, "/") {
+		Opts.Prefix = Opts.Prefix + "/"
+	}
+
 	return nil
 }