Convert calculate_output_groups.py into a Swift binary

Signed-off-by: Matt Pennig <[email protected]>

Author: pennig

distribution/BUILD

@@ -59,6 +59,9 @@ pkg_tar(
     remap_paths = dicts.add(
         {
             "MODULE.release.bazel": "MODULE.bazel",
+            "tools/calculate_output_groups/BUILD.release.bazel": (
+                "tools/calculate_output_groups/BUILD"
+            ),
             "tools/import_indexstores/BUILD.release.bazel": (
                 "tools/import_indexstores/BUILD"
             ),

examples/integration/test/fixtures/bwb.xcodeproj/project.pbxproj

@@ -14,16 +14,18 @@ struct CalculateOutputGroups: AsyncParsableCommand {
     @OptionGroup var arguments: OutputGroupsCalculator.Arguments
 
     func run() async throws {
+        var output = StdoutOutputStream()
         let logger = DefaultLogger(
             standardError: StderrOutputStream(),
-            standardOutput: StdoutOutputStream(),
+            standardOutput: output,
             colorize: colorDiagnostics
         )
 
-        let calculator = OutputGroupsCalculator()
+        let calculator = OutputGroupsCalculator(logger: logger)
 
         do {
-            try await calculator.calculateOutputGroups(arguments: arguments)
+            let groups = try await calculator.calculateOutputGroups(arguments: arguments)
+            print(groups, to: &output)
         } catch {
             logger.logError(error.localizedDescription)
             Darwin.exit(1)

examples/integration/test/fixtures/bwx.xcodeproj/project.pbxproj

@@ -3,7 +3,7 @@ import ToolCommon
 extension UsageError {
     static func buildMarker(_ path: String) -> Self {
         .init(message: """
-error: Build marker (\(path)) doesn't exist. If you manually cleared Derived \
+Build marker (\(path)) doesn't exist. If you manually cleared Derived \
 Data, you need to close and re-open the project for the file to be created \
 again. Using the "Clean Build Folder" command instead (⇧ ⌘ K) won't trigger \
 this error. If this error still happens after re-opening the project, please \
@@ -14,7 +14,7 @@ https://github.com/MobileNativeFoundation/rules_xcodeproj/issues/new?template=bu
 
     static func pifCache(_ path: String) -> Self {
         .init(message: """
-error: PIFCache (\(path)) doesn't exist. If you manually cleared Derived \
+PIFCache (\(path)) doesn't exist. If you manually cleared Derived \
 Data, you need to close and re-open the project for the PIFCache to be created \
 again. Using the "Clean Build Folder" command instead (⇧ ⌘ K) won't trigger \
 this error. If this error still happens after re-opening the project, please \
@@ -25,7 +25,7 @@ https://github.com/MobileNativeFoundation/rules_xcodeproj/issues/new?template=bu
 
     static func buildRequest(_ path: String) -> Self {
         .init(message: """
-error: Couldn't find a build-request.json file inside \(path)". Please file a bug \
+Couldn't find latest build-request.json file after 30 seconds. Please file a bug \
 report here: https://github.com/MobileNativeFoundation/rules_xcodeproj/issues/new?template=bug.md
 """)
     }

examples/rules_ios/test/fixtures/bwb.xcodeproj/project.pbxproj

@@ -3,7 +3,9 @@ import ToolCommon
 import ZippyJSON
 
 struct OutputGroupsCalculator {
-    func calculateOutputGroups(arguments: Arguments) async throws {
+    let logger: Logger
+    
+    func calculateOutputGroups(arguments: Arguments) async throws -> String {
         let pifCache = arguments.baseObjRoot.appendingPathComponent("XCBuildData/PIFCache")
         let projectCache = pifCache.appendingPathComponent("project")
         let targetCache = pifCache.appendingPathComponent("target")
@@ -30,12 +32,11 @@ struct OutputGroupsCalculator {
             targetCache: targetCache
         )
 
-        let output = try await outputGroups(
+        return try await outputGroups(
             buildRequest: buildRequest,
             targets: targetMap,
             prefixes: arguments.outputGroupPrefixes
         )
-        print(output)
     }
 
     private func loadBuildRequestFile(inPath path: URL, since: Date) async throws -> BuildRequest {
@@ -54,22 +55,33 @@ struct OutputGroupsCalculator {
         }
 
         // If the file was not immediately found, kick off a process to wait for the file to be created (or time out).
-        let findTask = Task {
-            while true {
-                try Task.checkCancellation()
-                try await Task.sleep(for: .seconds(1))
-                if let buildRequestURL = findBuildRequestURL() {
-                    return buildRequestURL
+        do {
+            let findTask = Task {
+                logger.logWarning("The latest build-request.json file has not been updated yet. Waiting…")
+                while true {
+                    try Task.checkCancellation()
+                    try await Task.sleep(for: .seconds(1))
+                    if let buildRequestURL = findBuildRequestURL() {
+                        return buildRequestURL
+                    }
                 }
             }
-        }
-        let timeoutTask = Task {
-            try await Task.sleep(for: .seconds(10))
-            findTask.cancel()
-        }
+            let waitingTask = Task {
+                try await Task.sleep(for: .seconds(10))
+                try Task.checkCancellation()
+                logger.logWarning("""
+The latest build-request.json file has still not been updated after 10 seconds. If this happens frequently, please file a bug report here:
+https://github.com/MobileNativeFoundation/rules_xcodeproj/issues/new?template=bug.md
+""")
+            }
+            let timeoutTask = Task {
+                try await Task.sleep(for: .seconds(30))
+                guard !Task.isCancelled else { return }
+                findTask.cancel()
+            }
 
-        do {
             let result = try await findTask.value
+            waitingTask.cancel()
             timeoutTask.cancel()
             return try result.decode(BuildRequest.self)
         } catch {

tools/calculate_output_groups/CalculateOutputGroups.swift

@@ -35,9 +35,6 @@ extension Generator {
 
         var buildSettings: BuildSettings = [
             "BAZEL_PACKAGE_BIN_DIR": "rules_xcodeproj",
-            "CALCULATE_OUTPUT_GROUPS_SCRIPT": """
-$(BAZEL_INTEGRATION_DIR)/calculate_output_groups.py
-""",
             "INDEXING_SUPPORTED_PLATFORMS__": """
 $(INDEXING_SUPPORTED_PLATFORMS__NO)
 """,

tools/calculate_output_groups/Errors.swift

@@ -283,7 +283,7 @@ $(BAZEL_OUT)\#(linkParams.path.string.dropFirst(9))
         buildSettings.set("TARGET_NAME", to: target.name)
 
         if !target.product.isResourceBundle {
-            // This is used in `calculate_output_groups.py`. We only want to set
+            // This is used in `calculate_output_groups`. We only want to set
             // it on buildable targets
             buildSettings.set("BAZEL_LABEL", to: target.label.description)
         }

tools/calculate_output_groups/OutputGroupsCalculator.swift

@@ -114,7 +114,7 @@ extension Generator.CalculateSharedBuildSettings {
         }
 
         if productType != .resourceBundle {
-            // This is used in `calculate_output_groups.py`. We only want to set
+            // This is used in `calculate_output_groups`. We only want to set
             // it on buildable targets.
             buildSettings.append(
                 .init(