sessionallocation

type AllocationRequest struct {
	SessionID          string                           `json:"session_id"`
	ManagerID          string                           `json:"manager_id,omitempty"`
	Request            *entities.RunServerRequest       `json:"request"`
	ProvisionSettings  *sessionsettings.SessionSettings `json:"provision_settings,omitempty"`
	WebhookPayload     []byte                           `json:"webhook_payload,omitempty"`
	Status             Status                           `json:"status"`
	Message            string                           `json:"message,omitempty"`
	AllocatedSessionID string                           `json:"allocated_session_id,omitempty"`
	Requirements       Requirements                     `json:"requirements"`
	UpdatedAt          time.Time                        `json:"updated_at"`
}

type AllocationResult struct {
	Status             Status `json:"status"`
	Message            string `json:"message,omitempty"`
	AllocatedSessionID string `json:"allocated_session_id,omitempty"`
	ProxyURL           string `json:"proxy_url,omitempty"`
}

type ExternalAllocatorClient interface {
	NextExternal(ctx context.Context, wait time.Duration) (*AllocationRequest, bool, error)
	CompleteExternal(ctx context.Context, sessionID string, result AllocationResult) error
}

type InternalAllocatorClient interface {
	Next(ctx context.Context, wait time.Duration) (*AllocationRequest, bool, error)
	Complete(ctx context.Context, sessionID string, result AllocationResult) error
}

type Notifier interface {
	Notify(ctx context.Context) error
	Subscribe(ctx context.Context) (<-chan struct{}, func(), error)
}

type Queue interface {
	SubmitExternalSessionAllocation(ctx context.Context, managerID, sessionID string, settings *sessionsettings.SessionSettings, req *entities.RunServerRequest) error
	NextSessionAllocation(ctx context.Context, wait time.Duration) (*AllocationRequest, bool, error)
	CompleteSessionAllocation(ctx context.Context, sessionID string, result AllocationResult) (*AllocationRequest, error)
	NextExternalSessionAllocation(ctx context.Context, managerID string, wait time.Duration) (*AllocationRequest, bool, error)
	CompleteExternalSessionAllocation(ctx context.Context, sessionID string, result AllocationResult) (*AllocationRequest, error)
}

type Requirements struct {
	AgentType string `json:"agent_type,omitempty"`
	// Sandbox is always true (network filter cannot be opted out).
	// The field is kept for backward compatibility with existing JSON.
	Sandbox bool `json:"sandbox"`
	DinD    bool `json:"dind"`
}

type Status string