from typing import Final

from pydantic import TypeAdapter
from pydantic.dataclasses import dataclass

from mdrsclient.api.base import BaseApi
from mdrsclient.api.utils import token_check


@dataclass(frozen=True)
class DoiFolderRef:
    """Nested folder reference returned inside a DOI response.

    The DOI endpoint only returns the folder ``id``; ``laboratory_id`` must be
    obtained by subsequently calling the folder retrieve endpoint.
    """

    id: str


@dataclass(frozen=True)
class DoiResponse:
    """Response from GET v3/doi/{id}/."""

    # The internal DOI suffix ID returned as a string (e.g. "20260429-001").
    id: str
    doi: str
    folder: DoiFolderRef


class DoiApi(BaseApi):
    ENTRYPOINT: Final[str] = "v3/doi/"

    def retrieve(self, doi_id: str) -> DoiResponse:
        """Retrieve the folder associated with a DOI suffix ID (GET v3/doi/{id}/)."""
        url = self.ENTRYPOINT + doi_id + "/"
        token_check(self.connection)
        response = self.connection.get(url)
        self._raise_response_error(response)
        return TypeAdapter(DoiResponse).validate_python(response.json())