Fetch the language servers

Author: SomeoneToIgnore

crates/language_tools/src/lsp_tool.rs

@@ -1,33 +1,75 @@
+use collections::HashSet;
+use editor::Editor;
+use gpui::{Entity, WeakEntity};
+use project::LspStore;
 use ui::{IconButtonShape, Tooltip, prelude::*};
-use workspace::StatusItemView;
+use workspace::{StatusItemView, Workspace};
 
-pub struct LspTool {}
+pub struct LspTool {
+    active_editor: Option<WeakEntity<Editor>>,
+    lsp_store: Entity<LspStore>,
+}
 
 impl LspTool {
-    pub fn new() -> Self {
-        Self {}
+    pub fn new(workspace: &Workspace, cx: &App) -> Self {
+        let lsp_store = workspace.project().read(cx).lsp_store();
+        Self {
+            active_editor: None,
+            lsp_store,
+        }
     }
 }
 
 impl StatusItemView for LspTool {
     fn set_active_pane_item(
         &mut self,
-        _active_pane_item: Option<&dyn workspace::ItemHandle>,
+        active_pane_item: Option<&dyn workspace::ItemHandle>,
         _window: &mut ui::Window,
-        _cx: &mut ui::Context<Self>,
+        cx: &mut ui::Context<Self>,
     ) {
-        // TODO kb see inline_completion_button.rs
+        self.active_editor = active_pane_item
+            .and_then(|item| item.downcast::<Editor>().map(|editor| editor.downgrade()));
+        cx.notify();
     }
 }
 
 impl Render for LspTool {
-    // TODO kb won't work for remote clients, disable for now
+    // TODO kb won't work for remote clients for now
     // TODO kb add a setting to remove this button out of the status bar
     fn render(
         &mut self,
         _window: &mut ui::Window,
         cx: &mut ui::Context<Self>,
     ) -> impl ui::IntoElement {
+        let Some(editor) = self
+            .active_editor
+            .as_ref()
+            .and_then(|editor| editor.upgrade())
+        else {
+            return div();
+        };
+
+        let buffers = editor.read(cx).buffer().read(cx).all_buffers();
+        let mut server_ids = HashSet::default();
+        let applicable_language_servers = self.lsp_store.update(cx, |lsp_store, cx| {
+            buffers
+                .iter()
+                .flat_map(|buffer| {
+                    buffer.update(cx, |buffer, cx| {
+                        lsp_store
+                            .language_servers_for_local_buffer(buffer, cx)
+                            .filter(|(_, server)| server_ids.insert(server.server_id()))
+                            .map(|(adapter, server)| (adapter.clone(), server.clone()))
+                            .collect::<Vec<_>>()
+                    })
+                })
+                .collect::<Vec<_>>()
+        });
+        dbg!(applicable_language_servers.len());
+        if applicable_language_servers.is_empty() {
+            return div();
+        }
+
         div().child(
             IconButton::new("zed-lsp-tool-button", IconName::Bolt)
                 .shape(IconButtonShape::Square)

crates/zed/src/zed.rs

@@ -237,7 +237,7 @@ pub fn initialize_workspace(
             cx.new(|cx| toolchain_selector::ActiveToolchain::new(workspace, window, cx));
         let vim_mode_indicator = cx.new(|cx| vim::ModeIndicator::new(window, cx));
         let image_info = cx.new(|_cx| ImageInfo::new(workspace));
-        let lsp_tool = cx.new(|_| LspTool::new());
+        let lsp_tool = cx.new(|cx| LspTool::new(workspace, cx));
 
         let cursor_position =
             cx.new(|_| go_to_line::cursor_position::CursorPosition::new(workspace));
@@ -246,8 +246,8 @@ pub fn initialize_workspace(
             status_bar.add_left_item(diagnostic_summary, window, cx);
             status_bar.add_left_item(activity_indicator, window, cx);
             status_bar.add_right_item(edit_prediction_button, window, cx);
-            status_bar.add_right_item(lsp_tool, window, cx);
             status_bar.add_right_item(active_buffer_language, window, cx);
+            status_bar.add_right_item(lsp_tool, window, cx);
             status_bar.add_right_item(active_toolchain_language, window, cx);
             status_bar.add_right_item(vim_mode_indicator, window, cx);
             status_bar.add_right_item(cursor_position, window, cx);