MoviePilot-Frontend/tests/support/factories/subscribe.ts

import type {
  DownloaderConf,
  FilterRuleGroup,
  MediaInfo,
  Site,
  Subscribe,
  TransferDirectoryConf,
} from '@/api/types'
import { createMediaInfo } from './media'

let subscribeSeed = 1000
let siteSeed = 100

/** 构造满足前端订阅契约的最小记录。 */
export function createSubscribe(overrides: Partial<Subscribe> = {}): Subscribe {
  subscribeSeed += 1
  return {
    best_version: 0,
    best_version_full: 0,
    current_priority: 0,
    date: '2026-07-16',
    downloader: '',
    episode_group: '',
    id: subscribeSeed,
    last_update: '2026-07-16 12:00:00',
    name: `测试订阅 ${subscribeSeed}`,
    show_edit_dialog: false,
    sites: [],
    state: 'R',
    tmdbid: subscribeSeed,
    type: '电影',
    username: 'tester',
    year: '2026',
    ...overrides,
  }
}

/** 构造电影媒体信息。 */
export function createSubscribeMovie(overrides: Partial<MediaInfo> = {}): MediaInfo {
  return createMediaInfo({ type: '电影', ...overrides })
}

/** 构造电视剧媒体信息。 */
export function createSubscribeTv(overrides: Partial<MediaInfo> = {}): MediaInfo {
  return createMediaInfo({
    season: 1,
    season_info: [
      { episode_count: 12, name: '第 1 季', season_number: 1 },
      { episode_count: 10, name: '第 2 季', season_number: 2 },
    ],
    type: '电视剧',
    ...overrides,
  })
}

/** 构造订阅站点选项。 */
export function createSubscribeSite(overrides: Partial<Site> = {}): Site {
  siteSeed += 1
  return {
    domain: `site-${siteSeed}.example.com`,
    downloader: '',
    id: siteSeed,
    is_active: true,
    name: `测试站点 ${siteSeed}`,
    url: `https://site-${siteSeed}.example.com`,
    ...overrides,
  }
}

/** 构造下载器选项。 */
export function createSubscribeDownloader(overrides: Partial<DownloaderConf> = {}): DownloaderConf {
  return {
    config: {},
    default: false,
    enabled: true,
    name: '测试下载器',
    type: 'qbittorrent',
    ...overrides,
  }
}

/** 构造下载目录配置。 */
export function createSubscribeDirectory(overrides: Partial<TransferDirectoryConf> = {}): TransferDirectoryConf {
  return {
    download_path: '/downloads',
    name: '测试目录',
    priority: 1,
    storage: 'local',
    transfer_type: 'link',
    ...overrides,
  }
}

/** 构造订阅过滤规则组。 */
export function createSubscribeRuleGroup(overrides: Partial<FilterRuleGroup> = {}): FilterRuleGroup {
  return {
    name: '默认规则组',
    rule_string: 'priority=1',
    ...overrides,
  }
}