from typing import Any, Dict, Optional, Union, cast

import httpx

from ...client import Client
from ...models.customer import Customer
from ...models.error import Error
from ...models.billing_info import BillingInfo
from ...types import Response

def _get_kwargs(
	body: BillingInfo,
	*,
	client: Client,
) -> Dict[str, Any]:
	url = "{}/user/payment".format(client.base_url)

	headers: Dict[str, Any] = client.get_headers()
	cookies: Dict[str, Any] = client.get_cookies()

	return {
		"url": url,
		"headers": headers,
		"cookies": cookies,
		"timeout": client.get_timeout(),
		"content": body,
	}


def _parse_response(*, response: httpx.Response) -> Optional[Union[Any, Customer, Error]]:
	if response.status_code == 200:
		response_200 = Customer.from_dict(response.json())
		return response_200
	if response.status_code == 400:
		response_4XX = Error.from_dict(response.json())
		return response_4XX
	if response.status_code == 500:
		response_5XX = Error.from_dict(response.json())
		return response_5XX
	return None


def _build_response(*, response: httpx.Response) -> Response[Union[Any, Customer, Error]]:
	return Response(
		status_code=response.status_code,
		content=response.content,
		headers=response.headers,
		parsed=_parse_response(response=response),
	)


def sync_detailed(
	body: BillingInfo,
	*,
	client: Client,
) -> Response[Union[Any, Customer, Error]]:
	kwargs = _get_kwargs(
		body=body,
		client=client,
	)

	response = httpx.put(
		verify=client.verify_ssl,
		**kwargs,
	)

	return _build_response(response=response)


def sync(
	body: BillingInfo,
	*,
	client: Client,
) -> Optional[Union[Any, Customer, Error]]:
	""" This includes billing address, phone, and name.
This endpoint requires authentication by any KittyCAD user. It updates the payment information for the authenticated user. """

	return sync_detailed(
		body=body,
		client=client,
	).parsed


async def asyncio_detailed(
	body: BillingInfo,
	*,
	client: Client,
) -> Response[Union[Any, Customer, Error]]:
	kwargs = _get_kwargs(
		body=body,
		client=client,
	)

	async with httpx.AsyncClient(verify=client.verify_ssl) as _client:
		response = await _client.put(**kwargs)

	return _build_response(response=response)


async def asyncio(
	body: BillingInfo,
	*,
	client: Client,
) -> Optional[Union[Any, Customer, Error]]:
	""" This includes billing address, phone, and name.
This endpoint requires authentication by any KittyCAD user. It updates the payment information for the authenticated user. """

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