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 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           interface{}   `json:"author"`
	Category         string        `json:"category"`
	Source           string        `json:"source"`
	Medium           interface{}   `json:"medium"`
	Standard         string        `json:"standard"`
	VideoCodec       string        `json:"videoCodec"`
	AudioCodec       string        `json:"audioCodec"`
	Team             string        `json:"team"`
	Processing       interface{}   `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         interface{}   `json:"infoHash"`
	Status           *Status       `json:"status"`
	EditedBy         interface{}   `json:"editedBy"`
	EditDate         interface{}   `json:"editDate"`
	Collection       bool          `json:"collection"`
	InRss            bool          `json:"inRss"`
	CanVote          bool          `json:"canVote"`
	ImageList        interface{}   `json:"imageList"`
	ResetBox         interface{}   `json:"resetBox"`
	OriginFileName   string        `json:"originFileName"`
	Descr            string        `json:"descr"`
	Nfo              interface{}   `json:"nfo"`
	MediaInfo        string        `json:"mediainfo"`
	CIDs             interface{}   `json:"cids"`
	AIDs             interface{}   `json:"aids"`
	ShowcaseList     []Showcase    `json:"showcaseList"`
	TagList          []interface{} `json:"tagList"`
	Scope            string        `json:"scope"`
	ScopeTeams       []interface{} `json:"scopeTeams"`
	Thanked          bool          `json:"thanked"`
	Rewarded         bool          `json:"rewarded"`
}

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

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"`
	DownloadSubPath string `json:"download_sub_path"`
}

type RSSPreset struct {
	Name            string
	URL             string
	Category        string
	Tag             string
	IntervalMinutes int32
	DownloadSubPath string
}

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"`
	DownloadSubPath string    `gorm:"size:256" json:"download_sub_path"`
	CreatedAt       time.Time `json:"created_at"`
	UpdatedAt       time.Time `json:"updated_at"`
}

type ResType interface {
	MTTorrentDetail | PHPTorrentInfo
	FreeDownChecker
}

type SettingsGlobal struct {
	ID                     uint      `gorm:"primaryKey" json:"id"`
	DefaultIntervalMinutes int32     `json:"default_interval_minutes"`
	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 SitePreset struct {
	Name       string
	AuthMethod string
	Enabled    bool
	APIUrl     string
	RSS        []RSSPreset
}

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 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   interface{}    `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"`
}