"""Search response schemas."""

from __future__ import annotations

from pydantic import BaseModel

from app.schemas.common import CollectionBriefOut, CompanyBriefOut
from app.schemas.movie import MovieSummaryOut
from app.schemas.person import PersonSummaryOut
from app.schemas.tv import SeriesSummaryOut


class SearchResultsOut(BaseModel):
    query: str
    movies: list[MovieSummaryOut] = []
    series: list[SeriesSummaryOut] = []
    people: list[PersonSummaryOut] = []
    collections: list[CollectionBriefOut] = []
    companies: list[CompanyBriefOut] = []
    total_results: int = 0