modsetter--surfsense/surfsense_backend/app/capabilities/youtube/comments/executor.py

"""``youtube.comments`` executor: verb input → scraper → comment items."""

from __future__ import annotations

from collections.abc import Awaitable, Callable

from app.capabilities.core import Executor
from app.capabilities.core.progress import emit_progress
from app.capabilities.youtube.comments.schemas import CommentsInput, CommentsOutput
from app.proprietary.platforms.youtube import (
    YouTubeCommentsInput,
    scrape_comments,
)

CommentsFn = Callable[[YouTubeCommentsInput], Awaitable[list[dict]]]


def build_comments_executor(scrape_fn: CommentsFn | None = None) -> Executor:
    """Bind the executor to a comments scraper fn (defaults to the proprietary actor)."""
    scrape_fn = scrape_fn or scrape_comments

    async def execute(payload: CommentsInput) -> CommentsOutput:
        actor_input = YouTubeCommentsInput(
            startUrls=[{"url": url} for url in payload.urls],
            maxComments=payload.max_comments,
            sortCommentsBy=payload.sort_by,
        )
        emit_progress(
            "starting",
            "Fetching YouTube comments",
            total=payload.max_comments,
            unit="comment",
        )
        items = await scrape_fn(actor_input)
        emit_progress(
            "done",
            f"Scraped {len(items)} comment(s)",
            current=len(items),
            unit="comment",
        )
        return CommentsOutput(items=items)

    return execute