1use proc_macro2::TokenStream;
2use quote::ToTokens;
3use syn::Ident;
4
5use crate::codegen::FromFieldImpl;
6use crate::options::{OuterFrom, ParseAttribute, ParseData};
7use crate::Result;
8
9#[derive(Debug)]
10pub struct FromFieldOptions {
11 pub base: OuterFrom,
12 pub vis: Option<Ident>,
13 pub ty: Option<Ident>,
14}
15
16impl FromFieldOptions {
17 pub fn new(di: &syn::DeriveInput) -> Result<Self> {
18 (FromFieldOptions {
19 base: OuterFrom::start(di)?,
20 vis: Default::default(),
21 ty: Default::default(),
22 })
23 .parse_attributes(&di.attrs)?
24 .parse_body(&di.data)
25 }
26}
27
28impl ParseAttribute for FromFieldOptions {
29 fn parse_nested(&mut self, mi: &syn::Meta) -> Result<()> {
30 self.base.parse_nested(mi)
31 }
32}
33
34impl ParseData for FromFieldOptions {
35 fn parse_variant(&mut self, variant: &syn::Variant) -> Result<()> {
36 self.base.parse_variant(variant)
37 }
38
39 fn parse_field(&mut self, field: &syn::Field) -> Result<()> {
40 match field
41 .ident
42 .as_ref()
43 .map(|v| v.to_string())
44 .as_ref()
45 .map(|v| v.as_str())
46 {
47 Some("vis") => {
48 self.vis = field.ident.clone();
49 Ok(())
50 }
51 Some("ty") => {
52 self.ty = field.ident.clone();
53 Ok(())
54 }
55 _ => self.base.parse_field(field),
56 }
57 }
58}
59
60impl<'a> From<&'a FromFieldOptions> for FromFieldImpl<'a> {
61 fn from(v: &'a FromFieldOptions) -> Self {
62 FromFieldImpl {
63 ident: v.base.ident.as_ref(),
64 vis: v.vis.as_ref(),
65 ty: v.ty.as_ref(),
66 attrs: v.base.attrs.as_ref(),
67 base: (&v.base.container).into(),
68 attr_names: &v.base.attr_names,
69 forward_attrs: v.base.forward_attrs.as_ref(),
70 from_ident: v.base.from_ident,
71 }
72 }
73}
74
75impl ToTokens for FromFieldOptions {
76 fn to_tokens(&self, tokens: &mut TokenStream) {
77 FromFieldImpl::from(self).to_tokens(tokens)
78 }
79}
80