1 files changed, 104 insertions, 98 deletions

diff --git a/crates/ra_ide_api/src/syntax_tree.rs b/crates/ra_ide_api/src/syntax_tree.rs
index 276f8a8c8..c7288220c 100644
--- a/crates/ra_ide_api/src/syntax_tree.rs
+++ b/crates/ra_ide_api/src/syntax_tree.rs
@@ -1,8 +1,9 @@
 use ra_db::SourceDatabase;

 use crate::db::RootDatabase;

 use ra_syntax::{

-    SourceFile, SyntaxNode, TextRange, AstNode,

-    algo::{self, visit::{visitor, Visitor}}, ast::{self, AstToken}

+    SourceFile, TextRange, AstNode, SyntaxToken, SyntaxElement,

+    algo,

+    SyntaxKind::{STRING, RAW_STRING},

 };

 

 pub use ra_db::FileId;

@@ -14,11 +15,15 @@ pub(crate) fn syntax_tree(
 ) -> String {

     if let Some(text_range) = text_range {

         let file = db.parse(file_id);

-        let node = algo::find_covering_node(file.syntax(), text_range);

-

-        if let Some(tree) = syntax_tree_for_string(node, text_range) {

-            return tree;

-        }

+        let node = match algo::find_covering_element(file.syntax(), text_range) {

+            SyntaxElement::Node(node) => node,

+            SyntaxElement::Token(token) => {

+                if let Some(tree) = syntax_tree_for_string(token, text_range) {

+                    return tree;

+                }

+                token.parent()

+            }

+        };

 

         node.debug_dump()

     } else {

@@ -28,19 +33,19 @@ pub(crate) fn syntax_tree(
 

 /// Attempts parsing the selected contents of a string literal

 /// as rust syntax and returns its syntax tree

-fn syntax_tree_for_string(node: &SyntaxNode, text_range: TextRange) -> Option<String> {

+fn syntax_tree_for_string(token: SyntaxToken, text_range: TextRange) -> Option<String> {

     // When the range is inside a string

     // we'll attempt parsing it as rust syntax

     // to provide the syntax tree of the contents of the string

-    visitor()

-        .visit(|node: &ast::String| syntax_tree_for_token(node, text_range))

-        .visit(|node: &ast::RawString| syntax_tree_for_token(node, text_range))

-        .accept(node)?

+    match token.kind() {

+        STRING | RAW_STRING => syntax_tree_for_token(token, text_range),

+        _ => None,

+    }

 }

 

-fn syntax_tree_for_token<T: AstToken>(node: &T, text_range: TextRange) -> Option<String> {

+fn syntax_tree_for_token(node: SyntaxToken, text_range: TextRange) -> Option<String> {

     // Range of the full node

-    let node_range = node.syntax().range();

+    let node_range = node.range();

     let text = node.text().to_string();

 

     // We start at some point inside the node

@@ -88,6 +93,8 @@ fn syntax_tree_for_token<T: AstToken>(node: &T, text_range: TextRange) -> Option
 

 #[cfg(test)]

 mod tests {

+    use test_utils::assert_eq_text;

+

     use crate::mock_analysis::{single_file, single_file_with_range};

 

     #[test]

@@ -96,23 +103,23 @@ mod tests {
         let (analysis, file_id) = single_file(r#"fn foo() {}"#);

         let syn = analysis.syntax_tree(file_id, None);

 

-        assert_eq!(

+        assert_eq_text!(

             syn.trim(),

             r#"

 SOURCE_FILE@[0; 11)

   FN_DEF@[0; 11)

-    FN_KW@[0; 2)

-    WHITESPACE@[2; 3)

+    FN_KW@[0; 2) "fn"

+    WHITESPACE@[2; 3) " "

     NAME@[3; 6)

       IDENT@[3; 6) "foo"

     PARAM_LIST@[6; 8)

-      L_PAREN@[6; 7)

-      R_PAREN@[7; 8)

-    WHITESPACE@[8; 9)

+      L_PAREN@[6; 7) "("

+      R_PAREN@[7; 8) ")"

+    WHITESPACE@[8; 9) " "

     BLOCK@[9; 11)

-      L_CURLY@[9; 10)

-      R_CURLY@[10; 11)

-    "#

+      L_CURLY@[9; 10) "{"

+      R_CURLY@[10; 11) "}"

+"#

             .trim()

         );

 

@@ -128,40 +135,40 @@ fn test() {
         );

         let syn = analysis.syntax_tree(file_id, None);

 

-        assert_eq!(

+        assert_eq_text!(

             syn.trim(),

             r#"

 SOURCE_FILE@[0; 60)

   FN_DEF@[0; 60)

-    FN_KW@[0; 2)

-    WHITESPACE@[2; 3)

+    FN_KW@[0; 2) "fn"

+    WHITESPACE@[2; 3) " "

     NAME@[3; 7)

       IDENT@[3; 7) "test"

     PARAM_LIST@[7; 9)

-      L_PAREN@[7; 8)

-      R_PAREN@[8; 9)

-    WHITESPACE@[9; 10)

+      L_PAREN@[7; 8) "("

+      R_PAREN@[8; 9) ")"

+    WHITESPACE@[9; 10) " "

     BLOCK@[10; 60)

-      L_CURLY@[10; 11)

-      WHITESPACE@[11; 16)

+      L_CURLY@[10; 11) "{"

+      WHITESPACE@[11; 16) "\n    "

       EXPR_STMT@[16; 58)

         MACRO_CALL@[16; 57)

           PATH@[16; 22)

             PATH_SEGMENT@[16; 22)

               NAME_REF@[16; 22)

                 IDENT@[16; 22) "assert"

-          EXCL@[22; 23)

+          EXCL@[22; 23) "!"

           TOKEN_TREE@[23; 57)

-            L_PAREN@[23; 24)

-            STRING@[24; 52)

-            COMMA@[52; 53)

-            WHITESPACE@[53; 54)

-            STRING@[54; 56)

-            R_PAREN@[56; 57)

-        SEMI@[57; 58)

-      WHITESPACE@[58; 59)

-      R_CURLY@[59; 60)

-    "#

+            L_PAREN@[23; 24) "("

+            STRING@[24; 52) "\"\n    fn foo() {\n     ..."

+            COMMA@[52; 53) ","

+            WHITESPACE@[53; 54) " "

+            STRING@[54; 56) "\"\""

+            R_PAREN@[56; 57) ")"

+        SEMI@[57; 58) ";"

+      WHITESPACE@[58; 59) "\n"

+      R_CURLY@[59; 60) "}"

+"#

             .trim()

         );

     }

@@ -171,22 +178,22 @@ SOURCE_FILE@[0; 60)
         let (analysis, range) = single_file_with_range(r#"<|>fn foo() {}<|>"#.trim());

         let syn = analysis.syntax_tree(range.file_id, Some(range.range));

 

-        assert_eq!(

+        assert_eq_text!(

             syn.trim(),

             r#"

 FN_DEF@[0; 11)

-  FN_KW@[0; 2)

-  WHITESPACE@[2; 3)

+  FN_KW@[0; 2) "fn"

+  WHITESPACE@[2; 3) " "

   NAME@[3; 6)

     IDENT@[3; 6) "foo"

   PARAM_LIST@[6; 8)

-    L_PAREN@[6; 7)

-    R_PAREN@[7; 8)

-  WHITESPACE@[8; 9)

+    L_PAREN@[6; 7) "("

+    R_PAREN@[7; 8) ")"

+  WHITESPACE@[8; 9) " "

   BLOCK@[9; 11)

-    L_CURLY@[9; 10)

-    R_CURLY@[10; 11)

-    "#

+    L_CURLY@[9; 10) "{"

+    R_CURLY@[10; 11) "}"

+"#

             .trim()

         );

 

@@ -201,7 +208,7 @@ FN_DEF@[0; 11)
         );

         let syn = analysis.syntax_tree(range.file_id, Some(range.range));

 

-        assert_eq!(

+        assert_eq_text!(

             syn.trim(),

             r#"

 EXPR_STMT@[16; 58)

@@ -210,16 +217,16 @@ EXPR_STMT@[16; 58)
       PATH_SEGMENT@[16; 22)

         NAME_REF@[16; 22)

           IDENT@[16; 22) "assert"

-    EXCL@[22; 23)

+    EXCL@[22; 23) "!"

     TOKEN_TREE@[23; 57)

-      L_PAREN@[23; 24)

-      STRING@[24; 52)

-      COMMA@[52; 53)

-      WHITESPACE@[53; 54)

-      STRING@[54; 56)

-      R_PAREN@[56; 57)

-  SEMI@[57; 58)

-    "#

+      L_PAREN@[23; 24) "("

+      STRING@[24; 52) "\"\n    fn foo() {\n     ..."

+      COMMA@[52; 53) ","

+      WHITESPACE@[53; 54) " "

+      STRING@[54; 56) "\"\""

+      R_PAREN@[56; 57) ")"

+  SEMI@[57; 58) ";"

+"#

             .trim()

         );

     }

@@ -238,23 +245,23 @@ fn bar() {
             .trim(),

         );

         let syn = analysis.syntax_tree(range.file_id, Some(range.range));

-        assert_eq!(

+        assert_eq_text!(

             syn.trim(),

             r#"

 SOURCE_FILE@[0; 12)

   FN_DEF@[0; 12)

-    FN_KW@[0; 2)

-    WHITESPACE@[2; 3)

+    FN_KW@[0; 2) "fn"

+    WHITESPACE@[2; 3) " "

     NAME@[3; 6)

       IDENT@[3; 6) "foo"

     PARAM_LIST@[6; 8)

-      L_PAREN@[6; 7)

-      R_PAREN@[7; 8)

-    WHITESPACE@[8; 9)

+      L_PAREN@[6; 7) "("

+      R_PAREN@[7; 8) ")"

+    WHITESPACE@[8; 9) " "

     BLOCK@[9; 12)

-      L_CURLY@[9; 10)

-      WHITESPACE@[10; 11)

-      R_CURLY@[11; 12)

+      L_CURLY@[9; 10) "{"

+      WHITESPACE@[10; 11) "\n"

+      R_CURLY@[11; 12) "}"

 "#

             .trim()

         );

@@ -272,23 +279,23 @@ fn bar() {
                 .trim(),

         );

         let syn = analysis.syntax_tree(range.file_id, Some(range.range));

-        assert_eq!(

+        assert_eq_text!(

             syn.trim(),

             r#"

 SOURCE_FILE@[0; 12)

   FN_DEF@[0; 12)

-    FN_KW@[0; 2)

-    WHITESPACE@[2; 3)

+    FN_KW@[0; 2) "fn"

+    WHITESPACE@[2; 3) " "

     NAME@[3; 6)

       IDENT@[3; 6) "foo"

     PARAM_LIST@[6; 8)

-      L_PAREN@[6; 7)

-      R_PAREN@[7; 8)

-    WHITESPACE@[8; 9)

+      L_PAREN@[6; 7) "("

+      R_PAREN@[7; 8) ")"

+    WHITESPACE@[8; 9) " "

     BLOCK@[9; 12)

-      L_CURLY@[9; 10)

-      WHITESPACE@[10; 11)

-      R_CURLY@[11; 12)

+      L_CURLY@[9; 10) "{"

+      WHITESPACE@[10; 11) "\n"

+      R_CURLY@[11; 12) "}"

 "#

             .trim()

         );

@@ -305,38 +312,37 @@ fn bar() {
                 .trim(),

         );

         let syn = analysis.syntax_tree(range.file_id, Some(range.range));

-        assert_eq!(

+        assert_eq_text!(

             syn.trim(),

             r#"

 SOURCE_FILE@[0; 25)

   FN_DEF@[0; 12)

-    FN_KW@[0; 2)

-    WHITESPACE@[2; 3)

+    FN_KW@[0; 2) "fn"

+    WHITESPACE@[2; 3) " "

     NAME@[3; 6)

       IDENT@[3; 6) "foo"

     PARAM_LIST@[6; 8)

-      L_PAREN@[6; 7)

-      R_PAREN@[7; 8)

-    WHITESPACE@[8; 9)

+      L_PAREN@[6; 7) "("

+      R_PAREN@[7; 8) ")"

+    WHITESPACE@[8; 9) " "

     BLOCK@[9; 12)

-      L_CURLY@[9; 10)

-      WHITESPACE@[10; 11)

-      R_CURLY@[11; 12)

-  WHITESPACE@[12; 13)

+      L_CURLY@[9; 10) "{"

+      WHITESPACE@[10; 11) "\n"

+      R_CURLY@[11; 12) "}"

+  WHITESPACE@[12; 13) "\n"

   FN_DEF@[13; 25)

-    FN_KW@[13; 15)

-    WHITESPACE@[15; 16)

+    FN_KW@[13; 15) "fn"

+    WHITESPACE@[15; 16) " "

     NAME@[16; 19)

       IDENT@[16; 19) "bar"

     PARAM_LIST@[19; 21)

-      L_PAREN@[19; 20)

-      R_PAREN@[20; 21)

-    WHITESPACE@[21; 22)

+      L_PAREN@[19; 20) "("

+      R_PAREN@[20; 21) ")"

+    WHITESPACE@[21; 22) " "

     BLOCK@[22; 25)

-      L_CURLY@[22; 23)

-      WHITESPACE@[23; 24)

-      R_CURLY@[24; 25)

-

+      L_CURLY@[22; 23) "{"

+      WHITESPACE@[23; 24) "\n"

+      R_CURLY@[24; 25) "}"

 "#

             .trim()

         );