nordic-dev.net/rust
nordic-dev.net/rust
2
0
Fork 0
mirror of https://github.com/rust-lang/rust.git synced 2024-11-29 18:23:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix proc macro token mapping

Browse Source
This commit is contained in:
Jonas Schievink 2020-12-03 18:38:05 +01:00
parent d46fce88f5
commit 3e6ffa5124
2 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
crates/hir_def/src/lib.rs
View File

@ -521,7 +521,7 @@ impl AsMacroCall for AstIdWithPath<ast::MacroCall> {
error_sink: &mut dyn FnMut(mbe::ExpandError), error_sink: &mut dyn FnMut(mbe::ExpandError),
) -> Option<MacroCallId> { ) -> Option<MacroCallId> {
let def: MacroDefId = resolver(self.path.clone()).or_else(|| { let def: MacroDefId = resolver(self.path.clone()).or_else(|| {
error_sink(mbe::ExpandError::Other("could not resolve macro".into())); error_sink(mbe::ExpandError::Other(format!("could not resolve macro `{}`", self.path)));
None None
})?; })?;
@ -556,7 +556,7 @@ impl AsMacroCall for AstIdWithPath<ast::Item> {
error_sink: &mut dyn FnMut(mbe::ExpandError), error_sink: &mut dyn FnMut(mbe::ExpandError),
) -> Option<MacroCallId> { ) -> Option<MacroCallId> {
let def: MacroDefId = resolver(self.path.clone()).or_else(|| { let def: MacroDefId = resolver(self.path.clone()).or_else(|| {
error_sink(mbe::ExpandError::Other("could not resolve macro".into())); error_sink(mbe::ExpandError::Other(format!("could not resolve macro `{}`", self.path)));
None None
})?; })?;

22
crates/hir_expand/src/lib.rs
View File

@ -143,7 +143,11 @@ impl HirFileId {
let loc: MacroCallLoc = db.lookup_intern_macro(lazy_id); let loc: MacroCallLoc = db.lookup_intern_macro(lazy_id);
let arg_tt = loc.kind.arg(db)?; let arg_tt = loc.kind.arg(db)?;
let def_tt = loc.def.ast_id?.to_node(db).token_tree()?;
let def = loc.def.ast_id.and_then(|id| {
let def_tt = id.to_node(db).token_tree()?;
Some(InFile::new(id.file_id, def_tt))
});
let macro_def = db.macro_def(loc.def)?; let macro_def = db.macro_def(loc.def)?;
let (parse, exp_map) = db.parse_macro_expansion(macro_file).value?; let (parse, exp_map) = db.parse_macro_expansion(macro_file).value?;
@ -152,7 +156,7 @@ impl HirFileId {
Some(ExpansionInfo { Some(ExpansionInfo {
expanded: InFile::new(self, parse.syntax_node()), expanded: InFile::new(self, parse.syntax_node()),
arg: InFile::new(loc.kind.file_id(), arg_tt), arg: InFile::new(loc.kind.file_id(), arg_tt),
def: InFile::new(loc.def.ast_id?.file_id, def_tt), def,
macro_arg, macro_arg,
macro_def, macro_def,
exp_map, exp_map,
@ -311,7 +315,8 @@ pub struct EagerCallLoc {
pub struct ExpansionInfo { pub struct ExpansionInfo {
expanded: InFile<SyntaxNode>, expanded: InFile<SyntaxNode>,
arg: InFile<SyntaxNode>, arg: InFile<SyntaxNode>,
def: InFile<ast::TokenTree>, /// The `macro_rules!` arguments.
def: Option<InFile<ast::TokenTree>>,
macro_def: Arc<(db::TokenExpander, mbe::TokenMap)>, macro_def: Arc<(db::TokenExpander, mbe::TokenMap)>,
macro_arg: Arc<(tt::Subtree, mbe::TokenMap)>, macro_arg: Arc<(tt::Subtree, mbe::TokenMap)>,
@ -348,9 +353,14 @@ impl ExpansionInfo {
let (token_id, origin) = self.macro_def.0.map_id_up(token_id); let (token_id, origin) = self.macro_def.0.map_id_up(token_id);
let (token_map, tt) = match origin { let (token_map, tt) = match origin {
mbe::Origin::Call => (&self.macro_arg.1, self.arg.clone()), mbe::Origin::Call => (&self.macro_arg.1, self.arg.clone()),
mbe::Origin::Def => { mbe::Origin::Def => (
(&self.macro_def.1, self.def.as_ref().map(|tt| tt.syntax().clone())) &self.macro_def.1,
} self.def
.as_ref()
.expect("`Origin::Def` used with non-`macro_rules!` macro")
.as_ref()
.map(|tt| tt.syntax().clone()),
),
}; };
let range = token_map.range_by_token(token_id)?.by_kind(token.value.kind())?; let range = token_map.range_by_token(token_id)?.by_kind(token.value.kind())?;