a
    î½jX!  ã                   @   s–   d Z ddlT dZd$dd„Zd%dd	„Zd&d
d„Zi dfdd„Zdd„ Zdd„ Zdd„ Z	d'dd„Z
dd„ Zdd„ Zd(dd„Zd)dd„Zd d!„ Zd"d#„ ZdS )*al  
This module contains variable stuff that is not worth splitting into separate modules.
Here is:
    DISCO client and agents-to-DISCO and browse-to-DISCO emulators.
    IBR and password manager.
    jabber:iq:privacy methods
All these methods takes 'disp' first argument that should be already connected
(and in most cases already authorised) dispatcher instance.
é   )Ú*zREGISTER DATA RECEIVEDNé    c                 C   s„   t |d|d}|r| |¡ |  |¡}|rFt|ƒsF|  t |dtd¡}|rft|ƒsf|  t |dtd¡}t|ƒr€dd„ | ¡ D ƒS g S )a   Try to obtain info from the remote object.
        If remote object doesn't support disco fall back to browse (if fb2b is true)
        and if it doesnt support browse (or fb2b is not true) fall back to agents protocol
        (if gb2a is true). Returns obtained info. Used internally. Úget)ÚtoÚtypZqueryNSc                 S   s   g | ]}t |tƒr|‘qS © )Ú
isinstanceÚNode)Ú.0Únr   r   úS/Users/admin/.hermes/hermes-agent/venv/lib/python3.9/site-packages/xmpp/features.pyÚ
<listcomp>,   ó    z_discover.<locals>.<listcomp>)ÚIqZsetQuerynodeÚSendAndWaitForResponseÚisResultNodeZ	NS_BROWSEZ	NS_AGENTSÚgetQueryPayload)ÚdispÚnsÚjidÚnodeZfb2bZfb2aÚiqÚrepr   r   r   Ú	_discover"   s    
  r   c                 C   sP   g }t | t||ƒD ]8}| ¡ dkr>| d¡r>| d| d¡¡ | |j¡ q|S )zJ Query remote object about any items that it contains. Return items list. ÚagentÚname)r   ZNS_DISCO_ITEMSÚgetNameÚgetTagÚsetAttrÚ
getTagDataÚappendÚattrs)r   r   r   ÚretÚir   r   r   ÚdiscoverItems/   s
    (r$   c                 C   sê   g g  }}t | t||ƒD ]È}| ¡ dkr6| |j¡ q| ¡ dkrT| | d¡¡ q| ¡ dkr| d¡r|| d| d¡¡ | d¡r˜| d| d¡¡ | |j¡ | d¡r¸| t	¡ | d¡rÌ| t
¡ | d	¡r| t¡ q||fS )
zY Query remote object about info that it publishes. Returns identities and features lists.ÚidentityÚfeatureÚvarr   r   ÚdescriptionZ	groupchatÚregisterÚsearch)r   ZNS_DISCO_INFOr   r    r!   ÚgetAttrr   r   r   ZNS_GROUPCHATÚNS_REGISTERZ	NS_SEARCH)r   r   r   Z
identitiesÚfeaturesr#   r   r   r   ÚdiscoverInfo;   s    
r.   Tc                 C   sf   t dt|d}t| ¡ ƒD ]}| ||| ¡ q|rP|  |¡}t| j||ƒ |S |  |td|i¡ dS )a   Gets registration form from remote host.
        You can pre-fill the info dictionary.
        F.e. if you are requesting info on registering user joey than specify
        info as {'username':'joey'}. See XEP-0077 for details.
        'disp' must be connected dispatcher instance.r   ©r   r   N)	r   r,   ÚlistÚkeysÚ
setTagDatar   Ú_ReceivedRegInfoZ
DispatcherZSendAndCallForResponse)r   ÚhostÚinfoÚsyncr   r#   Úrespr   r   r   Ú
getRegInfoO   s    "
r8   c                 C   sÆ   t dt|d}t|ƒsd S |jdtdjdtd}|rR|  tt|t|df¡ d S tdd}| ¡ D ]J}t	|ƒt	|ƒkrzqd| 
¡ d	kr–| | ¡ ¡ qd| | 
¡ ¡ | ¡ ¡ qd|  tt||f¡ d S )
Nr   r/   Úquery)Ú	namespaceÚx)r   Úform)r   Zinstructions)r   r,   r   r   ZNS_DATAÚEventÚREGISTER_DATA_RECEIVEDZDataFormr   Útyper   ZaddInstructionsZgetDataZsetFieldZsetValue)Úconr7   r   r   Zdfr#   r   r   r   r3   ]   s    
r3   c                 C   sh   t dt|d}t|ƒti ƒkr&| ¡ }t| ¡ ƒD ]}| d¡ ||| ¡ q2|  |¡}t	|ƒrddS dS )aB   Perform registration on remote server with provided info.
        disp must be connected dispatcher instance.
        Returns true or false depending on registration result.
        If registration fails you can get additional info from the dispatcher's owner
        attributes lastErrNode, lastErr and lastErrCode.
    Úsetr/   r9   r   N)
r   r,   r?   ZasDictr0   r1   ZsetTagr2   r   r   )r   r4   r5   r   r#   r7   r   r   r   r)   k   s
    (
r)   c                 C   s,   |   tdt|tdƒgd¡}t|ƒr(dS dS )z™ Unregisters with host (permanently removes account).
        disp must be connected and authorized dispatcher instance.
        Returns true on success.rA   Úremove©r   Úpayloadr   N)r   r   r,   r	   r   )r   r4   r7   r   r   r   Ú
unregisterx   s    rE   c                 C   sN   |s| j j}|  tdt|td| j jgdtd|gdgd¡}t|ƒrJdS dS )z¨ Changes password on specified or current (if not specified) server.
        disp must be connected and authorized dispatcher instance.
        Returns true on success.rA   Úusername©rD   ÚpasswordrC   r   N)Ú_ownerÚServerr   r   r,   r	   r   )r   Znewpasswordr4   r7   r   r   r   ÚchangePasswordTo   s    2rK   c                 C   s€   zndg i}|   tdtƒ¡}t|ƒs(W dS | ¡ D ]8}| ¡ dkrV|d  | d¡¡ q0| d¡|| ¡ < q0|W S    Y n0 dS )zg Requests privacy lists from connected server.
        Returns dictionary of existing lists on success.Úlistsr   Nr0   r   )r   r   Ú
NS_PRIVACYr   r   r   r    r+   )r   Údictr7   r0   r   r   r   ÚgetPrivacyLists‹   s    "rO   c              	   C   sL   z:|   tdttdd|iƒgd¡}t|ƒr8| ¡ d W S W n   Y n0 dS )zs Requests specific privacy list listname. Returns list of XML nodes (rules)
        taken from the server responce.r   r0   r   rG   r   N)r   r   rM   r	   r   r   ©r   Úlistnamer7   r   r   r   ÚgetPrivacyList˜   s     rR   Úactivec                 C   s>   |rd|i}ni }|   tdtt||ƒgd¡}t|ƒr:dS dS )zv Switches privacy list 'listname' to specified type.
        By default the type is 'active'. Returns true on success.r   rA   rG   r   N©r   r   rM   r	   r   )r   rQ   r   r!   r7   r   r   r   ÚsetActivePrivacyList    s    rU   c                 C   s   t | |dƒS )zF Sets the default privacy list as 'listname'. Returns true on success.Údefault)rU   )r   rQ   r   r   r   ÚsetDefaultPrivacyList¨   s    rW   c                 C   s&   |   tdt|gd¡}t|ƒr"dS dS )z½ Set the ruleset. 'list' should be the simpleXML node formatted
        according to RFC 3921 (XMPP-IM) (I.e. Node('list',{'name':listname},payload=[...]) )
        Returns true on success.rA   rG   r   N)r   r   rM   r   )r   r0   r7   r   r   r   ÚsetPrivacyList¬   s    rX   c              	   C   s0   |   tdttdd|iƒgd¡}t|ƒr,dS dS )z: Deletes privacy list 'listname'. Returns true on success.rA   r0   r   rG   r   NrT   rP   r   r   r   ÚdelPrivacyList³   s     rY   )Nr   r   )N)N)N)NrS   )N)Ú__doc__Úprotocolr>   r   r$   r.   r8   r3   r)   rE   rK   rO   rR   rU   rW   rX   rY   r   r   r   r   Ú<module>   s    






