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
extern crate proc_macro;
extern crate syn;
#[macro_use]
extern crate quote;



use proc_macro::TokenStream;



#[proc_macro_derive(MysqlEnum)]
pub fn toql_derive(input: TokenStream) -> TokenStream {
    let s = input.to_string();
    let ast = syn::parse_derive_input(&s).unwrap();
    let gen = impl_derive(&ast);

	//println!("GEN = {:?}", gen);
    gen.parse().unwrap()
}



fn impl_derive(ast: &syn::DeriveInput) -> quote::Tokens {
    let name = &ast.ident;

    quote! {
				impl mysql::prelude::ConvIr<#name> for mysql_enum::EnumIr<#name> {
						fn new(v: mysql::Value) -> Result< mysql_enum::EnumIr<#name>, mysql::FromValueError> {
							  mysql_enum::convert_enum::<#name>(v)
						}
						fn commit(self) -> #name {
								self.value
						}
						fn rollback(self) -> mysql::Value {
								mysql::Value::Bytes(self.string.into_bytes())
						}
				}
				impl mysql::prelude::FromValue for #name {
						type Intermediate = mysql_enum::EnumIr<#name>;
				}

				impl std::convert::From<#name> for mysql::Value {
					fn from (t: #name) -> Self {
						mysql::Value::from(t.to_string())
					}
				}
    }
}