package storage

import (
	"fmt"

	storenum "github.com/krau/SaveAny-Bot/pkg/enums/storage"
)

type TelegramStorageConfig struct {
	BaseConfig
	ChatID          int64 `toml:"chat_id" mapstructure:"chat_id" json:"chat_id"`
	ForceFile       bool  `toml:"force_file" mapstructure:"force_file" json:"force_file"`
	RateLimit       int   `toml:"rate_limit" mapstructure:"rate_limit" json:"rate_limit"`
	RateBurst       int   `toml:"rate_burst" mapstructure:"rate_burst" json:"rate_burst"`
	SkipLarge       bool  `toml:"skip_large" mapstructure:"skip_large" json:"skip_large"` // skip files larger than Telegram limit(2GB)
	SplitLargeVideo bool  `toml:"split_large_video" mapstructure:"split_large_video" json:"split_large_video"`
	// split files larger than the uploader account limit into parts of specified size, in MB
	// leave 0 to use the account limit (2000MB for bots/regular users, 4000MB for Premium users)
	// only effective when SkipLarge is false
	// use zip when splitting non-video files or when lossless video splitting is disabled/unavailable
	SplitSizeMB int64 `toml:"split_size_mb" mapstructure:"split_size_mb" json:"split_size_mb"`
}

func (m *TelegramStorageConfig) Validate() error {
	if m.ChatID == 0 {
		return fmt.Errorf("chat_id is required for telegram storage")
	}
	if m.RateLimit < 0 || m.RateBurst < 0 {
		return fmt.Errorf("rate_limit and rate_burst must be greater than 0 for telegram storage")
	}
	return nil
}

func (m *TelegramStorageConfig) GetType() storenum.StorageType {
	return storenum.Telegram
}

func (m *TelegramStorageConfig) GetName() string {
	return m.Name
}