models

type AWSConfig struct {
	Region                string
	VPCID                 string
	SubnetID              string
	DefaultRegion         string
	DefaultMachineType    string
	DefaultDiskSizeGB     int32
	DefaultCountPerRegion int
}

type AWSResourceState int

type AzureConfig struct {
	ResourceGroupName     string
	ResourceGroupLocation string
	SubscriptionID        string
	DefaultVMSize         string
	DefaultDiskSizeGB     int32
	DefaultLocation       string
	Tags                  map[string]string
	DefaultCountPerZone   int
}

type AzureEvent struct {
	Type       string
	ResourceID string
	Message    string
}

type BacalhauSettings struct {
	Key   string      `json:"key"`
	Value interface{} `json:"value"`
}

type CloudSpecificInfo struct {
	// Azure specific
	ResourceGroupName string

	// GCP specific
	Zone string

	// Both GCP and AWS
	Region string

	// AWS specific
	SpotMarketOptions *ec2Types.InstanceMarketOptionsRequest
}

type ClusterDetails struct {
	Name              string
	DeploymentType    DeploymentType
	KubernetesVersion string
	NodeCount         int
	NodeTypes         []string
}

type Deployment struct {
	Name           string
	ViperPath      string
	DeploymentType DeploymentType
	Azure          *AzureConfig
	GCP            *GCPConfig
	AWS            *AWSConfig
	Machines       map[string]Machiner
	UniqueID       string
	StartTime      time.Time
	EndTime        time.Time

	Locations              []string
	AllowedPorts           []int
	SSHUser                string
	SSHPort                int
	SSHPublicKeyPath       string
	SSHPublicKeyMaterial   string
	SSHPrivateKeyPath      string
	SSHPrivateKeyMaterial  string
	OrchestratorIP         string
	Tags                   map[string]string
	ProjectServiceAccounts map[string]ServiceAccountInfo

	BacalhauSettings []BacalhauSettings
	CustomScriptPath string
	// contains filtered or unexported fields
}

type DeploymentStatus string

type DeploymentType string

type Disk struct {
	Name   string
	ID     string
	SizeGB int32
	State  armcompute.DiskState
}

type DisplayStatus struct {
	ID              string
	Type            ResourceType
	Location        string
	StatusMessage   string
	DetailedStatus  string
	ElapsedTime     time.Duration
	StartTime       time.Time
	InstanceID      string
	PublicIP        string
	PrivateIP       string
	HighlightCycles int
	Name            string
	Progress        int
	Orchestrator    bool
	SSH             ServiceState
	Docker          ServiceState
	CorePackages    ServiceState
	Bacalhau        ServiceState
	CustomScript    ServiceState
	StageComplete   bool
}

type GCPConfig struct {
	ProjectID              string
	OrganizationID         string
	DefaultRegion          string
	DefaultZone            string
	DefaultMachineType     string
	DefaultDiskSizeGB      int32
	BillingAccountID       string
	ServiceAccountEmail    string
	ProjectServiceAccounts map[string]ServiceAccountInfo
	Tags                   map[string]string
	DefaultCountPerZone    int
}

type GCPResourceState int

type Machine struct {
	ID       string
	Name     string
	Type     ResourceType
	Location string
	Region   string
	Zone     string
	NodeType string

	StatusMessage string
	Parameters    Parameters
	PublicIP      string
	PrivateIP     string
	StartTime     time.Time
	Failed        bool

	VMSize          string
	DiskSizeGB      int
	DiskImageFamily string
	DiskImageURL    string
	ImageID         string
	ElapsedTime     time.Duration
	Orchestrator    bool
	OrchestratorIP  string

	SSHUser               string
	SSHPrivateKeyMaterial []byte
	SSHPrivateKeyPath     string
	SSHPublicKeyMaterial  []byte
	SSHPublicKeyPath      string
	SSHPort               int

	CreationStartTime time.Time
	CreationEndTime   time.Time
	SSHStartTime      time.Time
	SSHEndTime        time.Time
	DockerStartTime   time.Time
	DockerEndTime     time.Time
	BacalhauStartTime time.Time
	BacalhauEndTime   time.Time
	DeploymentEndTime time.Time

	CustomScriptExecuted bool

	DeploymentType DeploymentType
	CloudProvider  DeploymentType
	CloudSpecific  CloudSpecificInfo
	// contains filtered or unexported fields
}

type MachineResource struct {
	ResourceName  string
	ResourceType  ResourceType
	ResourceState MachineResourceState
	ResourceValue string
}

type MachineResourceState int

type Machiner interface {
	// Basic information
	GetID() string
	GetName() string
	SetName(string)
	GetType() ResourceType
	GetVMSize() string
	GetDiskSizeGB() int
	SetDiskSizeGB(size int)
	IsFailed() bool
	SetFailed(bool)
	GetDiskImageFamily() string
	SetDiskImageFamily(family string)
	GetDiskImageURL() string
	SetDiskImageURL(url string)
	GetImageID() string
	SetImageID(id string)
	GetLocation() string
	SetLocation(location string) error
	GetRegion() string
	GetZone() string
	GetPublicIP() string
	SetPublicIP(ip string)
	GetPrivateIP() string
	SetPrivateIP(ip string)

	// AWS Spot Market Options
	GetSpotMarketOptions() *ec2Types.InstanceMarketOptionsRequest
	SetSpotMarketOptions(options *ec2Types.InstanceMarketOptionsRequest)

	// SSH related
	GetSSHUser() string
	SetSSHUser(user string)
	GetSSHPublicKeyMaterial() []byte
	SetSSHPublicKeyMaterial([]byte)
	GetSSHPublicKeyPath() string
	SetSSHPublicKeyPath(path string)
	GetSSHPrivateKeyMaterial() []byte
	SetSSHPrivateKeyMaterial([]byte)
	GetSSHPrivateKeyPath() string
	SetSSHPrivateKeyPath(path string)
	GetSSHPort() int
	SetSSHPort(port int)

	// Status and state
	GetNodeType() string
	SetNodeType(nodeType string)
	IsOrchestrator() bool
	SetOrchestrator(orchestrator bool)
	IsComplete() bool
	SetComplete()
	GetOrchestratorIP() string
	SetOrchestratorIP(ip string)
	GetStatusMessage() string
	SetStatusMessage(message string)

	// Resource management
	GetMachineResource(resourceType string) MachineResource
	SetMachineResource(resourceType string, resourceState MachineResourceState)
	GetMachineResources() map[string]MachineResource
	GetMachineResourceState(resourceName string) MachineResourceState
	SetMachineResourceState(resourceName string, state MachineResourceState)
	ResourcesComplete() (int, int)

	// Service management
	GetServices() map[string]ServiceType
	GetServiceState(serviceName string) ServiceState
	SetServiceState(serviceName string, state ServiceState)
	SSHEnabled() bool
	DockerEnabled() bool
	BacalhauEnabled() bool
	CustomScriptEnabled() bool
	EnsureMachineServices() error
	ServicesComplete() (int, int)

	// Installation methods
	InstallDockerAndCorePackages(ctx context.Context) error

	// Cloud provider
	CloudType() DeploymentType

	// Timing information
	GetStartTime() time.Time
	SetStartTime(startTime time.Time)
	GetDeploymentEndTime() time.Time
	SetDeploymentEndTime(endTime time.Time)
	GetElapsedTime() time.Duration
	SetElapsedTime(elapsedTime time.Duration)

	// Logging
	LogTimingInfo(logger logger.Logger)
}

type NodeInfo struct {
	Name              string `yaml:"name"`
	NodeIP            string `yaml:"node_ip"`
	OrchestratorIP    string `yaml:"orchestrator_ip,omitempty"`
	DockerInstalled   bool   `yaml:"docker_installed"`
	BacalhauInstalled bool   `yaml:"bacalhau_installed"`
	CustomScript      bool   `yaml:"custom_script_installed"`
	SSHUsername       string `yaml:"ssh_username"`
	SSHKeyPath        string `yaml:"ssh_key_path"`
}

type NodeInfoList struct {
	Nodes []NodeInfo `yaml:"nodes"`
}

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

type Parameters struct {
	Count        int
	Type         string
	Orchestrator bool
}

type ProviderAbbreviation string

type ProvisionProgress struct {
	CurrentStep    *ProvisionStep
	CompletedSteps []*ProvisionStep
	TotalSteps     int
	// contains filtered or unexported fields
}

type ProvisionStep struct {
	Name        string
	Description string
	Status      string
	Error       error
}

type ProvisionStepStatus string

type ResourceType struct {
	ResourceString    string
	ShortResourceName string
}

type ServiceAccountInfo struct {
	Email string
	Key   string
}

type ServiceState int

type ServiceType struct {
	Name  string
	State ServiceState
}

type StatusCode string

type StatusUpdateMsg struct {
	Status *DisplayStatus
}

type TimeUpdateMsg struct{}