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
|
// This file is part of Windmark <https://github.com/gemrest/windmark>.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, version 3.
//
// This program is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
//
// Copyright (C) 2022-2023 Fuwn <[email protected]>
// SPDX-License-Identifier: GPL-3.0-only
use proc_macro::TokenStream;
use quote::quote;
use syn::punctuated::Punctuated;
pub fn fields(arguments: TokenStream, item: syn::ItemStruct) -> TokenStream {
let field_initializers = syn::parse_macro_input!(
arguments as super::parser::FieldInitializers<syn::Expr>
);
let router_identifier = item.ident;
let (named_fields, has_fields) = match item.fields {
syn::Fields::Named(fields) => (fields, true),
syn::Fields::Unit =>
(
syn::FieldsNamed {
brace_token: syn::token::Brace::default(),
named: Punctuated::default(),
},
false,
),
syn::Fields::Unnamed(_) =>
panic!(
"`#[rossweisse::router]` can only be used on `struct`s with named \
fields or unit structs"
),
};
let mut default_expressions = vec![];
let new_method_fields = named_fields.named.iter().map(|field| {
let name = &field.ident;
let initialiser = field_initializers
.0
.iter()
.find(|initialiser| initialiser.ident == name.clone().unwrap())
.map_or_else(
|| {
default_expressions.push({
let default_expression: syn::Expr =
syn::parse_quote! { ::std::default::Default::default() };
default_expression
});
default_expressions.last().unwrap()
},
|initialiser| &initialiser.expr,
);
quote! {
#name: #initialiser,
}
});
let new_methods = if has_fields {
quote! {
fn _new() -> Self {
Self {
#(#new_method_fields)*
router: ::windmark::router::Router::new(),
}
}
}
} else {
quote! {
fn _new() -> Self {
Self {
router: ::windmark::router::Router::new(),
}
}
}
};
let output_fields = named_fields.named;
let output = quote! {
struct #router_identifier {
#output_fields
router: ::windmark::router::Router,
}
impl #router_identifier {
#new_methods
pub async fn run(&mut self) -> Result<(), Box<dyn std::error::Error>> {
self.router.run().await
}
pub fn router(&mut self) -> &mut ::windmark::router::Router {
&mut self.router
}
}
};
output.into()
}
|