orama/core/pkg/cli/remotessh/config.go

package remotessh

import (
	"fmt"
	"os"
	"path/filepath"

	"github.com/DeBrosOfficial/network/pkg/inspector"
)

// FindNodesConf searches for the nodes.conf file
// in common locations relative to the current directory or project root.
func FindNodesConf() string {
	candidates := []string{
		"scripts/nodes.conf",
		"../scripts/nodes.conf",
		"network/scripts/nodes.conf",
	}

	// Also check from home dir
	home, _ := os.UserHomeDir()
	if home != "" {
		candidates = append(candidates, filepath.Join(home, ".orama", "nodes.conf"))
	}

	for _, c := range candidates {
		if _, err := os.Stat(c); err == nil {
			return c
		}
	}
	return ""
}

// LoadEnvNodes loads all nodes for a given environment from nodes.conf.
// SSHKey fields are NOT set — caller must call PrepareNodeKeys() after this.
func LoadEnvNodes(env string) ([]inspector.Node, error) {
	confPath := FindNodesConf()
	if confPath == "" {
		return nil, fmt.Errorf("nodes.conf not found (checked scripts/, ../scripts/, network/scripts/)")
	}

	nodes, err := inspector.LoadNodes(confPath)
	if err != nil {
		return nil, fmt.Errorf("failed to load %s: %w", confPath, err)
	}

	filtered := inspector.FilterByEnv(nodes, env)
	if len(filtered) == 0 {
		return nil, fmt.Errorf("no nodes found for environment %q in %s", env, confPath)
	}

	return filtered, nil
}

// PickHubNode selects the first node as the hub for fanout distribution.
func PickHubNode(nodes []inspector.Node) inspector.Node {
	return nodes[0]
}

// FilterByIP returns nodes matching the given IP address.
func FilterByIP(nodes []inspector.Node, ip string) []inspector.Node {
	var filtered []inspector.Node
	for _, n := range nodes {
		if n.Host == ip {
			filtered = append(filtered, n)
		}
	}
	return filtered
}