o
    6h                     @   s  d dl Z d dlZd dlZzd dlmZ d&ddZW n ey-   d dlm	Z d&ddZY nw d&ddZ
d	d
 Zdd Zdd Zdd Zdd Zdd Zdd Zh dZedddd eD  d Zdedefdd Zd!Zed"Zg d#Zdedefd$d%ZdS )'    N)int_to_bytesc                 C   s   t | |pd S N)_long_to_bytes)n	blocksize r   ^/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/jose/utils.pylong_to_bytes
      r	   )int_to_stringc                 C   s<   t | }|dkr
|S t||ksJ |t| }d| | S )Nr       )r   len)r   r   retpaddingr   r   r   r	      s   c                 C   s   t t| |dS )N   =)base64urlsafe_b64encoder	   strip)datasizer   r   r   long_to_base64   s   r   c                 C   s   t ddd | D dS )N c                 S   s   g | ]}d | qS )z%02xr   ).0byter   r   r   
<listcomp>   s    z#int_arr_to_long.<locals>.<listcomp>   )intjoin)arrr   r   r   int_arr_to_long   s   r   c                 C   s>   t | tr
| d} tt| d }ttdt	| |S )Nasciis   ==z%sB)

isinstancestrencoder   urlsafe_b64decodebytesr   structunpackr   )r   _dr   r   r   base64_to_long"   s   

r)   c                 C   s@   ||  d }tt|d }|d| }t|}|dS )a  Helper method for calculating an access token
    hash, as described in http://openid.net/specs/openid-connect-core-1_0.html#CodeIDToken

    Its value is the base64url encoding of the left-most half of the hash of the octets
    of the ASCII representation of the access_token value, where the hash algorithm
    used is the hash algorithm used in the alg Header Parameter of the ID Token's JOSE
    Header. For instance, if the alg is RS256, hash the access_token value with SHA-256,
    then take the left-most 128 bits and base64url encode them. The at_hash value is a
    case sensitive string.

    Args:
        access_token (str): An access token string.
        hash_alg (callable): A callable returning a hash object, e.g. hashlib.sha256

    utf-8   N)r#   digestr   r   base64url_encodedecode)access_tokenhash_alghash_digestcut_at	truncatedat_hashr   r   r   calculate_at_hash+   s
   
r5   c                 C   s.   t | d }|dkr| dd|  7 } t| S )zwHelper method to base64url_decode a string.

    Args:
        input (str): A base64url_encoded string to decode.

       r   r   )r   r   r$   )inputremr   r   r   base64url_decodeB   s   
r9   c                 C   s   t | ddS )zwHelper method to base64url_encode a string.

    Args:
        input (str): A base64url_encoded string to encode.

    r       )r   r   replace)r7   r   r   r   r-   Q   s   r-   c                 C   s   | j d d d | j S )zHelper method to determine the total number of seconds
    from a timedelta.

    Args:
        delta (timedelta): A timedelta to convert to seconds.
       <   )daysseconds)deltar   r   r   timedelta_total_seconds[   s   rA   c                 C   s8   t | tr| S t | tr| ddS tdt|  d)zCoerce **s** to bytes.r*   strictznot expecting type '')r!   r%   r"   r#   	TypeErrortype)sr   r   r   ensure_binarye   s
   

rG   >      X509 CRL
   PUBLIC KEY   CERTIFICATE   PRIVATE KEY   DH PARAMETERS   EC PRIVATE KEY   RSA PUBLIC KEY   DSA PRIVATE KEY   RSA PRIVATE KEY   SSH2 PUBLIC KEY   CERTIFICATE REQUEST   OPENSSH PRIVATE KEY   TRUSTED CERTIFICATE   ENCRYPTED PRIVATE KEY   NEW CERTIFICATE REQUEST   SSH2 ENCRYPTED PRIVATE KEYs   ----[- ]BEGIN (   |c                 c   s    | ]}t |V  qd S r   )reescape)r   pemr   r   r   	<genexpr>   s    r\   s	   )[- ]----keyreturnc                 C   s   t t| S r   )bool_PEM_REsearchr]   r   r   r   is_pem_format   r
   rc   s   -cert-v01@openssh.coms   \A(\S+)[ \t]+(\S+))s   ssh-ed25519s   ssh-rsas   ssh-dsss   ecdsa-sha2-nistp256s   ecdsa-sha2-nistp384s   ecdsa-sha2-nistp521c                    sP   t  fddtD rdS t }|r&|d}t|tt d  kr&dS dS )Nc                 3   s    | ]}| v V  qd S r   r   )r   string_valuerb   r   r   r\      s    zis_ssh_key.<locals>.<genexpr>T   F)any_SSH_KEY_FORMATS_SSH_PUBKEY_RCmatchgroup_CERT_SUFFIXr   )r]   ssh_pubkey_matchkey_typer   rb   r   
is_ssh_key   s   

rn   )r   )r   rY   r&   cryptography.utilsr   r   r	   ImportErrorecdsa.ecdsar   r   r   r)   r5   r9   r-   rA   rG   _PEMScompiler   r`   r%   r_   rc   rk   rh   rg   rn   r   r   r   r   <module>   s6    
	



