actix_derive 0.1.0

Actor framework for Rust
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

use quote;
use syn;
use rand::{Rng, thread_rng};

pub const MESSAGE_ATTR: &str = "rtype";

pub fn expand(ast: &syn::DeriveInput) -> quote::Tokens {
    let (item_type, error_type) = {
        match get_attribute_type_multiple(ast, MESSAGE_ATTR) {
            Some(ty) => {
                match ty.len() {
                    1 => (ty[0].clone(), None),
                    2 => (ty[0].clone(), ty[1].clone()),
                    _ => panic!("#[{}(type, type)] takes 2 parameters, given {}", MESSAGE_ATTR, ty.len()),
                }
            },
            None => {
                (None, None)
            }
        }

    };

    let name = &ast.ident;
    let (impl_generics, ty_generics, where_clause) = ast.generics.split_for_impl();

    let item_type = item_type.unwrap_or(syn::Ty::Tup(vec![]));
    let error_type = error_type.unwrap_or(syn::Ty::Tup(vec![]));
    let dummy_const = syn::Ident::new(format!("_IMPL_ACT_{}", name).to_lowercase());

    quote!{
        mod #dummy_const {
            extern crate actix;

            impl #impl_generics actix::ResponseType for #name #ty_generics #where_clause {
                type Item = #item_type;
                type Error = #error_type;
            }
        }
    }
}

fn get_attribute_type_multiple(ast: &syn::DeriveInput, name: &str) -> Option<Vec<Option<syn::Ty>>>
{
    let attr = ast.attrs.iter().find(|a| a.name() == name);

    if attr.is_none() {
        return None;
    }

    let attr = attr.unwrap();

    if let syn::MetaItem::List(_, ref vec) = attr.value {
        Some(vec.iter().map(|m| meta_item_to_ty(m, name)).collect())
    } else {
        panic!("The correct syntax is #[{}(type, type, ...)]", name);
    }
}

fn meta_item_to_ty(meta_item: &syn::NestedMetaItem, name: &str) -> Option<syn::Ty> {
    if let syn::NestedMetaItem::MetaItem(syn::MetaItem::Word(ref i)) = *meta_item {
        let ty = syn::parse::ty(i.as_ref());
        match ty {
            syn::parse::IResult::Done(_, ty) => Some(ty),
            _ => None,
        }
    } else {
        panic!("The correct syntax is #[{}(type)]", name);
    }
}

pub fn message_attr(ast: &mut syn::Item, types: Vec<syn::Path>) -> quote::Tokens {
    match ast.node {
        syn::ItemKind::Struct(_, ref gen) => {
            let (item, error) = match types.len() {
                0 => (None, None),
                1 => (Some(syn::Ty::Path(None, types[0].clone())), None),
                2 => (Some(syn::Ty::Path(None, types[0].clone())),
                      Some(syn::Ty::Path(None, types[1].clone()))),
                _ => panic!("#[msg(type, type)] takes 2 parameters, given {}", types.len()),
            };

            let name = &ast.ident;
            let (impl_generics, ty_generics, where_clause) = gen.split_for_impl();

            let item_type = item.unwrap_or(syn::Ty::Tup(vec![]));
            let error_type = error.unwrap_or(syn::Ty::Tup(vec![]));
            let dummy_const = syn::Ident::new(
                format!("_impl_act_{}_{}", name, thread_rng().gen::<u32>()));

            return quote!{
                #[allow(non_upper_case_globals, unused_attributes,
                        unused_qualifications, unused_variables, unused_imports)]
                const #dummy_const: () = {
                    extern crate actix;

                    impl #impl_generics actix::ResponseType for #name #ty_generics #where_clause {
                        type Item = #item_type;
                        type Error = #error_type;
                    }
                };
            }
        },
        _ => (),
    }
    panic!("#[msg] can only be used with Struct")
}