Merge pull request #10 from VoysAI/fix-build-rs-download-error

Fix build.rs `download_file` error case
ebraraktas · Aug 18, 2022 · a633d41 · a633d41
2 parents e4ef81b + f76b2c0
commit a633d41
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/build.rs b/build.rs
@@ -449,7 +449,10 @@ fn download_file(url: &str, path: &Path) {
     easy.url(url).unwrap();
     easy.write_function(move |data| Ok(writer.write(data).unwrap()))
         .unwrap();
-    easy.perform().unwrap();
+    easy.perform().unwrap_or_else(|e| {
+        std::fs::remove_file(path).unwrap(); // Delete corrupted or empty file
+        panic!("Error occurred while downloading from {}: {:?}", url, e);
+    });
 }
 
 fn main() {