39 lines
1.3 KiB
Go
39 lines
1.3 KiB
Go
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)
|
|
// split files larger than Telegram limit(2GB) into parts of specified size, in MB, leave 0 to set default(2000MB)
|
|
// only effective when SkipLarge is false
|
|
// use zip when splitting
|
|
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
|
|
}
|