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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

//! [`Default`](std::default::Default) implementation.

use proc_macro2::TokenStream;
use quote::quote;

use crate::{Data, DeriveTrait, Item, SimpleType, TraitImpl};

/// Dummy-struct implement [`Trait`](crate::Trait) for
/// [`Default`](std::default::Default).
pub struct Default;

impl TraitImpl for Default {
	fn as_str(&self) -> &'static str {
		"Default"
	}

	fn default_derive_trait(&self) -> DeriveTrait {
		DeriveTrait::Default
	}

	fn build_signature(
		&self,
		_item: &Item,
		_trait_: &DeriveTrait,
		body: &TokenStream,
	) -> TokenStream {
		quote! {
			fn default() -> Self {
				#body
			}
		}
	}

	fn build_body(&self, trait_: &DeriveTrait, data: &Data) -> TokenStream {
		if data.is_default() {
			let path = &data.path;

			match data.simple_type() {
				SimpleType::Struct(_) => {
					let fields = data.iter_field_ident(trait_);
					let trait_path = trait_.path();

					quote! { #path { #(#fields: #trait_path::default()),* } }
				}
				SimpleType::Tuple(_) => {
					let trait_path = trait_.path();
					let fields = data
						.iter_fields(trait_)
						.map(|_| quote! { #trait_path::default() });

					quote! { #path(#(#fields),*) }
				}
				SimpleType::Unit(_) => {
					quote! { #path }
				}
				SimpleType::Union(_) => unreachable!("unexpected trait for union"),
			}
		}
		// Skip `Default` implementation if variant isn't marked with a `default` attribute.
		else {
			TokenStream::new()
		}
	}
}