v0.3.8

internal/backend/server/context.go

@@ -0,0 +1,72 @@
+package server
+
+import (
+	"log/slog"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+)
+
+const HeaderServerUpstreamURL = "X-Server-Upstream-URL"
+
+type SourceKind string
+
+const (
+	SourceNative SourceKind = "native"
+	SourceMITM   SourceKind = "mitm"
+)
+
+type ProtocolClass string
+
+const (
+	ProtocolHTTP          ProtocolClass = "http"
+	ProtocolConnectUnary  ProtocolClass = "connect_unary"
+	ProtocolConnectStream ProtocolClass = "connect_stream"
+)
+
+type Context struct {
+	Writer    http.ResponseWriter
+	Request   *http.Request
+	RouteName string
+	Source    SourceKind
+	Protocol  ProtocolClass
+	StartedAt time.Time
+
+	UpstreamURL *url.URL
+	Mode        ExecutionMode
+	LastError   error
+
+	Logger *slog.Logger
+}
+
+func newContext(writer http.ResponseWriter, request *http.Request, route Route) *Context {
+	return &Context{
+		Writer:    writer,
+		Request:   request,
+		RouteName: route.Name,
+		Protocol:  route.Protocol,
+		StartedAt: time.Now(),
+		Logger:    slog.Default(),
+		Mode:      ModeLocal,
+	}
+}
+
+func (ctx *Context) ParseUpstreamURL() error {
+	if ctx == nil || ctx.Request == nil {
+		return nil
+	}
+	rawURL := strings.TrimSpace(ctx.Request.Header.Get(HeaderServerUpstreamURL))
+	if rawURL == "" {
+		ctx.Source = SourceNative
+		ctx.UpstreamURL = nil
+		return nil
+	}
+	parsed, err := ParseAndValidateRawURL(rawURL)
+	if err != nil {
+		return err
+	}
+	ctx.Source = SourceMITM
+	ctx.UpstreamURL = parsed
+	return nil
+}