create a serveconfig builder

This commit is contained in:
Evan Almloff committed 2023-03-30 11:03:07 -05:00
1 parent 1d395d572f
commit 1be48c4aa8
4 files changed
+74 -39

No files matched your search

+5 -21
View File
@@ -7,26 +7,18 @@ use axum::{
routing::{get, post},
Router,
};
use dioxus_core::Component;
use server_fn::{Payload, ServerFunctionRegistry};
use tokio::task::spawn_blocking;
use crate::{
dioxus_ssr_html,
serve::ServeConfig,
server_fn::{DioxusServerContext, DioxusServerFnRegistry, ServerFnTraitObj},
};
pub trait DioxusRouterExt {
fn register_server_fns(self, server_fn_route: &'static str) -> Self;
fn serve_dioxus_application(
self,
title: &'static str,
application_name: &'static str,
base_path: Option<&'static str>,
head: Option<&'static str>,
server_fn_route: &'static str,
app: Component,
) -> Self;
fn serve_dioxus_application(self, cfg: ServeConfig) -> Self;
}
impl DioxusRouterExt for Router {
@@ -45,24 +37,16 @@ impl DioxusRouterExt for Router {
router
}
fn serve_dioxus_application(
self,
title: &'static str,
application_name: &'static str,
base_path: Option<&'static str>,
head: Option<&'static str>,
server_fn_route: &'static str,
app: Component,
) -> Self {
fn serve_dioxus_application(self, cfg: ServeConfig) -> Self {
use tower_http::services::ServeDir;
// Serve the dist folder and the index.html file
let serve_dir = ServeDir::new("dist");
self.register_server_fns(server_fn_route)
self.register_server_fns(cfg.server_fn_route.unwrap_or_default())
.nest_service("/", serve_dir)
.fallback_service(get(move || {
let rendered = dioxus_ssr_html(title, application_name, base_path, head, app);
let rendered = dioxus_ssr_html(cfg);
async move { Full::from(rendered) }
}))
}