use proc_macro2::TokenStream;
use quote::{ToTokens, TokenStreamExt};
use syn;
#[derive(Debug, Clone)]
pub struct BuilderField<'a> {
pub field_ident: &'a syn::Ident,
pub field_type: &'a syn::Type,
pub field_enabled: bool,
pub field_visibility: syn::Visibility,
pub attrs: &'a [syn::Attribute],
}
impl<'a> ToTokens for BuilderField<'a> {
fn to_tokens(&self, tokens: &mut TokenStream) {
if self.field_enabled {
let vis = &self.field_visibility;
let ident = self.field_ident;
let ty = self.field_type;
let attrs = self.attrs;
tokens.append_all(quote!(
#(#attrs)* #vis #ident: ::derive_builder::export::core::option::Option<#ty>,
));
} else {
let ident = self.field_ident;
let ty = self.field_type;
let attrs = self.attrs;
tokens.append_all(quote!(
#(#attrs)* #ident: ::derive_builder::export::core::marker::PhantomData<#ty>,
));
}
}
}
impl<'a> BuilderField<'a> {
pub fn default_initializer_tokens(&self) -> TokenStream {
let ident = self.field_ident;
quote! { #ident : ::derive_builder::export::core::default::Default::default(), }
}
}
#[doc(hidden)]
#[macro_export]
macro_rules! default_builder_field {
() => {{
BuilderField {
field_ident: &syn::Ident::new("foo", ::proc_macro2::Span::call_site()),
field_type: &syn::parse_str("String").unwrap(),
field_enabled: true,
field_visibility: syn::parse_str("pub").unwrap(),
attrs: &[parse_quote!(#[some_attr])],
}
}};
}
#[cfg(test)]
mod tests {
#[allow(unused_imports)]
use super::*;
#[test]
fn setter_enabled() {
let field = default_builder_field!();
assert_eq!(
quote!(#field).to_string(),
quote!(
#[some_attr] pub foo: ::derive_builder::export::core::option::Option<String>,
)
.to_string()
);
}
#[test]
fn setter_disabled() {
let mut field = default_builder_field!();
field.field_enabled = false;
assert_eq!(
quote!(#field).to_string(),
quote!(
#[some_attr]
foo: ::derive_builder::export::core::marker::PhantomData<String>,
)
.to_string()
);
}
#[test]
fn private_field() {
let private = syn::Visibility::Inherited;
let mut field = default_builder_field!();
field.field_visibility = private;
assert_eq!(
quote!(#field).to_string(),
quote!(
#[some_attr]
foo: ::derive_builder::export::core::option::Option<String>,
)
.to_string()
);
}
}