(
+ cfg: ServeConfig,
+) -> BoxedFilter<(impl Reply,)> {
+ // Serve the dist folder and the index.html file
+ let serve_dir = warp::fs::dir("./dist");
- // transform it back into a string, saving us the allocation
- let msg = String::from_utf8(msg).map_err(|_| LiveViewError::SendingFailed)?;
-
- Ok(msg)
+ register_server_fns(cfg.server_fn_route.unwrap_or_default())
+ .or(warp::path::end()
+ .and(warp::get())
+ .map(move || warp::reply::html(dioxus_ssr_html(&cfg))))
+ .or(serve_dir)
+ .boxed()
}
-async fn transform_tx(message: String) -> Result {
- Ok(Message::text(message))
+#[derive(Debug)]
+struct FailedToReadBody(String);
+
+impl warp::reject::Reject for FailedToReadBody {}
+
+#[derive(Debug)]
+struct RecieveFailed(String);
+
+impl warp::reject::Reject for RecieveFailed {}
+
+async fn server_fn_handler(
+ server_context: DioxusServerContext,
+ function: Arc,
+ headers: HeaderMap,
+ body: Bytes,
+) -> Result, warp::Rejection> {
+ // Because the future returned by `server_fn_handler` is `Send`, and the future returned by this function must be send, we need to spawn a new runtime
+ let (resp_tx, resp_rx) = tokio::sync::oneshot::channel();
+ spawn_blocking({
+ move || {
+ tokio::runtime::Runtime::new()
+ .expect("couldn't spawn runtime")
+ .block_on(async {
+ let resp = match function(server_context, &body).await {
+ Ok(serialized) => {
+ // if this is Accept: application/json then send a serialized JSON response
+ let accept_header =
+ headers.get("Accept").and_then(|value| value.to_str().ok());
+ let mut res = Response::builder();
+ if accept_header == Some("application/json")
+ || accept_header
+ == Some(
+ "application/\
+ x-www-form-urlencoded",
+ )
+ || accept_header == Some("application/cbor")
+ {
+ res = res.status(StatusCode::OK);
+ }
+
+ let resp = match serialized {
+ Payload::Binary(data) => res
+ .header("Content-Type", "application/cbor")
+ .body(Bytes::from(data)),
+ Payload::Url(data) => res
+ .header(
+ "Content-Type",
+ "application/\
+ x-www-form-urlencoded",
+ )
+ .body(Bytes::from(data)),
+ Payload::Json(data) => res
+ .header("Content-Type", "application/json")
+ .body(Bytes::from(data)),
+ };
+
+ Box::new(resp.unwrap())
+ }
+ Err(e) => report_err(e),
+ };
+
+ if resp_tx.send(resp).is_err() {
+ eprintln!("Error sending response");
+ }
+ })
+ }
+ });
+ resp_rx.await.map_err(|err| {
+ warp::reject::custom(RecieveFailed(format!("Failed to recieve response {err}")))
+ })
+}
+
+fn report_err(e: E) -> Box {
+ Box::new(
+ Response::builder()
+ .status(StatusCode::INTERNAL_SERVER_ERROR)
+ .body(format!("Error: {}", e))
+ .unwrap(),
+ ) as Box
}
diff --git a/packages/server/src/lib.rs b/packages/server/src/lib.rs
index e7559363a..1cdd2f786 100644
--- a/packages/server/src/lib.rs
+++ b/packages/server/src/lib.rs
@@ -11,6 +11,8 @@ pub mod prelude {
pub use crate::adapters::axum_adapter::*;
#[cfg(feature = "salvo")]
pub use crate::adapters::salvo_adapter::*;
+ #[cfg(feature = "warp")]
+ pub use crate::adapters::warp_adapter::*;
#[cfg(feature = "ssr")]
pub use crate::serve::ServeConfig;
pub use crate::server_fn::{DioxusServerContext, ServerFn};