blob: e1a92b920aeeb4f344075e34571adf7b3ff22355 (
plain) (
blame)
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
|
// 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::parse_macro_input;
pub fn fields(arguments: TokenStream, item: syn::ItemStruct) -> TokenStream {
let field_initializers =
parse_macro_input!(arguments as super::parser::FieldInitializers);
let router_identifier = item.ident;
let named_fields = match item.fields {
syn::Fields::Named(fields) => fields,
_ =>
panic!(
"`#[rossweisse::router]` can only be used on `struct`s with named \
fields"
),
};
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(|initialiser| &initialiser.expr)
.unwrap_or_else(|| {
default_expressions.push({
let default_expression: syn::Expr =
syn::parse_quote! { ::std::default::Default::default() };
default_expression
});
default_expressions.last().unwrap()
});
quote! {
#name: #initialiser,
}
});
let new_methods = quote! {
fn _new() -> Self {
Self {
#(#new_method_fields)*
router: ::windmark::Router::new(),
}
}
pub async fn run(&mut self) -> Result<(), Box<dyn std::error::Error>> {
self.router.run().await
}
pub fn router(&mut self) -> &mut ::windmark::Router {
&mut self.router
}
};
let output_fields = named_fields.named;
let output = quote! {
struct #router_identifier {
#output_fields
router: ::windmark::Router,
}
impl #router_identifier {
#new_methods
}
};
output.into()
}
|