o
    ˷e                     @   s@   d Z ddlZddlmZ ddlmZ dd Zdd Zd	d
 ZdS )zUtilites for mutual TLS.    N)
exceptions)_mtls_helperc                  C   s   t t j} | duS )zCheck if default client SSL credentials exists on the device.

    Returns:
        bool: indicating if the default client cert source exists.
    N)r   _check_dca_metadata_pathCONTEXT_AWARE_METADATA_PATH)metadata_path r   Q/var/www/ideatree/venv/lib/python3.10/site-packages/google/auth/transport/mtls.pyhas_default_client_cert_source   s   r	   c                  C   s   t  stddd } | S )a  Get a callback which returns the default client SSL credentials.

    Returns:
        Callable[[], [bytes, bytes]]: A callback which returns the default
            client certificate bytes and private key bytes, both in PEM format.

    Raises:
        google.auth.exceptions.DefaultClientCertSourceError: If the default
            client SSL credentials don't exist or are malformed.
    z(Default client cert source doesn't existc               
   S   s`   zt  \} }}W ||fS  tttfy/ } zt|}t|| W Y d }~||fS d }~ww )N)	r   get_client_cert_and_keyOSErrorRuntimeError
ValueErrorr   MutualTLSChannelErrorsix
raise_from)_
cert_bytes	key_bytes
caught_excnew_excr   r   r   callback3   s   
z,default_client_cert_source.<locals>.callbackr	   r   r   )r   r   r   r   default_client_cert_source#   s   	r   c                    s"   t  std fdd}|S )at  Get a callback which returns the default encrpyted client SSL credentials.

    Args:
        cert_path (str): The cert file path. The default client certificate will
            be written to this file when the returned callback is called.
        key_path (str): The key file path. The default encrypted client key will
            be written to this file when the returned callback is called.

    Returns:
        Callable[[], [str, str, bytes]]: A callback which generates the default
            client certificate, encrpyted private key and passphrase. It writes
            the certificate and private key into the cert_path and key_path, and
            returns the cert_path, key_path and passphrase bytes.

    Raises:
        google.auth.exceptions.DefaultClientCertSourceError: If any problem
            occurs when loading or saving the client certificate and key.
    z2Default client encrypted cert source doesn't existc               
      s   z@t jdd\} }}}t d}|| W d    n1 s w   Y  td}|| W d    n1 s:w   Y  W n  tjtfy` } zt|}t	|| W Y d }~nd }~ww  |fS )NT)generate_encrypted_keywb)
r   get_client_ssl_credentialsopenwriter   ClientCertErrorr   r   r   r   )r   r   r   passphrase_bytes	cert_filekey_filer   r   	cert_pathkey_pathr   r   r   W   s(   


z6default_client_encrypted_cert_source.<locals>.callbackr   )r#   r$   r   r   r"   r   $default_client_encrypted_cert_source?   s   r%   )	__doc__r   google.authr   google.auth.transportr   r	   r   r%   r   r   r   r   <module>   s   