mirror of
https://github.com/tuono-labs/tuono
synced 2026-07-25 12:52:47 -07:00
feat: create route handler macro
This commit is contained in:
+3
-1
@@ -3,7 +3,9 @@ resolver = "2"
|
||||
members = [
|
||||
"crates/cli",
|
||||
"crates/axum_bundler",
|
||||
"crates/bundler"
|
||||
"crates/bundler",
|
||||
"crates/tuono_lib",
|
||||
"crates/tuono_lib_macros"
|
||||
]
|
||||
exclude = [
|
||||
"examples/playground"
|
||||
|
||||
@@ -16,6 +16,12 @@ Why Tuono? Just a badass name.
|
||||
cargo install tuono
|
||||
```
|
||||
|
||||
## Create a new project
|
||||
|
||||
```
|
||||
tuono new
|
||||
```
|
||||
|
||||
## Development
|
||||
|
||||
```
|
||||
|
||||
@@ -5,25 +5,18 @@ use std::io::prelude::*;
|
||||
use std::path::Path;
|
||||
use std::path::PathBuf;
|
||||
|
||||
const ROOT_FOLDER: &'static str = "src/routes";
|
||||
|
||||
const AXUM_ENTRY_POINT: &'static str = r##"
|
||||
// File automatically generated
|
||||
// Do not manually change it
|
||||
|
||||
use axum::response::Html;
|
||||
use axum::{http::Uri, routing::get, Router};
|
||||
use ssr_rs::Ssr;
|
||||
use std::cell::RefCell;
|
||||
use std::fs::read_to_string;
|
||||
use tower_http::services::ServeDir;
|
||||
use tuono_lib::{ssr, Ssr};
|
||||
|
||||
thread_local! {
|
||||
static SSR: RefCell<Ssr<'static, 'static>> = RefCell::new(
|
||||
Ssr::from(
|
||||
read_to_string("./out/server/server-main.js").unwrap(),
|
||||
""
|
||||
).unwrap()
|
||||
)
|
||||
}
|
||||
// MODULE_IMPORTS
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() {
|
||||
@@ -48,18 +41,13 @@ async fn catch_all(uri: Uri) -> Html<String> {
|
||||
}}"#
|
||||
);
|
||||
|
||||
let result = SSR.with(|ssr| ssr.borrow_mut().render_to_string(Some(&payload)));
|
||||
let result = ssr::Js::SSR.with(|ssr| ssr.borrow_mut().render_to_string(Some(&payload)));
|
||||
|
||||
match result {
|
||||
Ok(html) => Html(html),
|
||||
_ => Html("500 internal server error".to_string()),
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// ROUTE_DECLARATIONS
|
||||
|
||||
|
||||
"##;
|
||||
|
||||
fn create_main_file(base_path: &Path, bundled_file: &String) {
|
||||
@@ -71,8 +59,6 @@ fn create_main_file(base_path: &Path, bundled_file: &String) {
|
||||
.expect("write failed");
|
||||
}
|
||||
|
||||
fn clean_up_source_file(file: &mut String) {}
|
||||
|
||||
fn get_route_name<'a>(path: &PathBuf, base_path: &Path) -> String {
|
||||
let base_path_str = base_path.to_str().unwrap();
|
||||
if base_path_str == "." {
|
||||
@@ -96,9 +82,9 @@ fn get_route_name<'a>(path: &PathBuf, base_path: &Path) -> String {
|
||||
.replace("index", "")
|
||||
}
|
||||
|
||||
fn collect_handlers(base_path: &Path) -> (String, HashMap<String, String>) {
|
||||
let mut declarations = String::from("// ROUTE_DECLARATIONS\n");
|
||||
let mut fn_map: HashMap<String, String> = HashMap::new();
|
||||
fn collect_handlers(base_path: &Path) -> HashMap<String, String> {
|
||||
// <path, name>
|
||||
let mut mods_map: HashMap<String, String> = HashMap::new();
|
||||
|
||||
let _: Vec<_> = glob(base_path.join("src/routes/**/*.rs").to_str().unwrap())
|
||||
.unwrap()
|
||||
@@ -106,29 +92,45 @@ fn collect_handlers(base_path: &Path) -> (String, HashMap<String, String>) {
|
||||
let entry = entry.unwrap();
|
||||
|
||||
let route_name = get_route_name(&entry, base_path);
|
||||
let mut file_content = fs::read_to_string(entry).unwrap();
|
||||
clean_up_source_file(&mut file_content);
|
||||
let hash = fastmurmur3::hash(file_content.as_bytes());
|
||||
let new_fn_name = format!("fn_{}", hash);
|
||||
|
||||
let file_content = file_content.replace("get_server_side_props", &new_fn_name);
|
||||
// Remove first slash
|
||||
let mut module = route_name.as_str().chars();
|
||||
module.next();
|
||||
|
||||
fn_map.insert(route_name, new_fn_name);
|
||||
|
||||
declarations.push_str(&file_content);
|
||||
mods_map.insert(
|
||||
module.as_str().to_string(),
|
||||
entry
|
||||
.to_str()
|
||||
.unwrap()
|
||||
.replace(base_path.to_str().unwrap(), ""),
|
||||
);
|
||||
})
|
||||
.collect();
|
||||
|
||||
println!("{fn_map:?}");
|
||||
dbg!(&mods_map);
|
||||
|
||||
return (declarations, fn_map);
|
||||
return mods_map;
|
||||
}
|
||||
|
||||
fn create_axum_routes_declaration(routes: HashMap<String, String>) -> String {
|
||||
fn create_axum_routes_declaration(routes: &HashMap<String, String>) -> String {
|
||||
let mut route_declarations = String::from("// ROUTE_BUILDER\n");
|
||||
|
||||
for (route, handler) in routes.iter() {
|
||||
route_declarations.push_str(&format!(r#".route("{route}", get({handler}))"#))
|
||||
for (route, _path) in routes.iter() {
|
||||
route_declarations.push_str(&format!(r#".route("/{route}", get({route}::route))"#))
|
||||
}
|
||||
|
||||
route_declarations
|
||||
}
|
||||
|
||||
fn create_modules_declaration(routes: &HashMap<String, String>) -> String {
|
||||
let mut route_declarations = String::from("// MODULE_IMPORTS\n");
|
||||
|
||||
for (route, path) in routes.iter() {
|
||||
route_declarations.push_str(&format!(
|
||||
r#"#[path="..{path}"]
|
||||
mod {route};
|
||||
"#
|
||||
))
|
||||
}
|
||||
|
||||
route_declarations
|
||||
@@ -139,14 +141,11 @@ pub fn bundle() {
|
||||
|
||||
let base_path = std::env::current_dir().unwrap();
|
||||
|
||||
let (handlers, fn_map) = collect_handlers(&base_path);
|
||||
let mods = collect_handlers(&base_path);
|
||||
|
||||
let bundled_file = AXUM_ENTRY_POINT.replace(
|
||||
"// ROUTE_BUILDER\n",
|
||||
&create_axum_routes_declaration(fn_map),
|
||||
);
|
||||
|
||||
let bundled_file = bundled_file.replace("// ROUTE_DECLARATIONS", &handlers[..]);
|
||||
let bundled_file = AXUM_ENTRY_POINT
|
||||
.replace("// ROUTE_BUILDER\n", &create_axum_routes_declaration(&mods))
|
||||
.replace("// MODULE_IMPORTS\n", &create_modules_declaration(&mods));
|
||||
|
||||
create_main_file(&base_path, &bundled_file);
|
||||
}
|
||||
|
||||
@@ -9,6 +9,7 @@ use watchexec_supervisor::job::start_job;
|
||||
|
||||
#[tokio::main]
|
||||
pub async fn watch() -> Result<()> {
|
||||
bundle();
|
||||
let (job, _) = start_job(Arc::new(Command {
|
||||
program: Program::Exec {
|
||||
prog: "cargo".into(),
|
||||
|
||||
@@ -0,0 +1,13 @@
|
||||
[package]
|
||||
name = "tuono_lib"
|
||||
version = "0.0.1"
|
||||
edition = "2021"
|
||||
authors = ["V. Ageno <valerioageno@yahoo.it>"]
|
||||
|
||||
[lib]
|
||||
name = "tuono_lib"
|
||||
path = "src/lib.rs"
|
||||
|
||||
[dependencies]
|
||||
ssr_rs = "0.5.1"
|
||||
tuono_lib_macros = { path = "../tuono_lib_macros"}
|
||||
@@ -0,0 +1,3 @@
|
||||
# Tuonolib
|
||||
|
||||
This project exposes the interfaces to easily handle the backend build.
|
||||
@@ -0,0 +1,8 @@
|
||||
pub mod ssr;
|
||||
pub use ssr_rs::Ssr;
|
||||
pub use tuono_lib_macros::handler;
|
||||
|
||||
pub enum Response {
|
||||
Redirect(String),
|
||||
Props(String),
|
||||
}
|
||||
@@ -0,0 +1,15 @@
|
||||
use ssr_rs::Ssr;
|
||||
use std::cell::RefCell;
|
||||
use std::fs::read_to_string;
|
||||
|
||||
pub struct Js;
|
||||
impl Js {
|
||||
thread_local! {
|
||||
pub static SSR: RefCell<Ssr<'static, 'static>> = RefCell::new(
|
||||
Ssr::from(
|
||||
read_to_string("./out/server/server-main.js").unwrap(),
|
||||
""
|
||||
).unwrap()
|
||||
)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,12 @@
|
||||
[package]
|
||||
name = "tuono_lib_macros"
|
||||
version = "0.0.1"
|
||||
edition = "2021"
|
||||
|
||||
[lib]
|
||||
crate_type = ["proc-macro"]
|
||||
proc-marco = true
|
||||
|
||||
[dependencies]
|
||||
syn = { version = "1.0.59", features = ["full"] }
|
||||
quote = "1.0"
|
||||
@@ -0,0 +1 @@
|
||||
# Tuonolib Macro
|
||||
@@ -0,0 +1,35 @@
|
||||
use proc_macro::TokenStream;
|
||||
use quote::quote;
|
||||
use syn::{parse_macro_input, ItemFn};
|
||||
|
||||
pub fn handler_core(_args: TokenStream, item: TokenStream) -> TokenStream {
|
||||
let item = parse_macro_input!(item as ItemFn);
|
||||
|
||||
let fn_name = item.clone().sig.ident;
|
||||
|
||||
quote! {
|
||||
use tuono_lib::Response;
|
||||
use axum::response::Html;
|
||||
use axum::http::Uri;
|
||||
use tuono_lib::ssr;
|
||||
|
||||
#item
|
||||
|
||||
pub async fn route(uri: Uri) -> Html<String> {
|
||||
println!("Custom handler");
|
||||
let props = #fn_name();
|
||||
|
||||
let res = match props {
|
||||
Response::Props(val) => ssr::Js::SSR.with(|ssr| ssr.borrow_mut().render_to_string(Some(val.as_str()))),
|
||||
_ => Ok("500 Internal server error".to_string())
|
||||
};
|
||||
|
||||
match res {
|
||||
Ok(html) => Html(html),
|
||||
_ => Html("500".to_string())
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
.into()
|
||||
}
|
||||
@@ -0,0 +1,11 @@
|
||||
extern crate proc_macro;
|
||||
use proc_macro::TokenStream;
|
||||
|
||||
mod handler;
|
||||
|
||||
#[proc_macro_attribute]
|
||||
pub fn handler(args: TokenStream, item: TokenStream) -> TokenStream {
|
||||
handler::handler_core(args, item)
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user