diff --git a/docs/changelog.md b/docs/changelog.md index ca21062b..4b972c6f 100644 --- a/docs/changelog.md +++ b/docs/changelog.md @@ -5,6 +5,7 @@ ### New Features - Added `m1k1o/neko:vivaldi` tag (thanks @Xeddius). - Added `m1k1o/neko:opera` tag (thanks @prophetofxenu). +- Added `NEKO_PATH_PREFIX`. ## [n.eko v2.6](https://github.com/m1k1o/neko/releases/tag/v2.6) diff --git a/docs/getting-started/configuration.md b/docs/getting-started/configuration.md index 34008bc2..062b416e 100644 --- a/docs/getting-started/configuration.md +++ b/docs/getting-started/configuration.md @@ -134,6 +134,9 @@ nat1to1: #### `NEKO_PROXY`: - Enable reverse proxy mode, so that neko trusts `X-Forwarded-For` headers. - e.g. `false` +#### `NEKO_PATH_PREFIX`: + - Path prefix for HTTP requests. + - e.g. `/neko/` ### Expert settings diff --git a/server/internal/http/http.go b/server/internal/http/http.go index 533da496..bb702432 100644 --- a/server/internal/http/http.go +++ b/server/internal/http/http.go @@ -30,6 +30,12 @@ func New(conf *config.Server, webSocketHandler types.WebSocketHandler) *Server { router.Use(middleware.RequestLogger(&logformatter{logger})) router.Use(middleware.Recoverer) // Recover from panics without crashing server + if conf.PathPrefix != "/" { + router.Use(func(h http.Handler) http.Handler { + return http.StripPrefix(conf.PathPrefix, h) + }) + } + router.Get("/ws", func(w http.ResponseWriter, r *http.Request) { err := webSocketHandler.Upgrade(w, r) if err != nil { diff --git a/server/internal/types/config/server.go b/server/internal/types/config/server.go index fcb4b912..51f04659 100644 --- a/server/internal/types/config/server.go +++ b/server/internal/types/config/server.go @@ -1,15 +1,18 @@ package config import ( + "path" + "github.com/spf13/cobra" "github.com/spf13/viper" ) type Server struct { - Cert string - Key string - Bind string - Static string + Cert string + Key string + Bind string + Static string + PathPrefix string } func (Server) Init(cmd *cobra.Command) error { @@ -33,6 +36,11 @@ func (Server) Init(cmd *cobra.Command) error { return err } + cmd.PersistentFlags().String("path_prefix", "/", "path prefix for HTTP requests") + if err := viper.BindPFlag("path_prefix", cmd.PersistentFlags().Lookup("path_prefix")); err != nil { + return err + } + return nil } @@ -41,4 +49,5 @@ func (s *Server) Set() { s.Key = viper.GetString("key") s.Bind = viper.GetString("bind") s.Static = viper.GetString("static") + s.PathPrefix = path.Join("/", path.Clean(viper.GetString("path_prefix"))) }