feat: wrap with html for non curl clients

ozwaldorf · Oct 12, 2024 · f2596ba · f2596ba
1 parent dc827f4
commit f2596ba
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 9 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -23,4 +23,4 @@ pad = "0.1"
 humanize-bytes = "1.0"
 humantime = "2.1"
 urlencoding = "2.1.3"
-
+htmlescape = "0.3.1"
diff --git a/src/main.rs b/src/main.rs
@@ -1,4 +1,4 @@
-use std::io::{BufRead, Write};
+use std::io::{BufRead, Read, Write};
 use std::time::{Duration, SystemTime};
 
 use fastly::cache::simple::CacheEntry;
@@ -157,13 +157,31 @@ fn handle_get(req: Request) -> Result<Response, Error> {
         // Usage page
         Some("") => {
             let usage = get_usage(&host)?;
-            Ok(Response::from_body(usage)
-                .with_content_type(mime::TEXT_PLAIN_UTF_8)
-                .with_header(
-                    // Some browsers will set the title to this header
-                    header::CONTENT_DISPOSITION,
-                    r#"inline; filename="no bs pastebin.txt"; filename*=UTF-8''no%20bs%20pastebin.txt"#,
-                ))
+
+            // For all other clients other than curl, wrap with html (ie, browsers)
+            if let Some(agent) = req.get_header_str("user-agent") {
+                if !agent.starts_with("curl") {
+                    let wrapped = "\
+<!DOCTYPE html>
+<html>
+    <head>
+        <title>no bs pastebin</title>
+    </head>
+    <body>
+        <pre>"
+                        .to_string()
+                        + htmlescape::encode_minimal(&String::from_utf8_lossy(&usage.into_bytes()))
+                            .as_str()
+                        + "\
+        </pre>
+    </body>
+</html>";
+
+                    return Ok(Response::new().with_body_text_html(&wrapped));
+                }
+            }
+
+            Ok(Response::from_body(usage).with_content_type(mime::TEXT_PLAIN_UTF_8))
         },
 
         // Privacy policy page