zodiac_macro 0.1.1

Macros for zodiac
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

use proc_macro::TokenStream;
use quote::quote;
use syn::{ItemFn, ItemStatic, parse_macro_input};

#[proc_macro_attribute]
pub fn global_allocator(_attr: TokenStream, item: TokenStream) -> TokenStream {
    let global_allocator_item = parse_macro_input!(item as ItemStatic);
    let global_allocator_name = &global_allocator_item.ident;

    quote!(
        #[used]
        #[unsafe(no_mangle)]
        static __ZODIAC_GLOBAL_ALLOCATOR: &'static dyn ::zodiac::mem::Allocator = &#global_allocator_name;

        #global_allocator_item
    )
    .into()
}

#[proc_macro_attribute]
pub fn main(_attr: TokenStream, item: TokenStream) -> TokenStream {
    let main_fn = parse_macro_input!(item as ItemFn);
    let main_fn_name = &main_fn.sig.ident;

    quote!(
        #[unsafe(no_mangle)]
        extern "Rust" fn __zodiac_main() -> ! {
            let _: () = #main_fn_name();

            loop {}
        }

        #[expect(unused)]
        #main_fn
    )
    .into()
}

#[proc_macro_attribute]
pub fn panic_handler(_attr: TokenStream, item: TokenStream) -> TokenStream {
    let handler_fn = parse_macro_input!(item as ItemFn);
    let handler_fn_name = &handler_fn.sig.ident;

    quote!(
        #[unsafe(no_mangle)]
        extern "Rust" fn __zodiac_panic_handler(info: &core::panic::PanicInfo) -> ! {
            #handler_fn_name(info);
        }

        #[expect(unused)]
        #handler_fn
    )
    .into()
}