msg_channel_macro 0.1.0-beat.3

message channel
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125


use proc_macro2::{Ident, TokenStream};
use quote::{format_ident, quote, ToTokens};
use syn::{ImplItem, ItemImpl, Token, Type};
use syn::parse::{Parse, ParseStream};
use syn::punctuated::Punctuated;
use syn::spanned::Spanned;

pub struct MessageSetImpl {
    item_impl: ItemImpl,
    ident: Ident,
}

impl Parse for MessageSetImpl {
    fn parse(input: ParseStream) -> syn::Result<Self> {
        let item_impl: ItemImpl = input.parse()?;

        let ident = match item_impl.self_ty.as_ref() {
            Type::Path(type_path) => type_path
                .path
                .segments
                .last()
                .as_ref()
                .ok_or_else(|| syn::Error::new(type_path.path.span(), "missing ident from path"))?
                .ident
                .clone(),
            _ => {
                return Err(syn::Error::new(
                    item_impl.self_ty.span(),
                    "expected a path or ident",
                ))
            }
        };

        Ok(MessageSetImpl { item_impl, ident })
    }
}
impl ToTokens for MessageSetImpl {
    fn to_tokens(&self, tokens: &mut TokenStream) {
        let item_impl = &self.item_impl;

        let mut handler_ident: Option<Type> = None;
        let mut async_msg_idents: Punctuated<Type, Token![,]> = Punctuated::default();
        let mut sync_msg_idents: Punctuated<Type, Token![,]> = Punctuated::default();
        let mut async_concurrent_msg_idents: Punctuated<Type, Token![,]> = Punctuated::default();
        let mut sync_concurrent_msg_idents: Punctuated<Type, Token![,]> = Punctuated::default();

        for item in item_impl.items.iter() {
            match item {
                ImplItem::Type(item_type) => {
                    let item_type_name = item_type.ident.to_string();
                    match &*item_type_name {
                        "Handler" => {
                            handler_ident = Some(item_type.ty.clone());
                        }
                        "Async" => {
                            let Type::Tuple(tuple) = &item_type.ty else {
                                panic!("Async type must is tuple");
                            };
                            // tuple.elems.iter().map(|n| {
                            //     match n {
                            //         // Type::ImplTrait(_) => {}
                            //         // Type::TraitObject(_) => {}
                            //         // Type::Tuple(_) => {}
                            //         Type::Path(n) => {
                            //             n.path
                            //         }
                            //         _ => {
                            //             panic!("{n:?} msg type is supported");
                            //         }
                            //     }
                            // })
                            async_msg_idents = tuple.elems.clone();
                        }
                        "Sync" => {
                            let Type::Tuple(tuple) = &item_type.ty else {
                                panic!("Async type must is tuple");
                            };
                            sync_msg_idents = tuple.elems.clone();
                        }
                        "AsyncConcurrent" => {
                            let Type::Tuple(tuple) = &item_type.ty else {
                                panic!("Async type must is tuple");
                            };
                            async_concurrent_msg_idents = tuple.elems.clone();
                        }
                        "SyncConcurrent" => {
                            let Type::Tuple(tuple) = &item_type.ty else {
                                panic!("Async type must is tuple");
                            };
                            sync_concurrent_msg_idents = tuple.elems.clone();
                        }
                        _ => continue,
                    }
                }
                _ => continue,
            }
        }

        let handler_ident = handler_ident.unwrap();
        let ident = &self.ident;
        let actor_info_impl = {
            let async_msg = format_ident!("{}AsyncVariant", ident);
            let sync_msg = format_ident!("{}SyncVariant", ident);
            let async_concurrent_msg = format_ident!("{}AsyncConcurrentVariant", ident);
            let sync_concurrent_msg = format_ident!("{}SyncConcurrentVariant", ident);
            quote! {
                impl MessageVariantSet for #ident {
                    type AsyncVariant = #async_msg;
                    type SyncVariant = #sync_msg;
                    type AsyncConcurrentVariant = #async_concurrent_msg;
                    type SyncConcurrentVariant = #sync_concurrent_msg;
                }
            }
        };
        tokens.extend(quote! {
            #item_impl
            msg_channel::internal::impl_msg_handle_for_variant!(#ident;#handler_ident;Async;HandleAsync;#async_msg_idents);
            msg_channel::internal::impl_sync_msg_handle_for_variant!(#ident;#handler_ident;Sync;HandleSync;#sync_msg_idents);
            msg_channel::internal::impl_concurrent_msg_handle_for_variant!(#ident;#handler_ident;AsyncConcurrent;HandleAsyncConcurrent;#async_concurrent_msg_idents);
            msg_channel::internal::impl_sync_concurrent_msg_handle_for_variant!(#ident;#handler_ident;SyncConcurrent;HandleSyncConcurrent;#sync_concurrent_msg_idents);
            #actor_info_impl
        });
    }
}