81 lines
2.4 KiB
Go
81 lines
2.4 KiB
Go
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" default:"warn"`
|
|
LogFormat string `json:"logFormat" yaml:"logFormat" env:"LOG_FORMAT" default:"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" default:"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" default:"false"`
|
|
Timeout time.Duration `json:"timeout" yaml:"timeout" env:"TIMEOUT" default:"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
|
|
}
|