compio/compio-macros/src/lib.rs at 71157e5b7a69171b5ed562609087c75814b7799d · compio-rs/compio · GitHub

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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#![doc(
    html_logo_url = "https://github.com/compio-rs/compio-logo/raw/refs/heads/master/generated/colored-bold.svg"
)]
#![doc(
    html_favicon_url = "https://github.com/compio-rs/compio-logo/raw/refs/heads/master/generated/colored-bold.svg"
)]

mod item_fn;

mod main_fn;

use proc_macro::TokenStream;
use proc_macro_crate::{FoundCrate, crate_name};
use proc_macro2::{Ident, Span};
use quote::{ToTokens, quote};
use syn::parse_macro_input;

#[proc_macro_attribute]
pub fn main(args: TokenStream, item: TokenStream) -> TokenStream {
    parse_macro_input!(item as main_fn::CompioMain)
        .with_args(parse_macro_input!(args as item_fn::RawAttr))
        .into_token_stream()
        .into()
}

#[proc_macro_attribute]
pub fn test(args: TokenStream, item: TokenStream) -> TokenStream {
    parse_macro_input!(item as main_fn::CompioMain)
        .with_args(parse_macro_input!(args as item_fn::RawAttr))
        .with_test(true)
        .into_token_stream()
        .into()
}

fn retrieve_runtime_mod() -> proc_macro2::TokenStream {
    match crate_name("compio-runtime") {
        Ok(FoundCrate::Itself) => quote!(crate),
        Ok(FoundCrate::Name(name)) => {
            let ident = Ident::new(&name, Span::call_site());
            quote!(::#ident)
        }
        Err(_) => match crate_name("compio") {
            Ok(FoundCrate::Itself) => quote!(crate::runtime),
            Ok(FoundCrate::Name(name)) => {
                let ident = Ident::new(&name, Span::call_site());
                quote!(::#ident::runtime)
            }
            Err(_) => panic!("Cannot find compio or compio_runtime."),
        },
    }
}

fn retrieve_driver_mod() -> proc_macro2::TokenStream {
    match crate_name("compio-driver") {
        Ok(FoundCrate::Itself) => quote!(crate),
        Ok(FoundCrate::Name(name)) => {
            let ident = Ident::new(&name, Span::call_site());
            quote!(::#ident)
        }
        Err(_) => match crate_name("compio") {
            Ok(FoundCrate::Itself) => quote!(crate::driver),
            Ok(FoundCrate::Name(name)) => {
                let ident = Ident::new(&name, Span::call_site());
                quote!(::#ident::driver)
            }
            Err(_) => {
                let ident = Ident::new("compio_driver", Span::call_site());
                quote!(::#ident)
            }
        },
    }
}