feat: cache mime types for valid assets, use octet-stream content type

src/web/assets.rs

@@ -3,13 +3,9 @@
 //! This module handles serving static assets that are embedded into the binary
 //! at compile time using rust-embed.
 
-use axum::{
-    extract::Path,
-    http::{StatusCode, header},
-    response::{Html, IntoResponse, Response},
-};
+use dashmap::DashMap;
+use once_cell::sync::Lazy;
 use rust_embed::RustEmbed;
-use tracing::debug;
 
 /// Embedded web assets from the dist directory
 #[derive(RustEmbed)]
@@ -18,18 +14,24 @@ use tracing::debug;
 #[exclude = "*.map"]
 pub struct WebAssets;
 
-const ASSET_EXTENSIONS: &[&str] = &[
-    "js", "css", "png", "jpg", "jpeg", "gif", "svg", "ico", "woff", "woff2", "ttf", "eot",
-];
+/// Global cache for MIME types to avoid repeated mime_guess lookups
+static MIME_CACHE: Lazy<DashMap<String, Option<String>>> = Lazy::new(DashMap::new);
 
-/// Check if a path should be served as a static asset
-pub fn is_asset_path(path: &str) -> bool {
-    if !path.starts_with("/assets/") {
-        return path.eq("index.html");
+/// Get cached MIME type for a file path, caching on-demand
+/// Returns None if the MIME type is text/plain or if no MIME type could be determined
+pub fn get_mime_type_cached(path: &str) -> Option<String> {
+    // Check cache first
+    if let Some(cached) = MIME_CACHE.get(path) {
+        return cached.value().as_ref().cloned();
     }
 
-    match path.split_once('.') {
-        Some((_, extension)) => ASSET_EXTENSIONS.contains(&extension),
-        None => false,
-    }
+    // Perform MIME guess and cache the result
+    let result = mime_guess::from_path(path)
+        .first()
+        .map(|mime| mime.to_string());
+
+    // Cache the result
+    MIME_CACHE.insert(path.to_string(), result.clone());
+
+    result
 }

src/web/routes.rs

@@ -14,9 +14,9 @@ use tower_http::{
     cors::{Any, CorsLayer},
     trace::TraceLayer,
 };
-use tracing::{debug, info};
+use tracing::info;
 
-use crate::web::assets::WebAssets;
+use crate::web::assets::{WebAssets, get_mime_type_cached};
 
 use crate::banner::BannerApi;
 
@@ -82,9 +82,19 @@ async fn handle_spa_fallback(uri: Uri) -> Response {
     let path = uri.path().trim_start_matches('/');
 
     if let Some(content) = WebAssets::get(path) {
-        let mime_type = mime_guess::from_path(path).first_or_text_plain();
         let data = content.data.to_vec();
-        return ([(header::CONTENT_TYPE, mime_type.as_ref())], data).into_response();
+
+        // Use cached MIME type, only set Content-Type if we have a valid MIME type
+        let mime_type = get_mime_type_cached(path);
+        return (
+            [(
+                header::CONTENT_TYPE,
+                // For unknown types, set to application/octet-stream
+                mime_type.unwrap_or("application/octet-stream".to_string()),
+            )],
+            data,
+        )
+            .into_response();
     } else {
         // Any assets that are not found should be treated as a 404, not falling back to the SPA index.html
         if path.starts_with("assets/") {