studio

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

type Architecture struct {
	OS   string // darwin, linux, windows
	Arch string // arm64, amd64, 386
}

type AutoStartableBackend interface {
	Backend
	// IsInstalled checks if the backend software is installed (but not necessarily running)
	IsInstalled(ctx context.Context) bool
	// EnsureRunning ensures the backend is running, starting it if necessary
	EnsureRunning(ctx context.Context) error
}

type Backend interface {
	// Name returns the backend name
	Name() string

	// Mode returns the mode this backend handles
	Mode() Mode

	// IsAvailable checks if this backend is available on the system
	IsAvailable(ctx context.Context) bool

	// Create creates a new environment
	Create(ctx context.Context, opts *CreateOptions) (*Environment, error)

	// Start starts a stopped environment
	Start(ctx context.Context, envID string) error

	// Stop stops a running environment
	Stop(ctx context.Context, envID string) error

	// Remove removes an environment
	Remove(ctx context.Context, envID string) error

	// List lists all environments
	List(ctx context.Context) ([]*Environment, error)

	// Get gets an environment by ID or name
	Get(ctx context.Context, idOrName string) (*Environment, error)

	// Exec executes a command in an environment
	Exec(ctx context.Context, envID string, cmd []string) ([]byte, error)

	// Logs returns logs from an environment
	Logs(ctx context.Context, envID string, follow bool) (<-chan string, error)
}

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

type ContainerSetupConfig struct {
	// StudioName is the name of the studio (used for config file paths)
	StudioName string
	// GPUWorkerURL is the connection URL to the GPU worker
	GPUWorkerURL string
	// HardwareVendor is the GPU vendor (nvidia, amd, hygon)
	HardwareVendor string
	// MountUserHome indicates whether to mount the user's home directory
	MountUserHome bool
	// UserHomeContainerPath is the path to mount user home in container (default: /home/user/host)
	UserHomeContainerPath string
}

type ContainerSetupResult struct {
	// EnvVars are environment variables to set in the container
	EnvVars map[string]string
	// VolumeMounts are volumes to mount in the container
	VolumeMounts []VolumeMount
	// LibrariesDownloaded indicates if libraries were downloaded during setup
	LibrariesDownloaded bool
}

type CreateOptions struct {
	Name           string            `json:"name"`
	Mode           Mode              `json:"mode"`
	Image          string            `json:"image"`
	GPUWorkerURL   string            `json:"gpu_worker_url,omitempty"`  // TENSOR_FUSION_OPERATOR_CONNECTION_INFO
	HardwareVendor string            `json:"hardware_vendor,omitempty"` // nvidia, amd, hygon
	SSHPublicKey   string            `json:"ssh_public_key,omitempty"`
	WorkDir        string            `json:"work_dir,omitempty"`
	Ports          []PortMapping     `json:"ports,omitempty"`
	Envs           map[string]string `json:"envs,omitempty"`
	Volumes        []VolumeMount     `json:"volumes,omitempty"`
	Resources      ResourceSpec      `json:"resources,omitempty"`
	Labels         map[string]string `json:"labels,omitempty"`
	// NoUserVolume disables automatic mounting of user's home directory
	NoUserVolume bool `json:"no_user_volume,omitempty"`
	// Command specifies the container startup command or supplements ENTRYPOINT args
	Command []string `json:"command,omitempty"`
	// Endpoint overrides the GPU worker endpoint URL
	Endpoint string `json:"endpoint,omitempty"`
	// Platform specifies the container image platform (e.g., linux/amd64, linux/arm64)
	// If empty, auto-detect from VM/host architecture
	Platform string `json:"platform,omitempty"`
}

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

type Environment struct {
	ID            string            `json:"id"`
	Name          string            `json:"name"`
	Mode          Mode              `json:"mode"`
	Image         string            `json:"image"`
	Status        EnvironmentStatus `json:"status"`
	ConnectionURL string            `json:"connection_url,omitempty"`
	SSHHost       string            `json:"ssh_host,omitempty"`
	SSHPort       int               `json:"ssh_port,omitempty"`
	SSHUser       string            `json:"ssh_user,omitempty"`
	WorkDir       string            `json:"work_dir,omitempty"`
	GPUWorkerURL  string            `json:"gpu_worker_url,omitempty"`
	CreatedAt     time.Time         `json:"created_at"`
	Labels        map[string]string `json:"labels,omitempty"`
}

type EnvironmentStatus string

type GPUEnvConfig struct {
	Vendor        GPUVendor
	ConnectionURL string // TENSOR_FUSION_OPERATOR_CONNECTION_INFO value
	CachePath     string // Path to gpugo cache directory (for binaries like tensor-fusion-worker)
	LibsPath      string // Path to gpugo libs directory (for .so/.dll files, used in LD paths)
	LogPath       string // Path to logs directory
	StudioName    string // Name of the studio (for creating config files)
	IsContainer   bool   // Whether this is for a container (affects paths)
}

type GPUEnvResult struct {
	EnvVars         map[string]string
	LDSoConfPath    string        // Host path to ld.so.conf file
	LDSoPreloadPath string        // Host path to ld.so.preload file
	VolumeMounts    []VolumeMount // Additional volume mounts needed
}

type GPUVendor string

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

type LibraryInfo struct {
	Name     string
	Version  string
	Arch     Architecture
	Checksum string // SHA256 checksum
}

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

type Mode string

type PortMapping struct {
	HostPort      int    `json:"host_port"`
	ContainerPort int    `json:"container_port"`
	Protocol      string `json:"protocol,omitempty"` // tcp, udp
}

type ResourceSpec struct {
	CPUs   float64 `json:"cpus,omitempty"`
	Memory string  `json:"memory,omitempty"` // e.g., "8Gi"
}

type SSHConfig struct {
	Host         string
	HostName     string
	User         string
	Port         int
	IdentityFile string
	ExtraOptions map[string]string
}

type StudioImage struct {
	Name        string            `json:"name"`
	Tag         string            `json:"tag"`
	Description string            `json:"description"`
	Features    []string          `json:"features"` // torch, cuda, ssh, jupyter, etc.
	Size        string            `json:"size"`
	Registry    string            `json:"registry"`
	Labels      map[string]string `json:"labels,omitempty"`
}

type VolumeMount struct {
	HostPath      string `json:"host_path"`
	ContainerPath string `json:"container_path"`
	ReadOnly      bool   `json:"read_only,omitempty"`
}

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

type WSLStatus struct {
	WSLInstalled     bool
	HasDistribution  bool
	DistributionName string
	DockerInstalled  bool
	DockerRunning    bool
	ErrorMessage     string
}