package config

import (
	"io"
	"time"

	"github.com/spf13/viper"
)

var (
	v = viper.GetViper()
)

func init() {
	v.SetConfigName("gost")
	v.AddConfigPath("/etc/gost/")
	v.AddConfigPath("$HOME/.gost/")
	v.AddConfigPath(".")
}

type LogConfig struct {
	Output string
	Level  string
	Format string
}

type ProfilingConfig struct {
	Addr    string
	Enabled bool
}

type SelectorConfig struct {
	Strategy    string
	MaxFails    int
	FailTimeout time.Duration
}

type BypassConfig struct {
	Name     string
	Reverse  bool
	Matchers []string
}
type ListenerConfig struct {
	Type     string
	Chain    string
	Metadata map[string]interface{}
}

type HandlerConfig struct {
	Type     string
	Chain    string
	Bypass   string
	Metadata map[string]interface{}
}

type ForwarderConfig struct {
	Targets  []string
	Selector *SelectorConfig
}

type DialerConfig struct {
	Type     string
	Metadata map[string]interface{}
}

type ConnectorConfig struct {
	Type     string
	Metadata map[string]interface{}
}

type ServiceConfig struct {
	Name      string
	URL       string
	Addr      string
	Listener  *ListenerConfig
	Handler   *HandlerConfig
	Forwarder *ForwarderConfig
}

type ChainConfig struct {
	Name     string
	Selector *SelectorConfig
	Hops     []HopConfig
}

type HopConfig struct {
	Name     string
	Selector *SelectorConfig
	Nodes    []NodeConfig
}

type NodeConfig struct {
	Name      string
	URL       string
	Addr      string
	Dialer    *DialerConfig
	Connector *ConnectorConfig
	Bypass    string
}

type Config struct {
	Log       *LogConfig
	Profiling *ProfilingConfig
	Services  []ServiceConfig
	Chains    []ChainConfig
	Bypasses  []BypassConfig
}

func (c *Config) Load() error {
	if err := v.ReadInConfig(); err != nil {
		return err
	}

	return v.Unmarshal(c)
}

func (c *Config) Read(r io.Reader) error {
	if err := v.ReadConfig(r); err != nil {
		return err
	}

	return v.Unmarshal(c)
}

func (c *Config) ReadFile(file string) error {
	v.SetConfigFile(file)
	if err := v.ReadInConfig(); err != nil {
		return err
	}
	return v.Unmarshal(c)
}