Refactored file generation logic to be string-based and added validation tests

TTOzzi · TTOzzi · commit a48a3d43dbcd · 2025-07-17T22:47:08.000+09:00
diff --git a/Package.swift b/Package.swift
@@ -113,6 +113,7 @@ var targets: [Target] = [
     dependencies: [
       "SwiftFormat",
       "_SwiftFormatTestSupport",
+      "generate-swift-format",
       .product(name: "Markdown", package: "swift-markdown"),
     ] + swiftSyntaxDependencies(["SwiftOperators", "SwiftParser", "SwiftSyntax", "SwiftSyntaxBuilder"])
   ),
diff --git a/Sources/SwiftFormat/Core/Pipelines+Generated.swift b/Sources/SwiftFormat/Core/Pipelines+Generated.swift
@@ -2,7 +2,7 @@
 //
 // This source file is part of the Swift.org open source project
 //
-// Copyright (c) 2014 - 2019 Apple Inc. and the Swift project authors
+// Copyright (c) 2014 - 2025 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
diff --git a/Sources/SwiftFormat/Core/RuleNameCache+Generated.swift b/Sources/SwiftFormat/Core/RuleNameCache+Generated.swift
@@ -2,7 +2,7 @@
 //
 // This source file is part of the Swift.org open source project
 //
-// Copyright (c) 2014 - 2019 Apple Inc. and the Swift project authors
+// Copyright (c) 2014 - 2025 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
diff --git a/Sources/SwiftFormat/Core/RuleRegistry+Generated.swift b/Sources/SwiftFormat/Core/RuleRegistry+Generated.swift
@@ -2,7 +2,7 @@
 //
 // This source file is part of the Swift.org open source project
 //
-// Copyright (c) 2014 - 2019 Apple Inc. and the Swift project authors
+// Copyright (c) 2014 - 2025 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
diff --git a/Sources/generate-swift-format/FileGenerator.swift b/Sources/generate-swift-format/FileGenerator.swift
@@ -14,9 +14,8 @@ import Foundation
 
 /// Common behavior used to generate source files.
 protocol FileGenerator {
-  /// Types conforming to this protocol must implement this method to write their content into the
-  /// given file handle.
-  func write(into handle: FileHandle) throws
+  /// Generates the file content as a String.
+  func generateString() -> String
 }
 
 private struct FailedToCreateFileError: Error {
@@ -30,21 +29,7 @@ extension FileGenerator {
     if fm.fileExists(atPath: url.path) {
       try fm.removeItem(at: url)
     }
-
-    if !fm.createFile(atPath: url.path, contents: nil, attributes: nil) {
-      throw FailedToCreateFileError(url: url)
-    }
-    let handle = try FileHandle(forWritingTo: url)
-    defer { handle.closeFile() }
-
-    try write(into: handle)
-  }
-}
-
-extension FileHandle {
-  /// Writes the provided string as data to a file output stream.
-  public func write(_ string: String) {
-    guard let data = string.data(using: .utf8) else { return }
-    write(data)
+    let content = generateString()
+    try content.write(to: url, atomically: true, encoding: .utf8)
   }
 }
diff --git a/Sources/generate-swift-format/GenerateSwiftFormatPaths.swift b/Sources/generate-swift-format/GenerateSwiftFormatPaths.swift
@@ -0,0 +1,45 @@
+//===----------------------------------------------------------------------===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2014 - 2025 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 Foundation
+
+/// A namespace that provides paths for files generated by the `generate-swift-format` tool.
+enum GenerateSwiftFormatPaths {
+  private static let sourcesDirectory =
+    URL(fileURLWithPath: #file)
+    .deletingLastPathComponent()
+    .deletingLastPathComponent()
+  static let rulesDirectory =
+    sourcesDirectory
+    .appendingPathComponent("SwiftFormat")
+    .appendingPathComponent("Rules")
+  static let pipelineFile =
+    sourcesDirectory
+    .appendingPathComponent("SwiftFormat")
+    .appendingPathComponent("Core")
+    .appendingPathComponent("Pipelines+Generated.swift")
+  static let ruleRegistryFile =
+    sourcesDirectory
+    .appendingPathComponent("SwiftFormat")
+    .appendingPathComponent("Core")
+    .appendingPathComponent("RuleRegistry+Generated.swift")
+  static let ruleNameCacheFile =
+    sourcesDirectory
+    .appendingPathComponent("SwiftFormat")
+    .appendingPathComponent("Core")
+    .appendingPathComponent("RuleNameCache+Generated.swift")
+  static let ruleDocumentationFile =
+    sourcesDirectory
+    .appendingPathComponent("..")
+    .appendingPathComponent("Documentation")
+    .appendingPathComponent("RuleDocumentation.md")
+}
diff --git a/Sources/generate-swift-format/PipelineGenerator.swift b/Sources/generate-swift-format/PipelineGenerator.swift
@@ -23,14 +23,14 @@ final class PipelineGenerator: FileGenerator {
     self.ruleCollector = ruleCollector
   }
 
-  func write(into handle: FileHandle) throws {
-    handle.write(
-      """
+  func generateString() -> String {
+    var result = ""
+    result += """
       //===----------------------------------------------------------------------===//
       //
       // This source file is part of the Swift.org open source project
       //
-      // Copyright (c) 2014 - 2019 Apple Inc. and the Swift project authors
+      // Copyright (c) 2014 - 2025 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
@@ -65,58 +65,41 @@ final class PipelineGenerator: FileGenerator {
         }
 
       """
-    )
 
     for (nodeType, lintRules) in ruleCollector.syntaxNodeLinters.sorted(by: { $0.key < $1.key }) {
-      handle.write(
-        """
+      result += """
 
           override func visit(_ node: \(nodeType)) -> SyntaxVisitorContinueKind {
 
         """
-      )
-
       for ruleName in lintRules.sorted() {
-        handle.write(
-          """
+        result += """
               visitIfEnabled(\(ruleName).visit, for: node)
 
           """
-        )
       }
-
-      handle.write(
-        """
+      result += """
             return .visitChildren
           }
 
         """
-      )
-
-      handle.write(
-        """
+      result += """
           override func visitPost(_ node: \(nodeType)) {
 
         """
-      )
       for ruleName in lintRules.sorted() {
-        handle.write(
-          """
+        result += """
               onVisitPost(rule: \(ruleName).self, for: node)
 
           """
-        )
       }
-      handle.write(
-        """
+      result += """
           }
 
         """
-      )
     }
 
-    handle.write(
-      """
+    result += """
       }
 
       extension FormatPipeline {
@@ -125,24 +108,18 @@ final class PipelineGenerator: FileGenerator {
           var node = node
 
       """
-    )
-
     for ruleName in ruleCollector.allFormatters.map({ $0.typeName }).sorted() {
-      handle.write(
-        """
+      result += """
             node = \(ruleName)(context: context).rewrite(node)
 
         """
-      )
     }
-
-    handle.write(
-      """
+    result += """
           return node
         }
       }
 
       """
-    )
+    return result
   }
 }
diff --git a/Sources/generate-swift-format/RuleDocumentationGenerator.swift b/Sources/generate-swift-format/RuleDocumentationGenerator.swift
@@ -24,9 +24,9 @@ final class RuleDocumentationGenerator: FileGenerator {
     self.ruleCollector = ruleCollector
   }
 
-  func write(into handle: FileHandle) throws {
-    handle.write(
-      """
+  func generateString() -> String {
+    var result = ""
+    result += """
       <!-- This file is automatically generated with generate-swift-format. Do not edit! -->
 
       # `swift-format` Lint and Format Rules
@@ -41,29 +41,20 @@ final class RuleDocumentationGenerator: FileGenerator {
 
 
       """
-    )
-
     for detectedRule in ruleCollector.allLinters.sorted(by: { $0.typeName < $1.typeName }) {
-      handle.write(
-        """
-        - [\(detectedRule.typeName)](#\(detectedRule.typeName))
-
-        """
-      )
+      result += "- [\(detectedRule.typeName)](#\(detectedRule.typeName))\n"
     }
-
     for detectedRule in ruleCollector.allLinters.sorted(by: { $0.typeName < $1.typeName }) {
-      handle.write(
-        """
+      result += """
 
         ### \(detectedRule.typeName)
 
         \(detectedRule.description ?? "")
         \(ruleFormatSupportDescription(for: detectedRule))
 
         """
-      )
     }
+    return result
   }
 
   private func ruleFormatSupportDescription(for rule: RuleCollector.DetectedRule) -> String {
diff --git a/Sources/generate-swift-format/RuleNameCacheGenerator.swift b/Sources/generate-swift-format/RuleNameCacheGenerator.swift
@@ -23,14 +23,14 @@ final class RuleNameCacheGenerator: FileGenerator {
     self.ruleCollector = ruleCollector
   }
 
-  func write(into handle: FileHandle) throws {
-    handle.write(
-      """
+  func generateString() -> String {
+    var result = ""
+    result += """
       //===----------------------------------------------------------------------===//
       //
       // This source file is part of the Swift.org open source project
       //
-      // Copyright (c) 2014 - 2019 Apple Inc. and the Swift project authors
+      // Copyright (c) 2014 - 2025 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
@@ -45,11 +45,10 @@ final class RuleNameCacheGenerator: FileGenerator {
       public let ruleNameCache: [ObjectIdentifier: String] = [
 
       """
-    )
-
     for detectedRule in ruleCollector.allLinters.sorted(by: { $0.typeName < $1.typeName }) {
-      handle.write("  ObjectIdentifier(\(detectedRule.typeName).self): \"\(detectedRule.typeName)\",\n")
+      result += "  ObjectIdentifier(\(detectedRule.typeName).self): \"\(detectedRule.typeName)\",\n"
     }
-    handle.write("]\n")
+    result += "]\n"
+    return result
   }
 }
diff --git a/Sources/generate-swift-format/RuleRegistryGenerator.swift b/Sources/generate-swift-format/RuleRegistryGenerator.swift
@@ -23,14 +23,14 @@ final class RuleRegistryGenerator: FileGenerator {
     self.ruleCollector = ruleCollector
   }
 
-  func write(into handle: FileHandle) throws {
-    handle.write(
-      """
+  func generateString() -> String {
+    var result = ""
+    result += """
       //===----------------------------------------------------------------------===//
       //
       // This source file is part of the Swift.org open source project
       //
-      // Copyright (c) 2014 - 2019 Apple Inc. and the Swift project authors
+      // Copyright (c) 2014 - 2025 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
@@ -44,11 +44,10 @@ final class RuleRegistryGenerator: FileGenerator {
         public static let rules: [String: Bool] = [
 
       """
-    )
-
     for detectedRule in ruleCollector.allLinters.sorted(by: { $0.typeName < $1.typeName }) {
-      handle.write("    \"\(detectedRule.typeName)\": \(!detectedRule.isOptIn),\n")
+      result += "    \"\(detectedRule.typeName)\": \(!detectedRule.isOptIn),\n"
     }
-    handle.write("  ]\n}\n")
+    result += "  ]\n}\n"
+    return result
   }
 }
diff --git a/Sources/generate-swift-format/main.swift b/Sources/generate-swift-format/main.swift
diff --git a/Tests/SwiftFormatTests/Utilities/GeneratedFilesValidityTests.swift b/Tests/SwiftFormatTests/Utilities/GeneratedFilesValidityTests.swift
