abstractions

type AutoscaledInstance struct {
	Ctx                      context.Context
	AppConfig                types.AppConfig
	CancelFunc               context.CancelFunc
	Name                     string
	Rdb                      *common.RedisClient
	Lock                     *common.RedisLock
	IsActive                 bool
	FailedContainerThreshold int

	// DB objects
	Workspace  *types.Workspace
	Stub       *types.StubWithRelated
	StubConfig *types.StubConfigV1
	Object     *types.Object
	Token      *types.Token

	// Scheduling
	Scheduler          *scheduler.Scheduler
	ContainerEventChan chan types.ContainerEvent
	Containers         map[string]bool
	ScaleEventChan     chan int
	EntryPoint         []string
	Autoscaler         IAutoscaler

	// Repositories
	ContainerRepo    repository.ContainerRepository
	BackendRepo      repository.BackendRepository
	TaskRepo         repository.TaskRepository
	EventRepo        repository.EventRepository
	UsageMetricsRepo repository.UsageMetricsRepository

	// Keys
	InstanceLockKey string

	// Callbacks
	StartContainersFunc func(containersToRun int) error
	StopContainersFunc  func(containersToStop int) error
}

type AutoscaledInstanceConfig struct {
	Name                string
	AppConfig           types.AppConfig
	Workspace           *types.Workspace
	Stub                *types.StubWithRelated
	StubConfig          *types.StubConfigV1
	Object              *types.Object
	Token               *types.Token
	Scheduler           *scheduler.Scheduler
	Rdb                 *common.RedisClient
	InstanceLockKey     string
	ContainerRepo       repository.ContainerRepository
	BackendRepo         repository.BackendRepository
	EventRepo           repository.EventRepository
	TaskRepo            repository.TaskRepository
	UsageMetricsRepo    repository.UsageMetricsRepository
	StartContainersFunc func(containersToRun int) error
	StopContainersFunc  func(containersToStop int) error
}

type AutoscaledInstanceState struct {
	RunningContainers  int
	PendingContainers  int
	StoppingContainers int
	FailedContainers   []string
}

type Autoscaler[I IAutoscaledInstance, S AutoscalerSample] struct {
	// contains filtered or unexported fields
}

type AutoscalerResult struct {
	DesiredContainers int
	ResultValid       bool
}

type AutoscalerSample interface{}

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

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

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

type ContainerSecrets struct {
	Env      []string
	Bindings []types.SecretBinding
}

type ContainerStopper interface {
	Stop(*types.StopContainerArgs) error
}

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

type ContainerStreamOpts struct {
	SendCallback    func(o common.OutputMsg) error
	ExitCallback    func(exitCode int32) error
	ContainerRepo   repository.ContainerRepository
	Tailscale       *network.Tailscale
	Config          types.AppConfig
	KeyEventManager *common.KeyEventManager
	SyncQueue       chan *pb.SyncContainerWorkspaceRequest
}

type DurableDiskPlacementRepos struct {
	BackendRepo    repository.BackendRepository
	ComputeRepo    repository.ComputeRepository
	WorkerRepo     repository.WorkerRepository
	WorkerPoolRepo repository.WorkerPoolRepository
}

type GPUPoolChecker interface {
	HasManagedPoolForGPU(gpuType string) bool
}

type IAutoscaledInstance interface {
	ConsumeScaleResult(*AutoscalerResult)
	ConsumeContainerEvent(types.ContainerEvent)
	HandleScalingEvent(int) error
	Sync() error
}

type IAutoscaler interface {
	Start(ctx context.Context)
	Trigger()
}

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

type ProxyBufferPool struct{}

type RingBuffer[T any] struct {
	// contains filtered or unexported fields
}

type TunnelClient interface {
	Tunnel(context.Context, *pb.ContainerTunnelRequest) (*pb.ContainerTunnelResponse, error)
}