gravity

type CircuitBreaker = resilience.CircuitBreaker

type CircuitBreakerConfig = resilience.CircuitBreakerConfig

type CircuitBreakerState = resilience.CircuitBreakerState

type CircuitBreakerStats = resilience.CircuitBreakerStats

type ConnectionPoolConfig struct {
	// Connection pool size (4-8 connections as per PLAN.md)
	PoolSize int

	// Streams per connection for packet multiplexing
	StreamsPerConnection int

	// Stream allocation strategy
	AllocationStrategy StreamAllocationStrategy

	// Health check and failover settings
	HealthCheckInterval time.Duration
	FailoverTimeout     time.Duration

	// MaxGravityPeers caps how many Gravity servers Hadron connects to.
	// Default: DefaultMaxGravityPeers (3).
	MaxGravityPeers int

	// StreamsPerGravity is tunnel streams per Gravity host.
	// Default: DefaultStreamsPerGravity (2).
	StreamsPerGravity int
}

type EndpointSelector struct {
	// contains filtered or unexported fields
}

type FlowKey struct {
	SrcIP   [16]byte
	DstIP   [16]byte
	SrcPort uint16
	DstPort uint16
	Proto   uint8
}

type GravityClient struct {
	// contains filtered or unexported fields
}

type GravityConfig struct {
	Context  context.Context
	Logger   logger.Logger
	Provider provider.Provider
	URL      string
	// GravityURLs is a list of Gravity server URLs to connect to.
	// If set, this takes precedence over URL.
	// Hadron connects to up to MaxGravityPeers servers from this list.
	GravityURLs          []string
	CACert               string
	IP4Address           string
	IP6Address           string
	ECDSAPrivateKey      *ecdsa.PrivateKey
	InstanceID           string
	Region               string // Region where the instance is located, for display only
	AvailabilityZone     string // Availability zone where the instance is located, for display only
	CloudProvider        string // Type of cloud provider (e.g., aws, gcp, azure), for display only
	ClientVersion        string
	ClientName           string
	Capabilities         *pb.ClientCapabilities
	PingInterval         time.Duration
	WorkingDir           string
	TraceLogPackets      bool
	NetworkInterface     network.NetworkInterface
	ConnectionPoolConfig *ConnectionPoolConfig
	SkipAutoReconnect    bool
	InstanceTags         []string // Tags for display only
	InstanceType         string   // Type of instance (e.g., t2.micro)
	DefaultServerName    string   // Fallback TLS ServerName when connecting via IP address (default: "gravity.agentuity.com")
	UseMultiConnect      bool     // Use multiple connections to gravity

	// DiscoveryResolveFunc is called periodically to re-resolve the set of
	// available Gravity server URLs. If nil, no re-resolution or cycling occurs.
	// The function should return ALL available URLs (not capped by MaxGravityPeers).
	// The GravityClient handles selection and capping internally.
	DiscoveryResolveFunc func() []string

	// PeerDiscoveryInterval is how often to re-resolve DNS for new Gravity peers.
	// Default: 30 minutes.
	PeerDiscoveryInterval time.Duration

	// PeerCycleInterval is how often to rotate one connection when there are more
	// available Gravities than active connections. Only cycles if DNS returns more
	// IPs than MaxGravityPeers. Default: 2 hours.
	PeerCycleInterval time.Duration

	// MaxReconnectAttempts is the maximum number of reconnection attempts before
	// invoking ReconnectionFailedCallback. Default: 10
	MaxReconnectAttempts int

	// ReconnectAttemptTimeout is the timeout for each individual reconnection
	// attempt. If a single attempt takes longer than this, it is cancelled and
	// the next attempt begins. Default: 2 minutes
	ReconnectAttemptTimeout time.Duration

	// ReconnectionFailedCallback is invoked when all reconnection attempts are
	// exhausted. This is typically used to crash the process so a supervisor
	// (e.g. systemd) can perform a clean restart. If nil, the client simply
	// stops reconnecting.
	ReconnectionFailedCallback func(attempts int, lastErr error)
}

type GravityEndpoint struct {
	// URL is the address of this Gravity server.
	URL string

	// TLSServerName is the hostname to use for TLS SNI when the URL contains
	// an IP address (e.g., after DNS resolution). If empty, the hostname is
	// extracted from URL as usual.
	TLSServerName string
	// contains filtered or unexported fields
}

type IdentifyConfig struct {
	GravityURL        string
	InstanceID        string
	ECDSAPrivateKey   *ecdsa.PrivateKey
	CACert            string // Optional: additional CA certificate PEM to trust
	DefaultServerName string // Optional: fallback TLS ServerName when connecting via IP address
}

type PooledBuffer struct {
	Buffer []byte
	Length int
}

type RetryConfig = resilience.RetryConfig

type RetryStats = resilience.RetryStats

type RetryableFunc = resilience.RetryableFunc

type StreamAllocationStrategy int

type StreamInfo struct {
	// contains filtered or unexported fields
}

type StreamManager struct {
	// contains filtered or unexported fields
}

type StreamMetrics struct {
	PacketsSent     int64
	PacketsReceived int64
	BytesSent       int64
	BytesReceived   int64
	LastLatency     time.Duration
	ErrorCount      int64
	LastError       time.Time
	LastSendUs      int64 // unix microseconds of last successful send
	LastRecvUs      int64 // unix microseconds of last successful receive
}

type StreamMetricsSnapshot struct {
	StreamID        string
	ConnectionIndex int
	Healthy         bool
	PacketsSent     int64
	PacketsReceived int64
	BytesSent       int64
	BytesReceived   int64
	ErrorCount      int64
	LastSendUs      int64
	LastRecvUs      int64
}

type TunnelBinding struct {
	Endpoint *GravityEndpoint
	LastUsed time.Time
	IsReturn bool // true = reverse-flow binding from RecordInboundFlow
}

type TunnelStatsSnapshot struct {
	// Inbound (gravity → hadron → container)
	InboundReceived  uint64
	InboundDelivered uint64
	InboundDropped   uint64
	InboundBytes     uint64

	// Outbound (container → hadron → gravity)
	OutboundReceived uint64
	OutboundSent     uint64
	OutboundErrors   uint64
	OutboundBytes    uint64

	// Control plane
	PingsSent      uint64
	PongsReceived  uint64
	PingTimeouts   uint64
	LastPingSentUs int64
	LastPongRecvUs int64

	// Channel
	InboundChannelLen int
	InboundChannelCap int
	InboundHighWater  int32

	// Connection pool (point-in-time)
	TotalConnections      int
	HealthyConnections    int
	TotalTunnelStreams    int
	HealthyTunnelStreams  int
	TotalControlStreams   int
	HealthyControlStreams int

	// Per-stream metrics
	StreamMetrics map[string]StreamMetricsSnapshot
}