frame out cli tool

cmd/internal/config/config.go

@@ -0,0 +1,80 @@
+package config
+
+import (
+	"fmt"
+	"os"
+	"time"
+
+	"github.com/caarlos0/env/v11"
+	yaml "github.com/oasdiff/yaml3"
+)
+
+type ClientsConfig struct {
+	LogLevel  string `json:"logLevel" yaml:"logLevel" env:"LOG_LEVEL" envDefault:"warn"`
+	LogFormat string `json:"logFormat" yaml:"logFormat" env:"LOG_FORMAT" envDefault:"console"`
+
+	Clients []ClientConfig `json:"clients" yaml:"clients" envPrefix:"CLIENT_"`
+}
+
+type ClientType string
+
+const (
+	TypeEdgeOS      ClientType = "edgeOS"
+	TypeToughSwitch ClientType = "toughSwitch"
+)
+
+type ClientConfig struct {
+	Type     ClientType    `json:"type" yaml:"type" env:"TYPE"`
+	Name     string        `json:"name" yaml:"name" env:"NAME"`
+	Host     string        `json:"host" yaml:"host" env:"HOST"`
+	Scheme   string        `json:"scheme" yaml:"scheme" env:"SCHEME" envDefault:"https"`
+	User     string        `json:"user" yaml:"user" env:"USER"`
+	Pass     string        `json:"pass" yaml:"pass" env:"PASS"`
+	Insecure bool          `json:"insecure" yaml:"insecure" env:"INSECURE" envDefault:"false"`
+	Timeout  time.Duration `json:"timeout" yaml:"timeout" env:"TIMEOUT" envDefault:"10s"`
+}
+
+// LoadConfig will load a file if given, layering env on-top of the config
+// if present. Environment variables take the form:
+//   - LOG_LEVEL, LOG_FORMAT for top-level settings
+//   - CLIENT_0_NAME, CLIENT_0_HOST, CLIENT_0_TYPE, etc. for client array
+func LoadConfig(configPath *string) (*ClientsConfig, error) {
+	conf, err := env.ParseAs[ClientsConfig]()
+	if err != nil {
+		return nil, fmt.Errorf("could not parse env config: %w", err)
+	}
+
+	if configPath != nil && *configPath != "" {
+		file, err := os.Open(*configPath)
+		if err != nil {
+			return nil, fmt.Errorf("could not open config file: %w", err)
+		}
+		defer file.Close()
+
+		decoder := yaml.NewDecoder(file)
+		if err := decoder.Decode(&conf); err != nil {
+			return nil, fmt.Errorf("could not decode config file: %w", err)
+		}
+	}
+
+	return &conf, nil
+}
+
+// GetClientByName returns a client config by its name, or nil if not found.
+func (c *ClientsConfig) GetClientByName(name string) *ClientConfig {
+	for i := range c.Clients {
+		if c.Clients[i].Name == name {
+			return &c.Clients[i]
+		}
+	}
+	return nil
+}
+
+// GetClientNames returns a list of all configured client names.
+func (c *ClientsConfig) GetClientNames() []string {
+	names := make([]string, len(c.Clients))
+	for i, client := range c.Clients {
+		names[i] = client.Name
+	}
+	return names
+}