1 files changed, 57 insertions, 0 deletions
diff --git a/src/routes.rs b/src/routes.rs
new file mode 100644
index 0000000..b8f855a
--- /dev/null
+++ b/src/routes.rs
@@ -0,0 +1,57 @@
+use std::sync::Arc;
+use axum::{Router, response::{Response, IntoResponse}, extract::Query, Extension, http::{HeaderName, HeaderValue}, routing::get};
+use serde::Deserialize;
+use tera::{Context, Tera};
+use tower_http::services::ServeDir;
+use crate::State;
+
+pub fn router() -> Router {
+    Router::new()
+        .fallback(not_found)
+        .route("/", get(embed))
+        .nest_service("/static", ServeDir::new("public/static"))
+}
+
+#[derive(Deserialize, Debug)]
+struct Params {
+    name: String,
+    theme: Option<String>
+}
+
+async fn not_found() -> Response {
+    "do not the".into_response()
+}
+
+async fn embed(Extension(state): Extension<Arc<State>>, Query(params): Query<Params>) -> Response {
+    let Some(page) = state.pages.get(&params.name) else {
+        return not_found().await
+    };
+
+    let theme = match &params.theme {
+        Some(t) => t,
+        None => "dark"
+    };
+
+    let mut ctx = Context::default();
+    ctx.insert("prev", page.0.as_str());
+    ctx.insert("next", page.1.as_str());
+    ctx.insert("theme", theme);
+
+    render(&state.templates, &ctx, "embed.html")
+}
+
+pub fn render(tera: &Tera, ctx: &Context, template: &str) -> Response {
+    let rendered = match tera.render(template, ctx) {
+        Ok(r) => r,
+        Err(err) => {
+            return format!("{err:?}").into_response()
+        }
+    };
+    let mut res = rendered.into_response();
+    res.headers_mut().insert(
+        HeaderName::from_static("content-type"),
+        HeaderValue::from_static("text/html")
+    );
+    res
+}
+