kittybox-frontend-renderer: factor out the `assets` module into a file

Change-Id: I6138cfe8479ba8df9a1580049675c1dd84abdad1

1 files changed, 47 insertions, 0 deletions

diff --git a/templates/src/assets.rs b/templates/src/assets.rs
new file mode 100644
index 0000000..493c14d
--- /dev/null
+++ b/templates/src/assets.rs
@@ -0,0 +1,47 @@
+use axum::extract::Path;
+use axum::http::header::{CACHE_CONTROL, CONTENT_ENCODING, CONTENT_TYPE, X_CONTENT_TYPE_OPTIONS};
+use axum::http::StatusCode;
+use axum::response::{IntoResponse, Response};
+
+const ASSETS: include_dir::Dir<'static> = include_dir::include_dir!("$OUT_DIR/");
+const CACHE_FOR_A_DAY: &str = "max-age=86400";
+const GZIP: &str = "gzip";
+
+pub async fn statics(Path(path): Path<String>) -> Response {
+    let content_type: &'static str = if path.ends_with(".js") {
+        "application/javascript"
+    } else if path.ends_with(".css") {
+        "text/css"
+    } else if path.ends_with(".html") {
+        "text/html; charset=\"utf-8\""
+    } else {
+        "application/octet-stream"
+    };
+
+    match ASSETS.get_file(path.clone() + ".gz") {
+        Some(file) => (
+            StatusCode::OK,
+            [
+                (CONTENT_TYPE, content_type),
+                (CONTENT_ENCODING, GZIP),
+                (CACHE_CONTROL, CACHE_FOR_A_DAY),
+                (X_CONTENT_TYPE_OPTIONS, "nosniff"),
+            ],
+            file.contents(),
+        )
+            .into_response(),
+        None => match ASSETS.get_file(path) {
+            Some(file) => (
+                StatusCode::OK,
+                [
+                    (CONTENT_TYPE, content_type),
+                    (CACHE_CONTROL, CACHE_FOR_A_DAY),
+                    (X_CONTENT_TYPE_OPTIONS, "nosniff"),
+                ],
+                file.contents(),
+            )
+                .into_response(),
+            None => StatusCode::NOT_FOUND.into_response(),
+        },
+    }
+}