extern crate proc_macro;
use proc_macro::TokenStream;
use syn::{DeriveInput, parse_macro_input};
mod builder;
mod factory_observer;
mod models;
mod parser;
mod relationships;
#[proc_macro_derive(Orm, attributes(orm, sqlx))]
pub fn rullst_macro(input: TokenStream) -> TokenStream {
let input = parse_macro_input!(input as DeriveInput);
let parsed = match parser::parse(&input) {
Ok(p) => p,
Err(e) => return TokenStream::from(e.to_compile_error()),
};
let rels = relationships::generate(&parsed);
let builder_code = builder::generate(
&parsed,
&rels.flags,
&rels.inits,
&rels.methods,
&rels.eager_loads,
);
let factory_observer_code = factory_observer::generate(&parsed);
let model_code = models::generate(&parsed, &rels.model_methods);
let expanded = quote::quote! {
#builder_code
#factory_observer_code
#model_code
};
TokenStream::from(expanded)
}