parent module request

2 files changed, 22 insertions, 1 deletions

diff --git a/crates/server/src/main_loop/handlers.rs b/crates/server/src/main_loop/handlers.rs
index 675f69bec..16cc92464 100644
--- a/crates/server/src/main_loop/handlers.rs
+++ b/crates/server/src/main_loop/handlers.rs
@@ -3,7 +3,7 @@ use std::collections::HashMap;
 use languageserver_types::{
     Diagnostic, DiagnosticSeverity, Url, DocumentSymbol,
     Command, TextDocumentIdentifier, WorkspaceEdit,
-    SymbolInformation, Position,
+    SymbolInformation, Position, Location,
 };
 use libanalysis::{Query};
 use libeditor::{self, CursorPosition};
@@ -184,6 +184,23 @@ pub fn handle_goto_definition(
     Ok(Some(req::GotoDefinitionResponse::Array(res)))
 }
 
+pub fn handle_parent_module(
+    world: ServerWorld,
+    params: TextDocumentIdentifier,
+) -> Result<Vec<Location>> {
+    let file_id = params.try_conv_with(&world)?;
+    let mut res = Vec::new();
+    for (file_id, symbol) in world.analysis().parent_module(file_id) {
+        let line_index = world.analysis().file_line_index(file_id)?;
+        let location = to_location(
+            file_id, symbol.node_range,
+            &world, &line_index
+        )?;
+        res.push(location);
+    }
+    Ok(res)
+}
+
 pub fn handle_execute_command(
     world: ServerWorld,
     mut params: req::ExecuteCommandParams,
diff --git a/crates/server/src/main_loop/mod.rs b/crates/server/src/main_loop/mod.rs
index 752d6ddb2..9499b826c 100644
--- a/crates/server/src/main_loop/mod.rs
+++ b/crates/server/src/main_loop/mod.rs
@@ -26,6 +26,7 @@ use {
         handle_workspace_symbol,
         handle_goto_definition,
         handle_find_matching_brace,
+        handle_parent_module,
     },
 };
 
@@ -141,6 +142,9 @@ fn on_request(
     handle_request_on_threadpool::<req::GotoDefinition>(
         &mut req, pool, world, sender, handle_goto_definition,
     )?;
+    handle_request_on_threadpool::<req::ParentModule>(
+        &mut req, pool, world, sender, handle_parent_module,
+    )?;
     dispatch::handle_request::<req::ExecuteCommand, _>(&mut req, |params, resp| {
         io.send(RawMsg::Response(resp.into_response(Ok(None))?));