rossweisse/src/implementations/router/methods.rs


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

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()
}