1 files changed, 44 insertions, 55 deletions

diff --git a/crates/ra_ide_api/src/display/navigation_target.rs b/crates/ra_ide_api/src/display/navigation_target.rs
index f7ad08515..b30ef8e05 100644
--- a/crates/ra_ide_api/src/display/navigation_target.rs
+++ b/crates/ra_ide_api/src/display/navigation_target.rs
@@ -1,16 +1,17 @@
 //! FIXME: write short doc here
 
-use hir::{AssocItem, Either, FieldSource, HasSource, ModuleSource};
+use hir::{AssocItem, Either, FieldSource, HasSource, ModuleSource, Source};
 use ra_db::{FileId, SourceDatabase};
 use ra_syntax::{
     ast::{self, DocCommentsOwner, NameOwner},
     match_ast, AstNode, SmolStr,
     SyntaxKind::{self, BIND_PAT},
-    SyntaxNode, TextRange,
+    TextRange,
 };
 
+use crate::{db::RootDatabase, expand::original_range, FileSymbol};
+
 use super::short_label::ShortLabel;
-use crate::{db::RootDatabase, FileSymbol};
 
 /// `NavigationTarget` represents and element in the editor's UI which you can
 /// click on to navigate to a particular piece of code.
@@ -79,13 +80,13 @@ impl NavigationTarget {
     pub(crate) fn from_module_to_decl(db: &RootDatabase, module: hir::Module) -> NavigationTarget {
         let name = module.name(db).map(|it| it.to_string().into()).unwrap_or_default();
         if let Some(src) = module.declaration_source(db) {
-            let (file_id, text_range) = find_range_from_node(db, src.file_id, src.ast.syntax());
+            let frange = original_range(db, src.as_ref().map(|it| it.syntax()));
             return NavigationTarget::from_syntax(
-                file_id,
+                frange.file_id,
                 name,
                 None,
-                text_range,
-                src.ast.syntax(),
+                frange.range,
+                src.ast.syntax().kind(),
                 src.ast.doc_comment_text(),
                 src.ast.short_label(),
             );
@@ -140,22 +141,22 @@ impl NavigationTarget {
     /// Allows `NavigationTarget` to be created from a `NameOwner`
     pub(crate) fn from_named(
         db: &RootDatabase,
-        file_id: hir::HirFileId,
-        node: &impl ast::NameOwner,
+        node: Source<&dyn ast::NameOwner>,
         docs: Option<String>,
         description: Option<String>,
     ) -> NavigationTarget {
         //FIXME: use `_` instead of empty string
-        let name = node.name().map(|it| it.text().clone()).unwrap_or_default();
-        let focus_range = node.name().map(|it| find_range_from_node(db, file_id, it.syntax()).1);
-        let (file_id, full_range) = find_range_from_node(db, file_id, node.syntax());
+        let name = node.ast.name().map(|it| it.text().clone()).unwrap_or_default();
+        let focus_range =
+            node.ast.name().map(|it| original_range(db, node.with_ast(it.syntax())).range);
+        let frange = original_range(db, node.map(|it| it.syntax()));
 
         NavigationTarget::from_syntax(
-            file_id,
+            frange.file_id,
             name,
             focus_range,
-            full_range,
-            node.syntax(),
+            frange.range,
+            node.ast.syntax().kind(),
             docs,
             description,
         )
@@ -166,14 +167,14 @@ impl NavigationTarget {
         name: SmolStr,
         focus_range: Option<TextRange>,
         full_range: TextRange,
-        node: &SyntaxNode,
+        kind: SyntaxKind,
         docs: Option<String>,
         description: Option<String>,
     ) -> NavigationTarget {
         NavigationTarget {
             file_id,
             name,
-            kind: node.kind(),
+            kind,
             full_range,
             focus_range,
             container_name: None,
@@ -218,8 +219,7 @@ where
         let src = self.source(db);
         NavigationTarget::from_named(
             db,
-            src.file_id,
-            &src.ast,
+            src.as_ref().map(|it| it as &dyn ast::NameOwner),
             src.ast.doc_comment_text(),
             src.ast.short_label(),
         )
@@ -230,29 +230,29 @@ impl ToNav for hir::Module {
     fn to_nav(&self, db: &RootDatabase) -> NavigationTarget {
         let src = self.definition_source(db);
         let name = self.name(db).map(|it| it.to_string().into()).unwrap_or_default();
-        match src.ast {
+        match &src.ast {
             ModuleSource::SourceFile(node) => {
-                let (file_id, text_range) = find_range_from_node(db, src.file_id, node.syntax());
+                let frange = original_range(db, src.with_ast(node.syntax()));
 
                 NavigationTarget::from_syntax(
-                    file_id,
+                    frange.file_id,
                     name,
                     None,
-                    text_range,
-                    node.syntax(),
+                    frange.range,
+                    node.syntax().kind(),
                     None,
                     None,
                 )
             }
             ModuleSource::Module(node) => {
-                let (file_id, text_range) = find_range_from_node(db, src.file_id, node.syntax());
+                let frange = original_range(db, src.with_ast(node.syntax()));
 
                 NavigationTarget::from_syntax(
-                    file_id,
+                    frange.file_id,
                     name,
                     None,
-                    text_range,
-                    node.syntax(),
+                    frange.range,
+                    node.syntax().kind(),
                     node.doc_comment_text(),
                     node.short_label(),
                 )
@@ -264,14 +264,14 @@ impl ToNav for hir::Module {
 impl ToNav for hir::ImplBlock {
     fn to_nav(&self, db: &RootDatabase) -> NavigationTarget {
         let src = self.source(db);
-        let (file_id, text_range) = find_range_from_node(db, src.file_id, src.ast.syntax());
+        let frange = original_range(db, src.as_ref().map(|it| it.syntax()));
 
         NavigationTarget::from_syntax(
-            file_id,
+            frange.file_id,
             "impl".into(),
             None,
-            text_range,
-            src.ast.syntax(),
+            frange.range,
+            src.ast.syntax().kind(),
             None,
             None,
         )
@@ -282,22 +282,21 @@ impl ToNav for hir::StructField {
     fn to_nav(&self, db: &RootDatabase) -> NavigationTarget {
         let src = self.source(db);
 
-        match src.ast {
+        match &src.ast {
             FieldSource::Named(it) => NavigationTarget::from_named(
                 db,
-                src.file_id,
-                &it,
+                src.with_ast(it),
                 it.doc_comment_text(),
                 it.short_label(),
             ),
             FieldSource::Pos(it) => {
-                let (file_id, text_range) = find_range_from_node(db, src.file_id, it.syntax());
+                let frange = original_range(db, src.with_ast(it.syntax()));
                 NavigationTarget::from_syntax(
-                    file_id,
+                    frange.file_id,
                     "".into(),
                     None,
-                    text_range,
-                    it.syntax(),
+                    frange.range,
+                    it.syntax().kind(),
                     None,
                     None,
                 )
@@ -310,7 +309,12 @@ impl ToNav for hir::MacroDef {
     fn to_nav(&self, db: &RootDatabase) -> NavigationTarget {
         let src = self.source(db);
         log::debug!("nav target {:#?}", src.ast.syntax());
-        NavigationTarget::from_named(db, src.file_id, &src.ast, src.ast.doc_comment_text(), None)
+        NavigationTarget::from_named(
+            db,
+            src.as_ref().map(|it| it as &dyn ast::NameOwner),
+            src.ast.doc_comment_text(),
+            None,
+        )
     }
 }
 
@@ -360,21 +364,6 @@ impl ToNav for hir::Local {
     }
 }
 
-fn find_range_from_node(
-    db: &RootDatabase,
-    src: hir::HirFileId,
-    node: &SyntaxNode,
-) -> (FileId, TextRange) {
-    let text_range = node.text_range();
-    let (file_id, text_range) = src
-        .expansion_info(db)
-        .and_then(|expansion_info| expansion_info.find_range(text_range))
-        .unwrap_or((src, text_range));
-
-    // FIXME: handle recursive macro generated macro
-    (file_id.original_file(db), text_range)
-}
-
 pub(crate) fn docs_from_symbol(db: &RootDatabase, symbol: &FileSymbol) -> Option<String> {
     let parse = db.parse(symbol.file_id);
     let node = symbol.ptr.to_node(parse.tree().syntax());