Automatically make fields of pub enums pub

Author: jcaesar

src/imp.rs

@@ -56,7 +56,7 @@ pub(crate) fn recurse_through_definition(
     match &mut parsed {
         Declaration::Struct(s) => {
             strike_through_attributes(&mut s.attributes, &mut strike_attrs, ret);
-            recurse_through_struct_fields(&mut s.fields, &strike_attrs, ret, &None);
+            recurse_through_struct_fields(&mut s.fields, &strike_attrs, ret, &None, false);
             if make_pub {
                 s.vis_marker.get_or_insert_with(make_pub_marker);
             }
@@ -69,6 +69,7 @@ pub(crate) fn recurse_through_definition(
                     &strike_attrs,
                     ret,
                     &Some(v.name.clone()),
+                    is_plain_pub(&e.vis_marker),
                 );
             }
             if make_pub {
@@ -155,6 +156,7 @@ fn recurse_through_struct_fields(
     strike_attrs: &[Attribute],
     ret: &mut TokenStream,
     name_hint: &Option<Ident>,
+    in_pub_enum: bool,
 ) {
     match fields {
         StructFields::Named(n) => {
@@ -171,7 +173,7 @@ fn recurse_through_struct_fields(
                     strike_attrs,
                     ret,
                     &Some(name_hint),
-                    is_plain_pub(&field.vis_marker),
+                    is_plain_pub(&field.vis_marker) || in_pub_enum,
                     &mut field.ty.tokens,
                 );
             }
@@ -211,7 +213,7 @@ fn recurse_through_struct_fields(
                     strike_attrs,
                     ret,
                     name_hint,
-                    is_plain_pub(&field.vis_marker),
+                    is_plain_pub(&field.vis_marker) || in_pub_enum,
                     &mut field.ty.tokens,
                 );
             }

src/test.rs

@@ -446,7 +446,7 @@ fn pub_markers_sane() {
 #[test]
 fn issue_2() {
     let from = quote! {
-        pub enum Expr<'src> {
+        enum Expr<'src> {
             Binary(struct<'src> {
                  left: Box<Expr<'src>>,
                  operator: BinaryOp,
@@ -461,7 +461,26 @@ fn issue_2() {
     let out = quote! {
         struct Binary < 'src > { left : Box < Expr < 'src >> , operator : BinaryOp , right : Box < Expr < 'src >> , }
         enum Literal < 'src > { StringLit (& 'src str) , NumLit (& 'src str) , }
-        pub enum Expr < 'src > { Binary (Binary<'src>) , Literal (Literal<'src>) , }
+        enum Expr < 'src > { Binary (Binary<'src>) , Literal (Literal<'src>) , }
     };
     check(from, out)
 }
+
+#[test]
+fn pub_enum_autopubs() {
+    let from = quote! {
+        pub enum Outer {
+            An(struct ()),
+            Ny(struct {}),
+        };
+    };
+    let out = quote! {
+        pub struct An ();
+        pub struct Ny {}
+        pub enum Outer {
+            An(An),
+            Ny(Ny),
+        }
+    };
+    check(from, out);
+}