package config

import (
	"github.com/ansg191/ibd-trader-backend/internal/keys"
	"github.com/ansg191/ibd-trader-backend/internal/leader/manager/ibd"

	"github.com/spf13/viper"
)

type Config struct {
	// Logging configuration
	Log struct {
		// Log level
		Level string
		// Add source info to log messages
		AddSource bool
		// Enable colorized output
		Color bool
	}
	// Database configuration
	DB struct {
		// Database URL
		URL string
	}
	// Redis configuration
	Redis struct {
		// Redis address
		Addr string
		// Redis password
		Password string
	}
	// KMS configuration
	KMS struct {
		// GCP KMS configuration
		GCP *keys.GCPKeyName
	}
	// Server configuration
	Server struct {
		// Server port
		Port uint16
	}
	// OAuth 2.0 configuration
	Auth struct {
		// OAuth 2.0 domain
		Domain string
		// OAuth 2.0 client ID
		ClientID string
		// OAuth 2.0 client secret
		ClientSecret string
		// OAuth 2.0 callback URL
		CallbackURL string
	}
	// IBD configuration
	IBD struct {
		// Scraper API Key
		APIKey string
		// Proxy URL
		ProxyURL string
		// Scrape schedules. In cron format.
		Schedules ibd.Schedules
	}
	// Analyzer configuration
	Analyzer struct {
		// Use OpenAI for analysis
		OpenAI *struct {
			// OpenAI API Key
			APIKey string
		}
	}
}

func New() (*Config, error) {
	v := viper.New()

	v.SetDefault("log.level", "INFO")
	v.SetDefault("log.addSource", false)
	v.SetDefault("log.color", false)
	v.SetDefault("server.port", 8000)

	v.SetConfigName("config")
	v.AddConfigPath("/etc/ibd-trader/")
	v.AddConfigPath("$HOME/.ibd-trader")
	v.AddConfigPath(".")
	err := v.ReadInConfig()
	if err != nil {
		if _, ok := err.(viper.ConfigFileNotFoundError); ok {
			// Config file not found; ignore error
		} else {
			return nil, err
		}
	}

	v.MustBindEnv("db.url", "DATABASE_URL")
	v.MustBindEnv("redis.addr", "REDIS_ADDR")
	v.MustBindEnv("redis.password", "REDIS_PASSWORD")
	v.MustBindEnv("log.level", "LOG_LEVEL")
	v.MustBindEnv("server.port", "SERVER_PORT")
	v.MustBindEnv("auth.domain", "AUTH_DOMAIN")
	v.MustBindEnv("auth.clientID", "AUTH_CLIENT_ID")
	v.MustBindEnv("auth.clientSecret", "AUTH_CLIENT_SECRET")
	v.MustBindEnv("auth.callbackURL", "AUTH_CALLBACK_URL")

	config := new(Config)
	err = v.Unmarshal(config)
	if err != nil {
		return nil, err
	}

	return config, config.assert()
}

func (c *Config) assert() error {
	return nil
}