from http import HTTPStatus
from typing import Any

import httpx

from ... import errors
from ...client import AuthenticatedClient, Client
from ...models.error_response import ErrorResponse
from ...models.retriever_stage_definition import RetrieverStageDefinition
from ...types import Response


def _get_kwargs() -> dict[str, Any]:
    _kwargs: dict[str, Any] = {
        "method": "get",
        "url": "/v1/retrievers/stages",
    }

    return _kwargs


def _parse_response(
    *, client: AuthenticatedClient | Client, response: httpx.Response
) -> ErrorResponse | list[RetrieverStageDefinition] | None:
    if response.status_code == 200:
        response_200 = []
        _response_200 = response.json()
        for response_200_item_data in _response_200:
            response_200_item = RetrieverStageDefinition.from_dict(response_200_item_data)

            response_200.append(response_200_item)

        return response_200

    if response.status_code == 400:
        response_400 = ErrorResponse.from_dict(response.json())

        return response_400

    if response.status_code == 401:
        response_401 = ErrorResponse.from_dict(response.json())

        return response_401

    if response.status_code == 403:
        response_403 = ErrorResponse.from_dict(response.json())

        return response_403

    if response.status_code == 404:
        response_404 = ErrorResponse.from_dict(response.json())

        return response_404

    if response.status_code == 500:
        response_500 = ErrorResponse.from_dict(response.json())

        return response_500

    if client.raise_on_unexpected_status:
        raise errors.UnexpectedStatus(response.status_code, response.content)
    else:
        return None


def _build_response(
    *, client: AuthenticatedClient | Client, response: httpx.Response
) -> Response[ErrorResponse | list[RetrieverStageDefinition]]:
    return Response(
        status_code=HTTPStatus(response.status_code),
        content=response.content,
        headers=response.headers,
        parsed=_parse_response(client=client, response=response),
    )


def sync_detailed(
    *,
    client: AuthenticatedClient | Client,
) -> Response[ErrorResponse | list[RetrieverStageDefinition]]:
    """List Available Retriever Stages

     List all registered retriever stages with their configurations. Use this endpoint to discover
    available stages before creating retrievers. Each stage includes its ID, description, category, and
    full parameter schema. The parameter_schema field contains complete Pydantic JSON Schema with
    validation rules, descriptions, and examples for all stage parameters.

    Raises:
        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
        httpx.TimeoutException: If the request takes longer than Client.timeout.

    Returns:
        Response[ErrorResponse | list[RetrieverStageDefinition]]
    """

    kwargs = _get_kwargs()

    response = client.get_httpx_client().request(
        **kwargs,
    )

    return _build_response(client=client, response=response)


def sync(
    *,
    client: AuthenticatedClient | Client,
) -> ErrorResponse | list[RetrieverStageDefinition] | None:
    """List Available Retriever Stages

     List all registered retriever stages with their configurations. Use this endpoint to discover
    available stages before creating retrievers. Each stage includes its ID, description, category, and
    full parameter schema. The parameter_schema field contains complete Pydantic JSON Schema with
    validation rules, descriptions, and examples for all stage parameters.

    Raises:
        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
        httpx.TimeoutException: If the request takes longer than Client.timeout.

    Returns:
        ErrorResponse | list[RetrieverStageDefinition]
    """

    return sync_detailed(
        client=client,
    ).parsed


async def asyncio_detailed(
    *,
    client: AuthenticatedClient | Client,
) -> Response[ErrorResponse | list[RetrieverStageDefinition]]:
    """List Available Retriever Stages

     List all registered retriever stages with their configurations. Use this endpoint to discover
    available stages before creating retrievers. Each stage includes its ID, description, category, and
    full parameter schema. The parameter_schema field contains complete Pydantic JSON Schema with
    validation rules, descriptions, and examples for all stage parameters.

    Raises:
        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
        httpx.TimeoutException: If the request takes longer than Client.timeout.

    Returns:
        Response[ErrorResponse | list[RetrieverStageDefinition]]
    """

    kwargs = _get_kwargs()

    response = await client.get_async_httpx_client().request(**kwargs)

    return _build_response(client=client, response=response)


async def asyncio(
    *,
    client: AuthenticatedClient | Client,
) -> ErrorResponse | list[RetrieverStageDefinition] | None:
    """List Available Retriever Stages

     List all registered retriever stages with their configurations. Use this endpoint to discover
    available stages before creating retrievers. Each stage includes its ID, description, category, and
    full parameter schema. The parameter_schema field contains complete Pydantic JSON Schema with
    validation rules, descriptions, and examples for all stage parameters.

    Raises:
        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
        httpx.TimeoutException: If the request takes longer than Client.timeout.

    Returns:
        ErrorResponse | list[RetrieverStageDefinition]
    """

    return (
        await asyncio_detailed(
            client=client,
        )
    ).parsed
