from typing import Optional

from httpx import Client, Headers, HTTPStatusError, QueryParams, Response
from pydantic import ValidationError
from yarl import URL

from ..exceptions import StorageApiError, VectorBucketErrorMessage
from ..types import JSON, RequestMethod


class SyncRequestBuilder:
    def __init__(self, session: Client, base_url: URL, headers: Headers) -> None:
        self._session = session
        self._base_url = base_url
        self.headers = headers

    def send(
        self,
        http_method: RequestMethod,
        path: list[str],
        body: JSON = None,
        query_params: Optional[QueryParams] = None,
    ) -> Response:
        response = self._session.request(
            method=http_method,
            json=body,
            url=str(self._base_url.joinpath(*path)),
            headers=self.headers,
            params=query_params or QueryParams(),
        )
        try:
            response.raise_for_status()
            return response
        except HTTPStatusError as exc:
            try:
                error = VectorBucketErrorMessage.model_validate_json(response.content)
                raise StorageApiError(
                    message=error.message,
                    code=error.code or "400",
                    status=error.statusCode,
                ) from exc
            except ValidationError as exc:
                raise StorageApiError(
                    message=f"The request failed, but could not parse error message response:'{response.text}'",
                    code="LibraryError",
                    status=response.status_code,
                ) from exc