Merge pull request #3179 from artemcm/WarningControl-KeepTrackOfEnabled

[SwiftWarningControl] Keep track of enabled diagnostic group identifiers

Author: artemcm

Sources/SwiftWarningControl/WarningControlRegions.swift

@@ -103,8 +103,14 @@ public struct WarningControlRegionTree {
   /// Root region representing top-level (file) scope
   private var rootRegionNode: WarningControlRegionNode
 
+  /// All of the diagnostic group identifiers contained in this tree
+  /// which have at least one occurence with a non-`ignored` behavior
+  /// specifier
+  @_spi(ExperimentalLanguageFeatures)
+  public private(set) var enabledGroups: Set<DiagnosticGroupIdentifier> = []
+
   /// Inheritance tree among diagnostic group identifiers
-  let groupInheritanceTree: DiagnosticGroupInheritanceTree
+  private let groupInheritanceTree: DiagnosticGroupInheritanceTree
 
   init(
     range: Range<AbsolutePosition>,
@@ -130,6 +136,10 @@ public struct WarningControlRegionTree {
         let groupIdentifier = groups.removeFirst()
         processedGroups.insert(groupIdentifier)
         newNode.addWarningGroupControl(for: groupIdentifier, control: control)
+        if control != .ignored {
+          enabledGroups.insert(diagnosticGroupIdentifier)
+        }
+
         let newSubGroups = groupInheritanceTree.subgroups(of: groupIdentifier).filter { !processedGroups.contains($0) }
         // Ensure we add a corresponding control to each direct and
         // transitive sub-group of the one specified on this control.

Tests/SwiftWarningControlTest/WarningControlTests.swift

@@ -75,6 +75,40 @@ public class WarningGroupControlTests: XCTestCase {
     )
   }
 
+  func testEnabledGroupIdentifiers() throws {
+    let source =
+      """
+      @warn(Group1, as: warning)
+      @warn(Group2, as: error)
+      @warn(Group3, as: ignored)
+      func foo() {
+        @warn(Group4, as: warning)
+        func bar() {
+          @warn(Group5, as: ignored)
+          @warn(Group6, as: ignored)
+          func baz() {}
+          @warn(Group7, as: error)
+          func qux() {
+            @warn(Group8, as: warning)
+            func corge() {}
+          }
+        }
+      }
+      """
+    var parser = Parser(source)
+    let parseTree = SourceFileSyntax.parse(from: &parser)
+    let warningControlTree = parseTree.warningGroupControlRegionTree()
+    let enabledDiagnosticGroups = warningControlTree.enabledGroups
+    XCTAssertTrue(enabledDiagnosticGroups.contains("Group1"))
+    XCTAssertTrue(enabledDiagnosticGroups.contains("Group2"))
+    XCTAssertFalse(enabledDiagnosticGroups.contains("Group3"))
+    XCTAssertTrue(enabledDiagnosticGroups.contains("Group4"))
+    XCTAssertFalse(enabledDiagnosticGroups.contains("Group5"))
+    XCTAssertFalse(enabledDiagnosticGroups.contains("Group6"))
+    XCTAssertTrue(enabledDiagnosticGroups.contains("Group7"))
+    XCTAssertTrue(enabledDiagnosticGroups.contains("Group8"))
+  }
+
   func testNominalDeclWarningGroupControl() throws {
     try assertWarningGroupControl(
       """