:vjw<dZddlmZmZddlmZddlmZmZddl m Z ddl m Z ddlmZ ddlmZ dd lmZd d lmZd d lmZd dlmZdZGddeZGddeZGddeeeZGddeZ dZ!y#e$rd ZYWwxYw#e$r dd lmZYawxYw)zF tornado handler for managing and communicating with language servers )OptionalText) ensure_async) APIHandlerJupyterHandler) url_path_join)web)WebSocketHandler) authorizedc|S)z)A no-op fallback for `jupyter_server 1.x`)methods e/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/jupyter_lsp/handlers.pyr r s )WebSocketMixin)LanguageServerManager)SERVERS_RESPONSE) censored_speclspceZdZdZdefdZy) BaseHandlerNmanagerc||_yNrselfrs r initializezBaseHandler.initialize$  r__name__ __module__ __qualname__rrrr rrrr!G"7rrceZdZdZdefdZy)BaseJupyterHandlerNrc||_yrrrs rrzBaseJupyterHandler.initialize+r rr!r rrr'r'(r%rr'cXeZdZUdZeZdZeee d<dZ fdZ fdZ dZ dZxZS) LanguageServerWebSocketHandleradSetup tornado websocket to route to language server sessions. The logic of `get` and `pre_get` methods is derived from jupyter-server ws handlers, and should be kept in sync to follow best practice established by upstream; see: https://github.com/jupyter-server/jupyter_server/blob/v2.12.5/jupyter_server/services/kernels/websocket.py#L36 Nlanguage_serverc:K|j}|0|jjdtjdt |dsyt |jj||dtd{}|stjdy7w)zHandle a pre_get.Nz*Couldn't authenticate WebSocket connectioni authorizerexecute) current_userlogwarningr HTTPErrorhasattrrr- is_authorized AUTH_RESOURCE)ruserr4s rpre_getz&LanguageServerWebSocketHandler.pre_get=s   < HH  I J--$ $t\* + OO ) )$i O  --$ $ sA:B<B=Bc~K|jd{t||i|}| |d{yy7!7w)zGet an event socket.N)r7superget)rargskwargsres __class__s rr:z"LanguageServerWebSocketHandler.getPsEllngk4*6* ?II   s=9=;==cK|jjd{||_|jj||jj dj |jt|!y7iw)Nz[{}] Opened a handler) rreadyr+ subscriber0debugformatr9open)rr+r>s rrDz#LanguageServerWebSocketHandler.openWsgll  """. t$ .55d6J6JKL   #sB B A*B cK|jjdj|j|jj ||d{y7w)Nz[{}] Handling a message)r0rBrCr+ron_client_message)rmessages r on_messagez)LanguageServerWebSocketHandler.on_message^sB 0778L8LMNll,,Wd;;;sAAAAc|jj||jjdj |j y)Nz[{}] Closed a handler)r unsubscriber0rBrCr+)rs ron_closez'LanguageServerWebSocketHandler.on_closebs6   & .55d6J6JKLr)r"r#r$__doc__r5 auth_resourcer+rr__annotations__r7r:rDrHrK __classcell__)r>s@rr*r*/s7"M&*OXd^*%&<Mrr*cFeZdZdZeZeZeje dZ y)LanguageServersHandlerzsReports the status of all current servers Response should conform to schema in schema/servers.schema.json c TK|jjd{d|jjjDcic]\}}||j c}}|jj jDcic]\}}|t |c}}d}t|jj|}|r4|jjdjt|||j|y7cc}}wcc}}ww)z4finish with the JSON representations of the sessionsN)versionsessionsspecsz{} validation errors: {})rr@rUitemsto_jsonall_language_serversrlist validator iter_errorsr0r1rClenfinish)rr+sessionkeyspecresponseerrorss rr:zLanguageServersHandler.getpsll  """15 0E0E0K0K0M0M,OW !220M "&!B!B!H!H!J!JIC]4((!J  dnn00:;  HH  7>>s6{FS T H' #s.D(D,D( D(*D(D" (A3D( D(N) r"r#r$rLr5rMrr[r authenticatedr r:r rrrQrQgs3 "M IrrQc t|jd}d}d|ji}|jj dt|dt |ft|d|t |fgy)zAAdd Language Server routes to the notebook server web applicationrz(?P.*)rz.*statuswsN)ujoinbase_urllanguage_server_managerweb_app add_handlersrQr*)nbapplsp_urlre_langserversoptss rrlrlsjENNE*G.N u44 5D MM 7H %'=t Dgt^4.   rN)"rLtypingrrjupyter_core.utilsrjupyter_server.base.handlersrrjupyter_server.utilsrrhtornador tornado.websocketr jupyter_server.auth.decoratorr ImportErrorjupyter_server.base.websocketrjupyter_server.base.zmqhandlersrrschemar specs.utilsrr5rr'r*rQrlr rrr}s"+C7.8?<+$& *5M$&85Mp [ Fy?>?s"A?B ?B  B  BB