Add support for loading supported features from share/swift/features.json

Author: nkcsgexi

Sources/SwiftDriver/Driver/Driver.swift

@@ -330,6 +330,9 @@ public struct Driver {
   /// A collection of all the flags the selected toolchain's `swift-frontend` supports
   public let supportedFrontendFlags: Set<String>
 
+  /// A collection of all the features the selected toolchain's `swift-frontend` supports
+  public let supportedFrontendFeatures: Set<String>
+
   /// A global queue for emitting non-interrupted messages into stderr
   public static let stdErrQueue = DispatchQueue(label: "org.swift.driver.emit-to-stderr")
 
@@ -606,11 +609,12 @@ public struct Driver {
                                                                                outputFileMap: outputFileMap)
 
     self.supportedFrontendFlags =
-      try Self.computeSupportedCompilerFeatures(of: self.toolchain, hostTriple: self.hostTriple,
+      try Self.computeSupportedCompilerArgs(of: self.toolchain, hostTriple: self.hostTriple,
                                                 parsedOptions: &self.parsedOptions,
                                                 diagnosticsEngine: diagnosticEngine,
                                                 fileSystem: fileSystem, executor: executor,
                                                 env: env)
+    self.supportedFrontendFeatures = try Self.computeSupportedCompilerFeatures(of: self.toolchain, env: env)
 
     self.enabledSanitizers = try Self.parseSanitizerArgValues(
       &parsedOptions,

Sources/SwiftDriver/ExplicitModuleBuilds/ModuleDependencyScanning.swift

@@ -344,4 +344,24 @@ public extension Driver {
       .parentDirectory // bin
       .parentDirectory // toolchain root
   }
+
+  static func computeSupportedCompilerFeatures(of toolchain: Toolchain,
+                                               env: [String: String]) throws -> Set<String> {
+    struct FeatureInfo: Codable {
+      var name: String
+    }
+    struct FeatureList: Codable {
+      var features: [FeatureInfo]
+    }
+    let jsonPath = try getRootPath(of: toolchain, env: env)
+      .appending(component: "share")
+      .appending(component: "swift")
+      .appending(component: "features.json")
+    guard localFileSystem.exists(jsonPath) else {
+      return Set<String>()
+    }
+    let content = try localFileSystem.readFileContents(jsonPath)
+    let result = try JSONDecoder().decode(FeatureList.self, from: Data(content.contents))
+    return Set(result.features.map {$0.name})
+  }
 }

Sources/SwiftDriver/Jobs/EmitSupportedFeaturesJob.swift

@@ -52,7 +52,7 @@ extension Toolchain {
 }
 
 extension Driver {
-  static func computeSupportedCompilerFeatures(of toolchain: Toolchain, hostTriple: Triple,
+  static func computeSupportedCompilerArgs(of toolchain: Toolchain, hostTriple: Triple,
                                                parsedOptions: inout ParsedOptions,
                                                diagnosticsEngine: DiagnosticsEngine,
                                                fileSystem: FileSystem,

Tests/SwiftDriverTests/SwiftDriverTests.swift

@@ -5015,6 +5015,12 @@ final class SwiftDriverTests: XCTestCase {
     }
   }
 
+  func testSupportedFeatureJson() throws {
+    let driver = try Driver(args: ["swiftc", "-emit-module", "foo.swift"])
+    XCTAssertFalse(driver.supportedFrontendFeatures.isEmpty)
+    XCTAssertTrue(driver.supportedFrontendFeatures.contains("emit-localized-strings"))
+  }
+
   func testFilelist() throws {
     do {
       var driver = try Driver(args: ["swiftc", "-emit-module", "./a.swift", "./b.swift", "./c.swift", "-module-name", "main", "-target", "x86_64-apple-macosx10.9", "-driver-filelist-threshold=0", "-no-emit-module-separately"])