protocol

type EarlyAdapter interface {
	Name() string
	// ClaimEarly reports whether this adapter takes the connection.
	ClaimEarly(c *EarlyClaimCtx) bool
	// ServeEarly drives the connection; called only when ClaimEarly returned true.
	ServeEarly(ctx context.Context, c *EarlyClaimCtx)
}

type EarlyClaimCtx struct {
	// Peek holds the opening bytes from ClientReader; nil for a post-CONNECT stream.
	Peek []byte
	// TLSTerminated is true for a decrypted post-CONNECT stream.
	TLSTerminated bool
	// ALPN is the negotiated protocol ("h2"/"http/1.1"/""); empty at raw accept.
	ALPN string
	// Target is the fixed upstream for a post-CONNECT stream; nil at raw accept.
	Target *types.Target

	ClientConn   net.Conn
	ClientReader *bufio.Reader

	// UpstreamConn and UpstreamReader are pre-dialed for a post-CONNECT stream; nil at raw accept.
	UpstreamConn   net.Conn
	UpstreamReader *bufio.Reader
}

type Registry struct {
	Early   []EarlyAdapter
	Upgrade []UpgradeAdapter
	// contains filtered or unexported fields
}

type TLSEarlyAdapter interface {
	EarlyAdapter
	ClaimTLS(sni, host string, port int) bool
}

type UpgradeAdapter interface {
	Name() string
	// ClaimUpgrade reports whether this adapter takes the upgraded connection.
	ClaimUpgrade(c *UpgradeClaimCtx) bool
	// ServeUpgrade drives the connection; called only when ClaimUpgrade returned true.
	ServeUpgrade(ctx context.Context, c *UpgradeClaimCtx, conns UpgradeConns)
}

type UpgradeClaimCtx struct {
	Req    *types.RawHTTP1Request
	Target *types.Target
	// Signal is the upgrade signal driving this claim: "http_101" after a parsed
	// upgrade request, or "connect" after a CONNECT tunnel is established.
	Signal string
}

type UpgradeConns struct {
	ClientConn   net.Conn
	ClientReader *bufio.Reader
	// UpstreamConn and UpstreamReader are non-nil on the TLS path (reusing the
	// existing upstream) and nil on the plain path (the adapter dials its own).
	UpstreamConn   net.Conn
	UpstreamReader *bufio.Reader
}