Add annotations for structs, enums and variants. carllerche#34

Author: GothAck

src/enum.rs

@@ -72,6 +72,12 @@ impl Enum {
         self
     }
 
+    /// Add an arbitrary macro.
+    pub fn r#macro(&mut self, r#macro: &str) -> &mut Self {
+        self.type_def.r#macro(r#macro);
+        self
+    }
+
     /// Push a variant to the enum, returning a mutable reference to it.
     pub fn new_variant(&mut self, name: &str) -> &mut Variant {
         self.push_variant(Variant::new(name));

src/struct.rs

@@ -85,6 +85,12 @@ impl Struct {
         self
     }
 
+    /// Add an arbitrary macro.
+    pub fn r#macro(&mut self, r#macro: &str) -> &mut Self {
+        self.type_def.r#macro(r#macro);
+        self
+    }
+
     /// Push a named field to the struct.
     ///
     /// A struct can either set named fields with this function or tuple fields

src/variant.rs

@@ -10,6 +10,7 @@ use crate::r#type::Type;
 pub struct Variant {
     name: String,
     fields: Fields,
+    annotations: Vec<String>,
 }
 
 impl Variant {
@@ -18,6 +19,7 @@ impl Variant {
         Variant {
             name: name.to_string(),
             fields: Fields::Empty,
+            annotations: Vec::new(),
         }
     }
 
@@ -36,8 +38,18 @@ impl Variant {
         self
     }
 
+    /// Add an anotation to the variant.
+    pub fn annotation(&mut self, annotation: &str) -> &mut Self {
+        self.annotations.push(annotation.to_string());
+        self
+    }
+
     /// Formats the variant using the given formatter.
     pub fn fmt(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
+        for a in &self.annotations {
+            write!(fmt, "{}", a)?;
+            write!(fmt, "\n")?;
+        }
         write!(fmt, "{}", self.name)?;
         self.fields.fmt(fmt)?;
         write!(fmt, ",\n")?;