From 8ae56fa6d0e8a03d6ad75919d6be953f5fc27083 Mon Sep 17 00:00:00 2001
From: Aleksey Kladov <aleksey.kladov@gmail.com>
Date: Mon, 13 Aug 2018 16:35:17 +0300
Subject: Stupid goto definition

---
 crates/server/src/main_loop/handlers.rs | 27 +++++++++++++++++++--------
 crates/server/src/main_loop/mod.rs      |  4 ++++
 2 files changed, 23 insertions(+), 8 deletions(-)

(limited to 'crates/server/src/main_loop')

diff --git a/crates/server/src/main_loop/handlers.rs b/crates/server/src/main_loop/handlers.rs
index f51909280..e9dc78420 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, Location,
+    SymbolInformation,
 };
 use libanalysis::{World, Query};
 use libeditor;
@@ -13,7 +13,7 @@ use serde_json::{to_value, from_value};
 use ::{
     req::{self, Decoration}, Result,
     util::FilePath,
-    conv::{Conv, ConvWith, MapConvWith},
+    conv::{Conv, ConvWith, TryConvWith, MapConvWith},
 };
 
 pub fn handle_syntax_tree(
@@ -115,15 +115,10 @@ pub fn handle_workspace_symbol(
 
     for (path, symbol) in world.world_symbols(query).take(128) {
         let line_index = world.file_line_index(path)?;
-
         let info = SymbolInformation {
             name: symbol.name.to_string(),
             kind: symbol.kind.conv(),
-            location: Location::new(
-                Url::from_file_path(path)
-                    .map_err(|()| format_err!("invalid url"))?,
-                symbol.node_range.conv_with(&line_index),
-            ),
+            location: (path, symbol.node_range).try_conv_with(&line_index)?,
             container_name: None,
         };
         acc.push(info);
@@ -132,6 +127,22 @@ pub fn handle_workspace_symbol(
     Ok(Some(acc))
 }
 
+pub fn handle_goto_definition(
+    world: World,
+    params: req::TextDocumentPositionParams,
+) -> Result<Option<req::GotoDefinitionResponse>> {
+    let path = params.text_document.file_path()?;
+    let line_index = world.file_line_index(&path)?;
+    let offset = params.position.conv_with(&line_index);
+    let mut res = Vec::new();
+    for (path, symbol) in world.approximately_resolve_symbol(&path, offset)? {
+        let line_index = world.file_line_index(path)?;
+        let location = (path, symbol.node_range).try_conv_with(&line_index)?;
+        res.push(location)
+    }
+    Ok(Some(req::GotoDefinitionResponse::Array(res)))
+}
+
 pub fn handle_execute_command(
     world: World,
     mut params: req::ExecuteCommandParams,
diff --git a/crates/server/src/main_loop/mod.rs b/crates/server/src/main_loop/mod.rs
index e8b24355c..bc898c17b 100644
--- a/crates/server/src/main_loop/mod.rs
+++ b/crates/server/src/main_loop/mod.rs
@@ -26,6 +26,7 @@ use {
         handle_code_action,
         handle_execute_command,
         handle_workspace_symbol,
+        handle_goto_definition,
     },
 };
 
@@ -152,6 +153,9 @@ fn on_request(
     handle_request_on_threadpool::<req::WorkspaceSymbol>(
         &mut req, pool, world, sender, handle_workspace_symbol,
     )?;
+    handle_request_on_threadpool::<req::GotoDefinition>(
+        &mut req, pool, world, sender, handle_goto_definition,
+    )?;
     dispatch::handle_request::<req::ExecuteCommand, _>(&mut req, |params, resp| {
         io.send(RawMsg::Response(resp.into_response(Ok(None))?));
 
-- 
cgit v1.2.3