diff options
Diffstat (limited to 'crates/ra_hir_def/src/data.rs')
-rw-r--r-- | crates/ra_hir_def/src/data.rs | 38 |
1 files changed, 28 insertions, 10 deletions
diff --git a/crates/ra_hir_def/src/data.rs b/crates/ra_hir_def/src/data.rs index b0a3f1784..6a65633ae 100644 --- a/crates/ra_hir_def/src/data.rs +++ b/crates/ra_hir_def/src/data.rs | |||
@@ -226,21 +226,39 @@ fn collect_impl_items_in_macros( | |||
226 | let mut res = Vec::new(); | 226 | let mut res = Vec::new(); |
227 | 227 | ||
228 | for m in impl_block.value.syntax().children().filter_map(ast::MacroCall::cast) { | 228 | for m in impl_block.value.syntax().children().filter_map(ast::MacroCall::cast) { |
229 | if let Some((mark, items)) = expander.enter_expand(db, m) { | 229 | res.extend(collect_impl_items_in_macro(db, &mut expander, m, id)) |
230 | let items: InFile<ast::MacroItems> = expander.to_source(items); | ||
231 | expander.exit(db, mark); | ||
232 | res.extend(collect_impl_items( | ||
233 | db, | ||
234 | items.value.items().filter_map(|it| ImplItem::cast(it.syntax().clone())), | ||
235 | items.file_id, | ||
236 | id, | ||
237 | )); | ||
238 | } | ||
239 | } | 230 | } |
240 | 231 | ||
241 | res | 232 | res |
242 | } | 233 | } |
243 | 234 | ||
235 | fn collect_impl_items_in_macro( | ||
236 | db: &impl DefDatabase, | ||
237 | expander: &mut Expander, | ||
238 | m: ast::MacroCall, | ||
239 | id: ImplId, | ||
240 | ) -> Vec<AssocItemId> { | ||
241 | if let Some((mark, items)) = expander.enter_expand(db, m) { | ||
242 | let items: InFile<ast::MacroItems> = expander.to_source(items); | ||
243 | let mut res = collect_impl_items( | ||
244 | db, | ||
245 | items.value.items().filter_map(|it| ImplItem::cast(it.syntax().clone())), | ||
246 | items.file_id, | ||
247 | id, | ||
248 | ); | ||
249 | // Recursive collect macros | ||
250 | // Note that ast::ModuleItem do not include ast::MacroCall | ||
251 | // We cannot use ModuleItemOwner::items here | ||
252 | for it in items.value.syntax().children().filter_map(ast::MacroCall::cast) { | ||
253 | res.extend(collect_impl_items_in_macro(db, expander, it, id)) | ||
254 | } | ||
255 | expander.exit(db, mark); | ||
256 | res | ||
257 | } else { | ||
258 | Vec::new() | ||
259 | } | ||
260 | } | ||
261 | |||
244 | fn collect_impl_items( | 262 | fn collect_impl_items( |
245 | db: &impl DefDatabase, | 263 | db: &impl DefDatabase, |
246 | impl_items: impl Iterator<Item = ImplItem>, | 264 | impl_items: impl Iterator<Item = ImplItem>, |