Emit ABI descriptors in final module generation jobs

Author: nkcsgexi

Sources/SwiftDriver/Driver/Driver.swift

@@ -336,6 +336,9 @@ public struct Driver {
   /// A global queue for emitting non-interrupted messages into stderr
   public static let stdErrQueue = DispatchQueue(label: "org.swift.driver.emit-to-stderr")
 
+  enum KnownCompilerFeature: String {
+    case emit_abi_descriptor = "emit-abi-descriptor"
+  }
 
   lazy var sdkPath: VirtualPath? = {
     guard let rawSdkPath = frontendTargetInfo.sdkPath?.path else {
@@ -367,6 +370,10 @@ public struct Driver {
     }
   }
 
+  func isFeatureSupported(_ feature: KnownCompilerFeature) -> Bool {
+    return supportedFrontendFeatures.contains(feature.rawValue)
+  }
+
   /// Handler for emitting diagnostics to stderr.
   public static let stderrDiagnosticsHandler: DiagnosticsEngine.DiagnosticsHandler = { diagnostic in
     stdErrQueue.sync {

Sources/SwiftDriver/Jobs/EmitModuleJob.swift

@@ -87,9 +87,13 @@ extension Driver {
       commandLine.appendFlag(.parseAsLibrary)
     }
 
+    let outputPath = VirtualPath.lookup(moduleOutputPath)
     commandLine.appendFlag(.o)
-    commandLine.appendPath(VirtualPath.lookup(moduleOutputPath))
-
+    commandLine.appendPath(outputPath)
+    if isFeatureSupported(.emit_abi_descriptor) {
+      commandLine.appendFlag(.emitAbiDescriptorPath)
+      commandLine.appendPath(outputPath.replacingExtension(with: .jsonABIBaseline))
+    }
     return Job(
       moduleName: moduleOutputInfo.name,
       kind: .emitModule,

Sources/SwiftDriver/Jobs/MergeModuleJob.swift

@@ -72,9 +72,14 @@ extension Driver {
       try commandLine.appendLast(.disableIncrementalImports, from: &parsedOptions)
     }
 
+    let outputPath = VirtualPath.lookup(moduleOutputInfo.output!.outputPath)
     commandLine.appendFlag(.o)
-    commandLine.appendPath(VirtualPath.lookup(moduleOutputInfo.output!.outputPath))
+    commandLine.appendPath(outputPath)
 
+    if isFeatureSupported(.emit_abi_descriptor) {
+      commandLine.appendFlag(.emitAbiDescriptorPath)
+      commandLine.appendPath(outputPath.replacingExtension(with: .jsonABIBaseline))
+    }
     return Job(
       moduleName: moduleOutputInfo.name,
       kind: .mergeModule,