"""Contacts resource — contact lookup and management.

Backed by ``src/modules/contact/contact.controller.ts``.
"""

from __future__ import annotations

from typing import List, TYPE_CHECKING, TypedDict

from .._http import quote_segment
from ..types import (
    UpsertContactRequest,
    CheckNumberResponse,
    ContactPhoneResponse,
    ContactRecord,
    ProfilePictureResponse,
    ProfilePicturesResponse,
    SuccessResult,
)

if TYPE_CHECKING:
    from .._http import HttpExecutor


class ListContactsQuery(TypedDict, total=False):
    limit: int
    offset: int


class ContactsResource:
    def __init__(self, http: "HttpExecutor") -> None:
        self._http = http

    def list(self, session_id: str, query: ListContactsQuery | None = None) -> List[ContactRecord]:
        return self._http.request("GET", f"/api/sessions/{quote_segment(session_id)}/contacts", query=query)

    def get(self, session_id: str, contact_id: str) -> ContactRecord:
        return self._http.request("GET", f"/api/sessions/{quote_segment(session_id)}/contacts/{quote_segment(contact_id)}")

    def check(self, session_id: str, number: str) -> CheckNumberResponse:
        return self._http.request("GET", f"/api/sessions/{quote_segment(session_id)}/contacts/check/{quote_segment(number)}")

    def profile_picture(self, session_id: str, contact_id: str) -> ProfilePictureResponse:
        return self._http.request(
            "GET", f"/api/sessions/{quote_segment(session_id)}/contacts/{quote_segment(contact_id)}/profile-picture"
        )

    def profile_pictures(self, session_id: str, ids: List[str]) -> ProfilePicturesResponse:
        """Batch-resolve profile picture URLs for up to 50 contacts in one request.
        Returns a map of contact id → URL (None when a lookup fails)."""
        return self._http.request(
            "GET", f"/api/sessions/{quote_segment(session_id)}/contacts/profile-pictures", query={"ids": ",".join(ids)}
        )

    def phone(self, session_id: str, contact_id: str) -> ContactPhoneResponse:
        return self._http.request("GET", f"/api/sessions/{quote_segment(session_id)}/contacts/{quote_segment(contact_id)}/phone")

    def block(self, session_id: str, contact_id: str) -> SuccessResult:
        return self._http.request("POST", f"/api/sessions/{quote_segment(session_id)}/contacts/{quote_segment(contact_id)}/block")

    def upsert(self, session_id: str, contact_id: str, body: UpsertContactRequest) -> SuccessResult:
        """Save a contact to the addressbook, or edit an existing entry (OPERATOR)."""
        return self._http.request(
            "PUT", f"/api/sessions/{quote_segment(session_id)}/contacts/{quote_segment(contact_id)}", body=body
        )

    def delete(self, session_id: str, contact_id: str) -> SuccessResult:
        """Remove a contact from the addressbook (OPERATOR)."""
        return self._http.request(
            "DELETE", f"/api/sessions/{quote_segment(session_id)}/contacts/{quote_segment(contact_id)}"
        )

    def unblock(self, session_id: str, contact_id: str) -> SuccessResult:
        return self._http.request("DELETE", f"/api/sessions/{quote_segment(session_id)}/contacts/{quote_segment(contact_id)}/block")

    def list_blocked(self, session_id: str) -> List[str]:
        """List the JIDs this account has blocked.

        Session-wide, so it takes no contact id — unlike ``block`` and ``unblock``, which act on one
        contact. Returns a bare list of ids, not contact records.
        """
        return self._http.request("GET", f"/api/sessions/{quote_segment(session_id)}/contacts/blocked")