package config

import "math/rand"

type ConfigInterface interface {
	GetConfig() ZurgConfig
	GetVersion() string
	GetToken() string
	GetNumOfWorkers() int
	GetRefreshEverySeconds() int
	EnableRepair() bool
	GetHost() string
	GetPort() string
	GetDirectories() []string
	MeetsConditions(directory, torrentName string, torrentSize int64, torrentIDs, fileNames []string, fileSizes []int64) bool
	GetOnLibraryUpdate() string
	GetNetworkBufferSize() int
	EnableRetainFolderNameExtension() bool
	EnableRetainRDTorrentName() bool
	GetRandomPreferredHost() string
	ShouldServeFromRclone() bool
	ShouldForceIPv6() bool
	GetRealDebridTimeout() int
	GetRetriesUntilFailed() int
	EnableDownloadCache() bool
	GetRateLimitSleepSeconds() int
	ShouldDeleteRarFiles() bool
}

type ZurgConfig struct {
	Version                   string   `yaml:"zurg" json:"-"`
	Token                     string   `yaml:"token" json:"-"`
	Host                      string   `yaml:"host" json:"host"`
	Port                      string   `yaml:"port" json:"port"`
	NumOfWorkers              int      `yaml:"concurrent_workers" json:"concurrent_workers"`
	RefreshEverySeconds       int      `yaml:"check_for_changes_every_secs" json:"check_for_changes_every_secs"`
	CanRepair                 bool     `yaml:"enable_repair" json:"enable_repair"`
	OnLibraryUpdate           string   `yaml:"on_library_update" json:"on_library_update"`
	NetworkBufferSize         int      `yaml:"network_buffer_size" json:"network_buffer_size"`
	RateLimitSleepSeconds     int      `yaml:"rate_limit_sleep_secs" json:"rate_limit_sleep_secs"`
	RetainFolderNameExtension bool     `yaml:"retain_folder_name_extension" json:"retain_folder_name_extension"`
	RetainRDTorrentName       bool     `yaml:"retain_rd_torrent_name" json:"retain_rd_torrent_name"`
	PreferredHosts            []string `yaml:"preferred_hosts" json:"preferred_hosts"`
	ServeFromRclone           bool     `yaml:"serve_from_rclone" json:"serve_from_rclone"`
	ForceIPv6                 bool     `yaml:"force_ipv6" json:"force_ipv6"`
	RealDebridTimeout         int      `yaml:"realdebrid_timeout_secs" json:"realdebrid_timeout_secs"`
	RetriesUntilFailed        int      `yaml:"retries_until_failed" json:"retries_until_failed"`
	UseDownloadCache          bool     `yaml:"use_download_cache" json:"use_download_cache"`
	DeleteRarFiles            bool     `yaml:"delete_rar_files" json:"delete_rar_files"`
}

func (z *ZurgConfig) GetConfig() ZurgConfig {
	return *z
}

func (z *ZurgConfig) GetToken() string {
	return z.Token
}

func (z *ZurgConfig) GetHost() string {
	if z.Host == "" {
		return "[::]"
	}
	return z.Host
}

func (z *ZurgConfig) GetPort() string {
	if z.Port == "" {
		return "9999"
	}
	return z.Port
}

func (z *ZurgConfig) GetNumOfWorkers() int {
	if z.NumOfWorkers == 0 {
		return 50
	}
	return z.NumOfWorkers
}

func (z *ZurgConfig) GetRefreshEverySeconds() int {
	if z.RefreshEverySeconds == 0 {
		return 60
	}
	return z.RefreshEverySeconds
}

func (z *ZurgConfig) EnableRepair() bool {
	return z.CanRepair
}

func (z *ZurgConfig) GetOnLibraryUpdate() string {
	return z.OnLibraryUpdate
}

func (z *ZurgConfig) GetNetworkBufferSize() int {
	if z.NetworkBufferSize == 0 {
		return 64 * 1024 // 64kb
	}
	return z.NetworkBufferSize
}

func (z *ZurgConfig) EnableRetainFolderNameExtension() bool {
	return z.RetainFolderNameExtension
}

func (z *ZurgConfig) EnableRetainRDTorrentName() bool {
	return z.RetainRDTorrentName
}

func (z *ZurgConfig) GetRandomPreferredHost() string {
	if len(z.PreferredHosts) == 0 {
		return ""
	}
	randomIndex := rand.Intn(len(z.PreferredHosts))
	return z.PreferredHosts[randomIndex]
}

func (z *ZurgConfig) ShouldServeFromRclone() bool {
	return z.ServeFromRclone
}

func (z *ZurgConfig) ShouldForceIPv6() bool {
	return z.ForceIPv6
}

func (z *ZurgConfig) GetRetriesUntilFailed() int {
	if z.RetriesUntilFailed == 0 {
		return 5
	}
	return z.RetriesUntilFailed
}

func (z *ZurgConfig) EnableDownloadCache() bool {
	return z.UseDownloadCache
}

func (z *ZurgConfig) GetRealDebridTimeout() int {
	if z.RealDebridTimeout == 0 {
		return 60
	}
	return z.RealDebridTimeout
}

func (z *ZurgConfig) GetRateLimitSleepSeconds() int {
	if z.RateLimitSleepSeconds == 0 {
		return 4
	}
	return z.RateLimitSleepSeconds
}

func (z *ZurgConfig) ShouldDeleteRarFiles() bool {
	return z.DeleteRarFiles
}