models

type APIResponse[T ResType] struct {
	Message string `json:"message"`
	Data    T      `json:"data"`
	Code    any    `json:"code"`
}

type AdminUser struct {
	ID           uint   `gorm:"primaryKey"`
	Username     string `gorm:"uniqueIndex;size:64"`
	PasswordHash string `gorm:"size:255"`
	CreatedAt    time.Time
	UpdatedAt    time.Time
}

type Config struct {
	Global SettingsGlobal           `json:"global"`
	Qbit   QbitSettings             `json:"qbit"`
	Sites  map[SiteGroup]SiteConfig `json:"sites"`
}

type DiscountType string

type DownloaderDirectory struct {
	ID           uint      `gorm:"primaryKey" json:"id"`
	DownloaderID uint      `gorm:"index;not null" json:"downloader_id"` // 关联的下载器ID
	Path         string    `gorm:"size:512;not null" json:"path"`       // 目录路径
	Alias        string    `gorm:"size:128" json:"alias"`               // 目录别名（用于显示）
	IsDefault    bool      `json:"is_default"`                          // 是否为默认目录
	CreatedAt    time.Time `json:"created_at"`
	UpdatedAt    time.Time `json:"updated_at"`
}

type DownloaderSetting struct {
	ID          uint      `gorm:"primaryKey" json:"id"`
	Name        string    `gorm:"uniqueIndex;size:64;not null" json:"name"`
	Type        string    `gorm:"size:32;not null" json:"type"` // qbittorrent, transmission
	URL         string    `gorm:"size:512;not null" json:"url"`
	Username    string    `gorm:"size:128" json:"username"`
	Password    string    `gorm:"size:256" json:"password"`
	IsDefault   bool      `json:"is_default"`
	Enabled     bool      `json:"enabled"`
	AutoStart   bool      `json:"auto_start"`                              // 推送种子后自动开始下载
	ExtraConfig string    `gorm:"type:text" json:"extra_config,omitempty"` // JSON格式的额外配置
	CreatedAt   time.Time `json:"created_at"`
	UpdatedAt   time.Time `json:"updated_at"`
}

type DynamicSiteSetting struct {
	ID           uint      `gorm:"primaryKey" json:"id"`
	Name         string    `gorm:"uniqueIndex;size:64;not null" json:"name"`
	DisplayName  string    `gorm:"size:128" json:"display_name"`
	BaseURL      string    `gorm:"size:512" json:"base_url"`
	Enabled      bool      `json:"enabled"`
	AuthMethod   string    `gorm:"size:16;not null" json:"auth_method"` // cookie, api_key
	Cookie       string    `gorm:"size:2048" json:"cookie,omitempty"`
	APIKey       string    `gorm:"size:512" json:"api_key,omitempty"`
	APIURL       string    `gorm:"size:512" json:"api_url,omitempty"`
	DownloaderID *uint     `gorm:"index" json:"downloader_id,omitempty"` // 关联的下载器ID
	ParserConfig string    `gorm:"type:text" json:"parser_config"`       // JSON格式的解析器配置
	IsBuiltin    bool      `json:"is_builtin"`                           // 是否为内置站点
	TemplateID   *uint     `gorm:"index" json:"template_id,omitempty"`   // 关联的模板ID
	CreatedAt    time.Time `json:"created_at"`
	UpdatedAt    time.Time `json:"updated_at"`
}

type FaviconCache struct {
	ID          uint      `gorm:"primaryKey" json:"id"`
	SiteID      string    `gorm:"uniqueIndex;size:64;not null" json:"site_id"` // 站点标识，如 "hdsky", "mteam"
	SiteName    string    `gorm:"size:128" json:"site_name"`                   // 站点名称
	FaviconURL  string    `gorm:"size:512" json:"favicon_url"`                 // 原始 favicon URL
	Data        []byte    `gorm:"type:blob" json:"-"`                          // 图标二进制数据
	ContentType string    `gorm:"size:64" json:"content_type"`                 // MIME 类型
	ETag        string    `gorm:"size:128" json:"etag"`                        // 用于缓存验证
	LastFetched time.Time `json:"last_fetched"`                                // 最后获取时间
	CreatedAt   time.Time `json:"created_at"`
	UpdatedAt   time.Time `json:"updated_at"`
}

type FilterRule struct {
	ID          uint        `gorm:"primaryKey" json:"id"`
	Name        string      `gorm:"size:128;not null" json:"name"`
	Pattern     string      `gorm:"size:512;not null" json:"pattern"`
	PatternType PatternType `gorm:"size:16;not null;default:'keyword'" json:"pattern_type"`
	MatchField  MatchField  `gorm:"size:16;not null;default:'both'" json:"match_field"`
	RequireFree bool        `gorm:"default:true" json:"require_free"`
	Enabled     bool        `gorm:"default:true" json:"enabled"`
	SiteID      *uint       `gorm:"index" json:"site_id"`
	RSSID       *uint       `gorm:"index" json:"rss_id"`
	Priority    int         `gorm:"default:100" json:"priority"`
	CreatedAt   time.Time   `json:"created_at"`
	UpdatedAt   time.Time   `json:"updated_at"`
}

type FilterRuleDB struct {
	// contains filtered or unexported fields
}

type FreeDownChecker interface {
	IsFree() bool
	CanbeFinished(logger *zap.SugaredLogger, enabled bool, speedLimit, sizeLimitGB int) bool
	GetFreeEndTime() *time.Time
	GetFreeLevel() string
}

type MTTorrentDetail struct {
	ID               string     `json:"id"`
	CreatedDate      string     `json:"createdDate"`
	LastModifiedDate string     `json:"lastModifiedDate"`
	Name             string     `json:"name"`
	SmallDescr       string     `json:"smallDescr"`
	IMDb             string     `json:"imdb"`
	IMDbRating       *string    `json:"imdbRating"`
	Douban           string     `json:"douban"`
	DoubanRating     *string    `json:"doubanRating"`
	DmmCode          string     `json:"dmmCode"`
	Author           any        `json:"author"`
	Category         string     `json:"category"`
	Source           string     `json:"source"`
	Medium           any        `json:"medium"`
	Standard         string     `json:"standard"`
	VideoCodec       string     `json:"videoCodec"`
	AudioCodec       string     `json:"audioCodec"`
	Team             string     `json:"team"`
	Processing       any        `json:"processing"`
	Countries        []string   `json:"countries"`
	NumFiles         string     `json:"numfiles"`
	Size             string     `json:"size"`
	Tags             string     `json:"tags"`
	Labels           string     `json:"labels"`
	MsUp             string     `json:"msUp"`
	Anonymous        bool       `json:"anonymous"`
	InfoHash         any        `json:"infoHash"`
	Status           *Status    `json:"status"`
	EditedBy         any        `json:"editedBy"`
	EditDate         any        `json:"editDate"`
	Collection       bool       `json:"collection"`
	InRss            bool       `json:"inRss"`
	CanVote          bool       `json:"canVote"`
	ImageList        any        `json:"imageList"`
	ResetBox         any        `json:"resetBox"`
	OriginFileName   string     `json:"originFileName"`
	Descr            string     `json:"descr"`
	Nfo              any        `json:"nfo"`
	MediaInfo        string     `json:"mediainfo"`
	CIDs             any        `json:"cids"`
	AIDs             any        `json:"aids"`
	ShowcaseList     []Showcase `json:"showcaseList"`
	TagList          []any      `json:"tagList"`
	Scope            string     `json:"scope"`
	ScopeTeams       []any      `json:"scopeTeams"`
	Thanked          bool       `json:"thanked"`
	Rewarded         bool       `json:"rewarded"`
}

type MatchField string

type Migration struct {
	Version     int
	Description string
	Up          MigrationFunc
}

type MigrationFunc func(db *gorm.DB) error

type PHPTorrentInfo struct {
	Title     string       // 种子标题
	SubTitle  string       // 副标题/标签
	TorrentID string       // 种子 ID
	Discount  DiscountType // 优惠类型
	EndTime   time.Time    // 优惠结束时间
	SizeMB    float64      // 种子大小，单位为 MB
	Seeders   int          // 做种人数
	Leechers  int          // 下载人数
	Completed float64      // 最大完成百分比
	HR        bool         // 是否为 HR（Hit & Run）
}

type PatternType string

type PromotionRule struct {
	Categories  []string `json:"categories"`
	CreatedDate string   `json:"createdDate"`
	Discount    string   `json:"discount"`
}

type QbitSettings struct {
	ID        uint      `gorm:"primaryKey" json:"id"`
	Enabled   bool      `json:"enabled"`
	URL       string    `gorm:"not null" json:"url"`
	User      string    `gorm:"not null" json:"user"`
	Password  string    `gorm:"not null" json:"password"`
	CreatedAt time.Time `json:"created_at"`
	UpdatedAt time.Time `json:"updated_at"`
}

type RSSConfig struct {
	ID              uint   `json:"id"`
	Name            string `json:"name"`
	URL             string `json:"url"`
	Category        string `json:"category"`
	Tag             string `json:"tag"`
	IntervalMinutes int32  `json:"interval_minutes"` // RSS 执行间隔（分钟），0 表示使用全局设置
	Concurrency     int32  `json:"concurrency"`      // 并发数，0 表示使用全局设置
	DownloadSubPath string `json:"download_sub_path"`
	DownloadPath    string `json:"download_path"`   // 下载器中下载任务的目标下载路径（可选）
	DownloaderID    *uint  `json:"downloader_id"`   // 指定下载器，nil 表示使用默认下载器
	FilterRuleIDs   []uint `json:"filter_rule_ids"` // 关联的过滤规则 ID 列表
	IsExample       bool   `json:"is_example"`      // 是否为示例配置，示例配置不会被执行
	PauseOnFreeEnd  bool   `json:"pause_on_free_end"`
}

type RSSFilterAssociation struct {
	ID           uint      `gorm:"primaryKey" json:"id"`
	RSSID        uint      `gorm:"uniqueIndex:idx_rss_filter;not null" json:"rss_id"`
	FilterRuleID uint      `gorm:"uniqueIndex:idx_rss_filter;not null" json:"filter_rule_id"`
	CreatedAt    time.Time `json:"created_at"`
}

type RSSFilterAssociationDB struct {
	// contains filtered or unexported fields
}

type RSSSubscription struct {
	ID              uint      `gorm:"primaryKey" json:"id"`
	SiteID          uint      `gorm:"index" json:"site_id"`
	Name            string    `gorm:"size:128;not null" json:"name"`
	URL             string    `gorm:"size:1024;not null" json:"url"`
	Category        string    `gorm:"size:128" json:"category"`
	Tag             string    `gorm:"size:128" json:"tag"`
	IntervalMinutes int32     `gorm:"check:interval_minutes >= 1" json:"interval_minutes"` // RSS 执行间隔（分钟），0 表示使用全局设置
	Concurrency     int32     `json:"concurrency"`                                         // 并发数，0 表示使用全局设置
	DownloadSubPath string    `gorm:"size:256" json:"download_sub_path"`
	DownloadPath    string    `gorm:"size:512" json:"download_path"` // 下载器中下载任务的目标下载路径（可选）
	DownloaderID    *uint     `gorm:"index" json:"downloader_id"`    // 指定下载器，nil 表示使用默认下载器
	IsExample       bool      `json:"is_example"`                    // 是否为示例配置，示例配置不会被执行
	PauseOnFreeEnd  bool      `gorm:"default:false" json:"pause_on_free_end"`
	CreatedAt       time.Time `json:"created_at"`
	UpdatedAt       time.Time `json:"updated_at"`
}

type RegisteredSite struct {
	ID            string
	Name          string
	AuthMethod    string // "cookie" 或 "api_key"
	DefaultAPIUrl string
}

type ResType interface {
	MTTorrentDetail | PHPTorrentInfo
	FreeDownChecker
	TorrentMetadata
}

type SchemaManager struct {
	// contains filtered or unexported fields
}

type SchemaVersion struct {
	ID          uint      `gorm:"primaryKey" json:"id"`
	Version     int       `gorm:"uniqueIndex;not null" json:"version"` // 架构版本号
	Description string    `gorm:"size:255" json:"description"`         // 版本描述
	AppliedAt   time.Time `gorm:"not null" json:"applied_at"`          // 应用时间
	AppVersion  string    `gorm:"size:64" json:"app_version"`          // 应用版本
}

type SettingsGlobal struct {
	ID                     uint      `gorm:"primaryKey" json:"id"`
	DefaultIntervalMinutes int32     `json:"default_interval_minutes" gorm:"default:10"` // 默认 RSS 执行间隔（分钟）
	DefaultConcurrency     int32     `json:"default_concurrency" gorm:"default:3"`       // 默认并发数
	DefaultEnabled         bool      `json:"default_enabled"`
	DownloadDir            string    `gorm:"not null" json:"download_dir"`
	DownloadLimitEnabled   bool      `json:"download_limit_enabled"`
	DownloadSpeedLimit     int       `json:"download_speed_limit"`
	TorrentSizeGB          int       `json:"torrent_size_gb"`
	AutoStart              bool      `json:"auto_start"`
	RetainHours            int       `json:"retain_hours" gorm:"default:24"`
	MaxRetry               int       `json:"max_retry" gorm:"default:3"`
	CreatedAt              time.Time `json:"created_at"`
	UpdatedAt              time.Time `json:"updated_at"`
}

type Showcase struct {
	CreatedDate      string     `json:"createdDate"`
	LastModifiedDate string     `json:"lastModifiedDate"`
	ID               string     `json:"id"`
	Collection       bool       `json:"collection"`
	UserID           string     `json:"userid"`
	Username         string     `json:"username"`
	CnTitle          string     `json:"cntitle"`
	EnTitle          string     `json:"entitle"`
	Note             string     `json:"note"`
	Pic              string     `json:"pic"`
	Pic1             string     `json:"pic1"`
	Pic2             string     `json:"pic2"`
	Count            string     `json:"count"`
	Size             string     `json:"size"`
	View             string     `json:"view"`
	Statistics       Statistics `json:"statistics"`
}

type SiteConfig struct {
	Enabled    *bool       `json:"enabled"`
	AuthMethod string      `json:"auth_method"`
	Cookie     string      `json:"cookie"`
	APIKey     string      `json:"api_key"`
	APIUrl     string      `json:"api_url"`
	RSS        []RSSConfig `json:"rss"`
}

type SiteGroup string

type SiteSetting struct {
	ID         uint      `gorm:"primaryKey" json:"id"`
	Name       string    `gorm:"index;size:32" json:"name"`
	Enabled    bool      `json:"enabled"`
	AuthMethod string    `gorm:"size:16;not null" json:"auth_method"`
	Cookie     string    `gorm:"size:1024" json:"cookie"`
	APIKey     string    `gorm:"size:512" json:"api_key"`
	APIUrl     string    `gorm:"size:512" json:"api_url"`
	CreatedAt  time.Time `json:"created_at"`
	UpdatedAt  time.Time `json:"updated_at"`
}

type SiteTemplate struct {
	ID           uint      `gorm:"primaryKey" json:"id"`
	Name         string    `gorm:"uniqueIndex;size:64;not null" json:"name"`
	DisplayName  string    `gorm:"size:128" json:"display_name"`
	BaseURL      string    `gorm:"size:512;not null" json:"base_url"`
	AuthMethod   string    `gorm:"size:16;not null" json:"auth_method"` // cookie, api_key
	ParserConfig string    `gorm:"type:text" json:"parser_config"`      // JSON格式的解析器配置
	Description  string    `gorm:"size:1024" json:"description"`
	Version      string    `gorm:"size:32" json:"version"`
	Author       string    `gorm:"size:64" json:"author"`
	CreatedAt    time.Time `json:"created_at"`
	UpdatedAt    time.Time `json:"updated_at"`
}

type SiteTemplateExport struct {
	Name         string          `json:"name"`
	DisplayName  string          `json:"display_name"`
	BaseURL      string          `json:"base_url"`
	AuthMethod   string          `json:"auth_method"`
	ParserConfig json.RawMessage `json:"parser_config,omitempty"`
	Description  string          `json:"description,omitempty"`
	Version      string          `json:"version,omitempty"`
	Author       string          `json:"author,omitempty"`
}

type Statistics struct {
	CreatedDate      string `json:"createdDate"`
	LastModifiedDate string `json:"lastModifiedDate"`
	ID               string `json:"id"`
	Day              string `json:"day"`
	Week             string `json:"week"`
	Month            string `json:"month"`
	Year             string `json:"year"`
	DayClick         string `json:"dayClick"`
	WeekClick        string `json:"weekClick"`
	MonthClick       string `json:"monthClick"`
	YearClick        string `json:"yearClick"`
}

type Status struct {
	ID               string         `json:"id"`
	CreatedDate      string         `json:"createdDate"`
	LastModifiedDate string         `json:"lastModifiedDate"`
	PickType         string         `json:"pickType"`
	ToppingLevel     string         `json:"toppingLevel"`
	ToppingEndTime   string         `json:"toppingEndTime"`
	Discount         string         `json:"discount"`
	DiscountEndTime  string         `json:"discountEndTime"`
	TimesCompleted   string         `json:"timesCompleted"`
	Comments         string         `json:"comments"`
	LastAction       string         `json:"lastAction"`
	LastSeederAction string         `json:"lastSeederAction"`
	Views            string         `json:"views"`
	Hits             string         `json:"hits"`
	Support          string         `json:"support"`
	Oppose           string         `json:"oppose"`
	Status           string         `json:"status"`
	Seeders          string         `json:"seeders"`
	Leechers         string         `json:"leechers"`
	Banned           bool           `json:"banned"`
	Visible          bool           `json:"visible"`
	PromotionRule    *PromotionRule `json:"promotionRule"`
	MallSingleFree   any            `json:"mallSingleFree"`
}

type TorrentDB struct {
	DB *gorm.DB
}

type TorrentInfo struct {
	ID             uint       `gorm:"primaryKey" json:"id"`
	SiteName       string     `gorm:"uniqueIndex:idx_site_torrent" json:"siteName"`
	TorrentID      string     `gorm:"uniqueIndex:idx_site_torrent" json:"torrentId"`
	TorrentHash    *string    `gorm:"index" json:"torrentHash"`
	IsFree         bool       `gorm:"default:false" json:"isFree"`
	IsDownloaded   bool       `gorm:"default:false" json:"isDownloaded"`
	IsPushed       *bool      `gorm:"default:null" json:"isPushed"`
	IsSkipped      bool       `gorm:"default:false" json:"isSkipped"`
	FreeLevel      string     `gorm:"default:'normal'" json:"freeLevel"`
	FreeEndTime    *time.Time `gorm:"default:null" json:"freeEndTime"`
	PushTime       *time.Time `gorm:"default:null" json:"pushTime"`
	Title          string     `gorm:"default:''" json:"title"`
	Category       string     `gorm:"default:''" json:"category"`
	Tag            string     `gorm:"default:''" json:"tag"`
	CreatedAt      time.Time  `json:"createdAt"`
	UpdatedAt      time.Time  `json:"updatedAt"`
	IsExpired      bool       `gorm:"default:false" json:"isExpired"`
	LastCheckTime  *time.Time `gorm:"default:null" json:"lastCheckTime"`
	RetryCount     int        `gorm:"default:0" json:"retryCount"`
	LastError      string     `gorm:"default:''" json:"lastError"`
	DownloadSource string     `gorm:"size:32;default:'free_download'" json:"downloadSource"` // free_download or filter_rule
	FilterRuleID   *uint      `gorm:"index" json:"filterRuleId"`                             // ID of the matched filter rule

	// 免费结束管理相关字段
	DownloaderID     *uint      `gorm:"index" json:"downloaderId"`                   // 推送到的下载器 ID
	DownloaderName   string     `gorm:"size:64;default:''" json:"downloaderName"`    // 下载器名称（冗余存储便于查询）
	CompletedAt      *time.Time `gorm:"default:null" json:"completedAt"`             // 下载完成时间
	IsPausedBySystem bool       `gorm:"default:false" json:"isPausedBySystem"`       // 是否被系统自动暂停
	PauseOnFreeEnd   bool       `gorm:"default:false" json:"pauseOnFreeEnd"`         // 免费结束时是否暂停（来自 RSS 配置）
	PausedAt         *time.Time `gorm:"default:null" json:"pausedAt"`                // 系统暂停时间
	PauseReason      string     `gorm:"size:256;default:''" json:"pauseReason"`      // 暂停原因
	IsCompleted      bool       `gorm:"default:false;index" json:"isCompleted"`      // 下载是否已完成
	Progress         float64    `gorm:"default:0" json:"progress"`                   // 下载进度 (0-100)
	TorrentSize      int64      `gorm:"default:0" json:"torrentSize"`                // 种子大小（字节）
	DownloaderTaskID string     `gorm:"size:128;default:''" json:"downloaderTaskId"` // 下载器中的任务 ID（用于暂停/删除操作）
	CheckCount       int        `gorm:"default:0" json:"checkCount"`                 // 进度检查次数
}

type TorrentInfoArchive struct {
	ID                uint       `gorm:"primaryKey" json:"id"`
	OriginalID        uint       `gorm:"index" json:"originalId"`
	SiteName          string     `gorm:"index" json:"siteName"`
	TorrentID         string     `gorm:"index" json:"torrentId"`
	TorrentHash       *string    `gorm:"index" json:"torrentHash"`
	IsFree            bool       `json:"isFree"`
	IsDownloaded      bool       `json:"isDownloaded"`
	IsPushed          *bool      `json:"isPushed"`
	IsSkipped         bool       `json:"isSkipped"`
	FreeLevel         string     `json:"freeLevel"`
	FreeEndTime       *time.Time `json:"freeEndTime"`
	PushTime          *time.Time `json:"pushTime"`
	Title             string     `json:"title"`
	Category          string     `json:"category"`
	Tag               string     `json:"tag"`
	OriginalCreatedAt time.Time  `json:"originalCreatedAt"`
	OriginalUpdatedAt time.Time  `json:"originalUpdatedAt"`
	IsExpired         bool       `json:"isExpired"`
	LastCheckTime     *time.Time `json:"lastCheckTime"`
	RetryCount        int        `json:"retryCount"`
	LastError         string     `json:"lastError"`
	DownloadSource    string     `json:"downloadSource"`
	FilterRuleID      *uint      `json:"filterRuleId"`
	DownloaderID      *uint      `json:"downloaderId"`
	DownloaderName    string     `json:"downloaderName"`
	CompletedAt       *time.Time `json:"completedAt"`
	IsPausedBySystem  bool       `json:"isPausedBySystem"`
	PauseOnFreeEnd    bool       `json:"pauseOnFreeEnd"`
	PausedAt          *time.Time `json:"pausedAt"`
	PauseReason       string     `json:"pauseReason"`
	IsCompleted       bool       `json:"isCompleted"`
	Progress          float64    `json:"progress"`
	TorrentSize       int64      `json:"torrentSize"`
	DownloaderTaskID  string     `json:"downloaderTaskId"`
	CheckCount        int        `json:"checkCount"`
	ArchivedAt        time.Time  `gorm:"autoCreateTime" json:"archivedAt"`
}

type TorrentMetadata interface {
	// GetName 获取种子名称（优先返回中文名）
	GetName() string
	// GetSubTitle 获取副标题/标签信息
	GetSubTitle() string
}