Merge pull request #67826 from hamishknight/experimental

Author: hamishknight

include/swift/AST/CASTBridging.h

@@ -247,6 +247,9 @@ void Diagnostic_finish(BridgedDiagnostic cDiag);
 BridgedIdentifier ASTContext_getIdentifier(BridgedASTContext cContext,
                                            BridgedString cStr);
 
+_Bool ASTContext_langOptsHasFeature(BridgedASTContext cContext,
+                                    BridgedFeature feature);
+
 void *ImportDecl_create(BridgedASTContext cContext,
                         BridgedDeclContext cDeclContext,
                         BridgedSourceLoc cImportLoc, char kind,

include/swift/Basic/CBasicBridging.h

@@ -22,7 +22,6 @@
 // layering. i.e. Darwin overlay is created by Swift compiler.
 
 SWIFT_BEGIN_NULLABILITY_ANNOTATIONS
-SWIFT_BEGIN_ASSUME_NONNULL
 
 #ifdef __cplusplus
 extern "C" {
@@ -31,6 +30,14 @@ extern "C" {
 
 #endif
 
+typedef enum ENUM_EXTENSIBILITY_ATTR(open) BridgedFeature {
+#define LANGUAGE_FEATURE(FeatureName, SENumber, Description, Option) \
+FeatureName,
+#include "swift/Basic/Features.def"
+} BridgedFeature;
+
+SWIFT_BEGIN_ASSUME_NONNULL
+
 typedef struct BridgedData {
   const char *_Nullable baseAddress;
   unsigned long size;

lib/AST/CASTBridging.cpp

@@ -176,6 +176,11 @@ BridgedIdentifier ASTContext_getIdentifier(BridgedASTContext cContext,
   return {convertASTContext(cContext).getIdentifier(str).getAsOpaquePointer()};
 }
 
+bool ASTContext_langOptsHasFeature(BridgedASTContext cContext,
+                                   BridgedFeature feature) {
+  return convertASTContext(cContext).LangOpts.hasFeature((Feature)feature);
+}
+
 void *ImportDecl_create(BridgedASTContext cContext,
                         BridgedDeclContext cDeclContext,
                         BridgedSourceLoc cImportLoc, char kind,

lib/ASTGen/Sources/ASTGen/ASTGen.swift

@@ -1,5 +1,8 @@
 import CASTBridging
 import SwiftParser
+
+// Needed to use SyntaxTransformVisitor's visit method.
+@_spi(SyntaxTransformVisitor)
 import SwiftSyntax
 
 extension Array {

lib/ASTGen/Sources/ASTGen/Decls.swift

@@ -1,5 +1,8 @@
 import CASTBridging
 import SwiftParser
+
+// Needed to use SyntaxTransformVisitor's visit method.
+@_spi(SyntaxTransformVisitor)
 import SwiftSyntax
 
 extension ASTGenVisitor {

lib/ASTGen/Sources/ASTGen/Exprs.swift

@@ -1,5 +1,8 @@
 import CASTBridging
 import SwiftParser
+
+// Needed to use SyntaxTransformVisitor's visit method.
+@_spi(SyntaxTransformVisitor)
 import SwiftSyntax
 
 extension ASTGenVisitor {

lib/ASTGen/Sources/ASTGen/Generics.swift

@@ -1,5 +1,8 @@
 import CASTBridging
 import SwiftParser
+
+// Needed to use SyntaxTransformVisitor's visit method.
+@_spi(SyntaxTransformVisitor)
 import SwiftSyntax
 
 extension ASTGenVisitor {

lib/ASTGen/Sources/ASTGen/Macros.swift

@@ -14,12 +14,15 @@ import CASTBridging
 import SwiftDiagnostics
 import SwiftOperators
 import SwiftParser
-import SwiftSyntax
 import SwiftSyntaxBuilder
 import SwiftSyntaxMacros
 import SwiftSyntaxMacroExpansion
 import SwiftCompilerPluginMessageHandling
 
+// Needed to use SyntaxTransformVisitor's visit method.
+@_spi(SyntaxTransformVisitor)
+import SwiftSyntax
+
 extension SyntaxProtocol {
   func token(at position: AbsolutePosition) -> TokenSyntax? {
     // If the position isn't within this node at all, return early.

lib/ASTGen/Sources/ASTGen/Misc.swift

@@ -1,5 +1,8 @@
 import CASTBridging
 import SwiftParser
+
+// Needed to use SyntaxTransformVisitor's visit method.
+@_spi(SyntaxTransformVisitor)
 import SwiftSyntax
 
 extension ASTGenVisitor {

lib/ASTGen/Sources/ASTGen/SourceFile.swift

@@ -1,8 +1,13 @@
+import CBasicBridging
+import CASTBridging
+
 import SwiftDiagnostics
-import SwiftParser
 import SwiftSyntax
 import SwiftParserDiagnostics
 
+@_spi(ExperimentalLanguageFeatures)
+import SwiftParser
+
 /// Describes a source file that has been "exported" to the C++ part of the
 /// compiler, with enough information to interface with the C++ layer.
 struct ExportedSourceFile {
@@ -20,15 +25,31 @@ struct ExportedSourceFile {
   let syntax: SourceFileSyntax
 }
 
+extension Parser.ExperimentalFeatures {
+  init(from context: BridgedASTContext?) {
+    self = []
+    guard let context = context else { return }
+
+    func mapFeature(_ bridged: BridgedFeature, to feature: Self) {
+      if ASTContext_langOptsHasFeature(context, bridged) {
+        insert(feature)
+      }
+    }
+  }
+}
+
 /// Parses the given source file and produces a pointer to a new
 /// ExportedSourceFile instance.
 @_cdecl("swift_ASTGen_parseSourceFile")
 public func parseSourceFile(
   buffer: UnsafePointer<UInt8>, bufferLength: Int,
-  moduleName: UnsafePointer<UInt8>, filename: UnsafePointer<UInt8>
+  moduleName: UnsafePointer<UInt8>, filename: UnsafePointer<UInt8>,
+  ctxPtr: UnsafeMutableRawPointer?
 ) -> UnsafeRawPointer {
   let buffer = UnsafeBufferPointer(start: buffer, count: bufferLength)
-  let sourceFile = Parser.parse(source: buffer)
+
+  let ctx = ctxPtr.map { BridgedASTContext(raw: $0) }
+  let sourceFile = Parser.parse(source: buffer, experimentalFeatures: .init(from: ctx))
 
   let exportedPtr = UnsafeMutablePointer<ExportedSourceFile>.allocate(capacity: 1)
   exportedPtr.initialize(