aboutsummaryrefslogtreecommitdiff
path: root/tosin-macros/src
diff options
context:
space:
mode:
Diffstat (limited to 'tosin-macros/src')
-rw-r--r--tosin-macros/src/lib.rs41
1 files changed, 38 insertions, 3 deletions
diff --git a/tosin-macros/src/lib.rs b/tosin-macros/src/lib.rs
index 7efbac6..408378f 100644
--- a/tosin-macros/src/lib.rs
+++ b/tosin-macros/src/lib.rs
@@ -2,6 +2,8 @@
extern crate proc_macro;
+use std::fs;
+
use proc_macro::TokenStream;
use quote::quote;
@@ -19,9 +21,9 @@ fn impl_model(ast: &syn::DeriveInput) -> TokenStream {
let name = &ast.ident;
let gen = quote! {
impl #name {
- fn hello_macro() {
- println!("Hello, Macro! My name is {}!", stringify!(#name));
- }
+ pub const META: tosin::db::models::ModelMeta = tosin::db::models::ModelMeta {
+ name: stringify!(#name),
+ };
}
};
gen.into()
@@ -55,3 +57,36 @@ pub fn gather_migrations(_input: TokenStream) -> TokenStream {
gen.into()
}
+
+#[proc_macro]
+pub fn gather_models(_input: TokenStream) -> TokenStream {
+ let call_site = proc_macro::Span::call_site();
+ let call_site_file = call_site.source_file();
+ let call_site_path = call_site_file.path();
+ if !call_site_file.is_real() {
+ panic!("call site does not have a real path");
+ }
+
+ let call_site_ast = syn::parse_file(&fs::read_to_string(call_site_path).unwrap()).unwrap();
+ let models = call_site_ast.items.iter()
+ .filter_map(|item| if let syn::Item::Struct(item) = item { Some(item) } else { None })
+ .filter(|item| item.attrs.iter().any(|attr| {
+ let attr = if let Ok(syn::Meta::List(attr)) = attr.parse_meta() { attr } else { return false; };
+ if attr.path.get_ident().map_or(false, |hopefully_derive| hopefully_derive == "derive") {
+ let mut derived = attr.nested.iter()
+ .filter_map(|derived| if let syn::NestedMeta::Meta(derived) = derived { Some(derived) } else { None });
+ derived.any(|derived| derived.path().get_ident().map_or(false, |hopefully_model| hopefully_model == "Model"))
+ } else {
+ false
+ }
+ }))
+ .map(|item| &item.ident);
+
+ let gen = quote! {
+ pub const ALL: &[tosin::db::models::ModelMeta] = &[
+ #(#models::META),*
+ ];
+ };
+
+ gen.into()
+}