Portal: Support for portal to allow docs in enum fields

corigan01 · Feb 10, 2025 · e5c6fd6 · e5c6fd6
1 parent cd7860e
commit e5c6fd6
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/crates/portal-macro/src/ast.rs b/crates/portal-macro/src/ast.rs
@@ -141,6 +141,7 @@ pub struct ProtocolEnumDef {
 #[derive(Debug)]
 #[allow(unused)]
 pub struct ProtocolEnumVarient {
+    pub docs: Vec<Attribute>,
     pub ident: Ident,
     pub fields: ProtocolEnumFields,
 }

diff --git a/crates/portal-macro/src/parse.rs b/crates/portal-macro/src/parse.rs
@@ -466,9 +466,11 @@ impl TryFrom<&syn::Variant> for ast::ProtocolEnumVarient {
     type Error = syn::Error;
     fn try_from(value: &syn::Variant) -> Result<Self, Self::Error> {
         let ident = value.ident.clone();
+        let docs = value.attrs.clone();
         Ok(Self {
             ident,
             fields: (&value.fields).try_into()?,
+            docs,
         })
     }
 }

diff --git a/crates/portal-macro/src/rust_builder.rs b/crates/portal-macro/src/rust_builder.rs
@@ -272,8 +272,10 @@ impl ToTokens for ast::ProtocolEnumVarient {
     fn to_tokens(&self, tokens: &mut TokenStream2) {
         let ident = &self.ident;
         let fields = &self.fields;
+        let docs = &self.docs;
 
         tokens.append_all(quote! {
+            #(#docs)*
             #ident #fields
         });
     }