only collect the rsx and render macros when formatting

This commit is contained in:
Evan Almloff committed 2023-09-02 07:09:02 -05:00
1 parent 4511b37da8
commit f818d27fa6
1 file changed
+9 -1
+9 -1
View File
@@ -17,7 +17,15 @@ struct MacroCollector<'a, 'b> {
impl<'a, 'b> Visit<'b> for MacroCollector<'a, 'b> {
fn visit_macro(&mut self, i: &'b Macro) {
self.macros.push(i);
if let Some("rsx" | "render") = i
.path
.segments
.last()
.map(|i| i.ident.to_string())
.as_deref()
{
self.macros.push(i)
}
}
}