beet_router_parser 0.0.5

Tools for building reactive structures
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

pub use crate::prelude::*;
use anyhow::Result;
use quote::quote;
use std::path::Path;
use sweet::prelude::*;
use syn::File;




pub struct ParseDirRoutes;


impl ParseDirRoutes {
	pub fn build_string(config: &CollectRoutes, path: &Path) -> Result<String> {
		let routes_dir_name = config.routes_dir.to_string_lossy();
		let page_routes = ReadDir::files(path)?
			.into_iter()
			.map(|path| ParseFileRoutes::parse(&routes_dir_name, path))
			.collect::<Result<Vec<_>>>()?
			.into_iter()
			.flatten();

		let router_ident = &config.file_router_ident;
		let router_ident: syn::Type = syn::parse_str(router_ident)?;

		// let ident_tokens =
		// let ident = syn::Ident::new(router_ident, Span::call_site());
		// let syn_path: syn::Path = parse_quote!(#router_ident);
		// let syn_path = syn_path.to_token_stream();

		let prefix_tokens: File = config
			.file_router_tokens
			.as_ref()
			.map(|tokens| syn::parse_str(tokens))
			.unwrap_or_else(|| {
				Ok(syn::parse_quote! {
					use beet::prelude::*;
				})
			})?;

		let dir_idents = ReadDir::dirs(path)?
			.into_iter()
			.map(|path| {
				let name = path
					.file_stem()
					.expect("dir has no stem")
					.to_string_lossy();
				syn::Ident::new(&name, proc_macro2::Span::call_site())
			})
			.collect::<Vec<_>>();
		let include_files = ReadDir::files(path)?
			.into_iter()
			.filter(|path| {
				path.extension().map(|ext| ext == "rs").unwrap_or(false)
					&& path.file_name().unwrap() != "mod.rs"
			})
			.map(|path| {
				let name = path
					.file_stem()
					.expect("file has no stem")
					.to_string_lossy();
				let ident =
					syn::Ident::new(&name, proc_macro2::Span::call_site());
				quote! {pub mod #ident; }
			});
		let include_dirs =
			dir_idents.iter().map(|ident| quote! {pub mod #ident;});
		let collect_dirs = dir_idents.iter().map(|ident| {
			quote! {	#ident::collect_file_routes(router);}
		});

		let file: File = syn::parse_quote! {
			//! 🥁🥁🥁 This file has been auto generated by the Beet router.
			//! 🥁🥁🥁 Any changes will be overridden if the file is regenerated.
			// routes
			#(#include_files)*
			// Sub routes
			#(#include_dirs)*
			#prefix_tokens
			pub fn collect_file_routes(router: &mut #router_ident) {
				#(router.add_route(#page_routes);)*
				#(#collect_dirs)*
			}
		};
		let file = prettyplease::unparse(&file);

		Ok(file)
	}
}