api

type AgentEventEnvelope struct {
	SessionID string
	Event     agent.AgentEvent
}

type AgentEventReceiver interface {
	Recv() (*AgentEventEnvelope, error)
	Close() error
}

type AgentEventSender interface {
	Send(*AgentEventEnvelope) error
	Close() error
}

type AgentEventService interface {
	StreamAgentEvents(ctx context.Context, req *StreamAgentEventsRequest) (AgentEventReceiver, error)
}

type Capabilities struct {
	Snapshots         bool
	Rollback          bool
	Pause             bool
	StreamingProgress bool
	TierRouting       bool
}

type ContentBlock struct {
	Type              string
	Text              string
	TextSignature     string
	Thinking          string
	ThinkingSignature string
	Redacted          bool
	ImageData         string
	ImageMimeType     string
	ToolCallID        string
	ToolCallName      string
	ToolCallArguments map[string]any
	ThoughtSignature  string
}

type CreateSessionRequest struct {
	BaseSnapshot string
	SessionID    string
	Quota        int64
	Labels       map[string]string
}

type CreateSessionResponse struct {
	Session            *Session
	ServerCapabilities Capabilities
}

type CreateSnapshotRequest struct {
	SessionID string
	Name      string
}

type CreateSnapshotResponse struct {
	SnapshotID string
	TurnNumber int
	SpaceUsed  int64
}

type DestroySessionRequest struct {
	SessionID string
}

type DestroySessionResponse struct{}

type ExecuteToolRequest struct {
	SessionID  string
	ToolCallID string
	ToolName   string
	Params     map[string]any
	Resources  *ResourceSpec
}

type ExecuteToolResponse struct {
	SessionID     string
	ToolCallID    string
	ToolName      string
	Content       string
	ContentBlocks []ContentBlock
	SnapshotID    string
	ExitCode      *int
	Tier          int
	Duration      time.Duration
}

type ExecuteToolStreamMessage struct {
	Progress *ToolProgress
	Response *ExecuteToolResponse
}

type ExecuteToolStreamReceiver interface {
	Recv() (*ExecuteToolStreamMessage, error)
	Close() error
}

type GetProcessStatusRequest struct {
	SessionID string
	ProcessID string
}

type GetProcessStatusResponse struct {
	Status   ProcessStatus
	ExitCode *int
}

type GetSessionRequest struct {
	SessionID string
}

type GetSessionResponse struct {
	Session *Session
}

type HealthCheckRequest struct{}

type HealthCheckResponse struct {
	Status       string
	PoolState    string
	SessionCount int
	ActiveTools  int
	Uptime       time.Duration
	Errors       []string
}

type KillProcessRequest struct {
	SessionID string
	ProcessID string
	Signal    int
}

type KillProcessResponse struct{}

type LaunchProcessRequest struct {
	SessionID  string
	Binary     string
	Args       []string
	Env        map[string]string
	ExposePort int
}

type LaunchProcessResponse struct {
	ProcessID string
	Address   string
	Status    ProcessStatus
}

type ListSnapshotsRequest struct {
	SessionID string
}

type ListSnapshotsResponse struct {
	Snapshots []Snapshot
}

type PauseSessionRequest struct {
	SessionID string
}

type PauseSessionResponse struct{}

type ProcessStatus string

type ResourceSpec struct {
	CPUs  float64
	MemMB int
}

type ResumeSessionRequest struct {
	SessionID string
}

type ResumeSessionResponse struct{}

type RollbackSessionRequest struct {
	SessionID  string
	SnapshotID string
}

type RollbackSessionResponse struct{}

type SandboxService interface {
	CreateSession(ctx context.Context, req *CreateSessionRequest) (*CreateSessionResponse, error)
	GetSession(ctx context.Context, req *GetSessionRequest) (*GetSessionResponse, error)
	PauseSession(ctx context.Context, req *PauseSessionRequest) (*PauseSessionResponse, error)
	ResumeSession(ctx context.Context, req *ResumeSessionRequest) (*ResumeSessionResponse, error)
	DestroySession(ctx context.Context, req *DestroySessionRequest) (*DestroySessionResponse, error)
	LaunchProcess(ctx context.Context, req *LaunchProcessRequest) (*LaunchProcessResponse, error)
	KillProcess(ctx context.Context, req *KillProcessRequest) (*KillProcessResponse, error)
	GetProcessStatus(ctx context.Context, req *GetProcessStatusRequest) (*GetProcessStatusResponse, error)
	ExecuteTool(ctx context.Context, req *ExecuteToolRequest) (*ExecuteToolResponse, error)
	ExecuteToolStream(ctx context.Context, req *ExecuteToolRequest) (ExecuteToolStreamReceiver, error)
	TurnComplete(ctx context.Context, req *TurnCompleteRequest) (*TurnCompleteResponse, error)
	CreateSnapshot(ctx context.Context, req *CreateSnapshotRequest) (*CreateSnapshotResponse, error)
	RollbackSession(ctx context.Context, req *RollbackSessionRequest) (*RollbackSessionResponse, error)
	ListSnapshots(ctx context.Context, req *ListSnapshotsRequest) (*ListSnapshotsResponse, error)
	HealthCheck(ctx context.Context, req *HealthCheckRequest) (*HealthCheckResponse, error)
}

type Session struct {
	ID         string
	State      string
	Mountpoint string
	TurnCount  int
	SnapCount  int
	Created    time.Time
	Labels     map[string]string
	SpaceUsed  int64
}

type Snapshot struct {
	Name      string
	Dataset   string
	Used      int64
	Refer     int64
	CreatedAt time.Time
	Holds     int
}

type StreamAgentEventsRequest struct {
	SessionID string
}

type StreamTerminalRequest struct {
	SessionID string
}

type TerminalAttached struct {
	Rows       int    `json:"rows"`
	Cols       int    `json:"cols"`
	Scrollback []byte `json:"scrollback,omitempty"`
}

type TerminalClientMessage struct {
	Input  *TerminalInput  `json:"input,omitempty"`
	Resize *TerminalResize `json:"resize,omitempty"`
}

type TerminalDetached struct {
	Reason string `json:"reason"` // "session_ended", "kicked", "error"
}

type TerminalInput struct {
	Data []byte `json:"data"`
}

type TerminalOutput struct {
	Data []byte `json:"data"`
}

type TerminalResize struct {
	Rows int `json:"rows"`
	Cols int `json:"cols"`
}

type TerminalServerMessage struct {
	Attached *TerminalAttached `json:"attached,omitempty"`
	Output   *TerminalOutput   `json:"output,omitempty"`
	Detached *TerminalDetached `json:"detached,omitempty"`
}

type TerminalService interface {
	StreamTerminal(ctx context.Context, req *StreamTerminalRequest) (TerminalStreamHandle, error)
}

type TerminalStreamHandle interface {
	Send(*TerminalClientMessage) error
	Recv() (*TerminalServerMessage, error)
	Close() error
}

type ToolProgress struct {
	Content string
	IsError bool
}

type TurnCompleteRequest struct {
	SessionID string
}

type TurnCompleteResponse struct {
	SnapshotID string
	TurnNumber int
	SpaceUsed  int64
}