Set up an externally installed JSON config file for blocking driver features for given module names

Author: nkcsgexi

Sources/SwiftDriver/Jobs/Planning.swift

@@ -12,6 +12,7 @@
 
 import TSCBasic
 import SwiftOptions
+import Foundation
 
 public enum PlanningError: Error, DiagnosticData {
   case replReceivedInput
@@ -438,6 +439,46 @@ extension Driver {
     return try mergeModuleJob(inputs: moduleInputs, inputsFromOutputs: moduleInputsFromJobOutputs)
   }
 
+  func getAdopterConfigPathFromXcodeDefaultToolchain() -> AbsolutePath? {
+    let swiftPath = try? toolchain.resolvedTool(.swiftCompiler).path
+    guard var swiftPath = swiftPath else {
+      return nil
+    }
+    let toolchains = "Toolchains"
+    guard swiftPath.components.contains(toolchains) else {
+      return nil
+    }
+    while swiftPath.basename != toolchains  {
+      swiftPath = swiftPath.parentDirectory
+    }
+    assert(swiftPath.basename == toolchains)
+    return swiftPath.appending(component: "XcodeDefault.xctoolchain")
+      .appending(component: "usr")
+      .appending(component: "local")
+      .appending(component: "lib")
+      .appending(component: "swift")
+      .appending(component: "adopter_configs.json")
+  }
+
+  @_spi(Testing) public struct AdopterConfig: Decodable {
+    public let key: String
+    public let moduleNames: [String]
+  }
+
+  @_spi(Testing) public static func parseAdopterConfigs(_ config: AbsolutePath) -> [AdopterConfig] {
+    let results = try? localFileSystem.readFileContents(config).withData {
+      try JSONDecoder().decode([AdopterConfig].self, from: $0)
+    }
+    return results ?? []
+  }
+
+  func getAdopterConfigsFromXcodeDefaultToolchain() -> [AdopterConfig] {
+    if let config = getAdopterConfigPathFromXcodeDefaultToolchain() {
+      return Driver.parseAdopterConfigs(config)
+    }
+    return []
+  }
+
   private mutating func addVerifyJobs(emitModuleJob: Job, addJob: (Job) -> Void )
   throws {
     // Turn this flag on by default with the env var or for public frameworks.

Tests/SwiftDriverTests/SwiftDriverTests.swift

@@ -6003,6 +6003,34 @@ final class SwiftDriverTests: XCTestCase {
     }
   }
 
+  func testAdopterConfigFile() throws {
+    try withTemporaryFile { file in
+      try localFileSystem.writeFileContents(file.path) {
+        $0 <<< "["
+        $0 <<< "  {"
+        $0 <<< "  \"key\": \"SkipFeature1\","
+        $0 <<< "  \"moduleNames\": [\"foo\", \"bar\"]"
+        $0 <<< "  }"
+        $0 <<< "]"
+      }
+      let configs = Driver.parseAdopterConfigs(file.path)
+      XCTAssertEqual(configs.count, 1)
+      XCTAssertEqual(configs[0].key, "SkipFeature1")
+      XCTAssertEqual(configs[0].moduleNames, ["foo", "bar"])
+    }
+    try withTemporaryFile { file in
+      try localFileSystem.writeFileContents(file.path) {
+        $0 <<< "][ malformed }{"
+      }
+      let configs = Driver.parseAdopterConfigs(file.path)
+      XCTAssertEqual(configs.count, 0)
+    }
+    do {
+      let configs = Driver.parseAdopterConfigs(AbsolutePath("/abc/c/a.json"))
+      XCTAssertEqual(configs.count, 0)
+    }
+  }
+
   func testExtractLibraryLevel() throws {
     try withTemporaryFile { file in
       try localFileSystem.writeFileContents(file.path) { $0 <<< "// swift-module-flags: -library-level api" }