add mini relay

apps/mini/mini.go

@@ -0,0 +1,38 @@
+package mini
+
+import (
+	"log/slog"
+	"net"
+	"net/http"
+
+	"github.com/webteleport/relay"
+	"github.com/webteleport/ufo/apps/relay/envs"
+	"github.com/webteleport/utils"
+)
+
+func listenHTTP(handler http.Handler) error {
+	slog.Info("listening on HTTP http://" + envs.HOST + envs.HTTP_PORT)
+	ln, err := net.Listen("tcp4", envs.HTTP_PORT)
+	if err != nil {
+		return err
+	}
+	return http.Serve(ln, handler)
+}
+
+func Run([]string) (err error) {
+	sm := relay.NewSessionManager(envs.HOST)
+
+	var dsm http.Handler = sm
+	// Set the Alt-Svc header for UDP port discovery && http3 bootstrapping
+	dsm = AltSvcMiddleware(dsm)
+	dsm = utils.GinLoggerMiddleware(dsm)
+
+	return listenHTTP(dsm)
+}
+
+func AltSvcMiddleware(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Alt-Svc", envs.ALT_SVC)
+		next.ServeHTTP(w, r)
+	})
+}

cmd/ufo/main.go

@@ -25,6 +25,7 @@ import (
 	"github.com/webteleport/ufo/apps/intercept"
 	"github.com/webteleport/ufo/apps/ip"
 	"github.com/webteleport/ufo/apps/login"
+	"github.com/webteleport/ufo/apps/mini"
 	"github.com/webteleport/ufo/apps/mmdb"
 	"github.com/webteleport/ufo/apps/multi"
 	"github.com/webteleport/ufo/apps/naive"
@@ -122,6 +123,7 @@ var cmdRun multicall.RunnerFuncMap = map[string]multicall.RunnerFunc{
 	// relay
 	"hub":   relay.Run,
 	"relay": relay.Run,
+	"mini":  mini.Run,
 	// version info
 	"version": version.Run,
 	// binary upgrade