MoviePilot/app/adapters/system/rust.py

import logging
from functools import lru_cache
from typing import List, Optional, Tuple

from app.foundation.environment import is_free_threaded_runtime
from app.runtime.log import logger, log_settings
from app.runtime.settings import get_runtime_setting

try:
    import moviepilot_rust as _moviepilot_rust
except Exception as err:  # pragma: no cover - 取决于运行环境是否安装 Rust 扩展
    _moviepilot_rust = None
    _import_error = err
else:
    _import_error = None


def _rust_accel_enabled() -> bool:
    """读取 Rust 开关快照，组合根未装配时回退旧 Settings。"""
    return bool(get_runtime_setting('RUST_ACCEL'))


def is_required() -> bool:
    """free-threaded 运行时必须使用不会重新启用 GIL 的 Rust 快路径。"""
    return is_free_threaded_runtime()


def is_available() -> bool:
    """
    判断 Rust 扩展是否可用。
    """
    return bool(_moviepilot_rust and _moviepilot_rust.is_available())


def is_config_enabled() -> bool:
    """
    判断系统配置是否允许使用 Rust 加速。
    """
    return is_required() or _rust_accel_enabled()


def is_enabled() -> bool:
    """
    判断当前运行时是否实际启用 Rust 加速。
    """
    return is_config_enabled() and is_available()


def status() -> dict:
    """
    返回 Rust 加速能力与开关状态，供系统配置接口展示。
    """
    return {
        "available": is_available(),
        "enabled": is_enabled(),
        "required": is_required(),
        "import_error": str(_import_error) if _import_error else "",
    }


def import_error() -> Optional[Exception]:
    """
    返回 Rust 扩展导入失败的异常，便于调试构建问题。
    """
    return _import_error


def parse_filter_rule(expression: str) -> Optional[list]:
    """
    使用 Rust 解析过滤规则表达式，不可用时返回 None。
    """
    if not is_enabled():
        return None
    try:
        return _moviepilot_rust.parse_filter_rule_fast(expression)
    except BaseException as err:
        _raise_non_rust_panic(err)
        logger.debug(f"Rust 过滤规则解析失败，回退 Python：{err}")
        return None


def filter_torrents(
        groups: list,
        torrent_list: list,
        rule_set: dict,
        mediainfo=None,
        metainfo_options: Optional[dict] = None,
) -> Optional[Tuple[list, list]]:
    """
    使用 Rust 执行完整种子过滤入口，返回原列表下标、优先级和可选调试日志。
    """
    if not is_enabled():
        return None
    try:
        args = (
            groups,
            torrent_list,
            rule_set,
            mediainfo,
            metainfo_options or {},
        )
        if is_debug_log_enabled() and hasattr(_moviepilot_rust, "filter_torrents_with_trace_fast"):
            matched_orders, traces = _moviepilot_rust.filter_torrents_with_trace_fast(*args)
            return matched_orders, traces
        return _moviepilot_rust.filter_torrents_fast(*args), []
    except BaseException as err:
        _raise_non_rust_panic(err)
        logger.debug(f"Rust 种子过滤失败，回退 Python：{err}")
        return None


def is_debug_log_enabled() -> bool:
    """
    判断当前日志配置是否会实际输出 debug 日志。
    """
    if log_settings.DEBUG:
        return True
    return getattr(logging, log_settings.LOG_LEVEL.upper(), logging.INFO) <= logging.DEBUG

def parse_indexer_torrents(
        html_text: str,
        domain: str,
        list_config: dict,
        fields: dict,
        category: Optional[dict] = None,
        result_num: int = 100
) -> Optional[List[dict]]:
    """
    使用 Rust 批量解析普通配置站点种子列表，不可用时返回 None。
    """
    if not is_enabled():
        return None
    try:
        return _moviepilot_rust.parse_indexer_torrents_fast(
            html_text,
            domain,
            list_config,
            fields,
            category,
            result_num
        )
    except BaseException as err:
        _raise_non_rust_panic(err)
        logger.debug(f"Rust 站点列表解析失败，使用 Python 解析兜底：{err}")
        return None


def parse_indexer_subtitles(
        html_text: str,
        domain: str,
        list_config: dict,
        fields: dict,
        result_num: int = 100
) -> Optional[List[dict]]:
    """
    使用 Rust 批量解析普通配置站点字幕列表，不可用时返回 None。
    """
    if not is_enabled():
        return None
    try:
        return _moviepilot_rust.parse_indexer_subtitles_fast(
            html_text,
            domain,
            list_config,
            fields,
            result_num
        )
    except BaseException as err:
        _raise_non_rust_panic(err)
        logger.debug(f"Rust 字幕列表解析失败，使用 Python 解析兜底：{err}")
        return None


def parse_rss_items(xml_text: str, max_items: int = 1000) -> Optional[List[dict]]:
    """
    使用 Rust 解析 RSS/Atom 条目，不可用或异常时返回 None。
    """
    if not is_enabled():
        return None
    try:
        return _moviepilot_rust.parse_rss_items_fast(xml_text, max_items)
    except BaseException as err:
        _raise_non_rust_panic(err)
        logger.debug(f"Rust RSS解析失败，使用 Python 解析兜底：{err}")
        return None


def parse_metainfo(title: str, subtitle: Optional[str] = None, options: Optional[dict] = None) -> Optional[dict]:
    """
    使用 Rust 从标题入口解析 MetaInfo，不可用或异常时返回 None。
    """
    if not is_enabled():
        return None
    try:
        return _moviepilot_rust.parse_metainfo_fast(title, subtitle, options or {})
    except BaseException as err:
        _raise_non_rust_panic(err)
        logger.debug(f"Rust MetaInfo解析失败，使用 Python 解析兜底：{err}")
        return None


def parse_metainfo_path(path: str, options: Optional[dict] = None) -> Optional[dict]:
    """
    使用 Rust 从路径入口解析 MetaInfoPath，不可用或异常时返回 None。
    """
    if not is_enabled():
        return None
    try:
        return _moviepilot_rust.parse_metainfo_path_fast(path, options or {})
    except BaseException as err:
        _raise_non_rust_panic(err)
        logger.debug(f"Rust MetaInfoPath解析失败，使用 Python 解析兜底：{err}")
        return None


def parse_metamusic(
        title: str,
        artists: Optional[List[str]] = None,
        year: Optional[int] = None,
) -> Optional[dict]:
    """使用 Rust 解析音乐资源标题，旧扩展不支持时返回 None。

    :param title: 音乐资源标题或文件主干名
    :param artists: 调用方已有的高可信艺术家列表
    :param year: 调用方已有的高可信发行年份
    :return: Rust 解析字段，不可用、不支持或异常时返回 None
    """
    if not is_enabled():
        return None
    parser = getattr(_moviepilot_rust, "parse_metamusic_fast", None)
    if not callable(parser):
        return None
    try:
        result = parser(title, artists, year)
    except BaseException as err:
        _raise_non_rust_panic(err)
        logger.debug(f"Rust MetaMusic解析失败，使用 Python 解析兜底：{err}")
        return None
    return result if isinstance(result, dict) and result else None


def find_metainfo(title: str) -> Optional[dict]:
    """
    使用 Rust 提取标题中的显式媒体标签，不可用或异常时返回 None。
    """
    if not is_enabled():
        return None
    try:
        return _moviepilot_rust.find_metainfo_fast(title)
    except BaseException as err:
        _raise_non_rust_panic(err)
        logger.debug(f"Rust 显式媒体标签解析失败，使用 Python 解析兜底：{err}")
        return None


@lru_cache(maxsize=1)
def supports_extended_media_ids() -> bool:
    """
    判断当前 Rust 扩展是否支持 Bangumi 与 AniList 显式媒体标签。

    :return: 是否支持扩展数据源ID字段
    """
    if not is_enabled():
        return False
    try:
        result = _moviepilot_rust.find_metainfo_fast("test [anilist=1]")
    except BaseException as err:
        _raise_non_rust_panic(err)
        logger.debug(f"检测 Rust 扩展数据源ID能力失败：{err}")
        return False
    metainfo = result.get("metainfo") if isinstance(result, dict) else None
    return bool(
        metainfo
        and metainfo.get("media_source") == "anilist"
        and metainfo.get("media_id") == "1"
    )


def _raise_non_rust_panic(err: BaseException) -> None:
    """
    只吞掉 Rust 扩展 panic/异常，保留用户中断和进程退出语义。
    """
    if isinstance(err, (KeyboardInterrupt, SystemExit)):
        raise err