use async stdout

Author: QaidVoid

Cargo.toml

@@ -31,7 +31,7 @@ reqwest = { version = "0.12.15", default-features = false, features = ["rustls-t
 serde = { version = "1.0.219", features = ["derive"] }
 serde_json = "1.0.140"
 tar = "0.4.44"
-tokio = { version = "1.44.2", features = ["macros", "rt-multi-thread"] }
+tokio = { version = "1.44.2", features = ["io-std", "macros", "rt-multi-thread"] }
 url = "2.5.4"
 xz2 = "0.1.7"
 zip = "2.6.1"

src/downloader.rs

@@ -1,13 +1,12 @@
 use std::{
     collections::{HashMap, HashSet},
     fs::Permissions,
-    io::Write,
     os::unix::fs::PermissionsExt,
     path::{Path, PathBuf},
     sync::{Arc, Mutex},
 };
 
-use futures::{future::join_all, StreamExt, TryStreamExt};
+use futures::{future::join_all, TryStreamExt};
 use regex::Regex;
 use reqwest::header::{HeaderMap, CONTENT_DISPOSITION, ETAG, LAST_MODIFIED};
 
@@ -183,14 +182,16 @@ impl Downloader<'_> {
             }
 
             if options.output_path.as_deref() == Some("-") {
-                let stdout = std::io::stdout();
-                let mut stdout_lock = stdout.lock();
+                let mut stdout = tokio::io::stdout();
                 let mut stream = response.bytes_stream();
 
-                while let Some(chunk) = stream.next().await {
-                    let chunk = chunk.unwrap();
-                    stdout_lock.write_all(&chunk).unwrap();
-                    stdout_lock.flush().unwrap();
+                while let Some(chunk) = stream
+                    .try_next()
+                    .await
+                    .map_err(|_| DownloadError::ChunkError)?
+                {
+                    stdout.write_all(&chunk).await?;
+                    stdout.flush().await?;
                 }
                 return Ok("-".to_string());
             }