runnercontrol

type CommandClaimTiming struct {
	PoolAcquire time.Duration
	Query       time.Duration
	Decode      time.Duration
}

type CommandDelivery struct {
	ID          string
	Kind        string
	CreatedAt   time.Time
	DeliveredAt time.Time
	Message     *runnerv1.ControlPlaneToRunner
	ClaimTiming CommandClaimTiming
}

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

type CredentialAuthorityConfig struct {
	Credential    string
	CACertificate *x509.Certificate
}

type CredentialVerifier interface {
	VerifyClientCertificate(context.Context, *x509.Certificate, string) (RunnerIdentity, error)
}

type DataPlaneAdmission struct {
	ID                      string
	StreamID                string
	TenantRef               string
	SubjectRef              string
	SandboxID               string
	LeaseID                 string
	Generation              int64
	Kind                    string
	Operation               string
	RequestID               string
	IdempotencyKey          string
	RequestHash             string
	DeadlineAt              time.Time
	MaximumResponseBytes    int64
	MaximumRequestBytes     int64
	StreamWindowBytes       int64
	Priority                int64
	ExecOpen                *runnerv1.ExecOpen
	DeferResponseCredit     bool
	UseProfileRequestLimit  bool
	UseProfileResponseLimit bool
	UseProfileStreamWindow  bool
	Detachable              bool
	FileOpen                *runnerv1.FileOpen
	FileContent             []byte
	Request                 any
	CredentialDigest        []byte
	Now                     time.Time
}

type DataPlaneCompletion struct {
	TenantRef  string
	SubjectRef string
	SessionID  string
	Exec       *runnerv1.ExecBufferedResult
	File       *FileCompletion
	Now        time.Time
}

type DataPlaneSession struct {
	ID                       string
	StreamID                 string
	TenantRef                string
	SubjectRef               string
	SandboxID                string
	ProfileRevisionID        string
	AssignmentID             string
	InstanceID               string
	RunnerID                 string
	Generation               int64
	FencingToken             []byte
	RequestID                string
	LeaseID                  string
	Kind                     string
	Operation                string
	State                    string
	DeadlineAt               time.Time
	MaximumResponseBytes     int64
	MaximumRequestBytes      int64
	StreamWindowBytes        int64
	ResponseCreditBytes      int64
	RequestStreamBytes       int64
	RequestStreamClosed      bool
	Detachable               bool
	TerminalDetachSeconds    int64
	AttachmentID             string
	AttachedAt               *time.Time
	DetachedAt               *time.Time
	DetachExpiresAt          *time.Time
	OutboundBytes            int64
	InboundBytes             int64
	NextClientSequence       int64
	NextInboundSequence      int64
	NextOutboundSequence     int64
	TerminalKind             string
	TerminalDetail           string
	ExitCode                 int32
	Signal                   int32
	SpawnFailureReason       string
	ElapsedMilliseconds      int64
	LimitBytes               int64
	InfrastructureReason     string
	Retryable                bool
	TerminalMessage          string
	Stdout                   []byte
	Stderr                   []byte
	Content                  []byte
	Metadata                 *runnerv1.FileMetadata
	CreatedAt                time.Time
	UpdatedAt                time.Time
	CompletedAt              *time.Time
	RetainUntil              time.Time
	RequestJSON              []byte `json:"-"`
	Transport                string
	DataPlaneAddress         string
	DataPlaneCertificateSPKI string
}

type DirectDataPlaneAdmitter interface {
	ConsumeDirectDataPlane(context.Context, DirectDataPlaneConsumption) error
}

type DirectDataPlaneConsumption struct {
	SessionID        string
	AssignmentID     string
	Generation       int64
	FencingToken     []byte
	CredentialDigest []byte
	Now              time.Time
}

type DirectPortAdmitter interface {
	ConsumeDirectPortSession(context.Context, DirectPortConsumption) (PortTunnel, error)
}

type DirectPortConsumption struct {
	RunnerID         string
	SessionID        string
	AssignmentID     string
	Generation       int64
	FencingToken     []byte
	CredentialDigest []byte
	Now              time.Time
}

type Event struct {
	Kind         EventKind
	RunnerID     string
	ConnectionID string
	Response     *runnerv1.ControlPlaneToRunner
	Registration *runnerv1.RunnerRegistration
	Heartbeat    *runnerv1.RunnerHeartbeat
	Message      *runnerv1.RunnerToControlPlane
}

type EventKind string

type EventPersistenceRecord struct {
	Event      Event
	ReceivedAt time.Time
}

type ExecClientFrame struct {
	Sequence int64
	Input    []byte
	EndInput bool
	Credit   int64
	Cancel   bool
}

type ExecServerFrame struct {
	Sequence int64
	Output   *runnerv1.ExecOutput
	Terminal *runnerv1.ExecTerminal
}

type FileCompletion struct {
	Metadata *runnerv1.FileMetadata
	Content  []byte
	Terminal *runnerv1.FileTerminal
}

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

type LiveDataPlaneMetricsSnapshot struct {
	DroppedRouteNotFoundFrames uint64
}

type LiveDataPlaneSender interface {
	Send(*runnerv1.ControlPlaneToRunner) error
}

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

type PortSessionAdmission struct {
	Session        contracts.PortSession
	StreamID       string
	TenantRef      string
	SubjectRef     string
	RequestID      string
	LeaseID        string
	IdempotencyKey string
	RequestHash    string
	// CredentialDigest binds the single-use credential to this admission so the
	// home Runner can reject a mismatch locally without ever holding the
	// credential itself.
	CredentialDigest []byte
	Now              time.Time
}

type PortSessionFrameRecorder interface {
	RecordPortSessionFrame(context.Context, RunnerDataPlaneFrame, time.Time) (bool, error)
}

type PortSessionStore interface {
	AdmitPortSession(context.Context, PortSessionAdmission) (PortTunnel, bool, error)
	GetPortTunnel(context.Context, string, string, string, string, time.Time) (PortTunnel, error)
	ClosePortSession(context.Context, PortTunnelClose) (contracts.PortSession, error)
	ConsumePortSession(context.Context, string, string, string, time.Time) (PortTunnel, error)
	ConsumeDirectPortSession(context.Context, DirectPortConsumption) (PortTunnel, error)
	RecordPortClientBytes(context.Context, string, string, string, []byte, time.Time) error
	RecordPortTunnelAcknowledgement(context.Context, string, string, string, int64, time.Time) error
}

type PortTunnel struct {
	Session                     contracts.PortSession
	TenantRef                   string
	SubjectRef                  string
	RequestID                   string
	LeaseID                     string
	ProfileRevisionID           string
	AssignmentID                string
	InstanceID                  string
	RunnerID                    string
	StreamID                    string
	FencingToken                []byte
	GuestPort                   int64
	StreamWindowBytes           int64
	MaximumRequestBytes         int64
	MaximumResponseBytes        int64
	AcknowledgedInboundSequence int64
	// DataPlaneAddress is the home Runner's advertised caller-facing address. It
	// is returned only to an ingress holding the exact direct-endpoint grant.
	DataPlaneAddress string
	// DataPlaneCertificateSPKISHA256 is the admitted caller-facing certificate
	// public key. It is returned only with DataPlaneAddress.
	DataPlaneCertificateSPKISHA256 string
}

type PortTunnelClose struct {
	TenantRef      string
	SubjectRef     string
	SandboxID      string
	SessionID      string
	Generation     int64
	IdempotencyKey string
	RequestHash    string
	Reason         string
	Now            time.Time
}

type PortTunnelEvent struct {
	Sequence       int64
	Bytes          []byte
	TerminalKind   string
	TerminalDetail string
}

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

type PostgresDataPlaneStoreConfig struct {
	DatabaseURL         string
	Retention           time.Duration
	MaximumSessionBytes int64
}

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

type ProtocolStateStore interface {
	OpenConnection(context.Context, RunnerIdentity, string, uint32, time.Time) error
	CloseConnection(context.Context, string, string, time.Time) error
	RecordRegistration(context.Context, *runnerv1.RunnerRegistration, time.Time) (bool, error)
	RecordHeartbeat(context.Context, *runnerv1.RunnerHeartbeat, time.Time) (bool, error)
	RecordEvents(context.Context, []EventPersistenceRecord) error
	ClaimCommands(context.Context, string, string, int64, time.Time) ([]CommandDelivery, error)
	MarkCommandsDelivered(context.Context, []CommandDelivery, string) error
}

type PublicDataPlaneCancellation struct {
	TenantRef        string
	SubjectRef       string
	SandboxID        string
	SessionID        string
	SessionKind      string
	SessionOperation string
	IdempotencyKey   string
	RequestHash      string
	Reason           string
	Generation       int64
	Now              time.Time
	IdempotencyEnds  time.Time
}

type RunnerDataPlaneFrame struct {
	RunnerID     string
	ConnectionID string
	Message      *runnerv1.RunnerToControlPlane
}

type RunnerIdentity struct {
	RunnerID               string
	CredentialSerial       string
	CertificateFingerprint string
}

type Server struct {
	runnerv1.UnimplementedRunnerControlServer
	// contains filtered or unexported fields
}

type ServerConfig struct {
	CredentialVerifier  CredentialVerifier
	StateStore          ProtocolStateStore
	LiveDataPlane       *LiveDataPlaneBroker
	DirectPorts         DirectPortAdmitter
	PortSessions        PortSessionFrameRecorder
	DirectDataPlane     DirectDataPlaneAdmitter
	WorkspaceTransfers  WorkspaceTransferBroker
	SupportedVersions   VersionRange
	EnabledFeatures     []runnerv1.RunnerFeature
	HeartbeatInterval   time.Duration
	CommandPollInterval time.Duration
	CommandBatchSize    int64
	EventBatchSize      int
	EventBatchWait      time.Duration
	WorkWakeups         worknotify.Source
	Now                 func() time.Time
	NewConnectionID     func() string
}

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

type SessionConfig struct {
	AuthenticatedRunnerID string
	SupportedVersions     VersionRange
	EnabledFeatures       []runnerv1.RunnerFeature
	HeartbeatInterval     time.Duration
	ConnectionID          string
}

type TerminalCheckpoint struct {
	AttachmentID        string
	NextClientSequence  int64
	RequestBytes        int64
	ResponseCredit      int64
	InboundBytes        int64
	NextInboundSequence int64
	RecoveryAllowance   int64
	Cancel              bool
	Terminal            *runnerv1.ExecTerminal
}

type TerminalClientFrame struct {
	Sequence      int64
	Input         []byte
	ResizeRows    uint32
	ResizeColumns uint32
	Credit        int64
	Cancel        bool
}

type TerminalServerFrame struct {
	Sequence int64
	Output   []byte
	Terminal *runnerv1.ExecTerminal
}

type VersionRange struct {
	Minimum uint32
	Maximum uint32
}

type WorkspaceTransferBroker interface {
	Register(string, string, controlPlaneFrameSender)
	Unregister(string, string)
	Handle(context.Context, string, *runnerv1.WorkspaceTransferFrame, time.Time) error
}

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