use stream instead of buffer for uploading artifacts

src/commands/download.rs

@@ -41,13 +41,20 @@ pub async fn download(
 
     let output_get_res = output_get_req.get().await?;
 
-    assert!(output_get_res.status().is_success(), "Build artifacts could not be downloaded.");
+    assert!(
+        output_get_res.status().is_success(),
+        "Build artifacts could not be downloaded."
+    );
 
     println!("Build artifacts downloaded");
 
     let buf = &output_get_res.bytes().await?.to_vec();
 
-    assert!(vercel_cache_helper::utils::is_zstd_compressed(&buf), "Downloaded archive (Size: {}) is not zstd compressed.", &buf.len());
+    assert!(
+        vercel_cache_helper::utils::is_zstd_compressed(&buf),
+        "Downloaded archive (Size: {}) is not zstd compressed.",
+        &buf.len()
+    );
 
     output_dir_archive.write_all(buf)?;
 

src/commands/upload.rs

@@ -63,11 +63,15 @@ pub async fn upload(
 
     pb.set_message("Uploading artfiacts...");
 
-    let res = output_put_req
-        .buffer(&mut output_archive_buf, output_archive_size)
+    let mut output_archive_cursor = std::io::Cursor::new(output_archive_buf);
+    let response = output_put_req
+        .stream(&mut output_archive_cursor, output_archive_size)
         .await?;
 
-    assert!(res.status().is_success(), "Could not upload artifacts.");
+    if !response.status().is_success() {
+        println!("Could not upload artifacts.");
+        return Ok(());
+    }
 
     pb.finish_with_message("Artifacts uploaded successfully.");
 

src/main.rs

@@ -33,7 +33,10 @@ async fn main() -> vercel_cache_helper::Result<()> {
 
     let remote_client = vercel_cache_helper::get_remote_client(token, Some(team_id), product);
 
-    println!("🚀 fastn Vercel Cache Helper v{}", env!("CARGO_PKG_VERSION"));
+    println!(
+        "🚀 fastn Vercel Cache Helper v{}",
+        env!("CARGO_PKG_VERSION")
+    );
 
     let result_future: std::pin::Pin<
         Box<dyn std::future::Future<Output = Result<(), vercel_cache_helper::Error>>>,

src/vercel/artifact.rs

@@ -111,7 +111,6 @@ impl ArtifactPutRequest {
 
         let headers = self.0.get_headers("PUT", Some(content_len));
 
-
         let response = client
             .put(&self.0.url)
             .headers(headers)
@@ -148,11 +147,7 @@ impl ArtifactGetRequest {
 
         let headers = self.0.get_headers("GET", None);
 
-        let response = client
-            .get(&self.0.url)
-            .headers(headers)
-            .send()
-            .await?;
+        let response = client.get(&self.0.url).headers(headers).send().await?;
 
         Ok(response)
     }