tokio-rs · frederikhors · Aug 24, 2022
diff --git a/examples/rust-embed-spa/.gitignore b/examples/rust-embed-spa/.gitignore
@@ -0,0 +1 @@
+/target
diff --git a/examples/rust-embed-spa/Cargo.toml b/examples/rust-embed-spa/Cargo.toml
@@ -0,0 +1,12 @@
+[package]
+name = "rust_embed_spa"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+axum = "0.5.15"
+mime_guess = "2.0.4"
+rust-embed = "6.4.0"
+tokio = { version = "1.20.1", features = ["full"] }
diff --git a/examples/rust-embed-spa/README.md b/examples/rust-embed-spa/README.md
@@ -0,0 +1 @@
+A small example for hosting single page application (SPA) files with [axum](https://github.com/tokio-rs/axum) and [rust-embed](https://github.com/pyrossh/rust-embed).
diff --git a/examples/rust-embed-spa/assets/index.html b/examples/rust-embed-spa/assets/index.html
@@ -0,0 +1,15 @@
+<!DOCTYPE html>
+<html lang="en">
+
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Single Page Application</title>
+</head>
+
+<body>
+    I'm the body!
+    <script src="./js/script.js"></script>
+</body>
+
+</html>
diff --git a/examples/rust-embed-spa/assets/js/script.js b/examples/rust-embed-spa/assets/js/script.js
@@ -0,0 +1,3 @@
+var elem = document.createElement("div");
+elem.innerHTML = "I'm the JS script!<br>" + new Date();
+document.body.appendChild(elem);
diff --git a/examples/rust-embed-spa/src/main.rs b/examples/rust-embed-spa/src/main.rs
@@ -0,0 +1,73 @@
+use axum::{
+    body::{boxed, Full},
+    handler::Handler,
+    http::{header, StatusCode, Uri},
+    response::{IntoResponse, Response},
+    routing::Router,
+};
+use mime_guess;
+use rust_embed::RustEmbed;
+use std::net::SocketAddr;
+
+static INDEX_HTML: &str = "index.html";
+
+#[derive(RustEmbed)]
+#[folder = "assets/"]
+struct Assets;
+
+#[tokio::main]
+async fn main() {
+    let app = Router::new().fallback(static_handler.into_service());
+
+    let addr = SocketAddr::from(([127, 0, 0, 1], 3000));
+    println!("listening on {}", addr);
+    axum::Server::bind(&addr)
+        .serve(app.into_make_service())
+        .await
+        .unwrap();
+}
+
+async fn static_handler(uri: Uri) -> impl IntoResponse {
+    let path = uri.path().trim_start_matches('/').to_string();
+
+    if path.is_empty() || path == INDEX_HTML {
+        return index_html().await.into_response();
+    }
+
+    match Assets::get(path.as_str()) {
+        Some(content) => {
+            let body = boxed(Full::from(content.data));
+            let mime = mime_guess::from_path(path).first_or_octet_stream();
+
+            Response::builder()
+                .header(header::CONTENT_TYPE, mime.as_ref())
+                .body(body)
+                .unwrap()
+        }
+        None => {
+            if path.contains('.') {
+                return not_found().await.into_response();
+            }
+
+            index_html().await.into_response()
+        }
+    }
+}
+
+async fn index_html() -> impl IntoResponse {
+    match Assets::get(INDEX_HTML) {
+        Some(content) => {
+            let body = boxed(Full::from(content.data));
+
+            Response::builder()
+                .header(header::CONTENT_TYPE, "text/html")
+                .body(body)
+                .unwrap()
+        }
+        None => not_found().await.into_response(),
+    }
+}
+
+async fn not_found() -> impl IntoResponse {
+    (StatusCode::NOT_FOUND, "404")
+}