blob: 18f639bf32a8a305cfc30acc8aa03a60a82f5f57 (
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
|
// 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,
mut item: syn::ItemImpl,
) -> TokenStream {
let routes = item
.items
.iter_mut()
.filter_map(|item| {
let syn::ImplItem::Fn(method) = item else {
return None;
};
let route_attrribute = method
.attrs
.iter()
.find(|attribute| attribute.path().is_ident("route"))?;
let arguments = quote::ToTokens::into_token_stream(route_attrribute)
.to_string()
.trim_end_matches(")]")
.trim_start_matches("#[route(")
.to_string();
if arguments == "index" {
method.sig.ident =
syn::Ident::new("__router_index", method.sig.ident.span());
}
Some(method.sig.ident.clone())
})
.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!(
"/{}",
if route == "__router_index" {
String::new()
} else {
route.to_string()
}
)
})
.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()
}
|