package agent

import "strings"

const (
	DEF_WG_IFACE string = "wg0"
	DEF_WG_ETH   string = "eth"
)

type AgentWgConfig struct {
	WgInterfaces string `yaml:"interfaces"` //
	EthName      string `yaml:"eth"`        //
	Forward      bool   `yaml:"forward"`    //
	Iptables     bool   `yaml:"iptables"`   //
}

func (c *AgentWgConfig) GetWgInterfaces() []string {
	if c.WgInterfaces != "" {
		return strings.Split(c.WgInterfaces, ",")
	}

	return []string{DEF_WG_IFACE}
}

func (c *AgentWgConfig) GetWgInterface() string {
	ifaces := c.GetWgInterfaces()
	return ifaces[0]
}
func (c *AgentWgConfig) GetWgInterface4Connector() string {
	ifaces := c.GetWgInterfaces()
	if len(ifaces) > 1 {
		return ifaces[1]
	}
	return ifaces[0]
}
func (c *AgentWgConfig) GetWgInterface4Pop() string {
	ifaces := c.GetWgInterfaces()
	if len(ifaces) > 2 {
		return ifaces[2]
	}
	return ifaces[0]
}

func (c *AgentWgConfig) GetEth() string {
	eth := c.EthName
	if eth != "" {
		return eth
	}
	return DEF_WG_ETH
}