| David Tolnay | b247df1 | 2020-11-27 12:06:12 -0800 | [diff] [blame] | 1 | use proc_macro2::{Ident, Span}; |
| 2 | |
| 3 | #[derive(Copy, Clone)] |
| 4 | pub struct Derive { |
| 5 | pub what: Trait, |
| 6 | pub span: Span, |
| 7 | } |
| David Tolnay | 64703b4 | 2020-05-10 22:12:33 -0700 | [diff] [blame] | 8 | |
| 9 | #[derive(Copy, Clone, PartialEq)] |
| David Tolnay | b247df1 | 2020-11-27 12:06:12 -0800 | [diff] [blame] | 10 | pub enum Trait { |
| David Tolnay | 64703b4 | 2020-05-10 22:12:33 -0700 | [diff] [blame] | 11 | Clone, |
| 12 | Copy, |
| 13 | } |
| 14 | |
| 15 | impl Derive { |
| 16 | pub fn from(ident: &Ident) -> Option<Self> { |
| David Tolnay | b247df1 | 2020-11-27 12:06:12 -0800 | [diff] [blame] | 17 | let what = match ident.to_string().as_str() { |
| 18 | "Clone" => Trait::Clone, |
| 19 | "Copy" => Trait::Copy, |
| 20 | _ => return None, |
| 21 | }; |
| 22 | let span = ident.span(); |
| 23 | Some(Derive { what, span }) |
| 24 | } |
| 25 | } |
| 26 | |
| 27 | impl PartialEq<Trait> for Derive { |
| 28 | fn eq(&self, other: &Trait) -> bool { |
| 29 | self.what == *other |
| David Tolnay | 64703b4 | 2020-05-10 22:12:33 -0700 | [diff] [blame] | 30 | } |
| 31 | } |