add custom derives callback
This callback allows us to specify arbitrary derive attributes for each
named struct. This is useful for adding things that can't easily be
implemented separately, such as `serde::Deserialize` or
`zerocopy::FromBytes`.
diff --git a/src/callbacks.rs b/src/callbacks.rs
index e288af4..9b34544 100644
--- a/src/callbacks.rs
+++ b/src/callbacks.rs
@@ -95,4 +95,12 @@
) -> Option<ImplementsTrait> {
None
}
+
+ /// Provide a list of custom derive attributes.
+ ///
+ /// If no additional attributes are wanted, this function should return an
+ /// empty `Vec`.
+ fn add_derives(&self, _name: &str) -> Vec<String> {
+ vec![]
+ }
}
diff --git a/src/codegen/mod.rs b/src/codegen/mod.rs
index 515ebf1..c70c106 100644
--- a/src/codegen/mod.rs
+++ b/src/codegen/mod.rs
@@ -2005,6 +2005,15 @@
let mut derives: Vec<_> = derivable_traits.into();
derives.extend(item.annotations().derives().iter().map(String::as_str));
+ // The custom derives callback may return a list of derive attributes;
+ // add them to the end of the list.
+ let custom_derives;
+ if let Some(cb) = &ctx.options().parse_callbacks {
+ custom_derives = cb.add_derives(&canonical_name);
+ // In most cases this will be a no-op, since custom_derives will be empty.
+ derives.extend(custom_derives.iter().map(|s| s.as_str()));
+ };
+
if !derives.is_empty() {
attributes.push(attributes::derives(&derives))
}
