WarningControlDeclSyntax.swift

//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//

@_spi(ExperimentalLanguageFeatures) import SwiftSyntax

extension AttributeSyntax {
  var warningGroupControl: (DiagnosticGroupIdentifier, WarningGroupControl)? {
    // `@diagnose` attributes (also accepts the old `@warn` spelling)
    guard let attrName = attributeName.as(IdentifierTypeSyntax.self)?.name.text,
      attrName == "diagnose" || attrName == "warn"
    else {
      return nil
    }

    // First argument is the unquoted diagnostic group identifier
    guard
      let diagnosticGroupID = arguments?
        .as(LabeledExprListSyntax.self)?.first?.expression
        .as(DeclReferenceExprSyntax.self)?.baseName.text
    else {
      return nil
    }

    // Second argument is the `as: ` behavior control specifier
    guard
      let asParamExprSyntax = arguments?.as(LabeledExprListSyntax.self)?
        .dropFirst().first
    else {
      return nil
    }
    guard
      asParamExprSyntax.label?.text == "as",
      let controlText = asParamExprSyntax
        .expression.as(DeclReferenceExprSyntax.self)?
        .baseName.text,
      let control = WarningGroupControl(rawValue: controlText)
    else {
      return nil
    }

    return (DiagnosticGroupIdentifier(diagnosticGroupID), control)
  }
}

extension WithAttributesSyntax {
  /// Compute a dictionary of all `@diagnose` diagnostic group behavior controls
  /// specified on this warning control declaration scope.
  var allWarningGroupControls: [(DiagnosticGroupIdentifier, WarningGroupControl)] {
    attributes.reduce(into: [(DiagnosticGroupIdentifier, WarningGroupControl)]()) { result, attr in
      guard case .attribute(let attributeSyntax) = attr,
        let warningGroupControl = attributeSyntax.warningGroupControl
      else {
        return
      }
      result.append(warningGroupControl)
    }
  }
}

extension UsingDeclSyntax {
  var warningControl: (DiagnosticGroupIdentifier, WarningGroupControl)? {
    guard case .attribute(let attributeSyntax) = self.specifier,
      let warningGroupControl = attributeSyntax.warningGroupControl
    else {
      return nil
    }
    return warningGroupControl
  }
}