-
-
Notifications
You must be signed in to change notification settings - Fork 35
Adding support for build_from and build_from_clone builder methods.
#313
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: master
Are you sure you want to change the base?
Changes from all commits
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,158 @@ | ||
| use crate::builder::builder_gen::{BuilderGenCtx, member::Member}; | ||
| use crate::util::prelude::*; | ||
| use proc_macro2::{Ident, Span}; | ||
| use quote::quote; | ||
| use syn::{Type, spanned::Spanned}; | ||
|
|
||
| pub(super) fn emit(ctx: &BuilderGenCtx, target_ty: &Type) -> Result<TokenStream> { | ||
| let mut tokens = TokenStream::new(); | ||
| let ctor_args: Vec<_> = ctx | ||
| .members | ||
| .iter() | ||
| .map(|m| { | ||
| let ident = m.orig_ident(); | ||
| quote! { #ident } | ||
| }) | ||
| .collect(); | ||
|
|
||
| let base_name = ctx.finish_fn.ident.clone(); | ||
|
|
||
| if ctx.build_from { | ||
| tokens.extend(emit_build_from_method( | ||
| false, | ||
| &base_name, | ||
| target_ty, | ||
| &ctx.members, | ||
| &ctor_args, | ||
| )); | ||
| } | ||
|
|
||
| if ctx.build_from_clone { | ||
| tokens.extend(emit_build_from_method( | ||
| true, | ||
| &base_name, | ||
| target_ty, | ||
| &ctx.members, | ||
| &ctor_args, | ||
| )?); | ||
| } | ||
|
|
||
| Ok(tokens) | ||
| } | ||
|
|
||
| fn emit_build_from_method( | ||
| clone: bool, | ||
| base_name: &Ident, | ||
| target_ty: &Type, | ||
| members: &[Member], | ||
| ctor_args: &[TokenStream], | ||
| ) -> Result<TokenStream> { | ||
| let doc = if clone { | ||
| "Fills unset builder fields from an owned value of the target type and builds it." | ||
| } else { | ||
| "Fills unset builder fields from a reference to the target type and builds it." | ||
| }; | ||
|
|
||
| let method_name = if clone { | ||
| format_ident!("{}_from_clone", base_name) | ||
| } else { | ||
| format_ident!("{}_from", base_name) | ||
| }; | ||
|
|
||
| let arg_type = if clone { | ||
| quote!(&#target_ty) | ||
| } else { | ||
| quote!(#target_ty) | ||
| }; | ||
|
|
||
| let arg_pat = if clone { | ||
| quote!(mut from) | ||
| } else { | ||
| quote!(from) | ||
| }; | ||
|
|
||
| let ctor_path = extract_ctor_ident_path(target_ty, target_ty.span())?; | ||
| let field_vars = field_vars_from_members(members, clone); | ||
|
|
||
| Ok(quote! { | ||
| #[inline(always)] | ||
| #[doc = #doc] | ||
| pub fn #method_name(self, #arg_pat: #arg_type) -> #target_ty { | ||
| #( #field_vars )* | ||
| #ctor_path { | ||
| #( #ctor_args, )* | ||
| } | ||
| } | ||
| }) | ||
| } | ||
|
|
||
| fn field_vars_from_members(members: &[Member], clone: bool) -> Vec<TokenStream> { | ||
| members | ||
| .iter() | ||
| .map(|member| { | ||
| let ident = member.orig_ident(); | ||
| let ty = member.norm_ty(); | ||
| let default_expr = quote! { ::core::default::Default::default() }; | ||
|
|
||
| match member { | ||
| Member::Field(_) | Member::StartFn(_) => quote! { | ||
| let #ident: #ty = self.#ident; | ||
| }, | ||
| Member::Named(member) => { | ||
| let index = &member.index; | ||
| if clone { | ||
| quote! { | ||
| let #ident: #ty = match self.__unsafe_private_named.#index { | ||
| Some(value) => value, | ||
| None => from.#ident.clone(), | ||
| }; | ||
| } | ||
| } else { | ||
| quote! { | ||
| let #ident: #ty = match self.__unsafe_private_named.#index { | ||
| Some(value) => value, | ||
| None => from.#ident, | ||
| }; | ||
| } | ||
| } | ||
| } | ||
| Member::FinishFn(_) => { | ||
| if clone { | ||
| quote! { | ||
| let #ident: #ty = from.#ident.clone(); | ||
| } | ||
| } else { | ||
| quote! { | ||
| let #ident: #ty = from.#ident; | ||
| } | ||
| } | ||
| } | ||
| Member::Skip(_) => quote! { | ||
| let #ident: #ty = #default_expr; | ||
| }, | ||
| } | ||
| }) | ||
| .collect() | ||
| } | ||
|
|
||
| pub(crate) fn extract_ctor_ident_path(ty: &Type, span: Span) -> Result<TokenStream> { | ||
| use quote::quote_spanned; | ||
|
Check failure on line 139 in bon-macros/src/builder/builder_gen/build_from.rs
|
||
|
|
||
| let path = ty.as_path_no_qself().ok_or_else(|| { | ||
| err!( | ||
| &span, | ||
| "expected a concrete type path (like `MyStruct`) for constructor" | ||
| ) | ||
| })?; | ||
|
|
||
| let mut ident = path | ||
| .segments | ||
| .last() | ||
| .ok_or_else(|| err!(&span, "expected a named type, but found an empty path"))? | ||
| .ident | ||
| .clone(); | ||
|
|
||
| ident.set_span(span); | ||
|
|
||
| Ok(quote! { #ident }) | ||
| } | ||
Uh oh!
There was an error while loading. Please reload this page.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Unfortunatelly, this approach won't work in case if the builder is generated with the
#[builder]macro on top of a function. We have thebody.generate()infinish_fn.rshere that generates the proper body:bon/bon-macros/src/builder/builder_gen/finish_fn.rs
Line 139 in af6e925
It's only requirement is that members are assigned to variables of the same name in scope. Plus we need to inherit the potential
async/unsafe/constmodifiers and maybe handle the case of theResultfrom the original function