sandbox

type ContainerRuntime string

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

type ExecuteToolRequest struct {
	SessionID  string
	ToolName   string
	ToolCallID string
	Params     map[string]any
	Resources  *gvisor.ResourceSpec
	OnProgress func(content string, isError bool)
}

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

type GetProcessStatusRequest struct {
	SessionID string
	ProcessID string
}

type GetProcessStatusResponse struct {
	Status   ProcessStatus
	ExitCode *int
}

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

type KillProcessRequest struct {
	SessionID string
	ProcessID string
	Signal    int
}

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 ManagedProcess struct {
	// contains filtered or unexported fields
}

type ProcessStatus string

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

type ServiceConfig struct {
	StorageBackend   StorageBackend
	ContainerRuntime ContainerRuntime

	AdvertiseAddr string

	PoolName               string
	BasesDataset           string
	SessionsDataset        string
	SessionsRootDir        string
	MaxSessions            int
	DefaultSessionQuota    int64
	SnapshotPrefix         string
	MaxSnapshotsPerSession int
	DefaultResources       gvisor.ResourceSpec
	ToolTimeout            time.Duration
	PoolSpaceWarnThreshold float64
	PoolSpaceCritThreshold float64
	HealthCheckInterval    time.Duration
	PauseDrainTimeout      time.Duration
	ShutdownTimeout        time.Duration
	PerToolSnapshots       bool
}

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

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

type SessionState string

type SnapshotEntry struct {
	Name           string
	IsTurnSnapshot bool
}

type SnapshotResult struct {
	SnapshotID string
	TurnNumber int
	SpaceUsed  int64
}

type StorageBackend string