Fix statusbar message pooling data from all editors

Author: krassowski

atest/04_Interface/Statusbar.robot

@@ -18,3 +18,20 @@ Statusbar Popup Opens
     Element Should Contain    ${POPOVER}    python
     Element Should Contain    ${POPOVER}    initialized
     [Teardown]    Clean Up After Working With File    Python.ipynb
+
+Status Changes Correctly Between Editors
+    Prepare File for Editing    Python    status    example.py
+    Wait Until Fully Initialized
+    Open File    example.plain
+    Wait Until Element Contains    ${STATUSBAR}    Initialized (additional servers needed)    timeout=60s
+    Capture Page Screenshot    01-both-open.png
+    Switch To Tab    example.py
+    Wait Until Fully Initialized
+    Switch To Tab    example.plain
+    Wait Until Element Contains    ${STATUSBAR}    Initialized (additional servers needed)    timeout=60s
+    [Teardown]    Clean Up After Working With File    example.plain
+
+*** Keywords ***
+Switch To Tab
+    [Arguments]    ${file}
+    Click Element    ${JLAB XP DOCK TAB}\[contains(., '${file}')]

atest/Keywords.robot

@@ -285,8 +285,12 @@ Prepare File for Editing
     [Arguments]    ${Language}    ${Screenshots}    ${file}
     Set Tags    language:${Language.lower()}
     Set Screenshot Directory    ${OUTPUT DIR}${/}screenshots${/}${Screenshots}${/}${Language.lower()}
-    Copy File    examples${/}${file}    ${OUTPUT DIR}${/}home${/}${file}
     Try to Close All Tabs
+    Open File    ${file}
+
+Open File
+    [Arguments]    ${file}
+    Copy File    examples${/}${file}    ${OUTPUT DIR}${/}home${/}${file}
     Open ${file} in ${MENU EDITOR}
     Capture Page Screenshot    00-opened.png
 

packages/jupyterlab-lsp/src/components/statusbar.tsx

@@ -322,8 +322,7 @@ const classByStatus: StatusIconClass = {
 const shortMessageByStatus: StatusMap = {
   waiting: 'Waiting...',
   initialized: 'Fully initialized',
-  initialized_but_some_missing:
-    'Initialized (additional servers can be installed)',
+  initialized_but_some_missing: 'Initialized (additional servers needed)',
   initializing: 'Partially initialized',
   connecting: 'Connecting...'
 };
@@ -436,7 +435,22 @@ export namespace LSPStatus {
     }
 
     get status(): IStatus {
-      const detected_documents = this._connection_manager.documents;
+      let detected_documents: Map<string, VirtualDocument>;
+
+      if (!this.adapter?.virtual_editor) {
+        detected_documents = new Map();
+      } else {
+        let main_document = this.adapter.virtual_editor.virtual_document;
+        const all_documents = this._connection_manager.documents;
+        // detected documents that are open in the current virtual editor
+        const detected_documents_set = collect_documents(main_document);
+        detected_documents = new Map(
+          [...all_documents].filter(([id, doc]) =>
+            detected_documents_set.has(doc)
+          )
+        );
+      }
+
       let connected_documents = new Set<VirtualDocument>();
       let initialized_documents = new Set<VirtualDocument>();
       let absent_documents = new Set<VirtualDocument>();