mini: add LOGGIN gate

webteleport · Apr 18, 2024 · 71c2e6b · 71c2e6b
1 parent 1e452e3
commit 71c2e6b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/apps/mini/mini.go b/apps/mini/mini.go
@@ -4,6 +4,7 @@ import (
 	"log/slog"
 	"net"
 	"net/http"
+	"os"
 
 	"github.com/webteleport/relay"
 	"github.com/webteleport/ufo/apps/relay/envs"
@@ -21,13 +22,15 @@ func listenHTTP(handler http.Handler) error {
 
 func Run([]string) (err error) {
 	store := relay.NewSessionStore()
-	s := relay.NewWSServer(envs.HOST, store).
-		WithPostUpgrade(
+	s := relay.NewWSServer(envs.HOST, store)
+	if os.Getenv("LOGGIN") != "" {
+		s.WithPostUpgrade(
 			utils.GinLoggerMiddleware(
 				// Set the Alt-Svc header for UDP port discovery && http3 bootstrapping
 				AltSvcMiddleware(store),
 			),
 		)
+	}
 
 	return listenHTTP(s)
 }