from http import HTTPStatus
from typing import Any, cast
from urllib.parse import quote

import httpx

from ...client import AuthenticatedClient, Client
from ...types import Response, UNSET
from ... import errors

from ...models.http_validation_error import HTTPValidationError
from ...models.query_request import QueryRequest
from ...models.query_response import QueryResponse
from typing import cast



def _get_kwargs(
    *,
    body: QueryRequest,

) -> dict[str, Any]:
    headers: dict[str, Any] = {}


    

    

    _kwargs: dict[str, Any] = {
        "method": "post",
        "url": "/erp/v1/query",
    }

    _kwargs["json"] = body.to_dict()


    headers["Content-Type"] = "application/json"

    _kwargs["headers"] = headers
    return _kwargs



def _parse_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> HTTPValidationError | QueryResponse | None:
    if response.status_code == 200:
        response_200 = QueryResponse.from_dict(response.json())



        return response_200

    if response.status_code == 422:
        response_422 = HTTPValidationError.from_dict(response.json())



        return response_422

    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[HTTPValidationError | QueryResponse]:
    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,
    body: QueryRequest,

) -> Response[HTTPValidationError | QueryResponse]:
    r""" Execute Query

     Execute raw SQL query against Bejerman MSSQL database.

    **Security Features**:
    - Mutation queries (INSERT/UPDATE/DELETE/DROP) require `allow_mutations=true`
    - Query timeout enforcement (default 30s, max 300s)
    - Row limit enforcement (default 1000, max 10000)

    **Examples**:

    Safe SELECT query:
    ```json
    {
      \"company_id\": \"uuid\",
      \"sql\": \"SELECT * FROM Clientes WHERE cli_CUIT LIKE '%20-123%'\"
    }
    ```

    Mutation query (requires flag):
    ```json
    {
      \"company_id\": \"uuid\",
      \"sql\": \"UPDATE Clientes SET cli_EMail = 'new@email.com' WHERE cli_Cod = 'CLI001'\",
      \"allow_mutations\": true
    }
    ```

    Args:
        body (QueryRequest): Request schema for raw SQL queries.

    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[HTTPValidationError | QueryResponse]
     """


    kwargs = _get_kwargs(
        body=body,

    )

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

    return _build_response(client=client, response=response)

def sync(
    *,
    client: AuthenticatedClient | Client,
    body: QueryRequest,

) -> HTTPValidationError | QueryResponse | None:
    r""" Execute Query

     Execute raw SQL query against Bejerman MSSQL database.

    **Security Features**:
    - Mutation queries (INSERT/UPDATE/DELETE/DROP) require `allow_mutations=true`
    - Query timeout enforcement (default 30s, max 300s)
    - Row limit enforcement (default 1000, max 10000)

    **Examples**:

    Safe SELECT query:
    ```json
    {
      \"company_id\": \"uuid\",
      \"sql\": \"SELECT * FROM Clientes WHERE cli_CUIT LIKE '%20-123%'\"
    }
    ```

    Mutation query (requires flag):
    ```json
    {
      \"company_id\": \"uuid\",
      \"sql\": \"UPDATE Clientes SET cli_EMail = 'new@email.com' WHERE cli_Cod = 'CLI001'\",
      \"allow_mutations\": true
    }
    ```

    Args:
        body (QueryRequest): Request schema for raw SQL queries.

    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:
        HTTPValidationError | QueryResponse
     """


    return sync_detailed(
        client=client,
body=body,

    ).parsed

async def asyncio_detailed(
    *,
    client: AuthenticatedClient | Client,
    body: QueryRequest,

) -> Response[HTTPValidationError | QueryResponse]:
    r""" Execute Query

     Execute raw SQL query against Bejerman MSSQL database.

    **Security Features**:
    - Mutation queries (INSERT/UPDATE/DELETE/DROP) require `allow_mutations=true`
    - Query timeout enforcement (default 30s, max 300s)
    - Row limit enforcement (default 1000, max 10000)

    **Examples**:

    Safe SELECT query:
    ```json
    {
      \"company_id\": \"uuid\",
      \"sql\": \"SELECT * FROM Clientes WHERE cli_CUIT LIKE '%20-123%'\"
    }
    ```

    Mutation query (requires flag):
    ```json
    {
      \"company_id\": \"uuid\",
      \"sql\": \"UPDATE Clientes SET cli_EMail = 'new@email.com' WHERE cli_Cod = 'CLI001'\",
      \"allow_mutations\": true
    }
    ```

    Args:
        body (QueryRequest): Request schema for raw SQL queries.

    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[HTTPValidationError | QueryResponse]
     """


    kwargs = _get_kwargs(
        body=body,

    )

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

    return _build_response(client=client, response=response)

async def asyncio(
    *,
    client: AuthenticatedClient | Client,
    body: QueryRequest,

) -> HTTPValidationError | QueryResponse | None:
    r""" Execute Query

     Execute raw SQL query against Bejerman MSSQL database.

    **Security Features**:
    - Mutation queries (INSERT/UPDATE/DELETE/DROP) require `allow_mutations=true`
    - Query timeout enforcement (default 30s, max 300s)
    - Row limit enforcement (default 1000, max 10000)

    **Examples**:

    Safe SELECT query:
    ```json
    {
      \"company_id\": \"uuid\",
      \"sql\": \"SELECT * FROM Clientes WHERE cli_CUIT LIKE '%20-123%'\"
    }
    ```

    Mutation query (requires flag):
    ```json
    {
      \"company_id\": \"uuid\",
      \"sql\": \"UPDATE Clientes SET cli_EMail = 'new@email.com' WHERE cli_Cod = 'CLI001'\",
      \"allow_mutations\": true
    }
    ```

    Args:
        body (QueryRequest): Request schema for raw SQL queries.

    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:
        HTTPValidationError | QueryResponse
     """


    return (await asyncio_detailed(
        client=client,
body=body,

    )).parsed
