!Yw`z%ddlZddlZddlZddlmZddlmZmZmZm Z ej e Z dZ dZGddZdS) N)futures)JsonRpcExceptionJsonRpcRequestCancelledJsonRpcInternalErrorJsonRpcMethodNotFoundz2.0z$/cancelRequestcteZdZddfdZdZddZddZdZd Zd Z e d Z d Z d Z dZddZdS)EndpointcBttjSN)struuiduuid46lib/python3.11/site-packages/pylsp_jsonrpc/endpoint.pyzEndpoint.s#djllBSBSrc||_||_||_i|_i|_t j||_dS)aA JSON RPC endpoint for managing messages sent to/from the client. Args: dispatcher (dict): A dictionary of method name to handler function. The handler functions should return either the result or a callable that will be used to asynchronously compute the result. consumer (fn): A function that consumes JSON RPC message dicts and sends them to the client. id_generator (fn, optional): A function used to generate request IDs. Defaults to the string value of :func:`uuid.uuid4`. max_workers (int, optional): The number of workers in the asynchronous executor pool. ) max_workersN) _dispatcher _consumer _id_generator_client_request_futures_server_request_futuresrThreadPoolExecutor_executor_service)self dispatcherconsumer id_generatorrs r__init__zEndpoint.__init__sG&!)')$')$!(!; !T!T!Trc8|jdSr )rshutdown)rs rr$zEndpoint.shutdown's '')))))rNctd||t|d}|||d<||dS)zSend a JSON RPC notification to the client. Args: method (str): The method name of the notification to send params (any): The payload of the notification zSending notification: %s %s)jsonrpcmethodNparams)logdebugJSONRPC_VERSIONr)rr'r(messages rnotifyzEndpoint.notify*sY /@@@'     &GH  wrc@|}td|||t||d}|||d<t j}|||||j|<| ||S)a Send a JSON RPC request to the client. Args: method (str): The method name of the message to send params (any): The payload of the message Returns: Future that will resolve once a response has been received z!Sending request with id %s: %s %s)r&idr'Nr() rr)r*r+rFutureadd_done_callback_cancel_callbackrr)rr'r(msg_idr,request_futures rrequestzEndpoint.request<s##%% 5vvvNNN'     &GH  ))(()>)>v)F)FGGG/=$V, wrcfd}|S)z;Construct a cancellation callback for the given request ID.c|r@tdi|t dSdS)Nr/) cancelledr- CANCEL_METHOD set_exceptionr)future request_idrs rcallbackz+Endpoint._cancel_callback..callback[s^!! @ MD*+=>>>$$%<%>%>????? @ @rrrr<r=s`` rr2zEndpoint._cancel_callbackYs/ @ @ @ @ @ @rc `d|vs|dtkrtd|dSd|vrLtd|||d|ddSd|vr`td|||d|d |d dS td |||d|d|ddS#t$rb}t d |d| t|d| d Yd}~dSd}~wt$r~t d |d| t|dtjtj d YdSwxYw)zConsume a JSON RPC message from the client. Args: message (dict): The JSON RPC message sent by the client r&zUnknown message type %sNr/z$Handling notification from client %sr'r(z Handling response from client %sresulterrorzHandling request from client %sFailed to handle request %s)r&r/rA)r+r)warningr*_handle_notificationget_handle_response_handle_requestr exceptionrto_dict Exceptionrofsysexc_info)rr,es rconsumezEndpoint.consumeas? G # #wy'9_'L'L KK17 ; ; ; F w   II.callbackst [  I6SYZZZZZ [ [ [ I6SYZZZZZZ [s05&AAr)r'r(r=s`` rrUzEndpoint._notification_callbacks/ [ [ [ [ [ [ rc|j|d}|std|dS|rtd|dSdS)z-Handle a cancel notification from the client.Nz6Received cancel notification for unknown message id %szCancelled request with id %s)rpopr)rCcancelr*)rr3r4s rrQz$Endpoint._handle_cancel_notificationsx599&$GG  KKPRX Y Y Y F  " " > II4f = = = = = > >rc |j|}n'#t$r}tj||d}~wwxYw||}t |rit d||j|}||j |<| | |dSt|tjrOt d|||j |<| | |dSt d||t ||ddS)z!Handle a request from the client.Nz"Executing async request handler %sz&Request handler is already a future %sz/Got result from synchronous request handler: %s)r&r/r@)rrRrrKrSr)r*rrTrr1_request_callback isinstancerr0rr+)rr3r'r(rVrNrWr4s rrGzEndpoint._handle_requestsm :&v.GG : : :'*622 9 :! N # #  II:N K K K!3::>JJN3AD ( 0  , ,T-C-CF-K-K L L L L L  7 7  II> O O O3AD ( 0  , ,T-C-CF-K-K L L L L L IIG X X X NN*(     s  4/4cfd}|S)z6Construct a request callback for the given request ID.cjjd|r!|t t d} ||d<n#t$r<}t d| |d<Yd}~nhd}~wt$rXt dtj tj |d<YnwxYw|dS)N)r&r/r@rBrA)rr\r8r:rr+r@rr)rHrIrJrrKrLrMr)r;r,rNr<rs rr=z,Endpoint._request_callback..callbacksB  ( , ,Z > > >!! @$$%<%>%>???+ G  U$*MMOO!!# / / / ;ZHHH#$99;;       U U U ;ZHHH#7#:3<>>#J#J#R#R#T#T    U NN7 # # # # #sA55 D?2B66A"DDrr>s`` rr_zEndpoint._request_callbacks) $ $ $ $ $ $.rcj|j|d}|std|dS|Etd|||t j|dStd||||dS)z"Handle a response from the client.Nz*Received response to unknown message id %sz)Received error response to message %s: %sz"Received result for message %s: %s) rr\r)rCr*r:r from_dict set_result)rr3r@rAr4s rrFzEndpoint._handle_responses599&$GG  KKDf M M M F   IIA65 Q Q Q  ( ()9)CE)J)J K K K F 6GGG!!&)))))rr )NN)__name__ __module__ __qualname__r"r$r-r5r2rOrD staticmethodrUrQrGr_rFrrrr r s:S:SabUUUU(***    $:!!!F___.\ > > >46******rr )loggingrrL concurrentr exceptionsrrrr getLoggerrfr)r+r9r rrrrns FFFFFFFFFFFFg!!! d*d*d*d*d*d*d*d*d*d*r