use cfg_if::cfg_if; cfg_if! { if #[cfg(feature = "ssr")] { use axum::{ body::{boxed, Body, BoxBody}, extract::Extension, response::IntoResponse, http::{Request, Response, StatusCode, Uri}, }; use axum::response::Response as AxumResponse; use tower::ServiceExt; use tower_http::services::ServeDir; use std::sync::Arc; use leptos::{LeptosOptions, Errors, view}; use crate::app::{App, AppProps}; pub async fn file_and_error_handler(uri: Uri, Extension(options): Extension>, req: Request) -> AxumResponse { let options = &*options; let root = options.site_root.clone(); let res = get_static_file(uri.clone(), &root).await.unwrap(); if res.status() == StatusCode::OK { res.into_response() } else{ let handler = leptos_axum::render_app_to_stream( options.to_owned(), move |cx| view!{ cx, } ); handler(req).await.into_response() } } async fn get_static_file(uri: Uri, root: &str) -> Result, (StatusCode, String)> { let req = Request::builder().uri(uri.clone()).body(Body::empty()).unwrap(); // `ServeDir` implements `tower::Service` so we can call it with `tower::ServiceExt::oneshot` // This path is relative to the cargo root match ServeDir::new(root).oneshot(req).await { Ok(res) => Ok(res.map(boxed)), Err(err) => Err(( StatusCode::INTERNAL_SERVER_ERROR, format!("Something went wrong: {err}"), )), } } }}