feat(dir2config)

vendor/novit.nc/direktil/pkg/localconfig/host.go

@@ -0,0 +1,30 @@
+package localconfig
+
+import (
+	"io"
+
+	yaml "gopkg.in/yaml.v2"
+)
+
+type Host struct {
+	Name string
+	MACs []string
+	IPs  []string
+
+	IPXE string
+
+	Kernel   string
+	Initrd   string
+	Versions map[string]string
+
+	Config string
+}
+
+func (h *Host) WriteHashDataTo(w io.Writer) error {
+	return yaml.NewEncoder(w).Encode(Host{
+		Kernel:   h.Kernel,
+		Initrd:   h.Initrd,
+		Versions: h.Versions,
+		Config:   h.Config,
+	})
+}

vendor/novit.nc/direktil/pkg/localconfig/localconfig.go

@@ -0,0 +1,94 @@
+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
+}