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
//! [`Eq`](std::cmp::Eq) implementation.

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

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

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

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

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

	fn build_signature(
		&self,
		_item: &Item,
		_traits: &[DeriveTrait],
		_trait_: &DeriveTrait,
		body: &TokenStream,
	) -> TokenStream {
		quote! {
			#[inline]
			fn assert_receiver_is_total_eq(&self) {
				struct __AssertEq<__T: ::core::cmp::Eq + ?::core::marker::Sized>(::core::marker::PhantomData<__T>);

				#body
			}
		}
	}

	fn build_body(
		&self,
		_traits: &[DeriveTrait],
		trait_: &DeriveTrait,
		data: &Data,
	) -> TokenStream {
		let types = data.iter_fields(**trait_).map(|field| field.type_);

		quote! {
			#(let _: __AssertEq<#types>;)*
		}
	}
}