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
use proc_macro::TokenStream;
use proc_macro2::TokenStream as TokenStream2;
use quote::{
	format_ident,
	quote,
};
use syn::{
	parse_macro_input,
	Data,
	DeriveInput,
	Fields,
	Generics,
	Ident,
};

#[proc_macro_derive(Editable)]
pub fn derive_editable(input: TokenStream) -> TokenStream {
	let input = parse_macro_input!(input as DeriveInput);
	expand_derive_editable(&input).into()
}

fn expand_derive_editable(input: &DeriveInput) -> TokenStream2 {
	match &input.data {
		Data::Struct(data) => {
			implement_struct_editable(&input.ident, &input.generics, &data.fields)
		}
		_ => unimplemented!("Enum not supported"),
	}
}

fn implement_struct_editable(ident: &Ident, generics: &Generics, fields: &Fields) -> TokenStream2 {
	let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();

	let path = format_ident!("newport_editor");

	let mut tokens = Vec::new();
	match fields {
		Fields::Named(fields) => {
			for field in fields.named.iter() {
				match &field.ident {
					Some(ident) => {
						tokens.push(quote! {
							#path::Editable::edit(&mut self.#ident, stringify!(#ident), ui);
						});
					}
					None => {}
				}
			}
		}
		_ => panic!("Named are only supported"),
	}

	quote! {
		impl #impl_generics #path::Editable for #ident #ty_generics #where_clause {
			fn edit(&mut self, name: &str, builder: &mut #path::Builder) {
				// #path::CollapsingHeader::new(name)
					// .default_open(true)
					// .show(ui, |ui|{
						// #(#tokens)*
					// });
			}
		}
	}
}