From 8ba8dcbe7b8fd6760ec4ec21aac1662e20205b98 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Johannes=20L=C3=B6thberg?= <johannes.loethberg@elokon.com>
Date: Mon, 15 Jul 2024 18:36:06 +0200
Subject: [PATCH] Fix extended LLVM mode for LLVM < 17
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

The fix for extended LLVM mode with LLVM 17 accidentally broke it for
toolchain versions using LLVM 15 and 16.

Signed-off-by: Johannes Löthberg <johannes.loethberg@elokon.com>
---
 CHANGELOG.md          |  5 +++++
 src/toolchain/llvm.rs | 43 +++++++++++++++++++++++++++----------------
 2 files changed, 32 insertions(+), 16 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 0aa92f2b..eb7e212f 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -5,6 +5,11 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [Unreleased]
+
+### Fixed
+- Fix extended LLVM mode regression for LLVM versions < 17 introduced by #432. (#437)
+
 ## [0.12.1] - 2024-07-15
 
 ### Fixed
diff --git a/src/toolchain/llvm.rs b/src/toolchain/llvm.rs
index f37a5e03..5b9d143b 100644
--- a/src/toolchain/llvm.rs
+++ b/src/toolchain/llvm.rs
@@ -33,7 +33,7 @@ pub struct Llvm {
     // /// If `true`, full LLVM, instead of only libraries, are installed.
     extended: bool,
     /// LLVM libs-only toolchain file name.
-    pub file_name_libs: String,
+    pub file_name_libs: Option<String>,
     /// LLVM "full" toolchain file name.
     pub file_name_full: Option<String>,
     /// Host triple.
@@ -140,14 +140,23 @@ impl Llvm {
                 format!("libs-{file_name_full}")
             };
 
-            (
-                file_name_libs,
-                if version == DEFAULT_LLVM_15_VERSION || version == DEFAULT_LLVM_16_VERSION {
-                    None
+            // For LLVM 15 and 16 the "full" tarball was a superset of the "libs" tarball, so if
+            // we're in extended LLVM mode we only need the "full" tarballs for those versions.
+            //
+            // Later LLVM versions are built such that the "full" tarball has a statically linked
+            // `clang` binary and therefore doesn't contain libclang, and so then we need to fetch
+            // both tarballs.
+            if version == DEFAULT_LLVM_15_VERSION || version == DEFAULT_LLVM_16_VERSION {
+                if extended {
+                    (None, Some(file_name_full))
                 } else {
-                    extended.then_some(file_name_full)
-                },
-            )
+                    (Some(file_name_libs), None)
+                }
+            } else if extended {
+                (Some(file_name_libs), Some(file_name_full))
+            } else {
+                (Some(file_name_libs), None)
+            }
         };
 
         let repository_url = format!("{DEFAULT_LLVM_REPOSITORY}/{version}");
@@ -260,14 +269,16 @@ impl Installable for Llvm {
             );
         } else {
             info!("Installing Xtensa LLVM");
-            download_file(
-                format!("{}/{}", self.repository_url, self.file_name_libs),
-                "idf_tool_xtensa_elf_clang.libs.tar.xz",
-                self.path.to_str().unwrap(),
-                true,
-                false,
-            )
-            .await?;
+            if let Some(file_name_libs) = &self.file_name_libs {
+                download_file(
+                    format!("{}/{}", self.repository_url, file_name_libs),
+                    "idf_tool_xtensa_elf_clang.libs.tar.xz",
+                    self.path.to_str().unwrap(),
+                    true,
+                    false,
+                )
+                .await?;
+            }
             if let Some(file_name_full) = &self.file_name_full {
                 download_file(
                     format!("{}/{}", self.repository_url, file_name_full),