From 71c2e6b9127a9fc5e4081b4c7c864d114fd4687b Mon Sep 17 00:00:00 2001
From: navigaid <navigaid@gmail.com>
Date: Thu, 18 Apr 2024 06:39:21 +0000
Subject: [PATCH] mini: add LOGGIN gate

---
 apps/mini/mini.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/apps/mini/mini.go b/apps/mini/mini.go
index 0a5418e..4b60304 100644
--- 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)
 }