#[cfg(feature = "embed-ui")]
use axum::{
body::Body,
http::{header, Request, Response, StatusCode},
Router,
};
#[cfg(not(feature = "embed-ui"))]
use axum::Router;
#[cfg(feature = "embed-ui")]
use rust_embed::RustEmbed;
#[cfg(feature = "embed-ui")]
use std::path::Path;
#[cfg(feature = "embed-ui")]
#[derive(RustEmbed)]
#[folder = "ui"]
#[prefix = "suite"]
struct EmbeddedUi;
#[cfg(feature = "embed-ui")]
fn get_mime_type(path: &str) -> &'static str {
let ext = Path::new(path)
.extension()
.and_then(|e| e.to_str())
.unwrap_or("");
match ext {
"html" | "htm" => "text/html; charset=utf-8",
"css" => "text/css; charset=utf-8",
"js" | "mjs" => "application/javascript; charset=utf-8",
"json" => "application/json; charset=utf-8",
"png" => "image/png",
"jpg" | "jpeg" => "image/jpeg",
"gif" => "image/gif",
"svg" => "image/svg+xml",
"ico" => "image/x-icon",
"woff" => "font/woff",
"woff2" => "font/woff2",
"ttf" => "font/ttf",
"otf" => "font/otf",
"eot" => "application/vnd.ms-fontobject",
"webp" => "image/webp",
"mp4" => "video/mp4",
"webm" => "video/webm",
"mp3" => "audio/mpeg",
"wav" => "audio/wav",
"ogg" => "audio/ogg",
"pdf" => "application/pdf",
"xml" => "application/xml",
"txt" => "text/plain; charset=utf-8",
"md" => "text/markdown; charset=utf-8",
"wasm" => "application/wasm",
_ => "application/octet-stream",
}
}
#[cfg(feature = "embed-ui")]
async fn serve_embedded_file(req: Request) -> Response {
let path = req.uri().path().trim_start_matches('/');
let file_path = if path.is_empty() || path == "/" {
"index.html"
} else {
path
};
let file_path = file_path.strip_prefix("suite/").unwrap_or(file_path);
log::trace!("Serving embedded file: {}", file_path);
let try_paths = [
file_path.to_string(),
format!("{}/index.html", file_path.trim_end_matches('/')),
format!("{}.html", file_path),
];
for try_path in &try_paths {
if let Some(content) = EmbeddedUi::get(try_path) {
let mime = get_mime_type(try_path);
log::trace!("Found embedded file: {} with MIME type: {}", try_path, mime);
return Response::builder()
.status(StatusCode::OK)
.header(header::CONTENT_TYPE, mime)
.header(header::CACHE_CONTROL, "public, max-age=3600")
.body(Body::from(content.data.into_owned()))
.unwrap_or_else(|_| {
Response::builder()
.status(StatusCode::INTERNAL_SERVER_ERROR)
.body(Body::from("Internal Server Error"))
.unwrap()
});
}
}
log::warn!("Embedded file not found: {} (tried paths: {:?})", file_path, try_paths);
Response::builder()
.status(StatusCode::NOT_FOUND)
.header(header::CONTENT_TYPE, "text/html; charset=utf-8")
.body(Body::from(
r#"
404 Not Found
404 - Not Found
The requested file was not found in embedded UI.
Go to Home
"#,
))
.unwrap()
}
#[cfg(feature = "embed-ui")]
pub fn embedded_ui_router() -> Router {
use axum::routing::any;
Router::new().fallback(any(serve_embedded_file))
}
#[cfg(feature = "embed-ui")]
pub fn has_embedded_ui() -> bool {
let has_index = EmbeddedUi::get("index.html").is_some();
if has_index {
log::info!("Embedded UI detected - index.html found");
} else {
log::warn!("No embedded UI found - index.html not embedded");
}
has_index
}
#[cfg(feature = "embed-ui")]
pub fn list_embedded_files() -> Vec {
let files: Vec = EmbeddedUi::iter().map(|f| f.to_string()).collect();
log::debug!("Embedded UI contains {} files", files.len());
files
}
#[cfg(not(feature = "embed-ui"))]
pub fn has_embedded_ui() -> bool {
false
}
#[cfg(not(feature = "embed-ui"))]
pub fn list_embedded_files() -> Vec {
Vec::new()
}
#[cfg(not(feature = "embed-ui"))]
pub fn embedded_ui_router() -> Router {
Router::new()
}