Add macro_expansion_info in hir_expand

1 files changed, 34 insertions, 1 deletions

diff --git a/crates/ra_hir_expand/src/lib.rs b/crates/ra_hir_expand/src/lib.rs
index dd07a16b4..194020b45 100644
--- a/crates/ra_hir_expand/src/lib.rs
+++ b/crates/ra_hir_expand/src/lib.rs
@@ -16,7 +16,7 @@ use std::hash::{Hash, Hasher};
 use ra_db::{salsa, CrateId, FileId};
 use ra_syntax::{
     ast::{self, AstNode},
-    SyntaxNode,
+    SyntaxNode, TextRange,
 };
 
 use crate::ast_id_map::FileAstId;
@@ -112,6 +112,39 @@ impl MacroCallId {
     }
 }
 
+#[derive(Debug, Clone, PartialEq, Eq)]
+/// ExpansionInfo mainly describle how to map text range between src and expaned macro
+pub struct ExpansionInfo {
+    pub arg_map: Vec<(TextRange, TextRange)>,
+    pub def_map: Vec<(TextRange, TextRange)>,
+}
+
+impl ExpansionInfo {
+    pub fn find_range(
+        &self,
+        from: TextRange,
+        (arg_file_id, def_file_id): (HirFileId, HirFileId),
+    ) -> Option<(HirFileId, TextRange)> {
+        for (src, dest) in &self.arg_map {
+            dbg!((src, *dest, "arg_map"));
+            if src.is_subrange(&from) {
+                dbg!((arg_file_id, *dest));
+                return Some((arg_file_id, *dest));
+            }
+        }
+
+        for (src, dest) in &self.def_map {
+            dbg!((src, *dest, "def_map"));
+            if src.is_subrange(&from) {
+                dbg!((arg_file_id, *dest));
+                return Some((def_file_id, *dest));
+            }
+        }
+
+        None
+    }
+}
+
 /// `AstId` points to an AST node in any file.
 ///
 /// It is stable across reparses, and can be used as salsa key/value.