Capture code snippet from diagnostic compiler output

Sources/Basics/Observability.swift

@@ -56,6 +56,10 @@ public class ObservabilitySystem {
         func handleDiagnostic(scope: ObservabilityScope, diagnostic: Diagnostic) {
             self.underlying(scope, diagnostic)
         }
+
+        func printToOutput(message: String) {
+            self.diagnosticsHandler.printToOutput(message: message)
+        }
     }
 
     public static var NOOP: ObservabilityScope {
@@ -128,6 +132,10 @@ public final class ObservabilityScope: DiagnosticsEmitterProtocol, Sendable, Cus
         return parent?.errorsReportedInAnyScope ?? false
     }
 
+    public func print(message: String) {
+        self.diagnosticsHandler.printToOutput(message: message)
+    }
+
     // DiagnosticsEmitterProtocol
     public func emit(_ diagnostic: Diagnostic) {
         var diagnostic = diagnostic
@@ -150,6 +158,10 @@ public final class ObservabilityScope: DiagnosticsEmitterProtocol, Sendable, Cus
             self.underlying.handleDiagnostic(scope: scope, diagnostic: diagnostic)
         }
 
+        public func printToOutput(message: String) {
+            self.underlying.printToOutput(message: message)
+        }
+
         var errorsReported: Bool {
             self._errorsReported.get() ?? false
         }
@@ -160,6 +172,8 @@ public final class ObservabilityScope: DiagnosticsEmitterProtocol, Sendable, Cus
 
 public protocol DiagnosticsHandler: Sendable {
     func handleDiagnostic(scope: ObservabilityScope, diagnostic: Diagnostic)
+
+    func printToOutput(message: String)
 }
 
 /// Helper protocol to share default behavior.

Sources/CoreCommands/SwiftCommandObservabilityHandler.swift

@@ -116,6 +116,12 @@ public struct SwiftCommandObservabilityHandler: ObservabilityHandlerProvider {
             }
         }
 
+        func printToOutput(message: String) {
+            self.queue.async(group: self.sync) {
+                self.write(message)
+            }
+        }
+
         // for raw output reporting
         func print(_ output: String, verbose: Bool) {
             self.queue.async(group: self.sync) {