From 3ed87eaba2353e1f29d76e1c32d896a2cb6e1867 Mon Sep 17 00:00:00 2001
From: JMARyA <jmarya@hydrar.de>
Date: Tue, 31 Dec 2024 17:41:52 +0100
Subject: [PATCH] txt route

---
 src/archive.rs   |  2 +-
 src/main.rs      |  3 ++-
 src/pages/mod.rs | 16 ++++++++++++++++
 3 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/src/archive.rs b/src/archive.rs
index 361705b..4dea32f 100644
--- a/src/archive.rs
+++ b/src/archive.rs
@@ -198,7 +198,7 @@ impl Document {
 
         let mut buf = Vec::new();
         std::fs::File::open(file_path)
-            .unwrap()
+            .ok()?
             .read_to_end(&mut buf)
             .unwrap();
         let content = String::from_utf8_lossy(&buf);
diff --git a/src/main.rs b/src/main.rs
index 07cb17a..2ae11f7 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -41,7 +41,8 @@ async fn launch() -> _ {
                 pages::render_website,
                 pages::domain_info_route,
                 pages::favicon_route,
-                pages::vector_search
+                pages::vector_search,
+                pages::render_txt_website
             ],
         )
         .manage(arc)
diff --git a/src/pages/mod.rs b/src/pages/mod.rs
index 9cdc829..6a2f77e 100644
--- a/src/pages/mod.rs
+++ b/src/pages/mod.rs
@@ -135,6 +135,22 @@ pub async fn domain_info_route(
     render_page(content, ctx).await
 }
 
+#[get("/txt/<domain>/<path..>?<time>")]
+pub async fn render_txt_website(
+    domain: &str,
+    path: PathBuf,
+    time: Option<&str>,
+    arc: &State<WebsiteArchive>,
+) -> Option<String> {
+    let document = arc.get_domain(domain).path(path.to_str().unwrap());
+
+    let content = document
+        .render_local(time.map(|time| time.to_string()))
+        .await;
+
+    content.map(|content_html| html2md::parse_html(&content_html))
+}
+
 /// Return archived version of `domain` / `path` at `time`
 #[get("/s/<domain>/<path..>?<time>")]
 pub async fn render_website(