MoviePilot/app/startup/context.py

"""宿主启动阶段构建的类型化运行时上下文。"""

from collections.abc import AsyncGenerator, Callable, Generator
from dataclasses import dataclass, field
from typing import Protocol

from app.runtime.tasks import TaskRegistry

from app.application.messaging.chat import (
    AsyncAgentChatRepository,
    AgentChatPersistenceService,
    AsyncUnitOfWork,
)
from app.application.outbox import AsyncOutboxTransaction
from app.application.configuration import RuntimeConfiguration, RuntimeSettingsService
from app.application.subscription.delete import SubscribeDeletionRepository
from app.application.subscription.identity import SubscribeIdentityDeletionRepository
from app.application.subscription.mutation import (
    SubscriptionHistoryMutationRepository,
    SubscriptionMutationRepository,
)
from app.application.workflow import WorkflowCachePort


class AgentChatRepositoryFactory(Protocol):
    """由请求会话构造 Agent 会话仓储的工厂端口。"""

    def __call__(self, session: object) -> AsyncAgentChatRepository:
        """绑定请求会话并返回 Agent 会话仓储。"""
        ...


class AsyncUnitOfWorkFactory(Protocol):
    """由请求会话构造异步事务端口的工厂。"""

    def __call__(self, session: object) -> AsyncUnitOfWork:
        """绑定请求会话并返回异步事务端口。"""
        ...


class AsyncOutboxFactory(Protocol):
    """由请求会话构造异步 outbox 事务端口的工厂。"""

    def __call__(self, session: object) -> AsyncOutboxTransaction:
        """绑定请求会话并返回 outbox 暂存与收口端口。"""
        ...


class SubscriptionRepositoryFactory(Protocol):
    """由请求会话构造订阅写仓储的工厂。"""

    def __call__(
        self,
        session: object,
    ) -> (
        SubscriptionMutationRepository
        | SubscribeDeletionRepository
        | SubscribeIdentityDeletionRepository
    ):
        """绑定请求会话并返回订阅领域仓储。"""
        ...


class SubscriptionHistoryRepositoryFactory(Protocol):
    """由请求会话构造订阅历史写仓储的工厂。"""

    def __call__(self, session: object) -> SubscriptionHistoryMutationRepository:
        """绑定请求会话并返回订阅历史仓储。"""
        ...


class AsyncSessionProvider(Protocol):
    """FastAPI 请求级异步会话提供器。"""

    def __call__(self) -> AsyncGenerator[object, None]:
        """生成一个请求独占的异步数据库会话。"""
        ...


class SyncSessionProvider(Protocol):
    """兼容 API Facade 使用的同步会话提供器。"""

    def __call__(self) -> Generator[object, None, None]:
        """生成一个请求独占的同步数据库会话。"""
        ...


class RepositoryFactory(Protocol):
    """由请求 Session 构造某一明确领域仓储的通用工厂。"""

    def __call__(self, session: object) -> object:
        """绑定请求会话并返回领域仓储。"""
        ...


class StandaloneRepositoryFactory(Protocol):
    """构造自持有兼容事务边界的领域仓储。"""

    def __call__(self) -> object:
        """返回无需请求 Session 的领域仓储。"""
        ...


class SyncUnitOfWorkFactory(Protocol):
    """由同步请求 Session 构造事务端口的工厂。"""

    def __call__(self, session: object) -> object:
        """绑定请求会话并返回同步事务端口。"""
        ...


@dataclass(frozen=True, slots=True)
class AgentChatRuntime:
    """Agent 会话 API 可见的最小数据运行时。"""

    async_session: AsyncSessionProvider
    repository: AgentChatRepositoryFactory
    transaction: AsyncUnitOfWorkFactory
    persistence: AgentChatPersistenceService


@dataclass(frozen=True, slots=True)
class PersistenceRuntime:
    """全部 HTTP 业务领域共享的请求会话与事务工厂。"""

    sync_session: SyncSessionProvider
    async_session: AsyncSessionProvider
    sync_transaction: SyncUnitOfWorkFactory
    async_transaction: AsyncUnitOfWorkFactory


@dataclass(frozen=True, slots=True)
class AuthenticationRuntime:
    """认证、用户管理与 PassKey API 的显式数据工厂。"""

    user_repository: RepositoryFactory
    standalone_user: StandaloneRepositoryFactory
    system_config: StandaloneRepositoryFactory
    passkey: StandaloneRepositoryFactory


@dataclass(frozen=True, slots=True)
class MessagingRuntime:
    """消息历史 API 的显式仓储工厂。"""

    repository: RepositoryFactory


@dataclass(frozen=True, slots=True)
class HistoryRuntime:
    """下载、整理、媒体服务器与 Dashboard 领域的数据工厂。"""

    download_repository: RepositoryFactory
    transfer_repository: RepositoryFactory
    media_server_repository: RepositoryFactory


@dataclass(frozen=True, slots=True)
class SiteRuntime:
    """站点读写领域的显式仓储工厂。"""

    repository: RepositoryFactory


@dataclass(frozen=True, slots=True)
class WorkflowRuntime:
    """工作流定义、状态与缓存操作所需的数据工厂。"""

    repository: RepositoryFactory
    system_config: Callable[[], WorkflowCachePort]


@dataclass(frozen=True, slots=True)
class SubscriptionRuntime:
    """订阅 API 可见的请求级写事务运行时。"""

    async_session: AsyncSessionProvider
    repository: SubscriptionRepositoryFactory
    history_repository: SubscriptionHistoryRepositoryFactory
    transaction: AsyncUnitOfWorkFactory
    outbox: AsyncOutboxFactory


@dataclass(frozen=True, slots=True)
class HostRuntime:
    """宿主组合根构建且在一个 FastAPI lifespan 内共享的运行时对象。"""

    agent_chat: AgentChatRuntime
    persistence: PersistenceRuntime
    authentication: AuthenticationRuntime
    messaging: MessagingRuntime
    history: HistoryRuntime
    site: SiteRuntime
    subscription: SubscriptionRuntime
    workflow: WorkflowRuntime
    configuration: RuntimeConfiguration
    settings: RuntimeSettingsService
    tasks: TaskRegistry = field(default_factory=TaskRegistry)