U
    
ڲg$                      @   s  d Z ddlZddlZddlZddlZddlmZ ejejej	ej
ejejejejejejejejejejejejejejejejejejejejejej ej!ej"ej#ej$ej%ej&iZ'ejejejejejej	ejej!ej#d	Z(ejejejej%ejejejejej	ejejej!ej#ejejdZ)dd Z*dd	 Z+dd
dZ,dd Z-dd Z.dddZ/dd Z0dd Z1dd Z2dd Z3G dd dej4j5j6Z7dS )z)Internal utilities common to all modules.    N)
exceptions)	i  i  i  i  i  i  i  i  i  )                        	   
                  c                 C   sR   | dkrt  S t| t jr<t | j}| |k	r8td| S tdt| dS )z3Returns a reference to an initialized App instance.NzKIllegal app argument. App instance not initialized via the firebase module.zTIllegal app argument. Argument must be of type  firebase_admin.App, but given "{0}".)firebase_adminZget_app
isinstanceZAppname
ValueErrorformattype)appZinitialized_app r   9/tmp/pip-unpacked-wheel-p0r7i5ii/firebase_admin/_utils.py_get_initialized_appO   s    r   c                 C   s   t | } | ||S N)r   Z_get_service)r   r   Zinitializerr   r   r   get_app_service`   s    r   c                 C   s^   | j dkrt| S | j }|j }|j}t||\}}d}|rJ|| ||}|rR|S t| ||S )ag  Constructs a ``FirebaseError`` from the given requests error.

    This can be used to handle errors returned by Google Cloud Platform (GCP) APIs.

    Args:
        error: An error raised by the requests module while making an HTTP call to a GCP API.
        handle_func: A function that can be used to handle platform errors in a custom way. When
            specified, this function will be called with three arguments. It has the same
            signature as ```_handle_func_requests``, but may return ``None``.

    Returns:
        FirebaseError: A ``FirebaseError`` that can be raised to the user code.
    N)responsehandle_requests_errorcontentdecodestatus_code_parse_platform_error_handle_func_requests)errorZhandle_funcr   r    r"   
error_dictmessageexcr   r   r   #handle_platform_error_from_requestse   s    

r)   c                 C   sL   t | tstjd| | dS | d}| d}t|}t|}||dS )zConstructs a ``FirebaseError`` from the given operation error.

    Args:
        error: An error returned by a long running operation.

    Returns:
        FirebaseError: A ``FirebaseError`` that can be raised to the user code.
    5Unknown error while making a remote service call: {0}r'   causecoder'   )r'   )r   dictr   UnknownErrorr   get_rpc_code_to_error_code_error_code_to_exception_type)r%   rpc_coder'   Z
error_codeerr_typer   r   r   handle_operation_error   s    	


r5   c                 C   s   | d}t| ||S )a|  Constructs a ``FirebaseError`` from the given GCP error.

    Args:
        error: An error raised by the requests module while making an HTTP call.
        message: A message to be included in the resulting ``FirebaseError``.
        error_dict: Parsed GCP error response.

    Returns:
        FirebaseError: A ``FirebaseError`` that can be raised to the user code or None.
    status)r0   r   )r%   r'   r&   r-   r   r   r   r$      s    
r$   c                 C   s   t | tjjr"tjd| | dS t | tjjrDtjd| | dS | jdkrbtj	d| | dS |srt
| jj}|s~t| }t|}||| | jdS )ao  Constructs a ``FirebaseError`` from the given requests error.

    This method is agnostic of the remote service that produced the error, whether it is a GCP
    service or otherwise. Therefore, this method does not attempt to parse the error response in
    any way.

    Args:
        error: An error raised by the requests module while making an HTTP call.
        message: A message to be included in the resulting ``FirebaseError`` (optional). If not
            specified the string representation of the ``error`` argument is used as the message.
        code: A GCP error code that will be used to determine the resulting error type (optional).
            If not specified the HTTP status code on the error response is used to determine a
            suitable error code.

    Returns:
        FirebaseError: A ``FirebaseError`` that can be raised to the user code.
    z'Timed out while making an API call: {0}r+   z%Failed to establish a connection: {0}Nr*   )r'   r,   http_response)r   requestsr   TimeoutDeadlineExceededErrorr   ConnectionErrorUnavailableErrorr   r/   _http_status_to_error_coder"   strr2   )r%   r'   r-   r4   r   r   r   r      s*    
r   c                 C   s   t | tjS )z-Maps an HTTP status to a platform error code.)_HTTP_STATUS_TO_ERROR_CODEr0   r   UNKNOWN)r6   r   r   r   r=      s    r=   c                 C   s   t | tjS )z*Maps an RPC code to a platform error code.)_RPC_CODE_TO_ERROR_CODEr0   r   r@   )r3   r   r   r   r1      s    r1   c                 C   s   t | tjS )z0Maps a platform error code to an exception type.)_ERROR_CODE_TO_EXCEPTION_TYPEr0   r   r/   )r-   r   r   r   r2      s    r2   c                 C   sd   i }zt | }t|tr|}W n tk
r4   Y nX |di }|d}|s\d|| }||fS )a(  Parses an HTTP error response from a Google Cloud Platform API and extracts the error code
    and message fields.

    Args:
        content: Decoded content of the response body.
        status_code: HTTP status code.

    Returns:
        tuple: A tuple containing error code and message.
    r%   r'   z4Unexpected HTTP response with status: {0}; body: {1})jsonloadsr   r.   r   r0   r   )r    r"   dataZparsed_bodyr&   msgr   r   r   r#      s    


r#   c                   @   s    e Zd ZdZdd Zdd ZdS )EmulatorAdminCredentialsz Credentials for use with the firebase local emulator.

    This is used instead of user-supplied credentials or ADC.  It will silently do nothing when
    asked to refresh credentials.
    c                 C   s   t jjj|  d| _d S )Nowner)googleauthcredentialsCredentials__init__token)selfr   r   r   rM      s    z!EmulatorAdminCredentials.__init__c                 C   s   d S r   r   )rO   requestr   r   r   refresh  s    z EmulatorAdminCredentials.refreshN)__name__
__module____qualname____doc__rM   rQ   r   r   r   r   rG      s   rG   )N)NN)8rU   rC   Zgoogle.authrI   r8   r   r   ZINVALID_ARGUMENTZInvalidArgumentErrorZFAILED_PRECONDITIONZFailedPreconditionErrorZOUT_OF_RANGEZOutOfRangeErrorZUNAUTHENTICATEDZUnauthenticatedErrorZPERMISSION_DENIEDZPermissionDeniedError	NOT_FOUNDZNotFoundErrorZABORTEDZAbortedErrorZALREADY_EXISTSZAlreadyExistsErrorCONFLICTZConflictErrorZRESOURCE_EXHAUSTEDZResourceExhaustedErrorZ	CANCELLEDZCancelledErrorZ	DATA_LOSSZDataLossErrorr@   r/   ZINTERNALZInternalErrorUNAVAILABLEr<   ZDEADLINE_EXCEEDEDr:   rB   r?   rA   r   r   r)   r5   r$   r   r=   r1   r2   r#   rJ   rK   rL   rG   r   r   r   r   <module>   s                   

(