From 44b7dbd2ee3900319b30c9c16e500435048f7764 Mon Sep 17 00:00:00 2001
From: Yuta Saito <kateinoigakukun@gmail.com>
Date: Fri, 5 May 2023 00:30:34 +0900
Subject: [PATCH] Support jammy and amazonlinux2 for toolchain install (#397)

* Support Ubuntu 22.04 for toolchain download

* Support amazonlinux2 for toolchain install
---
 .../SwiftToolchain/ToolchainManagement.swift  | 39 +++++++++++--------
 1 file changed, 23 insertions(+), 16 deletions(-)

diff --git a/Sources/SwiftToolchain/ToolchainManagement.swift b/Sources/SwiftToolchain/ToolchainManagement.swift
index 3beeb82..89986cf 100644
--- a/Sources/SwiftToolchain/ToolchainManagement.swift
+++ b/Sources/SwiftToolchain/ToolchainManagement.swift
@@ -184,22 +184,7 @@ public class ToolchainSystem {
     #if os(macOS)
       let platformSuffixes = ["osx", "catalina", "macos"]
     #elseif os(Linux)
-      let releaseFile = AbsolutePath("/etc").appending(component: "lsb-release")
-      guard fileSystem.isFile(releaseFile) else {
-        throw ToolchainError.unsupportedOperatingSystem
-      }
-
-      let releaseData = try fileSystem.readFileContents(releaseFile).description
-      let ubuntuSuffix: String
-      if releaseData.contains("DISTRIB_RELEASE=18.04") {
-        ubuntuSuffix = "ubuntu18.04"
-      } else if releaseData.contains("DISTRIB_RELEASE=20.04") {
-        ubuntuSuffix = "ubuntu20.04"
-      } else {
-        throw ToolchainError.unsupportedOperatingSystem
-      }
-
-      let platformSuffixes = ["linux", ubuntuSuffix]
+      let platformSuffixes = ["linux", try self.inferLinuxDistributionSuffix()]
     #endif
 
     terminal.logLookup(
@@ -212,6 +197,28 @@ public class ToolchainSystem {
     }.first
   }
 
+  private func inferLinuxDistributionSuffix() throws -> String {
+    guard let releaseFile = [
+      AbsolutePath.root.appending(components: "etc", "lsb-release"),
+      AbsolutePath.root.appending(components: "etc", "os-release"),
+    ].first(where: fileSystem.isFile) else {
+      throw ToolchainError.unsupportedOperatingSystem
+    }
+
+    let releaseData = try fileSystem.readFileContents(releaseFile).description
+    if releaseData.contains("DISTRIB_RELEASE=18.04") {
+      return "ubuntu18.04"
+    } else if releaseData.contains("DISTRIB_RELEASE=20.04") {
+      return "ubuntu20.04"
+    } else if releaseData.contains("DISTRIB_RELEASE=22.04") {
+      return "ubuntu22.04"
+    } else if releaseData.contains(#"PRETTY_NAME="Amazon Linux 2""#) {
+      return "amazonlinux2"
+    } else {
+      throw ToolchainError.unsupportedOperatingSystem
+    }
+  }
+
   /** Infer `swift` binary path matching a given version if any is present, or infer the
    version from the `.swift-version` file. If neither version is installed, download it.
    */