MoviePilot/app/application/plugin/gateway.py

"""统一插件安装 Gateway。"""

from __future__ import annotations

from collections.abc import Awaitable, Callable
from dataclasses import dataclass
from datetime import datetime
from typing import Protocol

from app.application.plugin.admission import (
    PluginInstallAdmission,
    PluginInstallAdmissionRequest,
    PluginSourceAdmissionError,
    admit_plugin_install,
)
from app.application.plugin.identity import PluginIdentity
from app.application.plugin.install import PluginInstallResult
from app.application.plugin.inventory import PLUGIN_V3_GENERATIONS
from app.application.plugin.lifecycle import PluginStartupLease, plugin_lifecycle
from app.application.plugin.source import (
    Candidate,
    CandidateInventory,
    PluginLocalCandidate,
    PluginMarketCandidate,
    PluginSelection,
    get_effective_local_candidate,
    list_effective_online_candidates,
    select_plugin_candidate,
)

InventoryProvider = Callable[[bool], Awaitable[CandidateInventory]]
IdentityReader = Callable[[str], Awaitable[PluginIdentity | None]]
CandidateCompatibility = Callable[[Candidate], tuple[bool, str]]


@dataclass(frozen=True, slots=True)
class PluginSourceInspection:
    """前端与 Agent 选择来源所需的只读候选和当前身份快照。"""

    plugin_id: str
    inventory_complete: bool
    identity: PluginIdentity | None
    selection: PluginSelection
    online_candidates: tuple[PluginMarketCandidate, ...]
    local_candidate: PluginLocalCandidate | None


class PluginInstallExecutor(Protocol):
    """统一 Gateway 调用的可恢复安装执行端口。"""

    async def execute(
        self,
        *,
        admission: PluginInstallAdmission,
        release_version: str | None,
        force: bool,
        local_sync: bool = False,
    ) -> PluginInstallResult:
        """执行已通过来源准入的插件载荷事务。"""


class PluginInstallGateway:
    """让全部插件载荷写入共享同一来源策略和事务执行器。"""

    def __init__(
        self,
        *,
        inventory: InventoryProvider,
        identity: IdentityReader,
        candidate_compatibility: CandidateCompatibility,
        executor: PluginInstallExecutor,
        clock: Callable[[], datetime],
    ) -> None:
        """保存候选事实、身份读取、兼容校验、事务执行和时间端口。"""
        self.__inventory = inventory
        self.__identity = identity
        self.__candidate_compatibility = candidate_compatibility
        self.__executor = executor
        self.__clock = clock

    async def install(
        self,
        *,
        plugin_id: str,
        repo_url: str | None,
        package_version: str | None = None,
        release_version: str | None = None,
        force: bool = False,
        explicit_source: bool = False,
        source_change: bool = False,
        expected_revision: int | None = None,
        startup_token: PluginStartupLease | None = None,
        local_sync: bool = False,
    ) -> PluginInstallResult:
        """读取冻结库存并执行一次不能绕过来源身份的插件写入。"""
        try:
            inventory = await self.__inventory(force)
            async with plugin_lifecycle.hold(plugin_id, startup_token):
                identity = await self.__identity(plugin_id)
                admission = admit_plugin_install(
                    inventory,
                    request=PluginInstallAdmissionRequest(
                        plugin_id=plugin_id,
                        generations=_generation_order(package_version),
                        requested_repo_url=repo_url,
                        explicit_source=explicit_source,
                        source_change=source_change,
                        expected_revision=expected_revision,
                    ),
                    identity=identity,
                    now=self.__clock(),
                )
                compatible, message = self.__candidate_compatibility(
                    admission.candidate
                )
                if not compatible:
                    raise PluginSourceAdmissionError(
                        message or "插件候选与当前 MoviePilot 版本不兼容"
                    )
                return await self.__executor.execute(
                    admission=admission,
                    release_version=release_version,
                    force=force,
                    local_sync=local_sync,
                )
        except (TypeError, ValueError, PluginSourceAdmissionError) as error:
            return PluginInstallResult(
                success=False,
                message=str(error),
                failure_stage="source_admission",
            )

    async def inspect_source(
        self,
        *,
        plugin_id: str,
        package_version: str | None = None,
        force: bool = False,
    ) -> PluginSourceInspection:
        """读取与真实安装相同的库存和身份，返回脱敏来源选择快照。"""
        inventory = await self.__inventory(force)
        identity = await self.__identity(plugin_id)
        generations = _generation_order(package_version)
        selection = select_plugin_candidate(
            inventory,
            plugin_id=plugin_id,
            generations=generations,
            identity=identity,
        )
        return PluginSourceInspection(
            plugin_id=plugin_id,
            inventory_complete=inventory.complete,
            identity=identity,
            selection=selection,
            online_candidates=list_effective_online_candidates(
                inventory,
                plugin_id=plugin_id,
                generations=generations,
            ),
            local_candidate=get_effective_local_candidate(
                inventory,
                plugin_id=plugin_id,
                generations=generations,
            ),
        )


_plugin_install_gateway: PluginInstallGateway | None = None


def configure_plugin_install_service(gateway: PluginInstallGateway) -> None:
    """由启动组合根发布当前 lifespan 的唯一插件安装 Gateway。"""
    global _plugin_install_gateway
    _plugin_install_gateway = gateway


def get_plugin_install_service() -> PluginInstallGateway:
    """返回已装配 Gateway；启动未完成时拒绝任何载荷写入。"""
    if _plugin_install_gateway is None:
        raise RuntimeError("插件安装服务尚未完成初始化")
    return _plugin_install_gateway


def reset_plugin_install_service() -> None:
    """清除当前 lifespan 的 Gateway，供停机和隔离测试使用。"""
    global _plugin_install_gateway
    _plugin_install_gateway = None


def _generation_order(package_version: str | None) -> tuple[str, ...]:
    """把兼容入口的首选代际转换为来源选择优先序。"""
    normalized = (package_version or "v3").strip().lower()
    if normalized in {"", "v1"}:
        return ("v1",)
    if normalized == "v2":
        return ("v2", "v1")
    if normalized == "v3":
        return PLUGIN_V3_GENERATIONS
    raise ValueError("插件包代际必须为 v1、v2 或 v3")