Download button

src/cache.rs

@@ -29,6 +29,10 @@ impl Key {
     pub fn id(&self) -> String {
         self.id.to_string()
     }
+
+    pub fn extension(&self) -> String {
+        self.ext.clone()
+    }
 }
 
 impl TryFrom<Path<String>> for Key {

src/main.rs

@@ -21,6 +21,8 @@ mod web;
 
 #[derive(thiserror::Error, Debug)]
 pub enum Error {
+    #[error("axum http error: {0}")]
+    Axum(#[from] axum::http::Error),
     #[error("sqlite error: {0}")]
     Sqlite(#[from] rusqlite::Error),
     #[error("migrations error: {0}")]
@@ -68,7 +70,8 @@ impl From<Error> for StatusCode {
             | Error::TimeFormatting(_)
             | Error::Migration(_)
             | Error::SyntaxHighlighting(_)
-            | Error::SyntaxParsing(_) => StatusCode::INTERNAL_SERVER_ERROR,
+            | Error::SyntaxParsing(_)
+            | Error::Axum(_) => StatusCode::INTERNAL_SERVER_ERROR,
         }
     }
 }

src/rest.rs

@@ -1,8 +1,11 @@
 use crate::cache::Layer;
+use crate::highlight::DATA;
 use crate::id::Id;
 use crate::{Entry, Error, Router};
+use askama_axum::Response;
 use axum::extract::Path;
-use axum::http::StatusCode;
+use axum::headers::HeaderValue;
+use axum::http::{header, StatusCode};
 use axum::routing::{get, post};
 use axum::{Extension, Json};
 use rand::Rng;
@@ -57,9 +60,34 @@ async fn raw(Path(id): Path<String>, layer: Extension<Layer>) -> Result<String,
     Ok(layer.get_raw(Id::try_from(id.as_str())?).await?)
 }
 
+async fn download(
+    Path((id, extension)): Path<(String, String)>,
+    layer: Extension<Layer>,
+) -> Result<Response<String>, ErrorResponse> {
+    let raw_string = raw(Path(id.clone()), layer.clone()).await?;
+    let content_type = "text; charset=utf-8";
+
+    // validate Id and extension
+    Id::try_from(id.as_str())?;
+    let _ = DATA
+        .syntax_set
+        .find_syntax_by_extension(extension.as_str())
+        .ok_or(Error::IllegalCharacters)?;
+
+    let filename = format!("{}.{}", id, extension);
+    let content_disposition = format!("attachment; filename=\"{}\"", filename);
+
+    Ok(Response::builder()
+        .header(header::CONTENT_TYPE, HeaderValue::from_static(content_type))
+        .header(header::CONTENT_DISPOSITION, content_disposition)
+        .body(raw_string)
+        .map_err(Error::from)?)
+}
+
 pub fn routes() -> Router {
     Router::new()
         .route("/api/health", get(health))
         .route("/api/entries", post(insert))
         .route("/api/entries/:id", get(raw))
+        .route("/api/download/:id/:extension", get(download))
 }

src/web.rs

@@ -56,6 +56,7 @@ struct Paste<'a> {
     title: &'a str,
     id: String,
     formatted: String,
+    extension: String,
 }
 
 #[derive(Template)]
@@ -125,12 +126,14 @@ async fn show(
     let title = &TITLE;
     let key = Key::try_from(id_with_opt_ext)?;
     let id = key.id();
+    let extension = key.extension();
     let formatted = layer.get_formatted(key).await?;
 
     Ok(Paste {
         title,
         id,
         formatted,
+        extension,
     })
 }
 

templates/paste.html

@@ -24,7 +24,9 @@
       <pre class="code">{{ formatted|safe }}</pre>
     </div>
     <div class="paste-box">
-      <a class="punctuation definition tag" href="/api/entries/{{ id }}"/>raw</a> • <a class="punctuation definition tag" href="/">new</a>
+      <a class="punctuation definition tag" href="/api/download/{{ id }}/{{ extension }}">⤓</a>
+      • <a class="punctuation definition tag" href="/api/entries/{{ id }}"/>raw</a>
+      • <a class="punctuation definition tag" href="/">new</a>
     </div>
   </div>
 {% endblock %}