add cache cleaning queries

queries/records/delete_expired_records.sql

@@ -0,0 +1,5 @@
+DELETE FROM records
+WHERE
+    downloads >= max_downloads
+    OR julianday('now') - julianday(uploaded) > 3
+RETURNING cache_name;

queries/records/get_expired_records.sql

@@ -0,0 +1,7 @@
+SELECT
+    cache_name,
+    downloads,
+    max_downloads,
+    julianday('now') - julianday(uploaded) AS age
+FROM records
+WHERE downloads >= max_downloads OR age > 5;

queries/records/get_records.sql

@@ -0,0 +1,6 @@
+SELECT
+    cache_name,
+    uploaded,
+    downloads,
+    max_downloads
+FROM records;

queries/records/get_records_page.sql

@@ -0,0 +1,7 @@
+SELECT
+    cache_name,
+    uploaded,
+    downloads,
+    max_downloads
+FROM records
+LIMIT ? OFFSET ?;

src/router/link.rs

@@ -4,9 +4,14 @@ use axum::{
     routing::get,
     Router,
 };
+use axum_extra::TypedHeader;
 use reqwest::StatusCode;
 
-use crate::{templates::DownloadLinkTemplate, AppState, AsyncRemoveRecord};
+use crate::{
+    templates::{self, DownloadLinkTemplate},
+    util::headers::HxRequest,
+    AppState, AsyncRemoveRecord,
+};
 
 pub fn get_link_router() -> Router<AppState> {
     // Link pages
@@ -62,12 +67,22 @@ async fn link_delete(
 
 async fn remaining(
     State(state): State<AppState>,
+    hx_request: Option<TypedHeader<HxRequest>>,
     axum::extract::Path(id): axum::extract::Path<String>,
-) -> impl IntoResponse {
+) -> Result<impl IntoResponse, (StatusCode, String)> {
+    if hx_request.is_none() {
+        return Err((
+            StatusCode::BAD_REQUEST,
+            "Attempt to fetch html fragment from non-htmx request".to_string(),
+        ));
+    }
+
     let records = state.records.lock().await;
-    if let Some(record) = records.get(&id) {
+
-        Html(crate::templates::get_downloads_remaining_text(record))
+    Ok(Html(
-    } else {
+        records
-        Html("?".to_string())
+            .get(&id)
-    }
+            .map(templates::get_downloads_remaining_text)
+            .unwrap_or_else(|| "?".to_string()),
+    ))
 }

src/util/headers.rs

@@ -31,3 +31,38 @@ impl Header for ForwardedFor {
         values.extend(std::iter::once(HeaderValue::from_str(&self.0).unwrap()));
     }
 }
+
+#[derive(Debug)]
+pub struct HxRequest;
+
+pub static HXR_TEXT: &str = "hx-request";
+
+pub static HXR_NAME: HeaderName = HeaderName::from_static(HXR_TEXT);
+
+impl Header for HxRequest {
+    fn name() -> &'static HeaderName {
+        &FF_NAME
+    }
+
+    fn decode<'i, I>(values: &mut I) -> Result<Self, headers::Error>
+    where
+        Self: Sized,
+        I: Iterator<Item = &'i headers::HeaderValue>,
+    {
+        let value = values
+            .next()
+            .ok_or_else(headers::Error::invalid)?
+            .to_str()
+            .map_err(|_| headers::Error::invalid())?
+            .to_owned();
+
+        match &value[..] {
+            "true" => Ok(HxRequest),
+            _ => Err(headers::Error::invalid()),
+        }
+    }
+
+    fn encode<E: Extend<headers::HeaderValue>>(&self, values: &mut E) {
+        values.extend(std::iter::once(HeaderValue::from_static("true")));
+    }
+}

src/views/templates.rs

@@ -39,6 +39,12 @@ pub struct DownloadLinkFragment {
     pub record: UploadRecord,
 }
 
+#[derive(Template)]
+#[template(path = "link.html", block = "downloads_remaining")]
+pub struct DownloadsRemainingFragment {
+    pub record: UploadRecord,
+}
+
 #[derive(Template)]
 #[template(path = "linklist.html")]
 pub struct LinkListTemplate {

templates/link.html

@@ -10,9 +10,11 @@
       <div
         class="link-wrapper"
         hx-get="/link/{{ id }}/remaining"
-        hx-trigger="click from:#link delay:0.2s, every 10s"
+        hx-trigger="click from:#link delay:0.2s, every 60s"
       >
+        {% block downloads_remaining %}
           {{ self::get_downloads_remaining_text(record) }}
+        {% endblock downloads_remaining %}
       </div>
       <button class="return-button" onclick="clipboard()">
         Copy to Clipboard