hydra-macros 0.1.39

Macros for the hydra framework.
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
55
56
57
58
59
60
61
62
63
64
65
66

use proc_macro::TokenStream;

use quote::format_ident;
use quote::quote;

use syn::parse_macro_input;
use syn::ItemFn;

pub(crate) fn main(_: TokenStream, item: TokenStream) -> TokenStream {
    let input_fn = parse_macro_input!(item as ItemFn);
    let input_name = &input_fn.sig.ident;
    let input_block = &input_fn.block;
    let input_struct = format_ident!("{}_MainStruct", input_name);

    let output = quote! {
        #[allow(non_camel_case_types)]
        struct #input_struct;

        impl ::hydra::Application for #input_struct {
            async fn start(&self) -> Result<::hydra::Pid, ::hydra::ExitReason> {
                Ok(::hydra::Process::spawn_link(async move {
                    #input_block
                }))
            }
        }

        fn #input_name() {
            use ::hydra::Application;

            let main = #input_struct;
            main.run();
        }
    };

    output.into()
}

pub(crate) fn test(_: TokenStream, item: TokenStream) -> TokenStream {
    let input_fn = parse_macro_input!(item as ItemFn);
    let input_name = &input_fn.sig.ident;
    let input_block = &input_fn.block;
    let input_struct = format_ident!("{}_TestStruct", input_name);

    let output = quote! {
        #[allow(non_camel_case_types)]
        struct #input_struct;

        impl ::hydra::Application for #input_struct {
            async fn start(&self) -> Result<::hydra::Pid, ::hydra::ExitReason> {
                Ok(::hydra::Process::spawn_link(async move {
                    #input_block
                }))
            }
        }

        #[test]
        fn #input_name() {
            use ::hydra::Application;

            let test = #input_struct;
            test.test();
        }
    };

    output.into()
}