"""当事人查询服务组合。"""

from __future__ import annotations

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from django.contrib.auth.models import AbstractBaseUser as User
    from django.db.models import QuerySet

    from apps.client.models import Client

    from .query import ClientBatchQueryService, ClientGetQueryService, ClientListQueryService


class ClientQueryService:
    def __init__(
        self,
        list_query: ClientListQueryService | None = None,
        get_query: ClientGetQueryService | None = None,
        batch_query: ClientBatchQueryService | None = None,
    ) -> None:
        self._list_query = list_query
        self._get_query = get_query
        self._batch_query = batch_query

    @property
    def list_query(self) -> ClientListQueryService:
        if self._list_query is None:
            from .query import ClientListQueryService

            self._list_query = ClientListQueryService()
        return self._list_query

    @property
    def get_query(self) -> ClientGetQueryService:
        if self._get_query is None:
            from .query import ClientGetQueryService

            self._get_query = ClientGetQueryService()
        return self._get_query

    @property
    def batch_query(self) -> ClientBatchQueryService:
        if self._batch_query is None:
            from .query import ClientBatchQueryService

            self._batch_query = ClientBatchQueryService()
        return self._batch_query

    def list_clients(
        self,
        *,
        page: int = 1,
        page_size: int = 20,
        max_page_size: int = 100,
        client_type: str | None = None,
        is_our_client: bool | None = None,
        search: str | None = None,
        user: User | None = None,
    ) -> QuerySet[Client, Client]:
        return self.list_query.list_clients(
            page=page,
            page_size=page_size,
            max_page_size=max_page_size,
            client_type=client_type,
            is_our_client=is_our_client,
            search=search,
            user=user,
        )

    def get_client(self, *, client_id: int, user: User | None = None) -> Client:
        return self.get_query.get_client(client_id=client_id, user=user)

    def get_clients_by_ids(self, *, client_ids: list[int]) -> list[Client]:
        return self.batch_query.get_clients_by_ids(client_ids=client_ids)