sword-macros 0.3.0

Structured web framework built on top of tokio ecosystem, providing powerful features for building robust web applications.
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

pub mod shared;

#[cfg(feature = "event-handlers")]
pub mod event_handler;

#[cfg(feature = "grpc-controllers")]
pub mod grpc;

#[cfg(feature = "socketio-controllers")]
pub mod socketio;

#[cfg(feature = "socketio-controllers")]
pub use socketio::expand_on_handler;

#[cfg(feature = "web-controllers")]
pub mod web;

use proc_macro::TokenStream;
use quote::quote;
use shared::ControllerStruct;
#[cfg(any(
    feature = "web-controllers",
    feature = "socketio-controllers",
    feature = "grpc-controllers",
    feature = "event-handlers",
))]
use shared::ParsedControllerKind;
use syn::ItemStruct;

pub fn expand_controller(attr: TokenStream, item: TokenStream) -> syn::Result<TokenStream> {
    let input = syn::parse::<ItemStruct>(item.clone())?;
    let parsed_input = ControllerStruct::parse(attr, item)?;

    let controller_kind = &parsed_input.kind;

    let builder: proc_macro::TokenStream = match controller_kind {
        #[cfg(feature = "web-controllers")]
        ParsedControllerKind::Web { .. } => web::expand_web_controller(&parsed_input)?,

        #[cfg(feature = "socketio-controllers")]
        ParsedControllerKind::SocketIo { .. } => {
            socketio::expand_socketio_controller(&parsed_input)?
        }

        #[cfg(feature = "grpc-controllers")]
        ParsedControllerKind::Grpc { .. } => grpc::expand_grpc_controller(&parsed_input)?,

        #[cfg(feature = "event-handlers")]
        ParsedControllerKind::EventHandler { .. } => {
            event_handler::expand_event_handler(&parsed_input)?
        }

        #[cfg(not(any(
            feature = "web-controllers",
            feature = "socketio-controllers",
            feature = "grpc-controllers",
            feature = "event-handlers",
        )))]
        _ => {
            return Err(syn::Error::new(
                proc_macro2::Span::call_site(),
                "no controller feature is enabled on `sword-macros`; enable `web-controllers`, `socketio-controllers`, `grpc-controllers`, or `event-handlers`",
            ));
        }
    };

    let builder = proc_macro2::TokenStream::from(builder);

    let expanded = quote! {
        #input
        #builder
    };

    Ok(TokenStream::from(expanded))
}