95 lines
1.5 KiB
Go
95 lines
1.5 KiB
Go
package localconfig
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"strings"
|
|
|
|
yaml "gopkg.in/yaml.v2"
|
|
)
|
|
|
|
type Config struct {
|
|
Clusters []*Cluster
|
|
Hosts []*Host
|
|
SSLConfig string
|
|
}
|
|
|
|
type Cluster struct {
|
|
Name string
|
|
Addons string
|
|
}
|
|
|
|
func FromBytes(data []byte) (*Config, error) {
|
|
config := &Config{}
|
|
if err := yaml.Unmarshal(data, config); err != nil {
|
|
return nil, err
|
|
}
|
|
return config, nil
|
|
}
|
|
|
|
func FromFile(path string) (*Config, error) {
|
|
ba, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return FromBytes(ba)
|
|
}
|
|
|
|
func (c *Config) WriteTo(w io.Writer) error {
|
|
return yaml.NewEncoder(w).Encode(c)
|
|
}
|
|
|
|
func (c *Config) Cluster(name string) *Cluster {
|
|
for _, cluster := range c.Clusters {
|
|
if cluster.Name == name {
|
|
return cluster
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Config) ClusterByName(name string) *Cluster {
|
|
for _, cluster := range c.Clusters {
|
|
if cluster.Name == name {
|
|
return cluster
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Config) Host(name string) *Host {
|
|
for _, host := range c.Hosts {
|
|
if host.Name == name {
|
|
return host
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Config) HostByIP(ip string) *Host {
|
|
for _, host := range c.Hosts {
|
|
for _, hostIP := range host.IPs {
|
|
if hostIP == ip {
|
|
return host
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Config) HostByMAC(mac string) *Host {
|
|
// a bit of normalization
|
|
mac = strings.Replace(strings.ToLower(mac), "-", ":", -1)
|
|
|
|
for _, host := range c.Hosts {
|
|
for _, hostMAC := range host.MACs {
|
|
if strings.ToLower(hostMAC) == mac {
|
|
return host
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|