diff options
-rw-r--r-- | crates/ra_hir_def/src/data.rs | 46 | ||||
-rw-r--r-- | crates/ra_hir_ty/src/tests/macros.rs | 23 |
2 files changed, 59 insertions, 10 deletions
diff --git a/crates/ra_hir_def/src/data.rs b/crates/ra_hir_def/src/data.rs index b0a3f1784..1aa9a9b7d 100644 --- a/crates/ra_hir_def/src/data.rs +++ b/crates/ra_hir_def/src/data.rs | |||
@@ -225,22 +225,48 @@ fn collect_impl_items_in_macros( | |||
225 | let mut expander = Expander::new(db, impl_block.file_id, module_id); | 225 | let mut expander = Expander::new(db, impl_block.file_id, module_id); |
226 | let mut res = Vec::new(); | 226 | let mut res = Vec::new(); |
227 | 227 | ||
228 | // We set a limit to protect against infinite recursion | ||
229 | let limit = 100; | ||
230 | |||
228 | for m in impl_block.value.syntax().children().filter_map(ast::MacroCall::cast) { | 231 | for m in impl_block.value.syntax().children().filter_map(ast::MacroCall::cast) { |
229 | if let Some((mark, items)) = expander.enter_expand(db, m) { | 232 | res.extend(collect_impl_items_in_macro(db, &mut expander, m, id, limit)) |
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 | } | 233 | } |
240 | 234 | ||
241 | res | 235 | res |
242 | } | 236 | } |
243 | 237 | ||
238 | fn collect_impl_items_in_macro( | ||
239 | db: &impl DefDatabase, | ||
240 | expander: &mut Expander, | ||
241 | m: ast::MacroCall, | ||
242 | id: ImplId, | ||
243 | limit: usize, | ||
244 | ) -> Vec<AssocItemId> { | ||
245 | if limit == 0 { | ||
246 | return Vec::new(); | ||
247 | } | ||
248 | |||
249 | if let Some((mark, items)) = expander.enter_expand(db, m) { | ||
250 | let items: InFile<ast::MacroItems> = expander.to_source(items); | ||
251 | let mut res = collect_impl_items( | ||
252 | db, | ||
253 | items.value.items().filter_map(|it| ImplItem::cast(it.syntax().clone())), | ||
254 | items.file_id, | ||
255 | id, | ||
256 | ); | ||
257 | // Recursive collect macros | ||
258 | // Note that ast::ModuleItem do not include ast::MacroCall | ||
259 | // We cannot use ModuleItemOwner::items here | ||
260 | for it in items.value.syntax().children().filter_map(ast::MacroCall::cast) { | ||
261 | res.extend(collect_impl_items_in_macro(db, expander, it, id, limit - 1)) | ||
262 | } | ||
263 | expander.exit(db, mark); | ||
264 | res | ||
265 | } else { | ||
266 | Vec::new() | ||
267 | } | ||
268 | } | ||
269 | |||
244 | fn collect_impl_items( | 270 | fn collect_impl_items( |
245 | db: &impl DefDatabase, | 271 | db: &impl DefDatabase, |
246 | impl_items: impl Iterator<Item = ImplItem>, | 272 | impl_items: impl Iterator<Item = ImplItem>, |
diff --git a/crates/ra_hir_ty/src/tests/macros.rs b/crates/ra_hir_ty/src/tests/macros.rs index 7fdbf996f..69c695cc8 100644 --- a/crates/ra_hir_ty/src/tests/macros.rs +++ b/crates/ra_hir_ty/src/tests/macros.rs | |||
@@ -202,6 +202,29 @@ fn test() { S.foo()<|>; } | |||
202 | } | 202 | } |
203 | 203 | ||
204 | #[test] | 204 | #[test] |
205 | fn infer_impl_items_generated_by_macros_chain() { | ||
206 | let t = type_at( | ||
207 | r#" | ||
208 | //- /main.rs | ||
209 | macro_rules! m_inner { | ||
210 | () => {fn foo(&self) -> u128 {0}} | ||
211 | } | ||
212 | macro_rules! m { | ||
213 | () => {m_inner!();} | ||
214 | } | ||
215 | |||
216 | struct S; | ||
217 | impl S { | ||
218 | m!(); | ||
219 | } | ||
220 | |||
221 | fn test() { S.foo()<|>; } | ||
222 | "#, | ||
223 | ); | ||
224 | assert_eq!(t, "u128"); | ||
225 | } | ||
226 | |||
227 | #[test] | ||
205 | fn infer_macro_with_dollar_crate_is_correct_in_expr() { | 228 | fn infer_macro_with_dollar_crate_is_correct_in_expr() { |
206 | let (db, pos) = TestDB::with_position( | 229 | let (db, pos) = TestDB::with_position( |
207 | r#" | 230 | r#" |