Move file to SwiftSyntaxBuilder cod gen

Author: kimdv

CodeGeneration/Package.swift

@@ -13,7 +13,7 @@ let package = Package(
     .executable(name: "generate-swiftsyntaxbuilder", targets: ["generate-swiftsyntaxbuilder"]),
   ],
   dependencies: [
-    .package(url: "https://github.com/apple/swift-syntax.git", revision: "6e3dfb332553ad1462f0a3d45b4d91e349ce4013"),
+    .package(url: "https://github.com/apple/swift-syntax.git", revision: "7cbed94a9597cc12ec5adc8e63358a54391129fa"),
     .package(url: "https://github.com/apple/swift-argument-parser.git", .upToNextMinor(from: "1.1.4")),
   ],
   targets: [

CodeGeneration/Sources/SyntaxSupport/Node.swift

@@ -23,6 +23,7 @@ public class Node {
   public let nameForDiagnostics: String?
   public let description: String?
   public let baseKind: String
+  public let parserFunction: String?
   public let traits: [String]
   public let children: [Child]
   public let nonUnexpectedChildren: [Child]
@@ -77,6 +78,7 @@ public class Node {
        description: String? = nil,
        kind: String,
        traits: [String] = [],
+       parserFunction: String? = nil,
        children: [Child] = [],
        element: String = "",
        elementName: String? = nil,
@@ -88,7 +90,7 @@ public class Node {
     self.name = kindToType(kind: self.syntaxKind)
     self.nameForDiagnostics = nameForDiagnostics
     self.description = description
-
+    self.parserFunction = parserFunction
     self.traits = traits
     self.baseKind = kind
 

CodeGeneration/Sources/SyntaxSupport/gyb_generated/CommonNodes.swift

@@ -15,23 +15,28 @@
 public let COMMON_NODES: [Node] = [
   Node(name: "Decl",
        nameForDiagnostics: "declaration",
-       kind: "Syntax"),
+       kind: "Syntax",
+       parserFunction: "parseDeclaration"),
 
   Node(name: "Expr",
        nameForDiagnostics: "expression",
-       kind: "Syntax"),
+       kind: "Syntax",
+       parserFunction: "parseExpression"),
 
   Node(name: "Stmt",
        nameForDiagnostics: "statement",
-       kind: "Syntax"),
+       kind: "Syntax",
+       parserFunction: "parseStatement"),
 
   Node(name: "Type",
        nameForDiagnostics: "type",
-       kind: "Syntax"),
+       kind: "Syntax",
+       parserFunction: "parseType"),
 
   Node(name: "Pattern",
        nameForDiagnostics: "pattern",
-       kind: "Syntax"),
+       kind: "Syntax",
+       parserFunction: "parsePattern"),
 
   Node(name: "UnknownDecl",
        nameForDiagnostics: "declaration",

CodeGeneration/Sources/SyntaxSupport/gyb_generated/DeclNodes.swift

@@ -645,6 +645,7 @@ public let DECL_NODES: [Node] = [
        traits: [
          "Braced"
        ],
+       parserFunction: "parseMemberDeclList",
        children: [
          Child(name: "LeftBrace",
                kind: "LeftBraceToken",
@@ -693,6 +694,7 @@ public let DECL_NODES: [Node] = [
        traits: [
          "WithStatements"
        ],
+       parserFunction: "parseSourceFile",
        children: [
          Child(name: "Statements",
                kind: "CodeBlockItemList",

CodeGeneration/Sources/SyntaxSupport/gyb_generated/GenericNodes.swift

@@ -196,6 +196,7 @@ public let GENERIC_NODES: [Node] = [
   Node(name: "GenericParameterClause",
        nameForDiagnostics: "generic parameter clause",
        kind: "Syntax",
+       parserFunction: "parseGenericParameters",
        children: [
          Child(name: "LeftAngleBracket",
                kind: "LeftAngleToken",

CodeGeneration/Sources/SyntaxSupport/gyb_generated/StmtNodes.swift

@@ -560,6 +560,7 @@ public let STMT_NODES: [Node] = [
        traits: [
          "WithStatements"
        ],
+       parserFunction: "parseSwitchCase",
        children: [
          Child(name: "UnknownAttr",
                kind: "Attribute",
@@ -660,6 +661,7 @@ public let STMT_NODES: [Node] = [
        traits: [
          "WithCodeBlock"
        ],
+       parserFunction: "parseCatchClause",
        children: [
          Child(name: "CatchKeyword",
                kind: "CatchToken",

CodeGeneration/Sources/SyntaxSupport/gyb_helpers/utils.py

@@ -20,6 +20,9 @@ def make_swift_node(node):
     mapped_traits = map(lambda x: (' ' * spaces) + '"%s"' % x, node.traits)
     parameters += ['traits: [\n%s\n' % ',\n'.join(mapped_traits) + (' ' * (spaces - 2)) + ']']
 
+  if node.parser_function:
+    parameters += ['parserFunction: "%s"' % node.parser_function]
+
   if node.non_unexpected_children:
     children = []
     for child in node.non_unexpected_children:

CodeGeneration/Sources/generate-swiftsyntaxbuilder/GenerateSwiftSyntaxBuilder.swift

@@ -29,6 +29,7 @@ struct GenerateSwiftSyntaxBuilder: ParsableCommand {
         (buildableCollectionNodesFile, "BuildableCollectionNodes.swift"),
         (buildableNodesFile, "BuildableNodes.swift"),
         (resultBuildersFile, "ResultBuilders.swift"),
+        (syntaxExpressibleByStringInterpolationConformancesFile, "SyntaxExpressibleByStringInterpolationConformances.swift"),
         (tokenFile, "Token.swift"),
         (typealiasesFile, "Typealiases.swift")
       ],

CodeGeneration/Sources/generate-swiftsyntaxbuilder/SyntaxExpressibleByStringInterpolationConformancesFile.swift

@@ -0,0 +1,90 @@
+//===----------------------------------------------------------------------===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2014 - 2022 Apple Inc. and the Swift project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See https://swift.org/LICENSE.txt for license information
+// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
+//
+//===----------------------------------------------------------------------===//
+
+import SwiftSyntax
+import SyntaxSupport
+import SwiftSyntaxBuilder
+import Utils
+
+let syntaxExpressibleByStringInterpolationConformancesFile = SourceFile {
+  ImportDecl(
+    """
+    \(raw: generateCopyrightHeader(for: "generate-swiftsyntaxbuilder"))
+    import SwiftSyntax
+    """)
+  ImportDecl("import SwiftParser")
+  ImportDecl("import SwiftParserDiagnostics")
+  
+  ExtensionDecl("extension SyntaxParseable") {
+    InitializerDecl(
+      """
+      public init(stringInterpolationOrThrow stringInterpolation: SyntaxStringInterpolation) throws {
+        self = try performParse(source: stringInterpolation.sourceText, parse: { parser in
+          return Self.parse(from: &parser)
+        })
+      }
+      """)
+  }
+  
+  for node in SYNTAX_NODES where node.parserFunction != nil && node.isBase {
+    if node.parserFunction != nil && node.isBase {
+      ExtensionDecl("extension \(node.name)Protocol") {
+        InitializerDecl(
+          """
+          public init(stringInterpolationOrThrow stringInterpolation: SyntaxStringInterpolation) throws {
+            self = try performParse(source: stringInterpolation.sourceText, parse: { parser in
+              let node = \(raw: node.name).parse(from: &parser)
+              guard let result = node.as(Self.self) else {
+                throw SyntaxStringInterpolationError.producedInvalidNodeType(expectedType: Self.self, actualType: node.kind.syntaxNodeType)
+              }
+              return result
+            })
+          }
+          """)
+        
+      }
+      
+      ExtensionDecl("extension \(node.name): SyntaxExpressibleByStringInterpolation") {
+        InitializerDecl(
+          """
+          public init(stringInterpolationOrThrow stringInterpolation: SyntaxStringInterpolation) throws {
+            self = try performParse(source: stringInterpolation.sourceText, parse: { parser in
+              return Self.parse(from: &parser)
+            })
+          }
+          """)
+      }
+    } else if node.parserFunction != nil || (node.baseType.baseName != "Syntax" && node.baseType.baseName != "SyntaxCollection") {
+      ExtensionDecl("extension \(raw: node.name): SyntaxExpressibleByStringInterpolation")
+    }
+  }
+  
+  FunctionDecl(
+    """
+    // TODO: This should be fileprivate, but is currently used in
+    // `ConvenienceInitializers.swift`. See the corresponding TODO there.
+    func performParse<SyntaxType: SyntaxProtocol>(source: [UInt8], parse: (inout Parser) throws -> SyntaxType) throws -> SyntaxType {
+      return try source.withUnsafeBufferPointer { buffer in
+        var parser = Parser(buffer)
+        // FIXME: When the parser supports incremental parsing, put the
+        // interpolatedSyntaxNodes in so we don't have to parse them again.
+        let result = try parse(&parser)
+        if result.hasError {
+          let diagnostics = ParseDiagnosticsGenerator.diagnostics(for: result)
+          assert(!diagnostics.isEmpty)
+          throw SyntaxStringInterpolationError.diagnostics(diagnostics, tree: Syntax(result))
+        }
+        return result
+      }
+    }
+    """)
+}

Package.swift

@@ -88,7 +88,6 @@ let package = Package(
       dependencies: ["SwiftBasicFormat", "SwiftSyntax", "SwiftParser", "SwiftParserDiagnostics"],
       exclude: [
         "CMakeLists.txt",
-        "SyntaxExpressibleByStringInterpolationConformances.swift.gyb",
       ]
     ),
     .target(