o ^i@sddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZmZdd lmZd Zd d ZGdddZgdZGdddeZdefddZdS))ThreadPoolExecutor)Path)rmtree)List)run_on_executor)convert_yielded)lsp_message_listener)file_uri_to_path is_relative)LanguageServerManagerAPIc Cs4|D]}z||}WqttfyYdSw|SN)KeyError TypeError)objpathcrumbr_/home/ubuntu/hpml_nyu/venv/lib/python3.10/site-packages/jupyter_lsp/virtual_documents_shadow.pyextract_or_nones rc@seZdZeedZddZddZddZe dd Z e d d Z e d e d edefddZe defddZdefddZeddZdS) EditableFile) max_workerscCs"t|tr t||_dS||_dSr) isinstancestrrr)selfrrrr__init__s"zEditableFile.__init__cst|IdH|_dSr)r read_lineslinesrrrrread"szEditableFile.readcst|IdHSr)r write_linesrrrrwrite%szEditableFile.writecCs4dg}z |jjdd}W|StyY|Sw)Nutf-8encoding)r read_text splitlinesFileNotFoundError)rrrrrr(s zEditableFile.read_linescCs.|jjjddd|jjd|jdddS)NTparentsexist_ok r$r%)rparentmkdir write_textjoinrrrrrr!3szEditableFile.write_linesr charactersidecCs4d}|r|||d}|||krd}|||<|S)NFTr)rr2r3 needs_gluetrimmedrrrtrim8s zEditableFile.trimgluecCs.|sgS|r |dnd|r|dgSdgS)Nr#rr)leftrightr7rrrr1Bs&zEditableFile.jointextc Cs|jd|d}|j|dd}|j||ddd}|j||ddd}|d}|d|r1dnd||||||r@dnd|rEdnd||||||rUdnddp]dg|_dS) Nliner2r)rr2r3r8r-rr#)rr6splitr1) rr;startendbeforeafterneeds_glue_leftneeds_glue_rightinnerrrr apply_changeHs"   zEditableFile.apply_changecCs8ddd}t|j|jrt|jdndd}||dS)Nr)r<r2r8)r>r?)lenr)rr>r?rrr full_rangeYs  zEditableFile.full_rangeN)__name__ __module__ __qualname__r MAX_WORKERSexecutorrr r"rrr! staticmethodlistintr6boolr1rrEpropertyrGrrrrrs     r)ztextDocument/didOpentextDocument/didChangeztextDocument/didSavec@seZdZdZdS)ShadowFilesystemErrorz Error in the shadow file system.N)rHrIrJ__doc__rrrrrSfsrSvirtual_documents_uricsHds tddgtttdfdd}|S)Nzfile:/z6Virtual documents URI has to start with "file:/", got Fclientc s|j|}|ddsdS|dtvrdSt|ddg}|dur)td|t|dg}|s8td ||s?dSstd krIdSzjddd t t jddd Wn(t t t fy}z|td kr||jd WYd}~dSd}~wwdt|}t|std |dt|} | IdHt|dg} | durd| ig} n|ddkrdSd|dvrtd|dd} t| dkr|jd| ddD]} | d| j} | j| dfi| q| IdH|S)zIntercept a message with document contents creating a shadow file for it. Only create the shadow file if the URI matches the virtual documents URI. Returns the path on filesystem where the content was stored. requires_documents_on_diskTNmethodparams textDocumentz#Could not get textDocument from: {}urizCould not get URI from: {}r*a\[lsp] initialization of shadow filesystem failed three times check if the path set by `LanguageServerManager.virtual_documents_dir` or `JP_LSP_VIRTUAL_DIR` is correct; if this is happening with a server for which you control (or wish to override) jupyter-lsp specification you can try switching `requires_documents_on_disk` off. The errors were: %szPath z* is not relative to shadow filesystem rootr;rRcontentChangesz0textDocument/didChange is missing contentChangesrzBLSP warning: up to one change supported for textDocument/didChangerange)language_serversget WRITE_ONErrSformat startswithrFr/rrOSErrorPermissionErrorr)appendlogwarnr r rr rGrEr")scopemessagelanguage_servermanager server_specdocumentr[er editable_filer;changeschange change_rangefailures initializedshadow_filesystemrUrrshadow_virtual_documentsvsx                z9setup_shadow_filesystem..shadow_virtual_documents)rcrSrr r )rUrxrrtrsetup_shadow_filesystemjs  \ryN)concurrent.futuresrpathlibrshutilrtypingrtornado.concurrentr tornado.genrrlr pathsr r typesr rKrrra ValueErrorrSrryrrrrs         H