rullst_orm_macros/
lib.rs

1extern crate proc_macro;
2
3use proc_macro::TokenStream;
4use syn::{parse_macro_input, DeriveInput};
5
6mod parser;
7mod builder;
8mod relationships;
9mod models;
10mod factory_observer;
11
12#[proc_macro_derive(Orm, attributes(orm))]
13pub fn eloquent_macro(input: TokenStream) -> TokenStream {
14    let input = parse_macro_input!(input as DeriveInput);
15    
16    // Parse the input
17    let parsed = match parser::parse(&input) {
18        Ok(p) => p,
19        Err(e) => return TokenStream::from(e.to_compile_error()),
20    };
21    
22    // Generate relationships
23    let rels = relationships::generate(&parsed);
24    
25    // Generate the builder
26    let builder_code = builder::generate(
27        &parsed, 
28        &rels.flags, 
29        &rels.inits, 
30        &rels.methods, 
31        &rels.eager_loads
32    );
33    
34    // Generate factory and observers
35    let factory_observer_code = factory_observer::generate(&parsed);
36    
37    // Generate the model impl
38    let model_code = models::generate(&parsed, &rels.model_methods);
39    
40    // Combine
41    let expanded = quote::quote! {
42        #builder_code
43        #factory_observer_code
44        #model_code
45    };
46    
47    TokenStream::from(expanded)
48}
rullst_orm_macros/lib.rs

rullst_orm_macros/
lib.rs
