cursor-byok/internal/backend/server/context.go

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
}