o ^i@@szdZddlZddlZddlmZmZddlmZddlm Z m Z ddl m Z ddl mZddlmZd ZGd d d ZdS) zBase websocket classes.N)Optional no_type_check)urlparse)ioloopweb)IOStream)JupyterHandler)JupyterServerAuthWarningi0ucseZdZUdZdZdZdZdZee e d<e ddZ e ddZ edd eed efd d Zd dZeddZefddZefddZeddZddZZS)WebSocketMixinz"Mixin for common websocket optionsNgstreamcCs|jdtS)ziThe interval for websocket keep-alive pings. Set ws_ping_interval = 0 to disable pings. ws_ping_interval)settingsgetWS_PING_INTERVALselfrX/home/ubuntu/hpml_nyu/venv/lib/python3.10/site-packages/jupyter_server/base/websocket.py ping_intervalszWebSocketMixin.ping_intervalcCs|jdtd|jtS)zIf no ping is received in this many milliseconds, close the websocket connection (VPNs, etc. can fail to cleanly close ws connections). Default is max of 3 pings or 30 seconds. ws_ping_timeout)r rmaxrrrrrr ping_timeout"szWebSocketMixin.ping_timeoutoriginreturncCs|jdkst|dr|rdS|jjd}|dur|}|dus'|dur)dS|}t|j }||kr8dS|jrA|j|k}n|j rNt t |j |}nd}|sZ|jd|||S)zCheck Origin == Host or Access-Control-Allow-Origin. Tornado >= 4 calls this method automatically, raising 403 if it returns False. *skip_check_originTHostNFz>Blocking Cross Origin WebSocket Attempt. Origin: %s, Host: %s) allow_originhasattrrrequestheadersr get_originlowerrnetlocallow_origin_patboolrematchlogwarning)rrhost origin_hostallowrrr check_origin,s6   zWebSocketMixin.check_origincOsdS)zmeaningless for websocketsNr)rargskwargsrrr clear_cookieWszWebSocketMixin.clear_cookiecCsn|jdds1|jjstdt||jj}t|dds3|j}|dur5|j dtddSdSdS)z}Verify authentication if required. Only used when the websocket class does not inherit from JupyterHandler. allow_unauthenticated_accessFi__allow_unauthenticatedNz*Couldn't authenticate WebSocket connection) r rr methodr HTTPErrorgetattrr# current_userr)r*)rr4userrrr _maybe_authZs    zWebSocketMixin._maybe_authcslt|ts)|jdd }d|jvr|rtjdtdd|tj |i|Stj |i|ddiS)zHandle a get request.r2Fidentity_providerz~WebSocketMixin sub-class does not inherit from JupyterHandler preventing proper authentication using custom identity provider.) stacklevel_redirect_to_login) isinstancerr rwarningswarnr r9superprepare)rr/r0should_authenticate __class__rrrBls zWebSocketMixin.preparecsf|jd|jj|jdkr*tj}||_ |j |_ t |j |j|_ |j tj|i|S)zOpen the websocket.zOpening websocket %sr)r)debugr pathrrIOLoopcurrenttime last_ping last_pongPeriodicCallback send_ping ping_callbackstartrAopen)rr/r0looprDrrrQ|s    zWebSocketMixin.opencCs|jdur|jdur|jdS|jjr|dStj}d||j }d||j }|d|j krI||j krI|j d||dS|d||_ dS)z'send a ping to keep the websocket aliveNg@@r;z#WebSocket ping timeout after %i ms.) ws_connectionrOstopclient_terminatedcloserrHrIrJrLrKrrr)r*ping)rnowsince_last_pongsince_last_pingrrrrNs   zWebSocketMixin.send_pingcCstj|_dS)zHandle a pong message.N)rrHrIrJrL)rdatarrron_pongszWebSocketMixin.on_pong)N)__name__ __module__ __qualname____doc__rOrKrLr rr__annotations__propertyrrrstrr&r.r1r9rBrQrNr] __classcell__rrrDrr s,    *  r )rar'r?typingrr urllib.parsertornadorrtornado.iostreamrjupyter_server.base.handlersrjupyter_server.utilsr rr rrrrs