1 files changed, 37 insertions, 6 deletions
diff --git a/crates/chat-server/src/http.rs b/crates/chat-server/src/http.rs
index 1c7286f..324c593 100644
--- a/crates/chat-server/src/http.rs
+++ b/crates/chat-server/src/http.rs
@@ -17,6 +17,7 @@ use crate::service::Service;
 use anyhow::Result;
 use axum::{
     Json, Router,
+    body::Body,
     extract::{DefaultBodyLimit, State},
     http::{HeaderMap, HeaderValue, StatusCode},
     response::{IntoResponse, Response},
@@ -115,14 +116,25 @@ async fn asset(State(app): State<App>, uri: axum::http::Uri) -> Response {
             "text/css; charset=utf-8",
         ),
         "/crypto.wasm" => (
-            "target/wasm32-wasip1/release/stack-chat-crypto.wasm",
+            if app.root.join("bin/chat/crypto.wasm").is_file() {
+                "bin/chat/crypto.wasm"
+            } else {
+                "target/wasm32-wasip1/release/stack-chat-crypto.wasm"
+            },
             "application/wasm",
         ),
         "/javascript.html" => (
             "crates/chat-server/web/javascript.html",
             "text/html; charset=utf-8",
         ),
-        "/SOURCE.tar.gz" => ("var/build/chat-source.tar.gz", "application/gzip"),
+        "/SOURCE.tar.gz" => (
+            if app.root.join("bin/chat/SOURCE.tar.gz").is_file() {
+                "bin/chat/SOURCE.tar.gz"
+            } else {
+                "var/build/chat-source.tar.gz"
+            },
+            "application/gzip",
+        ),
         "/THIRD-PARTY-LICENSES.html" => {
             ("web/THIRD-PARTY-LICENSES.html", "text/html; charset=utf-8")
         }
@@ -130,10 +142,10 @@ async fn asset(State(app): State<App>, uri: axum::http::Uri) -> Response {
         "/LICENSE" => ("LICENSE", "text/plain; charset=utf-8"),
         _ => return secured(StatusCode::NOT_FOUND.into_response()),
     };
-    let Ok(bytes) = tokio::fs::read(app.root.join(path)).await else {
-        return secured(StatusCode::NOT_FOUND.into_response());
-    };
     if matches!(uri.path(), "/" | "/javascript.html") {
+        let Ok(bytes) = tokio::fs::read(app.root.join(path)).await else {
+            return secured(StatusCode::NOT_FOUND.into_response());
+        };
         let rendered = String::from_utf8(bytes)
             .map_err(anyhow::Error::from)
             .and_then(|source| stack_config::render(&source, &app.public));
@@ -142,7 +154,22 @@ async fn asset(State(app): State<App>, uri: axum::http::Uri) -> Response {
             |html| secured(([("content-type", mime)], html).into_response()),
         );
     }
-    secured(([("content-type", mime)], bytes).into_response())
+    let Ok(file) = tokio::fs::File::open(app.root.join(path)).await else {
+        return secured(StatusCode::NOT_FOUND.into_response());
+    };
+    let length = file.metadata().await.ok().map(|metadata| metadata.len());
+    let mut response = Response::new(Body::from_stream(tokio_util::io::ReaderStream::new(file)));
+    response
+        .headers_mut()
+        .insert("content-type", HeaderValue::from_static(mime));
+    if let Some(header_length) =
+        length.and_then(|value| HeaderValue::from_str(&value.to_string()).ok())
+    {
+        response
+            .headers_mut()
+            .insert("content-length", header_length);
+    }
+    secured(response)
 }
 
 pub(crate) async fn serve(
@@ -159,6 +186,10 @@ pub(crate) async fn serve(
         public,
     };
     let routes = Router::new()
+        .route(
+            "/healthz",
+            get(|| async { secured("ok\n".into_response()) }),
+        )
         .route("/api", post(api))
         .fallback(get(asset))
         .layer(DefaultBodyLimit::max(262_144))