Te'ddlZddlZddlZddlmZmZmZddlmZddl m Z m Z m Z m Z ejeZdZdZGdd Zy) N)AnyDictMapping)futures)JsonRpcExceptionJsonRpcRequestCancelledJsonRpcInternalErrorJsonRpcMethodNotFoundz2.0z$/cancelRequestc eZdZddfdZdZddZddZdZd Zd Z e d Z d Z e d e eefdedeeeffdZdZdZddZy)Endpointc<ttjSN)struuiduuid48/usr/lib/python3/dist-packages/pylsp_jsonrpc/endpoint.pyzEndpoint.s#djjlBSrc||_||_||_i|_i|_t j ||_y)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__s>&!)')$')$!(!;!; !Trc8|jjyr)r shutdown)r!s rr'zEndpoint.shutdown(s '')rNcttjd||t|d}|||d<|j|y)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)r!r*r+messages rnotifyzEndpoint.notify+sB /@'    &GH  wrc |j}tjd|||t||d}|||d<t j }|j |j|||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*r+) rr,r-r.rFutureadd_done_callback_cancel_callbackrr)r!r*r+msg_idr/request_futures rrequestzEndpoint.request=s##% 5vvvN'    &GH  )(()>)>v)FG/=$$V, wrcfd}|S)z;Construct a cancellation callback for the given request ID.c|jr2jtdi|jt yy)Nr2) cancelledr0 CANCEL_METHOD set_exceptionr )future request_idr!s rcallbackz+Endpoint._cancel_callback..callback\s9! MD*+=>$$%<%>?"rrr!r?r@s`` rr5zEndpoint._cancel_callbackZs @rc d|vs |dtk7rtjd|yd|vr;tjd||j |d|j dyd|vrKtjd||j |d|j d |j d y tjd ||j|d|d|j dy#t$rM}tjd |d|jt|d|jd Yd}~yd}~wt$rktjd |d|jt|dtjtj jd YywxYw)zConsume a JSON RPC message from the client. Args: message (dict): The JSON RPC message sent by the client r)zUnknown message type %sNr2z$Handling notification from client %sr*r+z Handling response from client %sresulterrorzHandling request from client %sFailed to handle request %s)r)r2rD)r.r,warningr-_handle_notificationget_handle_response_handle_requestr exceptionrto_dict Exceptionr ofsysexc_info)r!r/es rconsumezEndpoint.consumebsy G #wy'9_'L KK17 ;  w  IIC66 F>?AEA4F>=F>c|tk(r|j|dy |j|} ||}t|rStjd||jj|}|j|j||yy#t$rtj d|YywxYw#t $rtjd||YywxYw)z&Handle a notification from the client.r2Nz+Ignoring notification for unknown method %sz$Failed to handle notification %s: %sz'Executing async notification handler %s)r<_handle_cancel_notificationrKeyErrorr,rFrMrKcallabler-r submitr4_notification_callback)r!r*r+handlerhandler_resultnotification_futures rrGzEndpoint._handle_notifications ] "  , ,VD\ :  &&v.G  $V_N N # II? P"&"8"8"?"?"O   1 1$2M2MfV\2] ^ $  KKEv N     MM@&& Q  s"BB<B98B9< CCcfd}|S)z;Construct a notification callback for the given request ID.c |jtjdy#t$rtj dYywxYw)Nz-Successfully handled async notification %s %sz)Failed to handle async notification %s %s)rCr,r-rMrK)r>r*r+s rr@z1Endpoint._notification_callback..callbacksG [  I6SYZ [ I6SYZ [s'+ A Ar)r*r+r@s`` rrXzEndpoint._notification_callbacks [ rc|jj|d}|stjd|y|j rtj d|yy)z-Handle a cancel notification from the client.Nz6Received cancel notification for unknown message id %szCancelled request with id %s)rpopr,rFcancelr-)r!r6r7s rrTz$Endpoint._handle_cancel_notificationsR5599&$G KKPRX Y   " II4f = #rheaderrCreturnc|t|}t|trd|vsd|vr|j||S||d<|S)NrCrD)dict isinstanceupdate)rarCresponses r_make_response_payloadzEndpoint._make_response_payloadsI< fd #V);w&?P OOF #"(HX rc |j|}||}t |rat j d||jj|}||j|<|j|j|yt|tjrFt j d|||j|<|j|j|yt j d||jt |d|}|j#|y#t$r}tj||d}~wwxYw)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: %sr)r2)rrUr rNrVr,r-r rWrr4_request_callbackrerr3rhr.r) r!r6r*r+rYrQrZr7rgs rrJzEndpoint._handle_requests  :&&v.G! N # II:N K!33::>JN3AD ( ( 0  , ,T-C-CF-K L  7 II> O3AD ( ( 0  , ,T-C-CF-K L IIG X22.  H NN8 $/ :'**62 9 :sD&& E /EE cfd}|S)z6Construct a request callback for the given request ID.c8jjd|jr|jt t d} |j }j||}j#|y#t$r3}tjd|j|d<Yd}~Id}~wt$rQtjdtjtj j|d<YwxYw)NrjrErD)rr_r;r=r r.rCrhrr,rKrLrMr rNrOrPr)r>r/rCrQr?r!s rr@z,Endpoint._request_callback..callbacks  ( ( , ,Z >!$$%<%>?+ G  U55gvF NN7 #$ / ;ZH#$99;  U ;ZH#7#:#:3<<>#J#R#R#T  Us"B D)B<tCH~swsTWxGX%:8*rr )loggingrrOtypingrrr concurrentr exceptionsrr r r getLoggerrqr,r.r<r rrrrzsO %%FFg!! s*s*r