"""Search service: runs a `SearchProvider` and hydrates matches into full
response summaries via the entity services. API clients never see the
underlying search engine (section 16)."""

from __future__ import annotations

from sqlalchemy.ext.asyncio import AsyncSession

from app.schemas.search import SearchResultsOut
from app.services.collection_service import build_collection_summaries
from app.services.company_service import build_company_summaries
from app.services.movie_service import build_movie_summaries
from app.services.person_service import build_person_summaries
from app.services.search.base import SearchProvider
from app.services.tv_service import build_series_summaries


class SearchService:
    def __init__(self, provider: SearchProvider) -> None:
        self._provider = provider

    async def search(
        self, session: AsyncSession, query: str, *, limit_per_type: int = 10
    ) -> SearchResultsOut:
        query = query.strip()
        if not query:
            return SearchResultsOut(query=query)

        results = await self._provider.search(session, query, limit_per_type=limit_per_type)

        movies = await build_movie_summaries(session, [m.entity_id for m in results.movies])
        series = await build_series_summaries(session, [s.entity_id for s in results.series])
        people = await build_person_summaries(session, [p.entity_id for p in results.people])
        collections = await build_collection_summaries(
            session, [c.entity_id for c in results.collections]
        )
        companies = await build_company_summaries(
            session, [c.entity_id for c in results.companies]
        )

        return SearchResultsOut(
            query=query,
            movies=movies,
            series=series,
            people=people,
            collections=collections,
            companies=companies,
            total_results=len(movies) + len(series) + len(people) + len(collections) + len(companies),
        )