broken! feat: sqlx query get

sneaky not null id

.gitignore

@@ -3,3 +3,5 @@
 /.cache
 /node_modules
 testing.db
+testing.db-shm
+testing.db-wal

.sqlfluff

@@ -1,6 +1,10 @@
 [sqlfluff]
 dialect = sqlite
 
+[sqlfluff:rules:capitalisation.keywords]
+# Keywords
+capitalisation_policy = upper
+
 [sqlfluff:layout:type:column_constraint_segment]
 spacing_before = align
 align_within = create_table_statement

migrations/20241115182043_init.up.sql

@@ -1,7 +1,7 @@
 -- Add up migration script here
 CREATE TABLE IF NOT EXISTS records (
-    id            INTEGER PRIMARY KEY,
+    id            INTEGER NOT NULL PRIMARY KEY,
-    uploaded      TEXT    NOT NULL,
+    uploaded      TEXT    NOT NULL DEFAULT (datetime('now')),
     file_path     TEXT    NOT NULL,
     downloads     INTEGER NOT NULL DEFAULT 0,
     max_downloads INTEGER NOT NULL

queries/records/count_records.sql

@@ -0,0 +1 @@
+SELECT COUNT(*) FROM records;

queries/records/increment_download.sql

@@ -0,0 +1,3 @@
+UPDATE record
+SET downloads = downloads + 1
+WHERE id = ?

queries/records/new.sql

@@ -0,0 +1,2 @@
+INSERT INTO records (file_path, max_downloads)
+VALUES (?, ?);

shell.nix

@@ -12,7 +12,9 @@ mkShell rec {
     # Additional darwin specific inputs can be set here
     pkgs.darwin.apple_sdk.frameworks.SystemConfiguration
   ];
-  buildInputs = additionalBuildInputs;
+  buildInputs = [
+    sqlite
+  ] ++ additionalBuildInputs;
   LD_LIBRARY_PATH = lib.makeLibraryPath buildInputs;
   DATABASE_URL = "sqlite:testing.db";
 }

src/router/upload.rs

@@ -75,6 +75,19 @@ async fn upload_to_zip(
 
     let record = UploadRecord::new(archive_path);
     let mut records = state.records.lock().await;
+
+    if let Some(mut conn) = state.pool.try_acquire() {
+        let path = record.file.clone().into_os_string().into_string().unwrap();
+
+        let id = sqlx::query_file!("queries/records/new.sql", path, 5)
+            .execute(&mut *conn)
+            .await
+            .unwrap()
+            .last_insert_rowid();
+
+        // TODO: Looks like I actually gotta store cache_name lmfao
+    };
+
     records.insert(cache_name.clone(), record.clone());
 
     let records_cache = records.clone();

src/state.rs

@@ -2,12 +2,14 @@ use std::{
     collections::{hash_map::Entry, HashMap},
     io::ErrorKind,
     path::{Path, PathBuf},
+    str::FromStr,
     sync::Arc,
 };
 
 use async_trait::async_trait;
 use chrono::{DateTime, Duration, Utc};
 use serde::{Deserialize, Serialize};
+use sqlx::{sqlite::SqliteConnectOptions, SqlitePool};
 use tokio::sync::Mutex;
 
 use crate::cache;
@@ -54,12 +56,17 @@ impl Default for UploadRecord {
 #[derive(Clone)]
 pub struct AppState {
     pub records: Arc<Mutex<HashMap<String, UploadRecord>>>,
+    pub pool: SqlitePool,
 }
 
 impl AppState {
     pub fn new(records: HashMap<String, UploadRecord>) -> Self {
         Self {
             records: Arc::new(Mutex::new(records)),
+            pool: SqlitePool::connect_lazy_with(
+                SqliteConnectOptions::from_str("sqlite:testing.db")
+                    .expect("Invalid Database String"),
+            ),
         }
     }
 }