Share gnerate command between generate-swiftbasicformat and generate-swiftsyntaxbuilder

Author: ahoppen

Code-Generation/Sources/Utils/GenerateCommand.swift

@@ -0,0 +1,50 @@
+//===----------------------------------------------------------------------===//
+//
+// 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 ArgumentParser
+import Foundation
+import SwiftSyntax
+import SwiftSyntaxBuilder
+
+open class GenerateCommand: ParsableCommand {
+  @Argument(help: "The path to the destination directory where the source files are to be generated")
+  var generatedPath: String
+
+  @Flag(help: "Enable verbose output")
+  var verbose: Bool = false
+
+  open var sourceTemplates: [(SourceFile, String)] {
+    fatalError("Must be overridden by subclasses")
+  }
+
+  public required init() {}
+
+  public func run() throws {
+    let generatedURL = URL(fileURLWithPath: generatedPath)
+    let format = Format(indentWidth: 2)
+
+    try FileManager.default.createDirectory(
+      atPath: generatedURL.path,
+      withIntermediateDirectories: true,
+      attributes: nil
+    )
+
+    for (sourceFile, name) in sourceTemplates {
+      let fileURL = generatedURL.appendingPathComponent(name)
+      if verbose {
+        print("Generating \(fileURL.path)...")
+      }
+      let syntax = sourceFile.buildSyntax(format: format)
+      try "\(syntax)\n".write(to: fileURL, atomically: true, encoding: .utf8)
+    }
+  }
+}

Code-Generation/Sources/generate-swiftbasicformat/GenerateSwiftBasicFormat.swift

@@ -10,41 +10,14 @@
 //
 //===----------------------------------------------------------------------===//
 
-import ArgumentParser
-import Foundation
-import SwiftSyntax
+import Utils
 import SwiftSyntaxBuilder
 
-/// SwiftSyntaxBuilder sources to be generated
-private let sourceTemplates: [(SourceFile, String)] = [
-  (formatFile, "Format.swift"),
-]
-
 @main
-struct GenerateSwiftSyntaxBuilder: ParsableCommand {
-  @Argument(help: "The path to the destination directory where the source files are to be generated")
-  var generatedPath: String
-
-  @Flag(help: "Enable verbose output")
-  var verbose: Bool = false
-
-  func run() throws {
-    let generatedURL = URL(fileURLWithPath: generatedPath)
-    let format = Format(indentWidth: 2)
-
-    try FileManager.default.createDirectory(
-      atPath: generatedURL.path,
-      withIntermediateDirectories: true,
-      attributes: nil
-    )
-
-    for (sourceFile, name) in sourceTemplates {
-      let fileURL = generatedURL.appendingPathComponent(name)
-      if verbose {
-        print("Generating \(fileURL.path)...")
-      }
-      let syntax = sourceFile.buildSyntax(format: format)
-      try "\(syntax)\n".write(to: fileURL, atomically: true, encoding: .utf8)
-    }
+class GenerateSwiftStaticFormat: GenerateCommand {
+  open override var sourceTemplates: [(SourceFile, String)] {
+    [
+      (formatFile, "Format.swift"),
+    ]
   }
 }

Code-Generation/Sources/generate-swiftsyntaxbuilder/GenerateSwiftSyntaxBuilder.swift

@@ -10,45 +10,18 @@
 //
 //===----------------------------------------------------------------------===//
 
-import ArgumentParser
-import Foundation
-import SwiftSyntax
+import Utils
 import SwiftSyntaxBuilder
 
-/// SwiftSyntaxBuilder sources to be generated
-private let sourceTemplates: [(SourceFile, String)] = [
-  (buildableBaseProtocolsFile, "BuildableBaseProtocols.swift"),
-  (buildableCollectionNodesFile, "BuildableCollectionNodes.swift"),
-  (buildableNodesFile, "BuildableNodes.swift"),
-  (expressibleAsProtocolsFile, "ExpressibleAsProtocols.swift"),
-  (tokenFile, "Token.swift"),
-]
-
 @main
-struct GenerateSwiftSyntaxBuilder: ParsableCommand {
-  @Argument(help: "The path to the destination directory where the source files are to be generated")
-  var generatedPath: String
-
-  @Flag(help: "Enable verbose output")
-  var verbose: Bool = false
-
-  func run() throws {
-    let generatedURL = URL(fileURLWithPath: generatedPath)
-    let format = Format(indentWidth: 2)
-
-    try FileManager.default.createDirectory(
-      atPath: generatedURL.path,
-      withIntermediateDirectories: true,
-      attributes: nil
-    )
-
-    for (sourceFile, name) in sourceTemplates {
-      let fileURL = generatedURL.appendingPathComponent(name)
-      if verbose {
-        print("Generating \(fileURL.path)...")
-      }
-      let syntax = sourceFile.buildSyntax(format: format)
-      try "\(syntax)\n".write(to: fileURL, atomically: true, encoding: .utf8)
-    }
+class GenerateSwiftSyntaxBuilder: GenerateCommand {
+  open override var sourceTemplates: [(SourceFile, String)] {
+    [
+      (buildableBaseProtocolsFile, "BuildableBaseProtocols.swift"),
+      (buildableCollectionNodesFile, "BuildableCollectionNodes.swift"),
+      (buildableNodesFile, "BuildableNodes.swift"),
+      (expressibleAsProtocolsFile, "ExpressibleAsProtocols.swift"),
+      (tokenFile, "Token.swift"),
+    ]
   }
 }