o
    å‡jÃ  ã                   @   sT   d dl Z d dlZd dlZd dlmZ deeB defdd„ZdededB fdd	„ZdS )
é    N)ÚconfigÚuser_idÚreturnc                 C   sv   t t| ƒƒ}t tj d¡| d¡tj¡ 	¡ d d… }t
 |¡ d¡ d¡}|› d|› }t
 | d¡¡ d¡ d¡S )Núutf-8é   Úasciiú=ú:)ÚstrÚintÚhmacÚnewr   Ú	AUTH_SALTÚencodeÚhashlibÚsha256ÚdigestÚbase64Úurlsafe_b64encodeÚdecodeÚrstrip)r   ÚuidÚsigÚsig_b64Úpayload© r   ú1/home/dore/onebot/Bot_Ero/checkin_gallery/auth.pyÚmake_login_key   s   
ý
ür   Úkeyc              	   C   sÖ   | r|   ¡ sd S zV|   ¡ dt|   ¡ ƒ d   }t | d¡¡ d¡}| dd¡\}}|dt|ƒ d   }t | d¡¡}t t	j
 d¡| d¡tj¡ ¡ d d… }t ||¡r\|W S W d S  ttfyj   Y d S w )Nr   é   r   r   r	   é   r   )ÚstripÚlenr   Úurlsafe_b64decoder   r   Úsplitr   r   r   r   r   r   r   Úcompare_digestÚ
ValueErrorÚUnicodeDecodeError)r   Úpaddedr   r   r   Ú
sig_paddedr   Úexpectedr   r   r   Úverify_login_key   s,   
ý
üÿþÿr+   )	r   r   r   Úcheckin_galleryr   r   r
   r   r+   r   r   r   r   Ú<module>   s    