From faadd89fffb1832126de5bfdf89dc421409535bf Mon Sep 17 00:00:00 2001
From: ByteDream <bytedream@protonmail.com>
Date: Sun, 4 Dec 2022 18:54:19 +0100
Subject: [PATCH] Wait until buffer is empty when downloading

---
 crunchy-cli-core/src/cli/utils.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/crunchy-cli-core/src/cli/utils.rs b/crunchy-cli-core/src/cli/utils.rs
index 0bf6b45..758636d 100644
--- a/crunchy-cli-core/src/cli/utils.rs
+++ b/crunchy-cli-core/src/cli/utils.rs
@@ -169,7 +169,7 @@ pub async fn download_segments(
             data_pos += 1;
         }
 
-        if *count.lock().unwrap() >= total_segments {
+        if *count.lock().unwrap() >= total_segments && buf.is_empty() {
             break;
         }
     }