from typing import Any, Dict, Optional import httpx from ...client import Client from ...models.error import Error from ...models.uuid import Uuid from ...types import Response def _get_kwargs( provider_id: Uuid, *, client: Client, callback_url: Optional[str] = None, ) -> Dict[str, Any]: url = "{}/auth/saml/provider/{provider_id}/login".format( client.base_url, provider_id=provider_id, ) # noqa: E501 if callback_url is not None: if "?" in url: url = url + "&callback_url=" + str(callback_url) else: url = url + "?callback_url=" + str(callback_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(), } def _parse_response(*, response: httpx.Response) -> Optional[Error]: return None if response.status_code == 400: response_4XX = Error(**response.json()) return response_4XX if response.status_code == 500: response_5XX = Error(**response.json()) return response_5XX return Error(**response.json()) def _build_response(*, response: httpx.Response) -> Response[Optional[Error]]: return Response( status_code=response.status_code, content=response.content, headers=response.headers, parsed=_parse_response(response=response), ) def sync_detailed( provider_id: Uuid, *, client: Client, callback_url: Optional[str] = None, ) -> Response[Optional[Error]]: kwargs = _get_kwargs( provider_id=provider_id, callback_url=callback_url, client=client, ) response = httpx.get( verify=client.verify_ssl, **kwargs, ) return _build_response(response=response) def sync( provider_id: Uuid, *, client: Client, callback_url: Optional[str] = None, ) -> Optional[Error]: """The UI uses this to avoid having to ask the API anything about the IdP. It already knows the SAML IdP ID from the path, so it can just link to this path and rely on the API to redirect to the actual IdP.""" # noqa: E501 return sync_detailed( provider_id=provider_id, callback_url=callback_url, client=client, ).parsed async def asyncio_detailed( provider_id: Uuid, *, client: Client, callback_url: Optional[str] = None, ) -> Response[Optional[Error]]: kwargs = _get_kwargs( provider_id=provider_id, callback_url=callback_url, client=client, ) async with httpx.AsyncClient(verify=client.verify_ssl) as _client: response = await _client.get(**kwargs) return _build_response(response=response) async def asyncio( provider_id: Uuid, *, client: Client, callback_url: Optional[str] = None, ) -> Optional[Error]: """The UI uses this to avoid having to ask the API anything about the IdP. It already knows the SAML IdP ID from the path, so it can just link to this path and rely on the API to redirect to the actual IdP.""" # noqa: E501 return ( await asyncio_detailed( provider_id=provider_id, callback_url=callback_url, client=client, ) ).parsed