Delete object (#3)

Reviewed-on: #3

Cargo.lock

@@ -1370,7 +1370,7 @@ dependencies = [
 
 [[package]]
 name = "labyrinth"
-version = "0.0.2"
+version = "0.0.4"
 dependencies = [
  "aws-config",
  "aws-sdk-s3",

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "labyrinth"
-version = "0.0.2"
+version = "0.0.4"
 edition = "2024"
 license = "MIT"
 description = "Library for managing soaricarus storage"

src/lib.rs

@@ -37,8 +37,7 @@ pub async fn init_client(config: &config::Config) -> aws_sdk_s3::Client {
         .force_path_style(true)
         .build();
 
-    let client = aws_sdk_s3::Client::from_conf(s3_config);
-    client
+    aws_sdk_s3::Client::from_conf(s3_config)
 }
 
 #[derive(Debug)]
@@ -48,6 +47,7 @@ pub enum Error {
 }
 
 impl Labyrinth {
+    /// Upload object to the bucket
     pub async fn upload(
         &self,
         file_key: &str,
@@ -77,4 +77,51 @@ impl Labyrinth {
             Err(err) => Err(Error::Info(err.to_string())),
         }
     }
+
+    /// Download object from the bucket
+    pub async fn download(&self, file_key: &str) -> Result<Vec<u8>, Error> {
+        let client = init_client(&self.config).await;
+
+        match client
+            .get_object()
+            .bucket(self.config.bucket.clone())
+            .key(file_key)
+            .send()
+            .await
+        {
+            Ok(response) => {
+                let body = response.body;
+
+                // Get the body as ByteStream
+                match body.collect().await {
+                    Ok(data) => {
+                        let bytes = data.into_bytes();
+                        let raw_data = bytes.to_vec();
+                        Ok(raw_data)
+                    }
+                    Err(err) => Err(Error::Info(err.to_string())),
+                }
+            }
+            Err(err) => Err(Error::Info(err.to_string())),
+        }
+    }
+
+    /// Delete an object from the bucket
+    pub async fn delete(
+        &self,
+        file_key: &str,
+    ) -> Result<aws_sdk_s3::operation::delete_object::DeleteObjectOutput, Error> {
+        let client = init_client(&self.config).await;
+
+        match client
+            .delete_object()
+            .bucket(self.config.bucket.clone())
+            .key(file_key)
+            .send()
+            .await
+        {
+            Ok(response) => Ok(response),
+            Err(err) => Err(Error::Info(err.to_string())),
+        }
+    }
 }