MoviePilot/app/api/endpoints/submaintenance.py

"""
订阅维护命令端点。
"""

from typing import Annotated, Any, Optional

from fastapi import Depends, Query

from app.api.dependencies.auth import (
    get_current_active_user,
    get_current_active_user_async,
)
from app.api.dependencies.subscription import (
    get_search_subscriptions_command,
    get_subscription_mutation_service,
)
from app.api.principal import ApiPrincipal
from app.api.response import ResponseAPIRouter
from app.application.scheduling import get_scheduler
from app.application.subscription.mutation import (
    SubscriptionActor,
    SubscriptionMutationService,
)
from app.application.subscription.search import (
    SearchSubscriptionsCommand,
    SubscribeSearchActor,
    SubscriptionSearchSubmission,
)
from app.schemas.response import Response
from app.schemas.subscribe import (
    SubscriptionSearchSubmission as SubscriptionSearchSubmissionSchema,
)
from app.schemas.types import MediaType

router = ResponseAPIRouter()

SubscriptionTypeParam = Annotated[
    Optional[MediaType],
    Query(alias="type", description="仅处理指定媒体类型的订阅；省略时处理全部类型"),
]


def _subscription_job_kwargs(subscription_type: Optional[MediaType]) -> dict[str, str]:
    """
    把可选媒体类型转换为调度任务参数，未指定时保留全量任务语义。
    """
    if subscription_type is None:
        return {}
    return {"mtype": subscription_type.value}


def _search_submission_message(submission: SubscriptionSearchSubmission) -> str:
    """
    把搜索安排结果转换为用户能直接理解的提示。
    """
    if submission.target_count == 0:
        return "没有需要搜索的订阅"
    if submission.queued_count == 0:
        return (
            "这个订阅已经在搜索中，请稍候"
            if submission.single
            else f"{submission.ongoing_count} 个订阅已经在搜索中，无需重复提交"
        )
    if submission.ongoing_count:
        return (
            f"已安排 {submission.queued_count} 个订阅搜索，"
            f"另有 {submission.ongoing_count} 个正在处理中"
        )
    if submission.single:
        return "已安排搜索，很快开始"
    return f"已安排 {submission.queued_count} 个订阅搜索，系统会依次处理"


def _search_submission_schema(
    submission: SubscriptionSearchSubmission,
) -> SubscriptionSearchSubmissionSchema:
    """
    构造稳定的手工搜索响应数据。
    """
    return SubscriptionSearchSubmissionSchema(
        batch_id=submission.batch_id,
        batch_ids=list(submission.batch_ids),
        target_count=submission.target_count,
        queued_count=submission.queued_count,
        ongoing_count=submission.ongoing_count,
        single=submission.single,
    )


@router.get(
    "/refresh",
    summary="刷新订阅（兼容入口）",
    response_model=Response[None],
    include_in_schema=False,
    deprecated=True,
)
@router.post(
    "/refresh", summary="刷新订阅", response_model=Response[None]
)
def refresh_subscribes(
    subscription_type: SubscriptionTypeParam = None,
    current_user: ApiPrincipal = Depends(get_current_active_user),
) -> Any:
    """
    刷新订阅，可选按媒体类型限定处理范围。
    """
    if not current_user.is_superuser:
        return Response(success=False, message="订阅不存在")
    get_scheduler().start("subscribe_refresh", source="manual", **_subscription_job_kwargs(subscription_type))
    return Response(success=True)


@router.get(
    "/reset/{subid}",
    summary="重置订阅（兼容入口）",
    response_model=Response[None],
    include_in_schema=False,
    deprecated=True,
)
@router.post(
    "/reset/{subid}", summary="重置订阅", response_model=Response[None]
)
async def reset_subscribes(
    subid: int,
    mutation: SubscriptionMutationService = Depends(get_subscription_mutation_service),
    current_user: ApiPrincipal = Depends(get_current_active_user_async),
) -> Any:
    """
    重置一个订阅。
    """
    actor = SubscriptionActor(
        name=current_user.name,
        is_superuser=current_user.is_superuser,
    )
    change = await mutation.reset(subid, actor)
    if change:
        return Response(success=True)
    return Response(success=False, message="订阅不存在")


@router.get(
    "/check",
    summary="刷新订阅 TMDB 信息（兼容入口）",
    response_model=Response[None],
    include_in_schema=False,
    deprecated=True,
)
@router.post(
    "/check", summary="刷新订阅 TMDB 信息", response_model=Response[None]
)
def check_subscribes(
    subscription_type: SubscriptionTypeParam = None,
    current_user: ApiPrincipal = Depends(get_current_active_user),
) -> Any:
    """
    刷新订阅 TMDB 信息，可选按媒体类型限定处理范围。
    """
    if not current_user.is_superuser:
        return Response(success=False, message="订阅不存在")
    get_scheduler().start("subscribe_tmdb", **_subscription_job_kwargs(subscription_type))
    return Response(success=True)


@router.get(
    "/search",
    summary="搜索所有订阅（兼容入口）",
    response_model=Response[SubscriptionSearchSubmissionSchema],
    include_in_schema=False,
    deprecated=True,
)
@router.post(
    "/search",
    summary="搜索所有订阅",
    response_model=Response[SubscriptionSearchSubmissionSchema],
)
async def search_subscribes(
    subscription_type: SubscriptionTypeParam = None,
    command: SearchSubscriptionsCommand = Depends(get_search_subscriptions_command),
    current_user: ApiPrincipal = Depends(get_current_active_user_async),
) -> Any:
    """
    搜索当前用户可管理的订阅，可选按媒体类型限定处理范围。
    """
    submission = await command.execute(
        SubscribeSearchActor(
            username=current_user.name,
            is_superuser=current_user.is_superuser,
        ),
        mtype=subscription_type.value if subscription_type else None,
    )
    if submission is None:
        return Response(success=False, message="没有需要搜索的订阅")
    return Response(
        success=True,
        message=_search_submission_message(submission),
        data=_search_submission_schema(submission),
    )


@router.get(
    "/search/{subscribe_id}",
    summary="搜索订阅（兼容入口）",
    response_model=Response[SubscriptionSearchSubmissionSchema],
    include_in_schema=False,
    deprecated=True,
)
@router.post(
    "/search/{subscribe_id}",
    summary="搜索订阅",
    response_model=Response[SubscriptionSearchSubmissionSchema],
)
async def search_subscribe(
    subscribe_id: int,
    subscription_type: SubscriptionTypeParam = None,
    command: SearchSubscriptionsCommand = Depends(get_search_subscriptions_command),
    current_user: ApiPrincipal = Depends(get_current_active_user_async),
) -> Any:
    """
    根据订阅编号搜索一个订阅，可选校验媒体类型。
    """
    submission = await command.execute(
        SubscribeSearchActor(
            username=current_user.name,
            is_superuser=current_user.is_superuser,
        ),
        subscribe_id=subscribe_id,
        mtype=subscription_type.value if subscription_type else None,
    )
    if submission is None:
        return Response(success=False, message="订阅不存在")
    return Response(
        success=True,
        message=_search_submission_message(submission),
        data=_search_submission_schema(submission),
    )