feat(local-config): loaders

This commit is contained in:
Mikaël Cluseau 2018-12-10 17:34:40 +11:00
parent bc435a88a1
commit 3124a1bd36

View File

@ -1,8 +1,13 @@
package localconfig package localconfig
import "strings" import (
"io/ioutil"
"strings"
type LocalConfig struct { yaml "gopkg.in/yaml.v2"
)
type Config struct {
Clusters []*Cluster Clusters []*Cluster
Hosts []*Host Hosts []*Host
} }
@ -24,7 +29,24 @@ type Host struct {
Config []byte Config []byte
} }
func (c *LocalConfig) ClusterByName(name string) *Cluster { 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) ClusterByName(name string) *Cluster {
for _, cluster := range c.Clusters { for _, cluster := range c.Clusters {
if cluster.Name == name { if cluster.Name == name {
return cluster return cluster
@ -33,7 +55,7 @@ func (c *LocalConfig) ClusterByName(name string) *Cluster {
return nil return nil
} }
func (c *LocalConfig) HostByIP(ip string) *Host { func (c *Config) HostByIP(ip string) *Host {
for _, host := range c.Hosts { for _, host := range c.Hosts {
for _, hostIP := range host.IPs { for _, hostIP := range host.IPs {
if hostIP == ip { if hostIP == ip {
@ -44,7 +66,7 @@ func (c *LocalConfig) HostByIP(ip string) *Host {
return nil return nil
} }
func (c *LocalConfig) HostByMAC(mac string) *Host { func (c *Config) HostByMAC(mac string) *Host {
// a bit of normalization // a bit of normalization
mac = strings.Replace(strings.ToLower(mac), "-", ":", -1) mac = strings.Replace(strings.ToLower(mac), "-", ":", -1)