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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

// We support Rust 1.15 and struct initialization shorthand syntax is a Rust 1.17 feature
//
// NOTE: Rust 1.17 and older cannot parse `::` in attributes so we can't use
// `allow(clippy::redundant_field_names)`.
// Instead, we resort to using `allow(redundant_field_names)`, but this triggers
// `renamed_and_removed_lints` which we also want to allow.
#![cfg_attr(
    feature = "cargo-clippy",
    allow(renamed_and_removed_lints, redundant_field_names,)
)]

// We need to support Rust 1.15 to stable
#![allow(deprecated)]

extern crate proc_macro;
extern crate proc_macro2;
#[macro_use]
extern crate syn;

#[macro_use]
extern crate quote;

mod ast;
mod attr;
mod bound;
mod clone;
mod cmp;
mod debug;
mod default;
mod hash;
mod matcher;
mod utils;

use proc_macro::TokenStream;

fn derive_impls(input: &ast::Input) -> Result<proc_macro2::TokenStream, String> {
    let mut tokens = proc_macro2::TokenStream::new();

    if input.attrs.clone.is_some() {
        tokens.extend(clone::derive_clone(input));
    }
    if input.attrs.copy.is_some() {
        tokens.extend(clone::derive_copy(input));
    }
    if input.attrs.debug.is_some() {
        tokens.extend(debug::derive(input));
    }
    if let Some(ref default) = input.attrs.default {
        tokens.extend(default::derive(input, default));
    }
    if input.attrs.eq.is_some() {
        tokens.extend(cmp::derive_eq(input));
    }
    if input.attrs.hash.is_some() {
        tokens.extend(hash::derive(input));
    }
    if input.attrs.partial_eq.is_some() {
        tokens.extend(cmp::derive_partial_eq(input)?);
    }
    if input.attrs.partial_ord.is_some() {
        tokens.extend(cmp::derive_partial_ord(input)?);
    }
    if input.attrs.ord.is_some() {
        tokens.extend(cmp::derive_ord(input)?);
    }

    Ok(tokens)
}

fn detail(input: TokenStream) -> Result<TokenStream, String> {
    let parsed = syn::parse::<syn::DeriveInput>(input).map_err(|e| e.to_string())?;
    let output = derive_impls(&ast::Input::from_ast(&parsed)?)?;
    Ok(output.into())
}

#[cfg_attr(not(test), proc_macro_derive(Derivative, attributes(derivative)))]
pub fn derivative(input: TokenStream) -> TokenStream {
    match detail(input) {
        Ok(output) => output,
        Err(e) => panic!(e),
    }
}