File: //lib/python3/dist-packages/twisted/words/__pycache__/service.cpython-310.pyc
o
�b�� � @ s d Z ddlmZmZ ddlmZ ddlmZ ddlmZm Z
m Z ddl m
Z
mZ ddlmZmZmZ ddlmZ dd lmZ dd
lmZmZ dd lmZ eej �G d d
� d
��Z eej �G dd� d��Z dZ eej!�G dd� dej"��Z#G dd� dej$�Z%G dd� dej&�Z'eej!�G dd� dej(��Z)e�*e'e)� G dd� dej&�Z+eej �G dd � d ej(��Z,e�*e+e,� G d d � d ej&�Z-eej!�G d d � d ej&��Z.ee.ej ej/� G d!d"� d"ej(�Z0e�*e.e0� ee j1ej2�G d#d$� d$��Z3G d%d&� d&e3�Z4g d'�Z5d(S ))a
A module that needs a better name.
Implements new cred things for words.
How does this thing work?
- Network connection on some port expecting to speak some protocol
- Protocol-specific authentication, resulting in some kind of credentials object
- twisted.cred.portal login using those credentials for the interface
IUser and with something implementing IChatClient as the mind
- successful login results in an IUser avatar the protocol can call
methods on, and state added to the realm such that the mind will have
methods called on it as is necessary
- protocol specific actions lead to calls onto the avatar; remote events
lead to calls onto the mind
- protocol specific hangup, realm is notified, user is removed from active
play, the end.
� )�ctime�time)� implementer)� copyright)� credentials�error�portal)�defer�protocol)�failure�log�reflect)�registerAdapter)�pb)�ewords�iwords)�ircc @ sV e Zd Zdd� Zdd� Zdd� Zdd� Zdd
d �Zd d
� Zdd� Z dd� Z
dd� Z d S )�Groupc C s || _ i | _ddd�| _d S )N� )�topic� topic_author)�name�users�meta��selfr � r �7/usr/lib/python3/dist-packages/twisted/words/service.py�__init__. s
�zGroup.__init__c C s t �t||��S �N)r �Failure� Exception)r �err�pr r r � _ebUserCall6 � zGroup._ebUserCallc C s0 |D ]\}}|s|j \}}| �||�� � qd S r )�value�remove�getErrorMessage)r �results�success�result�userr" r r r � _cbUserCall9 s
��zGroup._cbUserCallc C s� t j�|�s
J d�|���|j| jvrEg }|| j|j< | j�� D ]}||ur;t�|j | |�}|j
| j |d� |� |� q t�
|��| j� t�d �S )Nz{!r} is not a chat client�r# )r � IChatClient�
providedBy�formatr r �valuesr �
maybeDeferred�
userJoined�
addErrbackr$ �append� DeferredList� addCallbackr- �succeed)r r, � additionsr# �dr r r �add? s �
�
z Group.addNc C s� z| j |j= W n ty Y n.w g }| j �� D ] }||ur4t�|j| ||�}|j| j|d� |� |� qt�
|�� | j � t�
d �S �Nr. )r r �KeyErrorr2 r r3 �userLeftr5 r$ r6 r7 r8 r- r9 )r r, �reason�removalsr# r; r r r r' N s �
�
z Group.removec C s t �t| j��S r )r r9 �lenr �r r r r �size] r% z
Group.sizec C sp || u sJ �g }| j �� D ] }||ur)t�|j|| |�}|j| j|d� |�|� q
t�|�� | j
� t� d �S r= ) r r2 r r3 �receiver5 r$ r6 r7 r8 r- r9 )r �sender� recipient�message�receivesr# r; r r r rE ` s
�
z
Group.receivec C s` || _ g }| j�� D ]}t�|j| |�}|j| j|d� |�|� q
t� |��
| j � t� d �S r= )
r r r2 r r3 �groupMetaUpdater5 r$ r6 r7 r8 r- r9 )r r �setsr# r; r r r � setMetadatak s
zGroup.setMetadatac C s t | j�� �S r )�iterr r2 rC r r r � iterusersu s zGroup.iterusersr ) �__name__�
__module__� __qualname__r r$ r- r< r' rD rE rL rN r r r r r , s
r c @ sN e Zd ZdZdZdd� Zdd� Zdd� Zddd �Zd
d � Z d d
� Z
dd� Z dS )�UserNc C s || _ g | _t� | _d S r )r �groupsr � lastMessager r r r r � z
User.__init__c C s || _ || _t� | _d S r )�realm�mindr �signOn)r rV rW r r r �loggedIn� rU z
User.loggedInc s � �fdd�}� � �j��|�S )Nc � �j �� � | S r )rS r6 �r+ ��groupr r r �cbJoin� � zUser.join.<locals>.cbJoin)r< rW r8 )r r] r^ r r\ r �join� s z User.joinc s" � �fdd�}� � �j|��|�S )Nc rZ r )rS r' r[ r\ r r �cbLeave� r_ zUser.leave.<locals>.cbLeave)r' rW r8 )r r] r@ ra r r\ r �leave� s z
User.leavec C s t � | _|�| j||�S r )r rT rE rW )r rG rH r r r �send� s z User.sendc C s
t | j�S r )rM rS rC r r r �
itergroups� �
zUser.itergroupsc C s" | j d d � D ]}| �|� qd S r )rS rb )r �gr r r �logout� s �z User.logoutr ) rO rP rQ rV rW r rY r` rb rc rd rg r r r r rR z s
rR zNickServ!NickServ@servicesc @ s� e Zd ZdZdZdZdZdZdZdd� Z dd� Z
dd � Z d
d � Z dPd d
�Z
dd� Zdd� ZdZdZdd� Zdd� Zdd� Zdd� Zdd� Zejd fejd fejd fejd fgZ ej d fej d!fgZ d"d#� Z d$d%� Z!d&d'� Z"d(d)� Z#dPd*d+�Z$dPd,d-�Z%d.d/� Z&d0d1� Z'd2d3� Z(d4d5� Z)d6d7� Z*d8d9� Z+d:d;� Z,d<d=� Z-d>d?� Z.d@dA� Z/dBdC� Z0dDdE� Z1dFdG� Z2dHdI� Z3dJdK� Z4dLdM� Z5dNdO� Z6dS )Q�IRCUserzM
Protocol instance representing an IRC user connected to the server.
N�utf-8c C s | j | _| jj| _| jj| _d S r )�irc_NICKSERV_PRIVMSG� irc_PRIVMSG�factoryrV r �hostnamerC r r r �connectionMade� s
zIRCUser.connectionMadec C s | j d ur| � � d | _d S d S r )rg �avatar�r r@ r r r �connectionLost� s
�zIRCUser.connectionLostc O s� d|vr | j |d< d|vr| j�| j�|d< | ||d gt|� }g }|D ]}t|t�r1|�d�}|�|� q%t j
j |i |�� d S )N�prefix�tori ) rm r �encode�encoding�list�
isinstance�bytes�decoder6 r �IRC� sendMessage)r �command�parameter_list�kw�arglist�arglistUnicode�argr r r r{ � s
zIRCUser.sendMessagec C s, | � |j� d|j� d| j� �d|j � d S )N�!�@�#)r` r rm �r r] r, r r r r4 � s ,zIRCUser.userJoinedc C s2 | � |j� d|j� d| j� �d|j |pd� d S )Nr� r� r� �leaving)�partr rm �r r] r, r@ r r r r? � s
�zIRCUser.userLeftc C s` t j�|�r d|j }n|j}|�dd�}|�� D ]}| �|j� d|j� d| j� �||� qd S )Nr� �textz <an unrepresentable message>r� r� )r �IGroupr0 r �get�
splitlines�privmsgrm )r rF rG rH �
recipientNamer� �Lr r r rE � s ��zIRCUser.receivec
C sN d|v r%|d }|� dd�}| �| jd|j ||� d|� d| j� �� d S d S )Nr r r r� r� r� )r� r r rm )r r] r r �authorr r r rJ � s ��zIRCUser.groupMetaUpdatec C � |d | _ dS )z�
Password message -- Register a password.
Parameters: <password>
[REQUIRED]
Note that IRC requires the client send this *before* NICK
and USER.
���N)�password�r rr �paramsr r r �irc_PASS s zIRCUser.irc_PASSc C s� |d }z
t |t�r|�| j�}W n ty) | �tt|�d� | j� � Y dS w || _
|| _ | j D ]\}}| �
||| jj � q3| jdu rP| �t|d� dS | j}d| _| �||� dS )zh
Nick message -- Set your nickname.
Parameters: <nickname>
[REQUIRED]
r z;Your nickname cannot be decoded. Please use ASCII or UTF-8.Nz Password?)rw rx ry ru �UnicodeDecodeErrorr� �NICKSERV�repr� transport�loseConnection�nicknamer �
_motdMessagesr{ rl � _serverInfor� �logInAs)r rr r� r� �coder� r� r r r �irc_NICK
s.
� �
�
zIRCUser.irc_NICKc C r� )zk
User message -- Set your realname.
Parameters: <user> <mode> <unused> <realname>
r� N)�realnamer� r r r �irc_USER/ s zIRCUser.irc_USERc C sd |d }|d }| j du r| j�� dS |�� dkr$| �t| j d� dS | j }d| _ | �||� dS )�^
Send a (private) message.
Parameters: <msgtarget> <text to be sent>
r r� N�nickservz1Denied. Please send me (NickServ) your password.)r� r� r� �lowerr� r� r� )r rr r� �targetr� r� r r r rj : s
�z IRCUser.irc_NICKSERV_PRIVMSGc C s6 | j j�t�||�| tj�}|j| j| j |fd� d S )N)� errbackArgs)
rl r �loginr �UsernamePasswordr �IUser� addCallbacks�_cbLogin�_ebLogin)r r� r� r; r r r r� Q s �zIRCUser.logInAsz:connected to Twisted IRCzA:Your host is %(serviceName)s, running version %(serviceVersion)sz,:This server was created on %(creationDate)sz&%(serviceName)s %(serviceVersion)s w nz(:- %(serviceName)s Message of the Day - z:End of /MOTD command.c C sZ |\}}}|t ju sJ d|����| `|| _|| _| jD ]\}}| �||| jj � q d S )NzRealm is buggy, got ) r r� rk ro rg �_welcomeMessagesr{ rl r� )r r+ �ifacero rg r� r� r r r r� n s
�zIRCUser._cbLoginc C sh |� tj�r| �t|d� n |� tj�r | �t|d� nt�d� t� |� | �t|d� | j
� � d S )Nz*Already logged in. No pod people allowed!zLogin failed. Goodbye.z Unhandled error during login:z"Server error during login. Sorry.) �checkr �AlreadyLoggedInr� r� �ecred�UnauthorizedLoginr �msgr" r� r� )r r" r� r r r r� z s �
zIRCUser._ebLoginc C s | j dur| �d| j� dS dS )zK
Ping message
Parameters: <server1> [ <server2> ]
N�PONG)rV r{ rm r� r r r �irc_PING� s
�zIRCUser.irc_PINGc C s | j �� dS )z>
Quit
Parameters: [ <Quit Message> ]
N)r� r� r� r r r �irc_QUIT� s zIRCUser.irc_QUITc G s0 |r | � tjd� d S | �| jd|j d� d S )N�:Unknown MODE flag.r� �+)r{ r �ERR_UNKNOWNMODE� channelModer )r r] �modes�argsr r r � _channelMode� s zIRCUser._channelModec C sD |r | � tjd� d S || ju r| � tjd� d S | � tjd� d S )Nr� r� z(:You can't look at someone else's modes.)r{ r r� ro � RPL_UMODEIS�ERR_USERSDONTMATCH)r r, r� r r r � _userMode� s
�zIRCUser._userModec s� z� d }t |t�r|��j�}W n ty% ��tj� d d� Y dS w |�d�rM� �fdd�}�j �
|dd� �}|j �j |t
� dd� �d� dS �fd d
�}�j �|�}|j �j|t
� dd� �d� dS ) z�
User mode message
Parameters: <nickname>
*( ( "+" / "-" ) *( "i" / "w" / "o" / "O" / "r" ) )
r z2:No such nickname (could not decode your unicode!)Nr� c �$ | � tj� ��tj� d d� d S )Nr � :That channel doesn't exist.��trapr � NoSuchGroupr{ r �ERR_NOSUCHCHANNEL�r" �r� r r r �ebGroup� s �z!IRCUser.irc_MODE.<locals>.ebGroup� )� callbackArgsc s � � tjd� d S )Nz:No such nickname.�r{ r �ERR_NOSUCHNICKr� rC r r �ebUser� � z IRCUser.irc_MODE.<locals>.ebUser)rw rx ry ru r� r{ r r� �
startswithrV � lookupGroupr� r� �tuple�
lookupUserr� )r rr r� �
channelOrUserr� r; r� r r� r �irc_MODE� s(
� ��
" "zIRCUser.irc_MODEc C s dS )zl
Userhost message
Parameters: <nickname> *( SPACE <nickname> )
[Optional]
Nr r� r r r � irc_USERHOST� s zIRCUser.irc_USERHOSTc s� z|d �t �t�r���j��W n ty% ��tj|d d� Y dS w |d � ��d�r:�j �
�dd� �}n �j � ��� dd� �}� �fd d
�}��fd d �}|�
||� dS )
r� r z6:No such nick/channel (could not decode your unicode!)Nr� r� r� c S s | j S r )rW �r, r r r �<lambda>� s z%IRCUser.irc_PRIVMSG.<locals>.<lambda>c s | d ur
�j �| d� i�S d S )Nr� )ro rc )�targ)� messageTextr r r �cbTarget� s �z%IRCUser.irc_PRIVMSG.<locals>.cbTargetc s � � tj�d� d S )Nz:No such nick/channel.r� r� )r �
targetNamer r �ebTarget� � z%IRCUser.irc_PRIVMSG.<locals>.ebTarget)rw rx ry ru r� r{ r r� r� rV r� r� r8 r� )r rr r� r� r� r� r )r� r r� r rk � s,
� ��
�zIRCUser.irc_PRIVMSGc � z|d � t � t�r� ��j�� W n ty% ��tj|d d� Y dS w � �d�r1� dd� � �fdd�}� �fdd �}�j �
� �� ||� dS )
zm
Join message
Parameters: ( <channel> *( "," <channel> ) [ <key> *( "," <key> ) ] )
r �1:No such channel (could not decode your unicode!)Nr� r� c s � �fdd�}�j �� ��|�S )Nc s> �� � �� ���jd� j dd� � �� D �� ��� � d S )Nr� c S � g | ]}|j �qS r �r ��.0r, r r r �
<listcomp> � zEIRCUser.irc_JOIN.<locals>.cbGroup.<locals>.cbJoin.<locals>.<listcomp>)r4 �namesr rN �
_sendTopic)�ignr\ r r r^ s �z1IRCUser.irc_JOIN.<locals>.cbGroup.<locals>.cbJoin�ro r` r8 )r] r^ rC �r] r �cbGroup s z!IRCUser.irc_JOIN.<locals>.cbGroupc s �� tjd� d� d S )Nr� z:No such channel.)r{ r r� r� �� groupNamer r r r� s �z!IRCUser.irc_JOIN.<locals>.ebGroup) rw rx ry ru r� r{ r r� r� rV �getGroupr� �r rr r� r� r� r r� r �irc_JOIN� s$
� ��
zIRCUser.irc_JOINc s� z|d � t |d t�r|d ��j�� W n ty) ��tj|d d� Y dS w � �d�r5� dd� � t |�dkrJ|d �t �t�rI��d��nd���fdd�}� �fd d
�}�j
� � �� ||� dS ) zc
Part message
Parameters: <channel> *( "," <channel> ) [ <Part Message> ]
r z :Could not decode your unicode!Nr� r� ri c s$ � ��fdd�}�j �� ���|�S )Nc s �� � ��� d S r )r? r[ )r] r@ r r r ra @ r� z2IRCUser.irc_PART.<locals>.cbGroup.<locals>.cbLeave)ro rb r8 )r] ra )r@ r r� r r� ? s z!IRCUser.irc_PART.<locals>.cbGroupc s, | � tj� ��tjd� d| �� � d S )Nr� �:)r� r r� r{ r �ERR_NOTONCHANNELr( r� r� r r r� E s �z!IRCUser.irc_PART.<locals>.ebGroup)
rw rx ry ru r� r{ r r� r� rB rV r� r� r� r )r� r@ r r �irc_PART% s, � ��
�zIRCUser.irc_PARTc r� )
zb
Names message
Parameters: [ <channel> *( "," <channel> ) [ <target> ] ]
r� r� Nr� r� c s( � � � jd| j dd� | �� D �� d S )Nr� c S r� r r� r� r r r r� g r� z6IRCUser.irc_NAMES.<locals>.cbGroup.<locals>.<listcomp>)r� r rN r� rC r r r� e s �z"IRCUser.irc_NAMES.<locals>.cbGroupc s$ | � tj� ���jd� g � d S )Nr� )r� r r� r� r r� ��channelr r r r� j s z"IRCUser.irc_NAMES.<locals>.ebGroup) rw rx ry ru r� r{ r r� r� rV r� r� r� r r� r � irc_NAMESM s$
� ��
zIRCUser.irc_NAMESc C s� z|d }t |d t�r|�| j�}W n ty$ | �tjd� Y dS w |�d�r0|dd� }t |�dkr@| �
||d � dS | � |� dS )zJ
Topic message
Parameters: <channel> [ <topic> ]
r z<:That channel doesn't exist (could not decode your unicode!)Nr� r� ) rw rx ry ru r� r{ r r� r� rB � _setTopic� _getTopic)r rr r� r� r r r � irc_TOPICq s" � ��
zIRCUser.irc_TOPICc C sd |j �d�}|r0|j �d�pd}|j �dd�}| �| jd|j |� | �| jd|j ||� dS dS )zP
Send the topic of the given group to this user, if it has one.
r r z<noone>�
topic_dater r� N)r r� r r � topicAuthor)r r] r r� �dater r r r� � s �zIRCUser._sendTopicc s( � �fdd�}�j �� ���j|� d S )Nc �" | � tj� ��tjd� d� d S �N�=r� r� r� r� r r r� � �
�z"IRCUser._getTopic.<locals>.ebGroup)rV r� r� r� )r r� r� r r� r r� � s zIRCUser._getTopicc s4 ��fdd�}� �fdd�}�j �� ��||� d S )Nc sH � j �� }�|d< �j|d< tt� �|d< � �fdd�}� �|��|�S )Nr r r� c s �� tjd� j d� d S )Nr� z.:You need to be a channel operator to do that.)r{ r �ERR_CHANOPRIVSNEEDEDr r� r\ r r �ebSet� s
�z1IRCUser._setTopic.<locals>.cbGroup.<locals>.ebSet)r �copyr �intr rL r5 )r] �newMetar )r r r� r r� � s
z"IRCUser._setTopic.<locals>.cbGroupc r r r� r� r� r r r� � r z"IRCUser._setTopic.<locals>.ebGroup)rV r� r� )r r� r r� r� r )r� r r r r� � s zIRCUser._setTopicc C s< |D ]\}}}| � tj|t|�d| � q| � tjd� dS )z�
Send a group of LIST response lines
@type channels: C{list} of C{(str, int, str)}
@param channels: Information about the channels being sent:
their name, the number of participants, and their topic.
r� z
:End of /LISTN)r{ r �RPL_LIST�str� RPL_LISTEND)r �channelsr rD r r r r rv � s z IRCUser.listc s� |rWz|d }t |t�r|�� j�}|�d�}W n ty, � �tj|d d� Y dS w g }|D ]}|� d�r>|dd� }|�
� j � |�� q1t
j|dd�}|�d d
� � n� j �� }� fd d �}|�|� dS )
z�
List query
Return information about the indicated channels, or about all
channels if none are specified.
Parameters: [ <channel> *( "," <channel> ) [ <target> ] ]
r �,r� Nr� r� T)�
consumeErrorsc S s dd� | D �S )Nc S � g | ]\}}|r|�qS r r �r� �s�rr r r r� � � z6IRCUser.irc_LIST.<locals>.<lambda>.<locals>.<listcomp>r )�gsr r r r� � s z"IRCUser.irc_LIST.<locals>.<lambda>c s6 dd� � t �� fdd�| D ��}|��fdd�� |S )Nc S s |j | |j�d�fS )Nr )r r r� )rD r] r r r �gotSize� r� z3IRCUser.irc_LIST.<locals>.cbGroups.<locals>.gotSizec s g | ]
}|� � �� |��qS r )rD r8 �r� r] �r r r r� � s z6IRCUser.irc_LIST.<locals>.cbGroups.<locals>.<listcomp>c s � � dd� | D ��S )Nc S r r r r r r r r� � r zHIRCUser.irc_LIST.<locals>.cbGroups.<locals>.<lambda>.<locals>.<listcomp>)rv )r) rC r r r� � s z4IRCUser.irc_LIST.<locals>.cbGroups.<locals>.<lambda>)r r7 r8 )rS r; rC r r �cbGroups� s �z"IRCUser.irc_LIST.<locals>.cbGroups)rw rx ry ru �splitr� r{ r r� r� r6 rV r� r r7 r8 rd )r rr r� � allChannelsr rS �chr r rC r �irc_LIST� s0
��
zIRCUser.irc_LISTc s, � � � jd|j � fdd�|�� D �� d S )Nr� c s* g | ]}|j � j� jj |j d d|j f�qS )�Hr )r rm rV )r� �mrC r r r� s ��z'IRCUser._channelWho.<locals>.<listcomp>)�whor rN �r r] r rC r � _channelWho� s
��zIRCUser._channelWhoc C s | � tjd� d S )Nz:User /WHO not implemented)r{ r � RPL_ENDOFWHO�r r, r r r �_userWho r� zIRCUser._userWhoc s� |s �� tjd� dS z|d � t� t�r� ��j�� W n ty0 �� tj|d d� Y dS w � �d�rP� �fdd�}�j �
� dd� �}|� �j |� dS � �fd d
�}�j �
� �}|� �j|� dS ) zC
Who query
Parameters: [ <mask> [ "o" ] ]
z:/WHO not supported.Nr z2:End of /WHO list (could not decode your unicode!)r� c � | � tj� ��tj� d� d S �Nz:End of /WHO list.)r� r r� r{ r r" r� �r� r r r r� * � z IRCUser.irc_WHO.<locals>.ebGroupr� c r% r& )r� r �
NoSuchUserr{ r r" r� r' r r r� 2 r( z IRCUser.irc_WHO.<locals>.ebUser)r{ r r" rw rx ry ru r� r� rV r� r� r! r� r$ )r rr r� r� r; r� r r' r �irc_WHO s.
� ��
zIRCUser.irc_WHOc s~ �fdd�}� �fdd�}z� d }t |t�r |��j�}W n ty2 ��tj� d d� Y dS w �j� |��
||� dS )zV
Whois query
Parameters: [ <target> ] <mask> *( "," <mask> )
c sN � � � j| j| j� jj| j� jjddtt� | j �| jdd� | �� D �� d S )NzHi mom!Fc S s g | ]}d |j �qS )r� r� r r r r r� L s z5IRCUser.irc_WHOIS.<locals>.cbUser.<locals>.<listcomp>)�whoisr rV r r rT rX rd r� rC r r �cbUser@ s �z!IRCUser.irc_WHOIS.<locals>.cbUserc r� )Nr �:No such nick/channel)r� r r) r{ r r� r� r� r r r� O s z!IRCUser.irc_WHOIS.<locals>.ebUserr r- N) rw rx ry ru r� r{ r r� rV r� r� )r rr r� r, r� r, r r� r � irc_WHOIS9 s
� �zIRCUser.irc_WHOISc C s | � tjd� dS )zE
Oper message
Parameters: <name> <password>
z:O-lines not applicableN)r{ r �ERR_NOOPERHOSTr� r r r �irc_OPER^ s zIRCUser.irc_OPERr )7rO rP rQ �__doc__rS rg ro rV ru rn rq r{ r4 r? rE rJ r� r� r� r� r� rj r� r � RPL_WELCOME� RPL_YOURHOST� RPL_CREATED�
RPL_MYINFOr� �
RPL_MOTDSTART�
RPL_ENDOFMOTDr� r� r� r� r� r� r� r� r� rk r� r� r� r� r� r� r� rv r r! r$ r* r. r0 r r r r rh � sp
" ����
&
#(($ 4
0 %rh c @ s e Zd ZdZeZdd� ZdS )�
IRCFactorya
IRC server that creates instances of the L{IRCUser} protocol.
@ivar _serverInfo: A dictionary mapping:
"serviceName" to the name of the server,
"serviceVersion" to the copyright version,
"creationDate" to the time that the server was started.
c C s& || _ || _| j jtjt� d�| _d S )N)� serviceName�serviceVersion� creationDate)rV r r r �versionr r� )r rV r r r r r s s �zIRCFactory.__init__N)rO rP rQ r1 rh r
r r r r r r8 g s r8 c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�PBMindc C � d S r r rC r r r r ~ � zPBMind.__init__c C s. t �t�}t|t�r|�d�}||j�| �fS �Nri )r
�qualr= rw r
rt �invoker�registerReference�r �jellierrA r r r �jellyFor� s
zPBMind.jellyForc C r> r r )r r, r] r r r �remote_userJoined� r? zPBMind.remote_userJoinedc C r> r r )r r, r] r@ r r r �remote_userLeft� r? zPBMind.remote_userLeftc C r> r r �r rF rG rH r r r �remote_receive� r? zPBMind.remote_receivec C r> r r �r r] r r r r �remote_groupMetaUpdate� r? z PBMind.remote_groupMetaUpdateN) rO rP rQ r rF rG rH rJ rL r r r r r= } s r= c @ s2 e Zd ZdZdd� Zdd� Zdd� Zd d d
�ZdS ) �PBMindReferencer c C sJ t j�|�rt| j| j|�}nt| j| j|�}| �dt| j| j|�||�S )NrE )r r� r0 �PBGrouprV ro �PBUser�
callRemote)r rF rG rH �recr r r rE � s �zPBMindReference.receivec C s | � dt| j| j|�|�S )NrJ )rP rN rV ro rK r r r rJ � s �z PBMindReference.groupMetaUpdatec C s& | � dt| j| j|�t| j| j|��S )Nr4 �rP rN rV ro rO r� r r r r4 � s
�zPBMindReference.userJoinedNc C s( | � dt| j| j|�t| j| j|�|�S )Nr? rR r� r r r r? � s �zPBMindReference.userLeftr )rO rP rQ r rE rJ r4 r? r r r r rM � s rM c @ s6 e Zd Zdd� Zdd� Zdd� Zd dd �Zd
d � ZdS )
rN c C � || _ || _|| _d S r )rV ro r] )r rV ro r] r r r r � �
zPBGroup.__init__c C � t | jj| jj| jjf�S r )�hashrV r ro r] rC r r r �processUniqueID� � zPBGroup.processUniqueIDc C sN t �| j�}t|t�r|�d�}| jj}t|t�r |�d�}|||j� | �fS r@ )
r
rA � __class__rw r
rt r] r rB rC )r rE rA r] r r r rF � s
zPBGroup.jellyForNc C � | j �| j|�S r )ro rb r] rp r r r � remote_leave� r% zPBGroup.remote_leavec C rZ r )ro rc r] �r rH r r r � remote_send� r% zPBGroup.remote_sendr )rO rP rQ r rW rF r[ r] r r r r rN � s
rN c @ sX e Zd Zdd� Zddd�Zdd� Zdd � Zd
d � Zd d
� Zddd�Z dd� Z
dd� Z dS )�PBGroupReferencec C sF |\}}}|| _ ttkrt| j t�r| j �d�| _ tj�| |||g�S r@ )r rx r
rw ry r �RemoteReference�
unjellyFor)r � unjellier� unjellyList�clsNamer �refr r r r` � s
zPBGroupReference.unjellyForNc C � | � d|�S )Nrb �rP rp r r r rb � � zPBGroupReference.leavec C re )Nrc rf r\ r r r rc � rg zPBGroupReference.sendc C r> r r r# r r r r< � � zPBGroupReference.addc C r> r r rC r r r rN � rh zPBGroupReference.iterusersc C r> r r rI r r r rE � rh zPBGroupReference.receivec C r> r r )r r, r@ r r r r' � rh zPBGroupReference.removec C r> r r )r r r r r rL � rh z PBGroupReference.setMetadatac C r> r r rC r r r rD � rh zPBGroupReference.sizer ) rO rP rQ r` rb rc r< rN rE r' rL rD r r r r r^ � s
r^ c @ � e Zd Zdd� Zdd� ZdS )rO c C rS r )rV ro r, )r rV ro r, r r r r � rT zPBUser.__init__c C rU r )rV rV r ro r, rC r r r rW rX zPBUser.processUniqueIDN)rO rP rQ r rW r r r r rO � s rO c @ s` e Zd Zdd� Zdd� Zdd� Zedd� �Zejd d� �Zd
d � Z d d
� Z
dd� Z ddd�Z dS )�
ChatAvatarc C �
|| _ d S r )ro )r ro r r r r
re zChatAvatar.__init__c C s0 t �| j�}t|t�r|�d�}||j�| �fS r@ )r
rA rY rw r
rt rB rC rD r r r rF
s
zChatAvatar.jellyForc s( � fdd�}� j j�|�}|�|� |S )Nc s( � �fdd�}�j �� �}|�|� |S )Nc s t �jj�j� �S r )rN ro rV )�ignoredr\ r r r^ r� z7ChatAvatar.remote_join.<locals>.cbGroup.<locals>.cbJoinr� )r] r^ r; rC r� r r� s
z'ChatAvatar.remote_join.<locals>.cbGroup)ro rV r� r8 )r r� r� r; r rC r � remote_join s
zChatAvatar.remote_joinc C r> r r rC r r r r � zChatAvatar.namec C r> r r )r r& r r r r % rn c C r> r r rK r r r rJ * rh zChatAvatar.groupMetaUpdatec C r> r r rI r r r rE . rh zChatAvatar.receivec C r> r r r� r r r r4 2 rh zChatAvatar.userJoinedNc C r> r r r� r r r r? 6 rh zChatAvatar.userLeftr )
rO rP rQ r rF rm �propertyr �setterrJ rE r4 r? r r r r rj s
rj c @ ri )�AvatarReferencec C re )Nr` rf )r r� r r r r` ? rg zAvatarReference.joinc s, t �� � | j�� fdd�� | jj�� � S )Nc s
� � d �S r )�callbackr �r; r r r� D s
z&AvatarReference.quit.<locals>.<lambda>)r �Deferred�broker�notifyOnDisconnectr� r� rC r rs r �quitB s zAvatarReference.quitN)rO rP rQ r` rw r r r r rq > s rq c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dZ
dZ dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )!�
WordsRealmri c C rk r r� r r r r r P re zWordsRealm.__init__c C � t |�S r )rR r r r r � userFactoryS � zWordsRealm.userFactoryc C ry r )r r r r r � groupFactoryV r{ zWordsRealm.groupFactoryc s � �fdd�}|S )Nc s" t �ddd� �� d � _� _d S )Nrg c S r> r r r r r r r� \ s z:WordsRealm.logoutFactory.<locals>.logout.<locals>.<lambda>)�getattrrV rW r �ro �facetr r rg Z s z(WordsRealm.logoutFactory.<locals>.logoutr )r ro r rg r r~ r �
logoutFactoryY s zWordsRealm.logoutFactoryc s8 t � t�r � ��j�� � ���fdd�}��� ��|�S )Nc sl | j d ur t�� ��D ]%}|| d �}|d ur0| ���� � �_��_ | �_||��| |�f S q t����r )rV r r� rY r ro r� �NotImplementedError)ro r� r ��avatarId�
interfacesrW r r r � gotAvatare s
�
z+WordsRealm.requestAvatar.<locals>.gotAvatar)rw rx ry � _encoding�getUserr8 )r r� rW r� r� r r� r �
requestAvatara s
zWordsRealm.requestAvatarc C r> r r rC r r r rd t rh zWordsRealm.itergroupsFTc C � t �r �r� r r r r r� | r? zWordsRealm.lookupUserc C r� r r� r r r r r� r? zWordsRealm.lookupGroupc C r� )a�
Add the given user to this service.
This is an internal method intended to be overridden by
L{WordsRealm} subclasses, not called by external code.
@type user: L{IUser}
@rtype: L{twisted.internet.defer.Deferred}
@return: A Deferred which fires with L{None} when the user is
added, or which fails with
L{twisted.words.ewords.DuplicateUser} if a user with the
same name exists already.
r� r# r r r �addUser� s zWordsRealm.addUserc C r� )aY
Add the given group to this service.
@type group: L{IGroup}
@rtype: L{twisted.internet.defer.Deferred}
@return: A Deferred which fires with L{None} when the group is
added, or which fails with
L{twisted.words.ewords.DuplicateGroup} if a group with the
same name exists already.
r� r r r r �addGroup� s zWordsRealm.addGroupc �. �j r� �fdd�}��� ��|�S ��� �S )Nc � | � tj� ��� �S r )r� r �DuplicateGroupr� r� �r r r r r� � �
z$WordsRealm.getGroup.<locals>.ebGroup)�createGroupOnRequest� createGroupr5 r� )r r r� r r� r r� � �
zWordsRealm.getGroupc r� )Nc r� r )r� r �
DuplicateUserr� r� r� r r r� � r� z"WordsRealm.getUser.<locals>.ebUser)�createUserOnRequest�
createUserr5 r� )r r r� r r� r r� � r� zWordsRealm.getUserc �H � fdd�}� �fdd�}� � � � ��� �}|�||� |��j� |S )Nc � t �t�� ��S r )r r r r� r� r� r r �cbLookup� r% z'WordsRealm.createUser.<locals>.cbLookupc r� r )r� r r) rz r� r� r r �ebLookup� r� z'WordsRealm.createUser.<locals>.ebLookup)r� r� r� r8 r� �r r r� r� r; r r� r r� � �
zWordsRealm.createUserc r� )Nc r� r )r r r r� r� r� r r r� � r% z(WordsRealm.createGroup.<locals>.cbLookupc r� r )r� r r� r| r� r� r r r� � r� z(WordsRealm.createGroup.<locals>.ebLookup)r� r� r� r8 r� r� r r� r r� � r� zWordsRealm.createGroupN)rO rP rQ r� r rz r| r� r� rd r� r� r� r� r� r� r� r� r� r� r r r r rx L s$
rx c sD e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Z� Z S )
�InMemoryWordsRealmc s" t � j|i |�� i | _i | _d S r )�superr r rS )r �ar~ �rY r r r � s
zInMemoryWordsRealm.__init__c C s t �| j�� �S r )r r9 rS r2 rC r r r rd � r% z InMemoryWordsRealm.itergroupsc C �6 |j | jv rt�t�t�� ��S || j|j < t�|�S r ) r r r �failr r r r� r9 r# r r r r� � �
zInMemoryWordsRealm.addUserc C r� r ) r rS r r� r r r r� r9 r r r r r� � r� zInMemoryWordsRealm.addGroupc
C �J |� � }z| j| }W n ty t�t�t�|��� Y S w t� |�S r )
r� r r> r r� r r r r) r9 )r r r, r r r r� � � �
z InMemoryWordsRealm.lookupUserc
C r� r )
r� rS r> r r� r r r r� r9 )r r r] r r r r� � r� z InMemoryWordsRealm.lookupGroup)
rO rP rQ r rd r� r� r� r� �
__classcell__r r r� r r� � s r� )r rR rx r� N)6r1 r r �zope.interfacer �twistedr � twisted.credr r r� r �twisted.internetr r
�twisted.pythonr r r
�twisted.python.componentsr �twisted.spreadr �
twisted.wordsr r �twisted.words.protocolsr r� r r� rR r� r/ rz rh �
ServerFactoryr8 �
Referenceabler= r_ rM �setUnjellyableForClassrN r^ rO rj � IPerspectiverq �IRealm� IChatServicerx r� �__all__r r r r �<module> sV M( E " '
2 (