package config

import (
	"io"
	"io/ioutil"

	yaml "gopkg.in/yaml.v2"
)

// Config represents the configuration for the exporter
type Config struct {
	Devices  []Device `yaml:"devices"`
	Features struct {
		BGP       bool `yaml:"bgp,omitempty"`
		Conntrack bool `yaml:"conntrack,omitempty"`
		DHCP      bool `yaml:"dhcp,omitempty"`
		DHCPL     bool `yaml:"dhcpl,omitempty"`
		DHCPv6    bool `yaml:"dhcpv6,omitempty"`
		Firmware  bool `yaml:"firmware,omitempty"`
		Health    bool `yaml:"health,omitempty"`
		Routes    bool `yaml:"routes,omitempty"`
		POE       bool `yaml:"poe,omitempty"`
		Pools     bool `yaml:"pools,omitempty"`
		Optics    bool `yaml:"optics,omitempty"`
		W60G      bool `yaml:"w60g,omitempty"`
		WlanSTA   bool `yaml:"wlansta,omitempty"`
		WlanIF    bool `yaml:"wlanif,omitempty"`
		Monitor   bool `yaml:"monitor,omitempty"`
		Ipsec     bool `yaml:"ipsec,omitempty"`
	} `yaml:"features,omitempty"`
}

// Device represents a target device
type Device struct {
	Name     string    `yaml:"name"`
	Address  string    `yaml:"address,omitempty"`
	Srv      SrvRecord `yaml:"srv,omitempty"`
	User     string    `yaml:"user"`
	Password string    `yaml:"password"`
	Port     string    `yaml:"port"`
}

type SrvRecord struct {
	Record string    `yaml:"record"`
	Dns    DnsServer `yaml:"dns,omitempty"`
}
type DnsServer struct {
	Address string `yaml:"address"`
	Port    int    `yaml:"port"`
}

// Load reads YAML from reader and unmashals in Config
func Load(r io.Reader) (*Config, error) {
	b, err := ioutil.ReadAll(r)
	if err != nil {
		return nil, err
	}

	c := &Config{}
	err = yaml.Unmarshal(b, c)
	if err != nil {
		return nil, err
	}

	return c, nil
}