blob: 4f9bbdfce84d45f5789d4355d672a4674e21f608 (
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
|
// 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;
pub fn methods(_arguments: TokenStream, item: syn::ItemImpl) -> TokenStream {
let routes = item
.items
.iter()
.filter_map(|item| {
if let syn::ImplItem::Fn(method) = item {
if method
.attrs
.iter()
.any(|attribute| attribute.path().is_ident("route"))
{
Some(method.sig.ident.clone())
} else {
None
}
} else {
None
}
})
.collect::<Vec<_>>();
let (implementation_generics, type_generics, where_clause) =
item.generics.split_for_impl();
let name = &item.self_ty;
let route_paths = routes
.iter()
.map(|route| format!("/{}", route))
.collect::<Vec<_>>();
quote::quote! {
#item
impl #implementation_generics #name #type_generics #where_clause {
pub fn new() -> Self {
let mut router = Self::_new();
#(
router.router.mount(#route_paths, |context| {
Self::#routes(context)
});
)*
router
}
}
}
.into()
}
|