package config

import (
	"fmt"

	"github.com/krau/SaveAny-Bot/types"
)

func (c *Config) GetStoragesByType(storageType types.StorageType) []StorageConfig {
	var storages []StorageConfig
	for _, storage := range c.Storages {
		if storage.GetType() == storageType {
			storages = append(storages, storage)
		}
	}
	return storages
}

func (c *Config) GetStorageByName(name string) StorageConfig {
	for _, storage := range c.Storages {
		if storage.GetName() == name {
			return storage
		}
	}
	return nil
}

type LocalStorageConfig struct {
	NewStorageConfig
	BasePath string `toml:"base_path" mapstructure:"base_path" json:"base_path"`
}

func (l *LocalStorageConfig) Validate() error {
	if l.BasePath == "" {
		return fmt.Errorf("path is required for local storage")
	}
	return nil
}

func (l *LocalStorageConfig) GetType() types.StorageType {
	return types.StorageTypeLocal
}

func (l *LocalStorageConfig) GetName() string {
	return l.Name
}

type AlistStorageConfig struct {
	NewStorageConfig
	URL      string `toml:"url" mapstructure:"url" json:"url"`
	Username string `toml:"username" mapstructure:"username" json:"username"`
	Password string `toml:"password" mapstructure:"password" json:"password"`
	Token    string `toml:"token" mapstructure:"token" json:"token"`
	BasePath string `toml:"base_path" mapstructure:"base_path" json:"base_path"`
	TokenExp int64  `toml:"token_exp" mapstructure:"token_exp" json:"token_exp"`
}

func (a *AlistStorageConfig) Validate() error {
	if a.URL == "" {
		return fmt.Errorf("url is required for alist storage")
	}
	if a.Token == "" && (a.Username == "" || a.Password == "") {
		return fmt.Errorf("username and password or token is required for alist storage")
	}
	if a.BasePath == "" {
		return fmt.Errorf("base_path is required for alist storage")
	}
	return nil
}

func (a *AlistStorageConfig) GetType() types.StorageType {
	return types.StorageTypeAlist
}

func (a *AlistStorageConfig) GetName() string {
	return a.Name
}

type WebdavStorageConfig struct {
	NewStorageConfig
	URL      string `toml:"url" mapstructure:"url" json:"url"`
	Username string `toml:"username" mapstructure:"username" json:"username"`
	Password string `toml:"password" mapstructure:"password" json:"password"`
	BasePath string `toml:"base_path" mapstructure:"base_path" json:"base_path"`
}

func (w *WebdavStorageConfig) Validate() error {
	if w.URL == "" {
		return fmt.Errorf("url is required for webdav storage")
	}
	if w.Username == "" || w.Password == "" {
		return fmt.Errorf("username and password is required for webdav storage")
	}
	if w.BasePath == "" {
		return fmt.Errorf("base_path is required for webdav storage")
	}
	return nil
}

func (w *WebdavStorageConfig) GetType() types.StorageType {
	return types.StorageTypeWebdav
}

func (w *WebdavStorageConfig) GetName() string {
	return w.Name
}