1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18

use proc_macro::TokenStream;

use syn::{parse_macro_input, DeriveInput};

mod ergol;
mod pgenum;

#[proc_macro_attribute]
pub fn ergol(_attr: TokenStream, input: TokenStream) -> TokenStream {
    let input = parse_macro_input!(input as DeriveInput);
    ergol::generate(input)
}

#[proc_macro_derive(PgEnum)]
pub fn derive_pgenum(input: TokenStream) -> TokenStream {
    let ast = syn::parse(input).unwrap();
    pgenum::generate(&ast)
}