add: bucket default images

crates/buckets-core/src/database/uploads.rs

@@ -2,7 +2,7 @@ use crate::{
     DataManager,
     model::{MediaUpload, UploadMetadata},
 };
-use oiseau::{PostgresRow, cache::Cache, execute, get, params, query_rows};
+use oiseau::{PostgresRow, cache::Cache, execute, get, params, query_row, query_rows};
 use tetratto_core::auto_method;
 use tetratto_core::model::{Error, Result};
 
@@ -20,6 +20,27 @@ impl DataManager {
 
     auto_method!(get_upload_by_id(usize as i64)@get_upload_from_row -> "SELECT * FROM uploads WHERE id = $1" --name="upload" --returns=MediaUpload --cache-key-tmpl="atto.upload:{}");
 
+    /// Get an upload by its ID and bucket.
+    pub async fn get_upload_by_id_bucket(&self, id: usize, bucket: String) -> Result<MediaUpload> {
+        let conn = match self.0.connect().await {
+            Ok(c) => c,
+            Err(e) => return Err(Error::DatabaseConnection(e.to_string())),
+        };
+
+        let res = query_row!(
+            &conn,
+            "SELECT * FROM uploads WHERE id = $1 AND bucket = $2",
+            &[&(id as i64), &bucket],
+            |x| { Ok(Self::get_upload_from_row(x)) }
+        );
+
+        if res.is_err() {
+            return Err(Error::GeneralNotFound("upload".to_string()));
+        }
+
+        Ok(res.unwrap())
+    }
+
     /// Get all uploads (paginated).
     ///
     /// # Arguments
@@ -102,6 +123,9 @@ impl DataManager {
 
     /// Create a new upload in the database.
     ///
+    /// Please note that the file must be manually written afterwards. You can use
+    /// the returned [`MediaUpload`] to retrieve the correct path to write the file.
+    ///
     /// # Arguments
     /// * `data` - a mock [`MediaUpload`] object to insert
     pub async fn create_upload(&self, data: MediaUpload) -> Result<MediaUpload> {
@@ -132,6 +156,11 @@ impl DataManager {
         Ok(data)
     }
 
+    /// Delete an upload given its `id`.
+    ///
+    /// # Warning
+    /// It's recommended that you use [`Self::delete_upload_with_id`] instead,
+    /// as the table's primary key is on `(id, bucket)`, not `id`.
     pub async fn delete_upload(&self, id: usize) -> Result<()> {
         // if !user.permissions.check(FinePermission::MANAGE_UPLOADS) {
         //     return Err(Error::NotAllowed);
@@ -163,5 +192,41 @@ impl DataManager {
         Ok(())
     }
 
+    /// Delete an upload given its `id` and `bucket`.
+    pub async fn delete_upload_with_bucket(&self, id: usize, bucket: String) -> Result<()> {
+        // if !user.permissions.check(FinePermission::MANAGE_UPLOADS) {
+        //     return Err(Error::NotAllowed);
+        // }
+
+        // delete file
+        // it's most important that the file gets off the file system first, even
+        // if there's an issue in the database
+        //
+        // the actual file takes up much more space than the database entry.
+        let upload = self.get_upload_by_id(id).await?;
+        upload.remove(&self.0.0.directory)?;
+
+        // delete from database
+        let conn = match self.0.connect().await {
+            Ok(c) => c,
+            Err(e) => return Err(Error::DatabaseConnection(e.to_string())),
+        };
+
+        let res = execute!(
+            &conn,
+            "DELETE FROM uploads WHERE id = $1 AND bucket = $2",
+            &[&(id as i64), &bucket]
+        );
+
+        if let Err(e) = res {
+            return Err(Error::DatabaseError(e.to_string()));
+        }
+
+        self.0.1.remove(format!("atto.upload:{}", id)).await;
+
+        // return
+        Ok(())
+    }
+
     auto_method!(update_upload_metadata(UploadMetadata) -> "UPDATE uploads SET metadata = $1 WHERE id = $2" --serde --cache-key-tmpl="atto.upload:{}");
 }