Skip to main content

toy_arms_derive/
lib.rs

1extern crate proc_macro;
2
3use proc_macro::TokenStream;
4use quote::quote;
5use syn;
6
7#[proc_macro_derive(GameObject)]
8pub fn game_object_derive(input: TokenStream) -> TokenStream {
9    let ast = syn::parse(input).unwrap();
10    impl_game_object(&ast)
11}
12
13fn impl_game_object(ast: &syn::DeriveInput) -> TokenStream {
14    let name = &ast.ident;
15    let gen = quote! {
16        impl GameObject for #name {
17            unsafe fn from_raw(address: *const usize) -> Option<*mut #name> {
18                let ptr = address as *mut Self;
19                if (ptr as *mut usize).is_null() {
20                    return None
21                }
22                Some(ptr)
23            }
24        }
25    };
26    gen.into()
27}