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

# -*- coding: utf-8 -*-
import json
from urllib.parse import urljoin
from typing import Optional, Tuple

from app.log import logger
from app.core.config import settings
from app.utils.http import RequestUtils
from app.utils.string import StringUtils
from app.modules.indexer.parser import SiteParserBase, SiteSchema


class RousiSiteUserInfo(SiteParserBase):
    """
    Rousi.pro 站点解析器
    使用 API v1 接口，通过 Passkey (Bearer Token) 进行认证
    """
    schema = SiteSchema.RousiPro
    request_mode = "apikey"

    def _parse_site_page(self, html_text: str):
        """
        配置 API 请求地址和请求头
        使用 API v1 的 /profile 接口获取用户信息
        """
        self._base_url = f"https://{StringUtils.get_url_domain(self._site_url)}"
        self._user_basic_page = "api/v1/profile?include_fields[user]=seeding_leeching_data"
        self._user_basic_params = {}
        self._user_basic_headers = {
            "Content-Type": "application/json",
            "Accept": "application/json",
            "Authorization": f"Bearer {self.apikey}"
        }

        # Rousi.pro API v1 在单个接口返回所有信息，无需额外页面
        self._user_traffic_page = None
        self._user_detail_page = None
        self._torrent_seeding_page = None
        self._user_mail_unread_page = None
        self._sys_mail_unread_page = None

    def _parse_logged_in(self, html_text):
        """
        判断是否登录成功
        API 认证模式下，通过 HTTP 状态码判断，此处始终返回 True
        """
        return True

    def _parse_user_base_info(self, html_text: str):
        """
        解析用户基本信息
        通过 API v1 接口获取用户完整信息，包括上传下载量、做种数据等

        API 响应示例：
        {
            "code": 0,
            "message": "success",
            "data": {
                "id": 1,
                "username": "example",
                "level_text": "Lv.5",
                "registered_at": "2024-01-01T00:00:00Z",
                "uploaded": 1073741824,
                "downloaded": 536870912,
                "ratio": 2.0,
                "karma": 1000.5,
                "seeding_leeching_data": {
                    "seeding_count": 10,
                    "seeding_size": 10737418240,
                    "leeching_count": 2,
                    "leeching_size": 2147483648
                }
            }
        }
        """
        if not html_text:
            return

        try:
            data = json.loads(html_text)
        except json.JSONDecodeError:
            logger.error(f"{self._site_name} JSON 解析失败")
            return

        if not data or data.get("code") != 0:
            self.err_msg = data.get("message", "未知错误")
            logger.warn(f"{self._site_name} API 错误: {self.err_msg}")
            return

        user_info = data.get("data")
        if not user_info:
            return

        # 基本信息
        self.userid = user_info.get("id")
        self.username = user_info.get("username")
        self.user_level = user_info.get("level_text") or user_info.get("role_text")

        # 注册时间：统一格式为 YYYY-MM-DD HH:MM:SS
        join_at = StringUtils.unify_datetime_str(user_info.get("registered_at"))
        if join_at:
            # 确保格式为 YYYY-MM-DD HH:MM:SS (19位)
            if len(join_at) >= 19:
                self.join_at = join_at[:19]
            else:
                self.join_at = join_at

        # 流量信息
        self.upload = int(user_info.get("uploaded") or 0)
        self.download = int(user_info.get("downloaded") or 0)
        self.ratio = round(float(user_info.get("ratio") or 0), 2)

        # 魔力值（站点称为 karma）
        self.bonus = float(user_info.get("karma") or 0)

        # 做种/下载中数据
        sl_data = user_info.get("seeding_leeching_data", {})
        self.seeding = int(sl_data.get("seeding_count") or 0)
        self.seeding_size = int(sl_data.get("seeding_size") or 0)
        self.leeching = int(sl_data.get("leeching_count") or 0)
        self.leeching_size = int(sl_data.get("leeching_size") or 0)

    def _parse_user_traffic_info(self, html_text: str):
        """
        解析用户流量信息
        Rousi.pro API v1 在 _parse_user_base_info 中已完成所有解析，此方法无需实现
        """
        pass

    def _parse_user_detail_info(self, html_text: str):
        """
        解析用户详细信息
        Rousi.pro API v1 在 _parse_user_base_info 中已完成所有解析，此方法无需实现
        """
        pass

    def _parse_user_torrent_seeding_info(self, html_text: str, multi_page: Optional[bool] = False) -> Optional[str]:
        """
        解析用户做种信息
        Rousi.pro API v1 在 _parse_user_base_info 中已通过 seeding_leeching_data 获取做种数据

        :param html_text: 页面内容
        :param multi_page: 是否多页数据
        :return: 下页地址（无下页返回 None）
        """
        return None

    def _parse_message_unread_links(self, html_text: str, msg_links: list) -> Optional[str]:
        """
        解析未读消息链接
        Rousi.pro API v1 暂未提供消息相关接口

        :param html_text: 页面内容
        :param msg_links: 消息链接列表
        :return: 下页地址（无下页返回 None）
        """
        return None

    def _parse_message_content(self, html_text) -> Tuple[Optional[str], Optional[str], Optional[str]]:
        """
        解析消息内容
        Rousi.pro API v1 暂未提供消息相关接口

        :param html_text: 页面内容
        :return: (标题, 日期, 内容)
        """
        return None, None, None

    def _pase_unread_msgs(self):
        """
        解析所有未读消息标题和内容
        Rousi.pro API v1 暂未提供消息相关接口，暂时以网页接口实现
        
        :return:
        """
        if not self.token:
            logger.warn(f"{self._site_name} 站点未配置 Authorization 请求头，跳过消息解析")
            return
        
        headers = {
            "User-Agent": self._ua,
            "Accept": "application/json, text/plain, */*",
            "Authorization": self.token if self.token.startswith("Bearer ") else f"Bearer {self.token}"
        }
        
        def __get_message_list(page: int):
            params = {
                "page": page,
                "page_size": 100,
                "unread_only": "true"
            }
            res = RequestUtils(
                headers=headers,
                timeout=60,
                proxies=settings.PROXY if self._proxy else None
            ).get_res(
                url=urljoin(self._base_url, "api/messages"),
                params=params
            )
            if not res or res.status_code != 200 or res.json().get("code", -1) != 0:
                logger.warn(f"{self._site_name} 站点解析消息失败，状态码: {res.status_code if res else '无响应'}")
                return {
                    "messages": [],
                    "total_pages": 0
                }
            return res.json().get("data")
        
        # 分页获取所有未读消息
        page = 0
        res = __get_message_list(page)
        page += 1
        messages = res.get("messages", [])
        total_pages = res.get("total_pages", 0)
        while page < total_pages:
            res = __get_message_list(page)
            messages.extend(res.get("messages", []))
            page += 1
        
        self.message_unread = len(messages)
        for messsage in messages:
            head = messsage.get("title")
            date = StringUtils.unify_datetime_str(messsage.get("created_at"))
            content = messsage.get("content")
            logger.debug(f"{self._site_name} 标题 {head} 时间 {date} 内容 {content}")
            self.message_unread_contents.append((head, date, content))
            
        # 更新消息为已读
        RequestUtils(
            headers=headers,
            timeout=60,
            proxies=settings.PROXY if self._proxy else None
        ).post_res(
            url=urljoin(self._base_url, "api/messages/read-all")
        )