cli/pkg/apmintegration/file.go

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

package lpmintegration

import (
	"os"
	"path/filepath"
	"strings"

	"github.com/luxfi/lpm/types"
	"github.com/luxfi/cli/v2/pkg/application"
	"gopkg.in/yaml.v3"
)

const yamlExt = ".yaml"

func GetRepos(app *application.Lux) ([]string, error) {
	repositoryDir := filepath.Join(app.ApmDir, "repositories")
	orgs, err := os.ReadDir(repositoryDir)
	if err != nil {
		return []string{}, err
	}

	output := []string{}

	for _, org := range orgs {
		repoDir := filepath.Join(repositoryDir, org.Name())
		repos, err := os.ReadDir(repoDir)
		if err != nil {
			return []string{}, err
		}
		for _, repo := range repos {
			output = append(output, org.Name()+"/"+repo.Name())
		}
	}

	return output, nil
}

func GetSubnets(app *application.Lux, repoAlias string) ([]string, error) {
	subnetDir := filepath.Join(app.ApmDir, "repositories", repoAlias, "subnets")
	subnets, err := os.ReadDir(subnetDir)
	if err != nil {
		return []string{}, err
	}
	subnetOptions := make([]string, len(subnets))
	for i, subnet := range subnets {
		// Remove the .yaml extension
		subnetOptions[i] = strings.TrimSuffix(subnet.Name(), filepath.Ext(subnet.Name()))
	}

	return subnetOptions, nil
}

type SubnetWrapper struct {
	Subnet types.Subnet `yaml:"subnet"`
}

type VMWrapper struct {
	VM types.VM `yaml:"vm"`
}

func LoadSubnetFile(app *application.Lux, subnetKey string) (types.Subnet, error) {
	repoAlias, subnetName, err := splitKey(subnetKey)
	if err != nil {
		return types.Subnet{}, err
	}

	subnetYamlPath := filepath.Join(app.ApmDir, "repositories", repoAlias, "subnets", subnetName+yamlExt)
	var subnetWrapper SubnetWrapper

	subnetYamlBytes, err := os.ReadFile(subnetYamlPath)
	if err != nil {
		return types.Subnet{}, err
	}

	err = yaml.Unmarshal(subnetYamlBytes, &subnetWrapper)
	if err != nil {
		return types.Subnet{}, err
	}

	return subnetWrapper.Subnet, nil
}

func getVMsInSubnet(app *application.Lux, subnetKey string) ([]string, error) {
	subnet, err := LoadSubnetFile(app, subnetKey)
	if err != nil {
		return []string{}, err
	}

	return subnet.VMs, nil
}

func LoadVMFile(app *application.Lux, repo, vm string) (types.VM, error) {
	vmYamlPath := filepath.Join(app.ApmDir, "repositories", repo, "vms", vm+yamlExt)
	var vmWrapper VMWrapper

	vmYamlBytes, err := os.ReadFile(vmYamlPath)
	if err != nil {
		return types.VM{}, err
	}

	err = yaml.Unmarshal(vmYamlBytes, &vmWrapper)
	if err != nil {
		return types.VM{}, err
	}

	return vmWrapper.VM, nil
}