improve tracking for structs and fields

This commit is contained in:
nora 2023-01-23 09:07:04 +01:00
parent 9ce2c89f4d
commit fa0b3021a9
2 changed files with 28 additions and 0 deletions

View file

@ -31,6 +31,18 @@ impl VisitMut for Visitor<'_> {
}
}
fn visit_field_mut(&mut self, field: &mut syn::Field) {
self.current_path.push(field.ident.to_string());
syn::visit_mut::visit_field_mut(self, method);
self.current_path.pop();
}
fn visit_item_struct_mut(&mut self, struct_: &mut syn::ItemStruct) {
self.current_path.push(struct_.ident.to_string());
syn::visit_mut::visit_item_struct_mut(self, method);
self.current_path.pop();
}
tracking!();
}

View file

@ -218,6 +218,8 @@ macro_rules! tracking {
tracking!(visit_impl_item_method_mut);
tracking!(visit_item_impl_mut);
tracking!(visit_item_mod_mut);
tracking!(visit_field_mut);
tracking!(visit_item_struct_mut);
};
(visit_item_fn_mut) => {
fn visit_item_fn_mut(&mut self, func: &mut syn::ItemFn) {
@ -248,5 +250,19 @@ macro_rules! tracking {
self.current_path.pop();
}
};
(visit_field_mut) => {
fn visit_field_mut(&mut self, field: &mut syn::Field) {
self.current_path.push(field.ident.to_string());
syn::visit_mut::visit_field_mut(self, method);
self.current_path.pop();
}
};
(visit_item_struct_mut) => {
fn visit_item_struct_mut(&mut self, struct_: &mut syn::ItemStruct) {
self.current_path.push(struct_.ident.to_string());
syn::visit_mut::visit_item_struct_mut(self, method);
self.current_path.pop();
}
};
}
pub(crate) use tracking;