sdk/models/cloud.go

// Copyright (C) 2025, Lux Industries, Inc. All rights reserved.
// See the file LICENSE for licensing terms.

package models

import (
	"maps"
	"slices"
)

type RegionConfig struct {
	InstanceIDs       []string
	APIInstanceIDs    []string
	PublicIPs         []string
	KeyPair           string
	SecurityGroup     string
	CertFilePath      string
	ImageID           string
	Prefix            string
	CertName          string
	SecurityGroupName string
	NumNodes          int
	InstanceType      string
}

type CloudConfig map[string]RegionConfig

// GetRegions returns a slice of strings representing the regions of the RegionConfig.
func (ccm *CloudConfig) GetRegions() []string {
	return slices.Collect(maps.Keys(*ccm))
}

// GetAllInstanceIDs returns all instance IDs
func (ccm *CloudConfig) GetAllInstanceIDs() []string {
	instanceIDs := []string{}
	for _, cloudConfig := range *ccm {
		instanceIDs = append(instanceIDs, cloudConfig.InstanceIDs...)
	}
	return instanceIDs
}

// GetAllAPIInstanceIDs returns all API instance IDs
func (ccm *CloudConfig) GetAllAPIInstanceIDs() []string {
	apiInstanceIDs := []string{}
	for _, cloudConfig := range *ccm {
		apiInstanceIDs = append(apiInstanceIDs, cloudConfig.APIInstanceIDs...)
	}
	return apiInstanceIDs
}

// GetInstanceIDsForRegion returns instance IDs for specific region
func (ccm *CloudConfig) GetInstanceIDsForRegion(region string) []string {
	if regionConf, ok := (*ccm)[region]; ok {
		return regionConf.InstanceIDs
	}
	return []string{}
}

// GetAPIInstanceIDsForRegion returns API instance IDs for specific region
func (ccm *CloudConfig) GetAPIInstanceIDsForRegion(region string) []string {
	if regionConf, ok := (*ccm)[region]; ok {
		return regionConf.APIInstanceIDs
	}
	return []string{}
}