Add support for include_str

Author: lnicola

crates/ra_hir_expand/src/builtin_macro.rs

@@ -1,4 +1,6 @@
 //! Builtin macro
+use std::ops::Deref;
+
 use crate::{
     db::AstDatabase, name, quote, AstId, CrateId, EagerMacroId, LazyMacroId, MacroCallId,
     MacroDefId, MacroDefKind, TextSize,
@@ -99,6 +101,7 @@ register_builtin! {
     EAGER:
     (concat, Concat) => concat_expand,
     (include, Include) => include_expand,
+    (include_str, IncludeStr) => include_str_expand,
     (env, Env) => env_expand,
     (option_env, OptionEnv) => option_env_expand
 }
@@ -331,6 +334,25 @@ fn include_expand(
     Ok((res, FragmentKind::Items))
 }
 
+fn include_str_expand(
+    db: &dyn AstDatabase,
+    arg_id: EagerMacroId,
+    tt: &tt::Subtree,
+) -> Result<(tt::Subtree, FragmentKind), mbe::ExpandError> {
+    dbg!(&tt);
+    let path = parse_string(tt)?;
+    dbg!(&path);
+    let file_id =
+        relative_file(db, arg_id.into(), &path).ok_or_else(|| mbe::ExpandError::ConversionError)?;
+    dbg!(&file_id);
+    let text = db.file_text(file_id);
+    dbg!(&text);
+    let text = text.deref();
+    // let text = "";
+
+    Ok((quote!(#text), FragmentKind::Expr))
+}
+
 fn get_env_inner(db: &dyn AstDatabase, arg_id: EagerMacroId, key: &str) -> Option<String> {
     let krate = db.lookup_intern_eager_expansion(arg_id).krate;
     db.crate_graph()[krate].env.get(key)

crates/ra_hir_expand/src/name.rs

@@ -191,6 +191,7 @@ pub mod known {
         stringify,
         concat,
         include,
+        include_str,
         format_args,
         format_args_nl,
         env,

crates/rust-analyzer/tests/heavy_tests/main.rs

@@ -494,6 +494,7 @@ fn main() {
 }
 //- /src/main.rs
 #[rustc_builtin_macro] macro_rules! include {}
+#[rustc_builtin_macro] macro_rules! include_str {}
 #[rustc_builtin_macro] macro_rules! concat {}
 #[rustc_builtin_macro] macro_rules! env {}
 
@@ -512,6 +513,7 @@ fn main() {
     let va = A;
     let vb = B;
     let should_be_str = message();
+    let another_str = include_str!("main.rs");
 }
 "###,
     )
@@ -523,32 +525,40 @@ fn main() {
     let res = server.send_request::<HoverRequest>(HoverParams {
         text_document_position_params: TextDocumentPositionParams::new(
             server.doc_id("src/main.rs"),
-            Position::new(18, 10),
+            Position::new(19, 10),
         ),
         work_done_progress_params: Default::default(),
     });
     assert!(res.to_string().contains("&str"));
+    let res = server.send_request::<HoverRequest>(HoverParams {
+        text_document_position_params: TextDocumentPositionParams::new(
+            server.doc_id("src/main.rs"),
+            Position::new(20, 10),
+        ),
+        work_done_progress_params: Default::default(),
+    });
+    assert!(dbg!(res.to_string()).contains("&str"));
     server.request::<GotoTypeDefinition>(
         GotoDefinitionParams {
             text_document_position_params: TextDocumentPositionParams::new(
                 server.doc_id("src/main.rs"),
-                Position::new(16, 9),
+                Position::new(17, 9),
             ),
             work_done_progress_params: Default::default(),
             partial_result_params: Default::default(),
         },
         json!([{
             "originSelectionRange": {
-                "end": { "character": 10, "line": 16 },
-                "start": { "character": 8, "line": 16 }
+                "end": { "character": 10, "line": 17 },
+                "start": { "character": 8, "line": 17 }
             },
             "targetRange": {
-                "end": { "character": 9, "line": 7 },
-                "start": { "character": 0, "line": 6 }
+                "end": { "character": 9, "line": 8 },
+                "start": { "character": 0, "line": 7 }
             },
             "targetSelectionRange": {
-                "end": { "character": 8, "line": 7 },
-                "start": { "character": 7, "line": 7 }
+                "end": { "character": 8, "line": 8 },
+                "start": { "character": 7, "line": 8 }
             },
             "targetUri": "file:///[..]src/main.rs"
         }]),
@@ -557,23 +567,23 @@ fn main() {
         GotoDefinitionParams {
             text_document_position_params: TextDocumentPositionParams::new(
                 server.doc_id("src/main.rs"),
-                Position::new(17, 9),
+                Position::new(18, 9),
             ),
             work_done_progress_params: Default::default(),
             partial_result_params: Default::default(),
         },
         json!([{
             "originSelectionRange": {
-                "end": { "character": 10, "line": 17 },
-                "start": { "character": 8, "line": 17 }
+                "end": { "character": 10, "line": 18 },
+                "start": { "character": 8, "line": 18 }
             },
             "targetRange": {
-                "end": { "character": 9, "line": 11 },
-                "start": { "character": 0, "line":10 }
+                "end": { "character": 9, "line": 12 },
+                "start": { "character": 0, "line":11 }
             },
             "targetSelectionRange": {
-                "end": { "character": 8, "line": 11 },
-                "start": { "character": 7, "line": 11 }
+                "end": { "character": 8, "line": 12 },
+                "start": { "character": 7, "line": 12 }
             },
             "targetUri": "file:///[..]src/main.rs"
         }]),