MoviePilot/app/modules/indexer/spider/rousi.py

import base64
import json
from typing import List, Optional, Tuple

from app.core.config import settings
from app.db.systemconfig_oper import SystemConfigOper
from app.log import logger
from app.schemas import MediaType
from app.utils.http import RequestUtils, AsyncRequestUtils
from app.utils.string import StringUtils


class RousiSpider:
    """
    Rousi.pro API v1 Spider

    使用 API v1 接口进行种子搜索
    - 认证方式：Bearer Token (Passkey)
    - 搜索接口：/api/v1/torrents
    - 详情接口：/api/v1/torrents/:id
    """
    _indexerid = None
    _domain = None
    _url = None
    _name = ""
    _proxy = None
    _cookie = None
    _ua = None
    _size = 100
    _searchurl = "https://%s/api/v1/torrents"
    _downloadurl = "https://%s/api/v1/torrents/%s"
    _timeout = 15

    # 分类定义
    # API 不支持多分类搜索，每次只使用一个分类
    _movie_category = 'movie'
    _tv_category = 'tv'

    # API KEY
    _apikey = None

    def __init__(self, indexer: dict):
        self.systemconfig = SystemConfigOper()
        if indexer:
            self._indexerid = indexer.get('id')
            self._url = indexer.get('domain')
            self._domain = StringUtils.get_url_domain(self._url)
            self._searchurl = self._searchurl % self._domain
            self._downloadurl = self._downloadurl % (self._domain, "%s")
            self._name = indexer.get('name')
            if indexer.get('proxy'):
                self._proxy = settings.PROXY
            self._cookie = indexer.get('cookie')
            self._ua = indexer.get('ua')
            self._apikey = indexer.get('apikey')
            self._timeout = indexer.get('timeout') or 15

    def __get_params(self, keyword: str, mtype: MediaType = None, cat: Optional[str] = None, page: Optional[int] = 0) -> dict:
        """
        构建 API 请求参数

        :param keyword: 搜索关键词
        :param mtype: 媒体类型 (MOVIE/TV)
        :param cat: 用户选择的分类 ID（逗号分隔的字符串）
        :param page: 页码（从 0 开始，API 需要从 1 开始）
        :return: 请求参数字典
        """
        params = {
            "page": int(page) + 1,
            "page_size": self._size
        }
        if keyword:
            params["keyword"] = keyword

        # API 不支持多分类搜索,只使用单个 category 参数
        # 优先使用用户选择的分类,如果用户未选择则根据 mtype 推断
        if cat:
            # 用户选择了特定分类,需要将分类 ID 映射回 API 的 category name
            category_names = self.__get_category_names_by_ids(cat)
            if category_names:
                # 如果用户选择了多个分类,只取第一个
                params["category"] = category_names[0]
        elif mtype:
            # 用户未选择分类,根据媒体类型推断
            if mtype == MediaType.MOVIE:
                params["category"] = self._movie_category
            elif mtype == MediaType.TV:
                params["category"] = self._tv_category

        return params

    def __get_category_names_by_ids(self, cat: str) -> Optional[list]:
        """
        根据用户选择的分类 ID 获取 API 的 category names

        :param cat: 用户选择的分类 ID（逗号分隔的多个ID，如 "1,2,3"）
        :return: API 的 category names 列表（如 ["movie", "tv", "documentary"]）
        """
        if not cat:
            return None

        # ID 到 category name 的映射
        id_to_name = {
            '1': 'movie',
            '2': 'tv',
            '3': 'documentary',
            '4': 'animation',
            '6': 'variety'
        }

        # 分割多个分类 ID 并映射为 category names
        cat_ids = [c.strip() for c in cat.split(',') if c.strip()]
        category_names = [id_to_name.get(cat_id) for cat_id in cat_ids if cat_id in id_to_name]

        return category_names if category_names else None

    def __process_response(self, res) -> Tuple[bool, List[dict]]:
        """
        处理 API 响应

        :param res: 请求响应对象
        :return: (是否发生错误, 种子列表)
        """
        if res and res.status_code == 200:
            try:
                data = res.json()
                if data.get('code') == 0:
                    results = data.get('data', {}).get('torrents', [])
                    return False, self.__parse_result(results)
                else:
                    logger.warn(f"{self._name} 搜索失败，错误信息：{data.get('message')}")
                    return True, []
            except Exception as e:
                logger.warn(f"{self._name} 解析响应失败：{e}")
                return True, []
        elif res is not None:
            logger.warn(f"{self._name} 搜索失败，HTTP 错误码：{res.status_code}")
            return True, []
        else:
            logger.warn(f"{self._name} 搜索失败，无法连接 {self._domain}")
            return True, []

    def __parse_result(self, results: List[dict]) -> List[dict]:
        """
        解析搜索结果

        将 API 返回的种子数据转换为 MoviePilot 标准格式

        :param results: API 返回的种子列表
        :return: 标准化的种子信息列表
        """
        torrents = []
        if not results:
            return torrents

        for result in results:
            # 解析分类信息
            raw_cat = result.get('category')
            cat_val = None

            category = MediaType.UNKNOWN.value

            if isinstance(raw_cat, dict):
                cat_val = raw_cat.get('slug') or raw_cat.get('name')
            elif isinstance(raw_cat, str):
                cat_val = raw_cat

            if cat_val:
                cat_val = str(cat_val).lower()
                if cat_val == self._movie_category:
                    category = MediaType.MOVIE.value
                elif cat_val == self._tv_category:
                    category = MediaType.TV.value
                else:
                    category = MediaType.UNKNOWN.value

            # 解析促销信息
            # API 后端已处理全站促销优先级，直接使用返回的 promotion 数据
            downloadvolumefactor = 1.0
            uploadvolumefactor = 1.0
            freedate = None

            promotion = result.get('promotion')
            if promotion and promotion.get('is_active'):
                downloadvolumefactor = float(promotion.get('down_multiplier', 1.0))
                uploadvolumefactor = float(promotion.get('up_multiplier', 1.0))
                # 促销到期时间，格式化为 YYYY-MM-DD HH:MM:SS
                if promotion.get('until'):
                    freedate = StringUtils.unify_datetime_str(promotion.get('until'))

            torrent = {
                'title': result.get('title'),
                'description': result.get('subtitle'),
                'enclosure': self.__get_download_url(result.get('id')),
                'pubdate': StringUtils.unify_datetime_str(result.get('created_at')),
                'size': int(result.get('size') or 0),
                'seeders': int(result.get('seeders') or 0),
                'peers': int(result.get('leechers') or 0),
                'grabs': int(result.get('downloads') or 0),
                'downloadvolumefactor': downloadvolumefactor,
                'uploadvolumefactor': uploadvolumefactor,
                'freedate': freedate,
                'page_url': f"https://{self._domain}/torrent/{result.get('uuid')}",
                'labels': [],
                'category': category
            }
            torrents.append(torrent)
        return torrents

    def search(self, keyword: str, mtype: MediaType = None, cat: Optional[str] = None, page: Optional[int] = 0) -> Tuple[bool, List[dict]]:
        """
        同步搜索种子

        :param keyword: 搜索关键词
        :param mtype: 媒体类型 (MOVIE/TV)
        :param cat: 用户选择的分类 ID（逗号分隔）
        :param page: 页码（从 0 开始）
        :return: (是否发生错误, 种子列表)
        """
        if not self._apikey:
            logger.warn(f"{self._name} 未配置 API Key (Passkey)")
            return True, []

        params = self.__get_params(keyword, mtype, cat, page)
        headers = {
            "Authorization": f"Bearer {self._apikey}",
            "Accept": "application/json"
        }

        res = RequestUtils(
            headers=headers,
            proxies=self._proxy,
            timeout=self._timeout
        ).get_res(url=self._searchurl, params=params)

        return self.__process_response(res)

    async def async_search(self, keyword: str, mtype: MediaType = None, cat: Optional[str] = None, page: Optional[int] = 0) -> Tuple[bool, List[dict]]:
        """
        异步搜索种子

        :param keyword: 搜索关键词
        :param mtype: 媒体类型 (MOVIE/TV)
        :param cat: 用户选择的分类 ID（逗号分隔）
        :param page: 页码（从 0 开始）
        :return: (是否发生错误, 种子列表)
        """
        if not self._apikey:
            logger.warn(f"{self._name} 未配置 API Key (Passkey)")
            return True, []

        params = self.__get_params(keyword, mtype, cat, page)
        headers = {
            "Authorization": f"Bearer {self._apikey}",
            "Accept": "application/json"
        }

        res = await AsyncRequestUtils(
            headers=headers,
            proxies=self._proxy,
            timeout=self._timeout
        ).get_res(url=self._searchurl, params=params)

        return self.__process_response(res)

    def __get_download_url(self, torrent_id: int) -> str:
        """
        构建种子下载链接

        使用 base64 编码的方式告诉 MoviePilot 如何获取真实下载地址
        MoviePilot 会先请求详情接口，然后从响应中提取 data.download_url

        :param torrent_id: 种子 ID
        :return: base64 编码的请求配置字符串 + 详情接口 URL
        """
        url = self._downloadurl % torrent_id
        # MoviePilot 会解析这个特殊格式的 URL：
        # 1. 使用指定的 method 和 header 请求 URL
        # 2. 从 JSON 响应中提取 result 指定的字段值作为真实下载地址
        params = {
            'method': 'get',
            'header': {
                'Authorization': f'Bearer {self._apikey}',
                'Accept': 'application/json'
            },
            'result': 'data.download_url'
        }
        base64_str = base64.b64encode(json.dumps(params).encode('utf-8')).decode('utf-8')
        return f"[{base64_str}]{url}"