a ën§af4ã@s0UdZddlmZddlZe e¡ZddlZddlZ ddl m Z m Z m Z mZmZddlmZddlmZmZddlmZmZddlmZdd lmZmZmZd d lmZd d l m!Z!m"Z"m#Z#d d l$m%Z%d dl&m'Z'd dl(m)Z)ddl*m+Z+ddl,m-Z-dZ.Gdd„de-eƒZ/e)Gdd„dƒƒZ0dZ1de2d<dS)zA Provide a web socket handler for the Bokeh Server application. é)Ú annotationsN)ÚAnyÚDictÚListÚUnionÚcast)Úurlparse)ÚlocksÚweb)ÚWebSocketClosedErrorÚWebSocketHandler)Úsettings)Úcheck_token_signatureÚget_session_idÚget_token_payloadé)ÚProtocol)Ú MessageErrorÚ ProtocolErrorÚValidationError)ÚMessage)ÚReceiver)Ú dataclassé)ÚProtocolHandleré)Ú AuthMixin)Ú WSHandlercs(eZdZdZddœ‡fdd„ Zdd„Zdd d œd d „Zejddœd d„ƒZ dddœdd„Z ddœdd„Z dddœdd„Z dddœdd„Z dddœd d!„Zd"dd#œd$d%„ZdZ sign_sessionsrCrrAZio_loopZ add_callbackÚ _async_openÚ ExceptionÚdebug)r/ÚtokenZnowZpayloadÚ session_idÚer4r4r5Úopen|s6      þ zWSHandler.openz List[str]z str | None)Ú subprotocolsrcCs8t d¡t d|¡t|ƒdks&dS|d|_|dS)NzSubprotocol header receivedz Supplied subprotocol headers: %rrrr)r@rKÚtraceÚlenr))r/rPr4r4r5Úselect_subprotocol¥s     zWSHandler.select_subprotocolzDict[str, Any] | NonecCs0|jdurdSd|ji}|jdur,|j|d<|S)Nr"r#)r+r,)r/Zoptionsr4r4r5Úget_compression_options­s     z!WSHandler.get_compression_options)rLrc Ãsàz€t|ƒ}|j ||j|¡IdH|j |¡}tƒ}t|ƒ|_t  d|¡t ƒ|_ t  d|¡|j   |||j|¡|_t d¡Wn<ty¼}z$t d|¡| ¡|‚WYd}~n d}~00|jj d¡}| |¡IdHdS)a Perform the specific steps needed to open a connection to a Bokeh session Specifically, this method coordinates: * Getting a session for a session ID (creating a new one if needed) * Creating a protocol receiver and handler * Opening a new ServerConnection and sending it an ACK Args: session_id (str) : A session ID to for a session to connect to If no session exists with the given ID, a new session is made Returns: None NzReceiver created for %rzProtocolHandler created for %rzServerConnection createdz/Could not create new server session, reason: %sZACK)rr Zcreate_session_if_neededZrequestZ get_sessionrrr$r@rKrr%r>Znew_connectionr&rDrrArEÚprotocolZcreateÚ send_message)r/rLrMZsessionrUrNÚmsgr4r4r5rIµs$     zWSHandler._async_openz str | bytes)Úfragmentrc Ãsàz| |¡IdH}WnDtyX}z,tjd||dd| d¡d}WYd}~n d}~00z@|r˜tdurttj |¡| |¡IdH}|r˜|  |¡IdHWn@tyÚ}z(tjd||dd| d¡WYd}~n d}~00dS)a Process an individual wire protocol fragment. The websocket RFC specifies opcodes for distinguishing text frames from binary frames. Tornado passes us either a text or binary string depending on that opcode, we have to look at the type of the fragment to see what we got. Args: fragment (unicode or bytes) : wire fragment to process Nz/Unhandled exception receiving a message: %r: %rT©Úexc_infoz server failed to parse a messagez.Handler or its work threw an exception: %r: %rz!server failed to handle a message) Ú_receiverJr@rAÚ_internal_errorÚ_message_test_portÚreceivedÚappendÚ_handleÚ _schedule)r/rXÚmessagerNÚworkr4r4r5Ú on_messageás"   zWSHandler.on_messageÚbytes)ÚdatarcCs\zt| d¡ƒ|_WnBty6tjd|ddYn"tyVtjd|ddYn0dS)Nzutf-8z#received invalid unicode in pong %rTrYz#received invalid integer in pong %r)ÚintÚdecoder'ÚUnicodeDecodeErrorr@rQÚ ValueError)r/rfr4r4r5Úon_pongs   zWSHandler.on_pongz Message[Any])rbrcÃsJz(tdurtj |¡| |¡IdHWntyDt d¡Yn0dS)z‰ Send a Bokeh Server protocol message to the connected client. Args: message (Message) : a message to send Nz/Failed sending message as connection was closed)r]Úsentr_Úsendr r@Zwarning©r/rbr4r4r5rVs  zWSHandler.send_messageFTz!Union[bytes, str, Dict[str, Any]])rbÚbinaryÚlockedrcƒsb|rJ|j ¡IdH$tƒ ||¡IdHWdƒq^1s>0Yntƒ ||¡IdHdS)zj Override parent write_message with a version that acquires a write lock before writing. N)r(Úacquirer-Ú write_message)r/rbrorpr2r4r5rr"s4zWSHandler.write_messagecCs.t d|j|j¡|jdur*|j |j¡dS)z2 Clean up when the connection is closed. z/WebSocket connection closed: code=%s, reason=%rN)r@rDZ close_codeZ close_reasonr&r>Z client_lost)r/r4r4r5Úon_close.s zWSHandler.on_closezMessage[Any] | Nonec ÃsXz|j |¡IdH}|WStttfyR}z| t|ƒ¡WYd}~dSd}~00dSr6)r$ZconsumerrrÚ_protocol_errorr8)r/rXrbrNr4r4r5r[6s zWSHandler._receivez Any | Nonec Ãs\z|j ||j¡IdH}|WStttfyV}z| t|ƒ¡WYd}~dSd}~00dSr6)r%Zhandler&rrrr\r8)r/rbrcrNr4r4r5r`?s zWSHandler._handler)rcrcÃs:t|tƒr&| ttt|ƒ¡IdHn| d|›¡dS)Nzexpected a Message not )Ú isinstancerrVrrr\)r/rcr4r4r5raHs zWSHandler._schedulecCst d|¡| d|¡dS)Nz3Bokeh Server internal error: %s, closing connectioni'©r@rArErnr4r4r5r\Ps zWSHandler._internal_errorcCst d|¡| d|¡dS)Nz3Bokeh Server protocol error: %s, closing connectioni'rvrnr4r4r5rtTs zWSHandler._protocol_error)FT)Ú__name__Ú __module__Ú __qualname__Ú__doc__r.r7rBr Z authenticatedrOrSrTrIrdrkrVrrrsr[r`rar\rtÚ __classcell__r4r4r2r5rDs((,' ÿ   rc@seZdZUded<ded<dS)ÚMessageTestPortzList[Message[Any]]rlr^N)rwrxryÚ__annotations__r4r4r4r5r|_s r|zMessageTestPort | Noner])3rzZ __future__rZloggingZ getLoggerrwr@rFrHrGÚtypingrrrrrZ urllib.parserZtornador r Ztornado.websocketr r Zbokeh.settingsr Zbokeh.util.tokenrrrrUrZprotocol.exceptionsrrrZprotocol.messagerZprotocol.receiverrZutil.dataclassesrZprotocol_handlerrZ auth_mixinrÚ__all__rr|r]r}r4r4r4r5Ús2