
    cjh}                       d Z ddlmZ ddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlZddlmZmZ ddlZddlmZ ddlmZ ddlmZ dd	lmZmZ  ej4                  e      Zd
Z G d de      Z G d de      Z  edd      Z! G d d      Z" G d de	jF                        Z$ G d de$      Z% G d dejL                        Z' G d d      Z( G d d      Z)y)ae  
The MIT License (MIT)

Copyright (c) 2015-2021 Rapptz
Copyright (c) 2021-present Pycord Development

Permission is hereby granted, free of charge, to any person obtaining a
copy of this software and associated documentation files (the "Software"),
to deal in the Software without restriction, including without limitation
the rights to use, copy, modify, merge, publish, distribute, sublicense,
and/or sell copies of the Software, and to permit persons to whom the
Software is furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
DEALINGS IN THE SOFTWARE.
    )annotationsN)deque
namedtuple   )utils)BaseActivity)SpeakingState)ConnectionClosedInvalidArgument)DiscordWebSocketKeepAliveHandlerVoiceKeepAliveHandlerDiscordVoiceWebSocketReconnectWebSocketc                      e Zd ZdZdddZy)r   z*Signals to safely reconnect the WebSocket.Tresumec               B    || _         || _        |rd| _        y d| _        y )NRESUMEIDENTIFY)shard_idr   op)selfr   r   s      B/var/www/html/venv/lib/python3.12/site-packages/discord/gateway.py__init__zReconnectWebSocket.__init__<   s     $(*    N)__name__
__module____qualname____doc__r    r   r   r   r   9   s    4+/ 5r   r   c                      e Zd ZdZy)WebSocketClosurezIAn exception to make up for the fact that aiohttp doesn't signal closure.N)r   r   r   r    r!   r   r   r#   r#   B   s    Sr   r#   EventListenerzpredicate event result futurec                  &    e Zd ZddZd Zd Zd Zy)GatewayRatelimiterc                |    || _         || _        d| _        || _        t	        j
                         | _        d | _        y )N        )max	remainingwindowperasyncioLocklockr   )r   countr,   s      r   r   zGatewayRatelimiter.__init__J   s2    LLN	r   c                    t        j                          }|| j                  | j                  z   kD  ry| j                  dk(  S )NFr   )timer+   r,   r*   r   currents     r   is_ratelimitedz!GatewayRatelimiter.is_ratelimitedS   s4    ))+T[[488++~~""r   c                r   t        j                          }|| j                  | j                  z   kD  r| j                  | _        | j                  | j                  k(  r|| _        | j                  dk(  r| j                  || j                  z
  z
  S | xj                  dz  c_        | j                  dk(  r|| _        y)Nr   r   r(   )r2   r+   r,   r)   r*   r3   s     r   	get_delayzGatewayRatelimiter.get_delayY   s    ))+T[[488++!XXDN>>TXX%!DK>>Q88w455!>>Q!DKr   c                ,  K   | j                   4 d {    | j                         }|r>t        j                  d| j                  |       t        j                  |       d {    d d d       d {    y 7 e7 7 	# 1 d {  7  sw Y   y xY ww)Nz=WebSocket in shard ID %s is ratelimited, waiting %.2f seconds)r/   r7   _logwarningr   r-   sleep)r   deltas     r   blockzGatewayRatelimiter.blockk   s~     99 	+ 	+NN$ESMM
 mmE***	+ 	+ 	+ +	+ 	+ 	+ 	+sW   BA9BAA?#A;$A?(B3A=4B;A?=B?BBBBN)n         N@)r   r   r   r   r5   r7   r=   r!   r   r   r&   r&   I   s    #$	+r   r&   c                  0    e Zd Zd Zd Zd Zd Zd Zd Zy)r   c                D   |j                  dd       }|j                  dd       }|j                  dd       }t        j                  j                  | g|i | || _        |j
                  | _        || _        d| _        || _	        d| _
        d| _        d| _        t        j                         | _        t        j                          | _        t        j                          | _        t        j                          | _        t)        d      | _        |j,                  | _        y )	Nwsintervalr   Tz5Keeping shard ID %s websocket alive with sequence %s.z7Shard ID %s heartbeat blocked for more than %s seconds.z5Can't keep up, shard ID %s websocket is %.1fs behind.inf)pop	threadingThreadr   rB   	thread_id_main_thread_idrC   daemonr   msg	block_msg
behind_msgEvent_stop_evr2   perf_counter	_last_ack
_last_send
_last_recvfloatlatency_max_heartbeat_timeoutheartbeat_timeout)r   argskwargsrB   rC   r   s         r   r   zKeepAliveHandler.__init__x   s    ZZd#::j$/::j$/!!$888!||  JRQ!)**,++-++-U|!#!:!:r   c                   | j                   j                  | j                        s| j                  | j                  z   t        j                         k  rt        j                  d| j                         | j                  j                  d      }t        j                  || j                  j                        }	 |j                          | j%                          y | j'                         }t        j)                  | j*                  | j                  |d          | j                  j-                  |      }t        j                  || j                  j                        }	 d}	 	 |j                  d       	 t        j                         | _"        | j                   j                  | j                        sy y # t         $ r t        j#                  d       Y w xY w# | j%                           Y y xY w# t.        j0                  j2                  $ r |dz  }	 t5        j6                         | j8                     }dj;                  t=        j>                  |            }| j@                   d	| }n# tB        $ r | j@                  }Y nw xY wt        j                  || j                  |       Y nw xY wO# t         $ r | j%                          Y ?w xY w)
NzJShard ID %s has stopped responding to the gateway. Closing and restarting.  loopz7An error occurred while stopping the gateway. Ignoring.dr   
    z0
Loop thread traceback (most recent call last):
)#rO   waitrC   rS   rW   r2   rP   r9   r:   r   rB   closer-   run_coroutine_threadsafer]   result	Exception	exceptionstopget_payloaddebugrK   send_heartbeat
concurrentfuturesTimeoutErrorsys_current_framesrI   join	tracebackformat_stackrL   KeyErrorrR   )r   corofdatatotalframestackrK   s           r   runzKeepAliveHandler.run   s?   --$$T]]3!7!77$:K:K:MM' MM ww}}T*44TMHHJ IIK##%DJJtxxS	:77))$/D00DGGLLIA6@$ #'"3"3"5e --$$T]]33 ! NNQ
 IIK &--:: @	$'$7$7$9$:N:N$OE %'GGI,B,B5,I$JE#'>>"2 3116!9  	  ( 1"&..C1 S$--?@	 $  		sx   <F: "J4 &G6 :GG GG G36"J/!I0:6J/0JJ/J$J/,J4 .J//J4 4KKc                \    | j                   j                  | j                   j                  dS )Nr   r^   )rB   	HEARTBEATsequencer   s    r   rh   zKeepAliveHandler.get_payload   s!    gg''dgg.>.>??r   c                8    | j                   j                          y N)rO   setr   s    r   rg   zKeepAliveHandler.stop   s    r   c                6    t        j                         | _        y r   )r2   rP   rS   r   s    r   tickzKeepAliveHandler.tick   s    ++-r   c                    t        j                         }|| _        || j                  z
  | _        | j                  dkD  r6t
        j                  | j                  | j                  | j                         y y )Nr_   )	r2   rP   rQ   rR   rU   r9   r:   rM   r   r   ack_times     r   ackzKeepAliveHandler.ack   sT    $$&!$//1<<"LL$--F r   N)	r   r   r   r   rz   rh   rg   r   r   r!   r   r   r   r   w   s#    ;(36j@.Gr   r   c                  *     e Zd Z fdZd Zd Z xZS )r   c                p    t        |   |i | t        d      | _        d| _        d| _        d| _        y )N   )maxlenz<Keeping shard ID %s voice websocket alive with timestamp %s.z<Shard ID %s voice heartbeat blocked for more than %s secondsz:High socket latency, shard ID %s heartbeat is %.1fs behind)superr   r   recent_ack_latenciesrK   rL   rM   )r   rX   rY   	__class__s      r   r   zVoiceKeepAliveHandler.__init__   s7    $)&)$)$4!QWVr   c                p    | j                   j                  t        t        j                         dz        dS )N  r|   )rB   r}   intr2   r   s    r   rh   z!VoiceKeepAliveHandler.get_payload   s'    gg''c$))+2D.EFFr   c                    t        j                         }|| _        || _        || j                  z
  | _        | j                  j                  | j
                         y r   )r2   rP   rQ   rS   rR   rU   r   appendr   s     r   r   zVoiceKeepAliveHandler.ack   sF    $$&!"$//1!!((6r   )r   r   r   r   rh   r   __classcell__r   s   @r   r   r      s    WG7r   r   c                  (     e Zd Zdddd fdZ xZS )DiscordClientWebSocketResponser[   r   codemessagec               B   K   t         |   ||       d {   S 7 w)Nr   )r   rb   )r   r   r   r   s      r   rb   z$DiscordClientWebSocketResponse.close   s      W]g]>>>>s   )r   r   r   bytesreturnbool)r   r   r   rb   r   r   s   @r   r   r      s    )- ? ?r   r   c                     e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZd Zed        Zd Zd Zd Zedddddddd       Zd*dZd Zd Zd Zed+d       Zd Zd Zd Z d  Z!d! Z"d" Z#ddd#d$d%Z$	 d*dddd&d'Z%d,d(Z&d-d)Z'y).r   a   Implements a WebSocket for Discord's gateway v6.

    Attributes
    ----------
    DISPATCH
        Receive only. Denotes an event to be sent to Discord, such as READY.
    HEARTBEAT
        When received tells Discord to keep the connection alive.
        When sent asks if your connection is currently alive.
    IDENTIFY
        Send only. Starts a new session.
    PRESENCE
        Send only. Updates your presence.
    VOICE_STATE
        Send only. Starts a new connection to a voice guild.
    VOICE_PING
        Send only. Checks ping time to a voice guild, do not use.
    RESUME
        Send only. Resumes an existing connection.
    RECONNECT
        Receive only. Tells the client to reconnect to a new gateway.
    REQUEST_MEMBERS
        Send only. Asks for the full member list of a guild.
    INVALIDATE_SESSION
        Receive only. Tells the client to optionally invalidate the session
        and IDENTIFY again.
    HELLO
        Receive only. Tells the client the heartbeat interval.
    HEARTBEAT_ACK
        Receive only. Confirms receiving of a heartbeat. Not having one implies
        a connection issue.
    GUILD_SYNC
        Send only. Requests a guild sync.
    gateway
        The gateway we are currently connected to.
    token
        The authentication token for discord.
    r   r                        	   r_         c               $   || _         || _        d | _        g | _        d | _        t        j                         | _        d | _        d | _	        d | _
        t        j                         | _        t               | _        d | _        t#               | _        y )Nc                      y r   r!   )rX   s    r   <lambda>z+DiscordWebSocket.__init__.<locals>.<lambda>%  s    r   )socketr]   	_dispatch_dispatch_listeners_keep_aliverF   	get_identrH   
session_idr~   resume_gateway_urlzlibdecompressobj_zlib	bytearray_buffer_close_coder&   _rate_limiter)r   r   r]   s      r   r   zDiscordWebSocket.__init__   s}    	 ,#% ",,. "&'')
 {/1r   c                0    | j                   j                   S r   )r   closedr   s    r   openzDiscordWebSocket.open5  s    ;;%%%%r   c                6    | j                   j                         S r   )r   r5   r   s    r   r5   zDiscordWebSocket.is_ratelimited9  s    !!0022r   c               (    | j                  d|       y )Nsocket_raw_receive)r   r   rv   s     r   debug_log_receivez"DiscordWebSocket.debug_log_receive<  s    +T2r   c                    y r   r!   )r   _s     r   log_receivezDiscordWebSocket.log_receive?  s    r   FN)initialgatewayr   sessionr~   r   c                 K   |xs" |j                   j                          d{   }|j                   j                  |       d{   } | ||j                        }	|j                   j                  |	_        |j
                  |	_        |j
                  j                  |	_        |j                  |	_	        ||	_
        |j
                  j                  |	_        ||	_        ||	_        ||	j                  _        |j
                  j                  |	_        ||	_        ||	_        |j
                  j$                  |	_        |j(                  r"|	j*                  |	_        |	j.                  |	_        |j
                  j3                  |	       t4        j7                  d|       |	j9                          d{    |s|	j;                          d{    |	S |	j=                          d{    |	S 7 7 7 @7 (7 w)zmCreates a main websocket for Discord from a :class:`Client`.

        This is for internal use only.
        Nr\   z!Created websocket connected to %s)httpget_gateway
ws_connectr]   token_connectionparsers_discord_parsersdispatchr   r   
call_hooks_initial_identifyr   r   shard_countr   r~   rW   rV   _enable_debug_events
debug_sendsendr   r   _update_referencesr9   ri   
poll_eventidentifyr   )
clsclientr   r   r   r   r~   r   r   rB   s
             r   from_clientzDiscordWebSocket.from_clientB  s      <6;;#:#:#<<{{--g66fkk* ;;$$++$0088
**55&$,!++77$*$6$6$H$H!&&mmBG11BN--b1

6@ mmo++-Iiik	G =66 	   	sX   "G6G*"G6G-E'G6/G00G6	G2
G6#G4$G6-G60G62G64G6c                    | j                   j                         }t        ||||      }| j                  j	                  |       |S )a`  Waits for a DISPATCH'd event that meets the predicate.

        Parameters
        ----------
        event: :class:`str`
            The event name in all upper case to wait for.
        predicate
            A function that takes a data parameter to check for event
            properties. The data parameter is the 'd' key in the JSON message.
        result
            A function that takes the same data parameter and executes to send
            the result to the future. If ``None``, returns the data.

        Returns
        -------
        asyncio.Future
            A future to wait for.
        )event	predicaterd   future)r]   create_futurer$   r   r   )r   r   r   rd   r   entrys         r   wait_forzDiscordWebSocket.wait_forw  sD    ( ((*9VF
 	  ''.r   c                  K   | j                   | j                  t        j                  dddddddd}| j                  *| j
                  | j                  | j
                  g|d	   d
<   | j                  }|j                  |j                  !|j                  |j                  ddd|d	   d<   |j                  |j                  j                  |d	   d<   | j                  d| j                  | j                         d{    | j                  |       d{    t        j                  d| j                         y7 >7 'w)zSends the IDENTIFY packet.pycord)osbrowserdeviceT   r   )r   
propertiescompresslarge_thresholdvr|   Nr^   shardr   F)statusgamesinceafkpresenceintentsbefore_identify)r   z*Shard ID %s has sent the IDENTIFY payload.)r   r   rn   platformr   r   r   	_activity_status_intentsvaluer   r   send_as_jsonr9   info)r   payloadstates      r   r   zDiscordWebSocket.identify  s9     --,,'&
 !#&

 ==$)9)9)E%)]]D4D4D$EGCL!  ??&%--*C--	(GCL$ >>%&+nn&:&:GCL#oot}}d6L6L  
 	
 	
 (((		>N		
 	)s$   D EEEE&EEc                   K   | j                   | j                  | j                  | j                  dd}| j	                  |       d{    t
        j                  d| j                         y7 %w)zSends the RESUME packet.)seqr   r   r|   Nz(Shard ID %s has sent the RESUME payload.)r   r~   r   r   r   r9   r   r   r   r   s     r   r   zDiscordWebSocket.resume  s^      ++}}"oo
 (((		<dmmL 	)s   AA0A.	&A0c               z  K   t        |      t        u rw| j                  j                  |       t	        |      dk  s|dd  dk7  ry | j
                  j                  | j                        }|j                  d      }t               | _        | j                  |       t        j                  |      }t        j                  d| j                  |       |j                  d      }|r| j!                  d|       |j                  d      }|j                  d	      }|j                  d
      }||| _        | j$                  r| j$                  j'                          || j(                  k7  r|| j*                  k(  rBt        j                  d       | j-                          d {    t/        | j                        || j0                  k(  r'| j$                  r| j$                  j3                          y || j4                  k(  r@| j$                  r3| j$                  j7                         }| j9                  |       d {    y || j:                  k(  rq|d   dz  }t=        | || j                        | _        | j9                  | j$                  j7                                d {    | j$                  j?                          y || j@                  k(  r|du r-| j-                          d {    t/        | j                        d | _        d | _!        t        jE                  d| j                         | j-                  d       d {    t/        | j                  d      t        jG                  d|       y |dk(  r|j                  dg       x| _$        }|d
   | _        |d   | _!        |d   | _%        | j                  |d<   t        jE                  d| j                  djM                  |      | jB                         nh|dk(  rc|j                  dg       x| _$        }| j                  |d<   t        jE                  d| j                  | jB                  djM                  |             	 | jN                  |   }	 |	|       g }
tS        | jT                        D ]  \  }}|jV                  |k7  r|jX                  }|j[                         r|
j]                  |       D	 |j_                  |      }|sY|j`                  |n|ja                  |      }|jc                  |       |
j]                  |        ti        |
      D ]  }| jT                  |=  y 7 7 7 7 q7 # tP        $ r t        j                  d|       Y w xY w# td        $ r-}|jg                  |       |
j]                  |       Y d }~ d }~ww xY ww) Nr   s     zutf-8z$For Shard ID %s: WebSocket Event: %stsocket_event_typer   r^   szReceived RECONNECT opcode.heartbeat_interval     @@)rB   rC   r   Tz)Shard ID %s session has been invalidated.r   r   Fr   zUnknown OP code %s.READY_tracer   r   __shard_id__z:Shard ID %s has connected to Gateway: %s (Session ID: %s)., RESUMEDz?Shard ID %s has successfully RESUMED session %s under trace %s.zUnknown event %s.)5typer   r   extendlenr   
decompressdecoder   r   r   
_from_jsonr9   ri   r   getr   r~   r   r   DISPATCH	RECONNECTrb   r   HEARTBEAT_ACKr   r}   rh   r   HELLOr   startINVALIDATE_SESSIONr   r   r:   r  r   rp   r   rs   	enumerater   r   r   	cancelledr   r   rd   
set_resultre   set_exceptionreversed)r   rK   r   r   rv   r  beatrC   tracefuncremovedindexr   r   validretexcs                    r   received_messagez!DiscordWebSocket.received_message  s    9LL$3x!|s23x+>>**''5C**W%C$;DLs#

94==#NNN.6WWT]wws|ggcl?DM!!#T^^# 

78jjl""(77T'''##$$((*T^^###++779D++D111TZZ 45>#3h$  ''(8(8(D(D(FGGG  &&(T,,,4<**,&&,T]];; $"&		Et}}Ujjdj+++(uEELL.3G"&((8R"88DK%HDM"<0DO&*+?&@D##'==D IIL		% 	 i"&((8R"88DK%#'==D IIQ		% 		((/D J %d&>&>? 	*LE5{{e#\\F!u%	*-
 "',,"6$ELL<NC%%c*NN5)%	*( g& 	0E((/	0y # 2 H ' ,B  	3JJ*E2	3$  &$$S)u%%&s   FV; U!BV;9U:A%V;U AV;%U&AV;UD!V;%U 4A#V;V)V;,A"V;V;V;V;V;U?;V;>U??V;	V8"V3-V;3V88V;c                L    | j                   }|t        d      S |j                  S )zDMeasures latency between a HEARTBEAT and a HEARTBEAT_ACK in seconds.rD   r   rT   rU   r   	heartbeats     r   rU   zDiscordWebSocket.latencyF  )     $$	(0uU|Gi6G6GGr   c                R    | j                   xs | j                  j                  }|dvS )N)r   i  i  i  i  i  i  )r   r   
close_coder   r   s     r   _can_handle_closez"DiscordWebSocket._can_handle_closeL  s(    94;;#9#9EEEr   c                  K   	 | j                   j                  | j                         d{   }|j                  t        j
                  j                  u r$| j                  |j                         d{    y|j                  t        j
                  j                  u r$| j                  |j                         d{    y|j                  t        j
                  j                  t        j
                  j                  t        j
                  j                  t        j
                  j                  fv rt        j                  d|       t         y7 (7 7 # t"        j$                  t         f$ r}| j&                  r!| j&                  j)                          d| _        t+        |t"        j$                        r+t        j-                  d       t/        | j0                        d| j2                  xs | j                   j4                  }| j7                         r,t        j-                  d|       t/        | j0                        dt        j-                  d|       t9        | j                   | j0                  |      dd}~ww xY ww)zPolls for a DISPATCH event and handles the general gateway loop.

        Raises
        ------
        ConnectionClosed
            The websocket connection was terminated for unhandled reasons.
        timeoutNReceived %sz3Timed out receiving packet. Attempting a reconnect.z1Websocket closed with %s, attempting a reconnect.z+Websocket closed with %s, cannot reconnect.r   r   )r   receiverV   r  aiohttp	WSMsgTypeTEXTr+  rv   BINARYCLOSEDCLOSINGCLOSEERRORr9   ri   r#   r-   rm   r   rg   
isinstancer   r   r   r   r2  r4  r
   )r   rK   er   s       r   r   zDiscordWebSocket.poll_eventP  s     	++D4O4O+PPCxx7,,111++CHH555W..555++CHH555!!((!!))!!''!!''	  

=#.&& Q55 $$&67 	  %%'#' !W112		OP(7TA##=t{{'='=D%%'		MtT(7TA		GN&KK$--d!	sx   J	)E EAE 6E7E ;J	<AE  EE J	BE J	E E E J6DJJJ	c                  K   | j                   j                          d {    | j                  d|       | j                  j	                  |       d {    y 7 :7 w)Nsocket_raw_send)r   r=   r   r   send_strr   s     r   r   zDiscordWebSocket.debug_sendz  sN       &&(((($/kk""4((( 	)(s!   AA4AAAAc                  K   | j                   j                          d {    | j                  j                  |       d {    y 7 (7 wr   )r   r=   r   rG  r   s     r   r   zDiscordWebSocket.send  s?       &&(((kk""4((( 	)(s!   AA	"AAAAc                   K   	 | j                  t        j                  |             d {    y 7 # t        $ r<}| j	                         s"t        | j                  | j                        |Y d }~y d }~ww xY wwNr   )r   r   _to_jsonRuntimeErrorr4  r
   r   r   r   rv   r*  s      r   r   zDiscordWebSocket.send_as_json  sa     	U))ENN40111 	U))+&t{{T]]KQTT ,	Us6   A;'3 13 A;3 	A82A3.A;3A88A;c                  K   	 | j                   j                  t        j                  |             d {    y 7 # t        $ r<}| j                         s"t        | j                   | j                        |Y d }~y d }~ww xY wwrJ  )r   rG  r   rL  rM  r4  r
   r   rN  s      r   rj   zDiscordWebSocket.send_heartbeat  sg     	U++&&u~~d';<<< 	U))+&t{{T]]KQTT ,	Us7   B1= ;= B= 	B2A=8B=BBr(   )activityr   r   c               l  K   |-t        |t              st        d      |j                         g}ng }|dk(  r t	        t        j
                         dz        }| j                  |d||dd}t        j                  |      }t        j                  d|       | j                  |       d {    y 7 w)Nz'activity must derive from BaseActivity.idler   F)
activitiesr   r   r   r|   zSending "%s" to change status)rC  r   r   to_dictr   r2   PRESENCEr   rL  r9   ri   r   )r   rP  r   r   r   sents         r   change_presencez DiscordWebSocket.change_presence  s     h5%&OPP ((*+HHV		d*+E --& 	
 ~~g&

2D9iios   B*B4,B2-B4)user_ids	presencesnoncec                  K   | j                   |||dd}|r||d   d<   |r||d   d<   |||d   d<   | j                  |       d {    y 7 w)N)guild_idrY  limitr|   r^   rZ  rX  query)REQUEST_MEMBERSr   )r   r\  r^  r]  rX  rY  rZ  r   s           r   request_chunkszDiscordWebSocket.request_chunks  sn      &&&YO

 $)GCL!'/GCL$$)GCL!(((s   AAA	Ac                   K   | j                   ||||dd}t        j                  d|       | j                  |       d {    y 7 w)N)r\  
channel_id	self_mute	self_deafr|   zUpdating our voice state to %s.)VOICE_STATEr9   ri   r   )r   r\  rb  rc  rd  r   s         r   voice_statezDiscordWebSocket.voice_state  sL     ""$(&&	
 	

4g>(((s   ?A	AA	c                   K   | j                   r!| j                   j                          d | _         || _        | j                  j	                  |       d {    y 7 wNr  )r   rg   r   r   rb   r3  s     r   rb   zDiscordWebSocket.close  sL     !!##DkkT***s   AAAAr   r   rT   )FF)r[   )(r   r   r   r    r  r}   r   rU  re  
VOICE_PINGr   r  r_  r  r  r  
GUILD_SYNCr   propertyr   r5   r   r   classmethodr   r   r   r   r+  rU   r4  r   r   r   r   rj   rW  r`  rf  rb   r!   r   r   r   r      s   %N HIHHKJFIOEMJ2* & &33 
 2 2h6$OLM~0@ H H
F(T)
)UU 15T 4 #)7;uTX)&)+r   r   c                      e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdddZd Zd ZeZd Zd Zedddd       Zd Zd Zej6                  fdZd Zd Zed"d       Z ed#d       Z!d Z"d  Z#d$d!Z$y)%r   aG  Implements the WebSocket protocol for handling voice connections.

    Attributes
    ----------
    IDENTIFY
        Send only. Starts a new voice session.
    SELECT_PROTOCOL
        Send only. Tells discord what encryption mode and how to connect for voice.
    READY
        Receive only. Tells the websocket that the initial connection has completed.
    HEARTBEAT
        Send only. Keeps your websocket connection alive.
    SESSION_DESCRIPTION
        Receive only. Gives you the secret key required for voice.
    SPEAKING
        Send only. Notifies the client if you are currently speaking.
    HEARTBEAT_ACK
        Receive only. Tells you your heartbeat has been acknowledged.
    RESUME
        Sent only. Tells the client to resume its session.
    HELLO
        Receive only. Tells you that your websocket connection was acknowledged.
    RESUMED
        Sent only. Tells you that your RESUME request has succeeded.
    CLIENT_CONNECT
        Indicates a user has connected to voice.
    CLIENT_DISCONNECT
        Receive only.  Indicates a user has disconnected from voice.
    r   r   r   r   r   r   r   r   r   r   r      N)hookc               l    || _         || _        d | _        d | _        d | _        i | _        |r|| _        y y r   )rB   r]   r   r   
secret_keyssrc_map_hook)r   r   r]   rp  s       r   r   zDiscordVoiceWebSocket.__init__  s=    	DJ r   c                   K   y wr   r!   )r   rX   s     r   rt  zDiscordVoiceWebSocket._hook  s	     s   c                   K   t         j                  d|       | j                  j                  t	        j
                  |             d {    y 7 w)Nz"Sending voice websocket frame: %s.)r9   ri   rB   rG  r   rL  r   s     r   r   z"DiscordVoiceWebSocket.send_as_json  s6     

7>ggu~~d3444s   AA
AAc                   K   | j                   }| j                  |j                  t        |j                        |j
                  dd}| j                  |       d {    y 7 w)N)r   	server_idr   r|   )r   r   r   strrx  r   r   r   r   r   s      r   r   zDiscordVoiceWebSocket.resume  sV       ++ 1#..
 (((s   AA%A#A%c                
  K   | j                   }| j                  t        |j                        t        |j                  j
                        |j                  |j                  dd}| j                  |       d {    y 7 w)N)rx  user_idr   r   r|   )	r   r   ry  rx  useridr   r   r   rz  s      r   r   zDiscordVoiceWebSocket.identify   sf       -- 1uzz}}-#..	
 (((s   A9B;B<BF)r   rp  c                 K   d|j                    d}|j                  j                  }|j                  |d       d{   } | ||j                  |      }||_        ||_        d|_        t        j                         |_
        |r|j                          d{    |S |j                          d{    |S 7 }7 "7 
w)z7Creates a voice websocket for the :class:`VoiceClient`.zwss://z/?v=4   )r   N)r]   rp  r?   )endpoint_stater   r   r]   r   r   rV   rF   r   rH   r   r   )r   r   r   rp  r   r   r   rB   s           r   r   z!DiscordVoiceWebSocket.from_client-  s      6??+51}}!!w<<fkk5
$(! **,))+ 	 ++-	 = s5   =CB= ACB?C6C7C?CCc                n   K   | j                   d|||ddd}| j                  |       d {    y 7 w)Nudp)addressportmode)protocolrv   r|   )SELECT_PROTOCOLr   )r   ipr  r  r   s        r   select_protocolz%DiscordVoiceWebSocket.select_protocol@  s>     &&!$&dC
 (((s   +535c                   K   | j                   d| j                  j                  id}| j                  |       d {    y 7 w)N
audio_ssrcr|   )CLIENT_CONNECTr   ssrcr   r  s     r   client_connectz$DiscordVoiceWebSocket.client_connectK  s@     %% 0 0 5 56

 (((s   :AAAc                x   K   | j                   t        |      ddd}| j                  |       d {    y 7 w)Nr   )speakingdelayr|   )SPEAKINGr   r   rz  s      r   speakzDiscordVoiceWebSocket.speakS  s1     #e*q-QR(((s   0:8:c                  K   t         j                  d|       |d   }|j                  d      }|| j                  k(  r| j	                  |       d {    nC|| j
                  k(  r| j                  j                          n|| j                  k(  rt         j                  d       n|| j                  k(  r.|d   | j                  _        | j                  |       d {    n|| j                  k(  r?|d   dz  }t        | t!        |d      	      | _        | j                  j#                          nh|| j$                  k(  rY|d
   }t'        |d         }|d   }|| j(                  v r|| j(                  |   d<   n | j(                  j+                  |||di       | j-                  | |       d {    y 7 e7 7 
w)Nz"Voice websocket frame received: %sr   r^   zVoice RESUME succeeded.r  r	  r
  g      @)rB   rC   r  r|  r  )r|  r  )r9   ri   r  r  initial_connectionr  r   r   r  r   SESSION_DESCRIPTIONr   r  load_secret_keyr  r   minr  r  r   rs  updatert  )r   rK   r   rv   rC   r  r}  r  s           r   r+  z&DiscordVoiceWebSocket.received_messageX  s    

7=Ywws|))$///4%%%  "4<<II/04+++$(LD!&&t,,,4::01F:H4#h"4 D ""$4== <DtI'DJ'Ht}}$2:d#J/$$d(,S%TUjjs###1 0 -" 	$s8   AF>F7BF> F:!CF>1F<2F>:F><F>c                T  K   | j                   }|d   |_        |d   |_        |d   |_        t	        d      }t        j                  d|dd       t        j                  d|dd	       t        j                  d
|d|j                         |j                  j                  ||j                  |j                  f       | j                  j                  |j                  d       d {   }t        j                  d|       d}|j                  d|      }||| j                  d      |_        t        j                   d|t#        |      dz
        d   |_        t        j                  d|j                  |j$                         |d   D cg c]  }|| j                   j&                  v s| }}t        j                  ddj)                  |             |d   }| j+                  |j                  |j$                  |       d {    t        j-                  d|       y 7 9c c}w 7 #w)Nr  r  r  J   z>Hr   r   r   F   z>Ir   z)received packet in initial_connection: %sr   asciizdetected ip: %s port: %smodesz'received supported encryption modes: %sr  z(selected the voice protocol for use (%s))r   r  
voice_portendpoint_ipr   struct	pack_intor   sendtor]   	sock_recvr9   ri   r'  r  r  unpack_fromr  r  supported_modesrp   r  r   )	r   rv   r   packetrecvip_startip_endr  r  s	            r   r  z(DiscordVoiceWebSocket.initial_connectionx  s      &\
< J2vq!,vq"-vq%**5FU%6%68H8H$IJYY((r::

>E Ax((//8''dCIMB1E


-uxxD "']
dd6F6F6V6V.VD
 
 	

<dii>NOQx""588UZZ>>>		<dC' ;
 	?s8   C#H(%H&B$H(
H!(H!,AH(H&H(!H(c                L    | j                   }|t        d      S |j                  S )z=Latency between a HEARTBEAT and its HEARTBEAT_ACK in seconds.rD   r-  r.  s     r   rU   zDiscordVoiceWebSocket.latency  r0  r   c                    | j                   }||j                  st        d      S t        |j                        t	        |j                        z  S )z'Average of last 20 HEARTBEAT latencies.rD   )r   r   rT   sumr  r.  s     r   average_latencyz%DiscordVoiceWebSocket.average_latency  sH     $$	I$B$B<9112S9W9W5XXXr   c                   K   t         j                  d       |j                  d      x| _        | j                  _        | j                          d {    | j                  d       d {    y 7 7 w)Nz(received secret key for voice connectionrr  F)r9   r   r  rr  r   r  r   s     r   r  z%DiscordVoiceWebSocket.load_secret_key  s[     		<=8<8NN$**5jjljj 	s$   AA5A1A5+A3,A53A5c                  K   t        j                  | j                  j                         d       d {   }|j                  t
        j                  j                  u r7| j                  t        j                  |j                               d {    y |j                  t
        j                  j                  u r8t        j                  d|       t        | j                  d       |j                  |j                  t
        j                  j                   t
        j                  j"                  t
        j                  j$                  fv r8t        j                  d|       t        | j                  d | j&                        y 7 R7 w)Ng      >@r6  r8  rK  r9  )r-   r   rB   r:  r  r;  r<  r=  r+  r   r  rv   rB  r9   ri   r
   r?  rA  r@  r   )r   rK   s     r   r   z DiscordVoiceWebSocket.poll_event  s    $$TWW__%6EE88w((---''(8(8(BCCCXX**000JJ}c*"477T:HXX$$##%%
 

 JJ}c*"477T@P@PQQ
 FCs#   3FFAFF
C6F
Fc                   K   | j                   | j                   j                          || _        | j                  j	                  |       d {    y 7 wrh  )r   rg   r   rB   rb   r3  s     r   rb   zDiscordVoiceWebSocket.close  sD     '!!#ggmmm&&&s   AAAAri  )r   zlist[float] | float)r   )%r   r   r   r    r   r  r  r}   r  r  r  r   r  r  r  CLIENT_DISCONNECTr   rt  r   rj   r   r   rm  r   r  r  r	   voicer  r+  r  rl  rU   r  r  r   rb   r!   r   r   r   r     s    < HOEIHMFEGN-1 5 "N
)) 16T  $	)) !. 3 3 )
$@D@ H H
 Y Y R 'r   r   )*r    
__future__r   r-   concurrent.futuresrk   loggingr  rn   rF   r2   rq   r   collectionsr   r   r;  r`   r   rP  r   enumsr	   errorsr
   r   	getLoggerr   r9   __all__re   r   r#   r$   r&   rG   r   r   ClientWebSocketResponser   r   r   r!   r   r   <module>r     s   2 #     
     )   "   5w"5 5Ty T ?,KL++ ++\XGy'' XGv7, 7&?W%D%D ?
h+ h+Vm' m'r   