pub-fields 0.1.1

Provides a proc macro attribute that defaults all struct fields to public.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

use proc_macro::TokenStream;
use quote::ToTokens;
use syn::{parse::Nothing, parse_macro_input, parse_quote, ItemStruct};

/// A simple attribute that makes all fields public on a struct.
///
/// Usage:
/// ```rust
/// use pub_fields::pub_fields;
///
/// #[pub_fields]
/// pub struct MyStruct {
///     a: usize,
///     b: usize,
///     c: usize,
/// }
/// ```
/// =>
/// ```rust
/// pub struct MyStruct {
///     pub a: usize,
///     pub b: usize,
///     pub c: usize,
/// }
/// ```
#[proc_macro_attribute]
pub fn pub_fields(attr: TokenStream, tokens: TokenStream) -> TokenStream {
    parse_macro_input!(attr as Nothing);
    let mut item_struct = parse_macro_input!(tokens as ItemStruct);
    for field in &mut item_struct.fields {
        field.vis = match &field.vis {
            syn::Visibility::Public(p) => syn::Visibility::Public(*p),
            syn::Visibility::Restricted(res) => syn::Visibility::Restricted(res.clone()),
            syn::Visibility::Inherited => parse_quote!(pub),
        };
    }
    item_struct.to_token_stream().into()
}