From 81962a59f4e1b656d5ebcb1ed9e413825392ba2b Mon Sep 17 00:00:00 2001
From: lovasoa <pere.jobs@gmail.com>
Date: Tue, 28 May 2024 10:43:31 +0200
Subject: [PATCH] build script download improvements

---
 build.rs | 42 +++++++++++++++++++++++++++++-------------
 1 file changed, 29 insertions(+), 13 deletions(-)

diff --git a/build.rs b/build.rs
index dd55e6d3..e3cd4269 100644
--- a/build.rs
+++ b/build.rs
@@ -42,7 +42,11 @@ async fn download_deps(filename: &str) {
 }
 
 async fn process_input_file(path_out: &Path, original: File) {
-    let client = awc::Client::default();
+    let client = awc::ClientBuilder::new()
+        .timeout(core::time::Duration::from_secs(3))
+        .max_http_version(awc::http::Version::HTTP_11)
+        .no_default_headers()
+        .finish();
     let mut outfile = gzip::Encoder::new(File::create(path_out).unwrap()).unwrap();
     for l in BufReader::new(original).lines() {
         let line = l.unwrap();
@@ -50,18 +54,7 @@ async fn process_input_file(path_out: &Path, original: File) {
             let url = line
                 .trim_start_matches("/* !include ")
                 .trim_end_matches(" */");
-            let mut resp = client.get(url).send().await.expect(
-                "We need to download external frontend dependencies to build the static frontend.",
-            );
-            if resp.status() != 200 {
-                panic!("Received {} status code from {}", resp.status(), url);
-            }
-            while let Some(b) = resp.next().await {
-                let chunk = b.unwrap_or_else(|_| panic!("Failed to read data from {}", url));
-                outfile
-                    .write_all(&chunk)
-                    .expect("Failed to write external frontend dependency to local file");
-            }
+            download_url_to_opened_file(&client, url, &mut outfile).await;
             outfile.write_all(b"\n").unwrap();
         } else {
             writeln!(outfile, "{}", line).unwrap();
@@ -73,6 +66,29 @@ async fn process_input_file(path_out: &Path, original: File) {
         .expect("Unable to write compressed frontend asset");
 }
 
+async fn download_url_to_opened_file(
+    client: &awc::Client,
+    url: &str,
+    outfile: &mut impl std::io::Write,
+) {
+    let mut resp = client.get(url).send().await.unwrap_or_else(|err| {
+        panic!(
+            "We need to download external frontend dependencies to build the static frontend. \
+                Could not download {url} \
+                {err}"
+        )
+    });
+    if resp.status() != 200 {
+        panic!("Received {} status code from {}", resp.status(), url);
+    }
+    while let Some(b) = resp.next().await {
+        let chunk = b.unwrap_or_else(|err| panic!("Failed to read data from {url}: {err}"));
+        outfile
+            .write_all(&chunk)
+            .expect("Failed to write external frontend dependency to local file");
+    }
+}
+
 // Given a filename, creates a new unique filename based on the file contents
 fn hashed_filename(path: &Path) -> String {
     let mut file = File::open(path).unwrap();