feat: db downloads

queries/records/get_and_update.sql

@@ -1,8 +1,6 @@
-UPDATE record
+UPDATE records
 SET downloads = downloads + 1
-WHERE cache_name = ?
-RETURNING
-    cache_name,
-    uploaded,
-    downloads,
-    max_downloads;
+WHERE
+    cache_name = ?
+    AND downloads < max_downloads
+    AND julianday('now') - julianday(uploaded) > 5;

src/lib.rs

@@ -8,3 +8,7 @@ pub mod views;
 pub use router::*;
 pub use state::*;
 pub use views::*;
+
+use std::{path::PathBuf, sync::LazyLock};
+
+pub static CACHE_DIR: LazyLock<PathBuf> = LazyLock::new(|| PathBuf::from("./.cache/serve"));

src/router/download.rs

@@ -8,7 +8,7 @@ use axum::{
 use reqwest::StatusCode;
 use tokio_util::io::ReaderStream;
 
-use crate::{AppState, AsyncRemoveRecord};
+use crate::{AppState, CACHE_DIR};
 
 pub fn get_download_router() -> Router<AppState> {
     Router::new().route("/:id", get(download))
@@ -18,23 +18,25 @@ async fn download(
     axum::extract::Path(id): axum::extract::Path<String>,
     State(state): State<AppState>,
 ) -> Result<axum::response::Response, (StatusCode, String)> {
-    {
-        let mut records = state.records.lock().await;
-        if let Some(record) = records
-            .get_mut(&id)
-            .filter(|record| record.can_be_downloaded())
-        {
-            record.downloads += 1;
+    let mut conn = state.pool.acquire().await.unwrap();
 
-            let file = tokio::fs::File::open(&record.file).await.unwrap();
+    let rows_affected = sqlx::query_file!("queries/records/get_and_update.sql", id)
+        .execute(&mut *conn)
+        .await
+        .map_err(|err| (StatusCode::INTERNAL_SERVER_ERROR, err.to_string()))?
+        .rows_affected();
+
+    drop(conn);
+
+    if rows_affected > 0 {
+        let file = tokio::fs::File::open(CACHE_DIR.join(id))
+            .await
+            .map_err(|err| (StatusCode::INTERNAL_SERVER_ERROR, err.to_string()))?;
 
         return Ok(axum::response::Response::builder()
             .header("Content-Type", "application/zip")
             .body(Body::from_stream(ReaderStream::new(file)))
             .unwrap());
-        } else {
-            records.remove_record(&id).await.unwrap()
-        }
     }
 
     Ok(Redirect::to("/404.html").into_response())