[frontend] Add an option called -verify-additional-prefix to add additional check prefixes to the DiagnosticVerifier.

This enables one to use varying prefixes when checking diagnostics with the
DiagnosticVerifier. So for instance, I can make a test work both with and
without send-non-sendable enabled by adding additional prefixes. As an example:

```swift
// RUN: %target-swift-frontend ... -verify-additional-prefix no-sns-
// RUN: %target-swift-frontend ... -verify-additional-prefix sns-

let x = ... // expected-error {{This is always checked no matter what prefixes I added}}
let y = ... // expected-no-sns-error {{This is only checked if send non sendable is disabled}}
let z = ... // expected-sns-error {{This is only checked if send non sendable is enabled}}
let w = ... // expected-no-sns-error {{This is checked for a specific error when sns is disabled...}}
// expected-sns-error @-1 {{and for a different error when sns is enabled}}
```

rdar://114643840

Author: gottesmm

include/swift/Basic/DiagnosticOptions.h

@@ -83,6 +83,17 @@ class DiagnosticOptions {
   /// Path to a directory of diagnostic localization tables.
   std::string LocalizationPath = "";
 
+  /// A list of post-fixes to expected- that the diagnostic verifier should
+  /// check for diagnostics.
+  ///
+  /// For example, if one placed the phrase "name", the verifier will check for:
+  /// expected-name-{error,note,warning,remark} as well as the normal expected-
+  /// prefixes.
+  ///
+  /// NOTE: These are checked in order so if a postfix a is a prefix of another
+  /// postfix b and a is before b in the list, b will never be checked.
+  std::vector<std::string> AdditionalDiagnosticVerifierPostfixes;
+
   /// Return a hash code of any components from these options that should
   /// contribute to a Swift Bridging PCH hash.
   llvm::hash_code getPCHHashComponents() const {

include/swift/Frontend/DiagnosticVerifier.h

@@ -94,12 +94,15 @@ class DiagnosticVerifier : public DiagnosticConsumer {
   SmallVector<unsigned, 4> AdditionalBufferIDs;
   bool AutoApplyFixes;
   bool IgnoreUnknown;
+  ArrayRef<std::string> AcceptedExpectedPostfixes;
 
 public:
   explicit DiagnosticVerifier(SourceManager &SM, ArrayRef<unsigned> BufferIDs,
-                              bool AutoApplyFixes, bool IgnoreUnknown)
+                              bool AutoApplyFixes, bool IgnoreUnknown,
+                              ArrayRef<std::string> AcceptedExpectedPostfixes)
       : SM(SM), BufferIDs(BufferIDs), AutoApplyFixes(AutoApplyFixes),
-        IgnoreUnknown(IgnoreUnknown) {}
+        IgnoreUnknown(IgnoreUnknown),
+        AcceptedExpectedPostfixes(AcceptedExpectedPostfixes) {}
 
   void appendAdditionalBufferID(unsigned bufferID) {
     AdditionalBufferIDs.push_back(bufferID);

include/swift/Option/FrontendOptions.td

@@ -111,6 +111,8 @@ def verify : Flag<["-"], "verify">,
            "annotations">;
 def verify_additional_file : Separate<["-"], "verify-additional-file">,
   HelpText<"Verify diagnostics in this file in addition to source files">;
+def verify_additional_prefix : Separate<["-"], "verify-additional-prefix">,
+  HelpText<"Check for diagnostics with the prefix expected-<PREFIX> as well as expected-">;
 def verify_apply_fixes : Flag<["-"], "verify-apply-fixes">,
   HelpText<"Like -verify, but updates the original source file">;
 def verify_ignore_unknown: Flag<["-"], "verify-ignore-unknown">,

lib/Frontend/CompilerInvocation.cpp

@@ -1795,6 +1795,8 @@ static bool ParseDiagnosticArgs(DiagnosticOptions &Opts, ArgList &Args,
 
   for (Arg *A : Args.filtered(OPT_verify_additional_file))
     Opts.AdditionalVerifierFiles.push_back(A->getValue());
+  for (Arg *A : Args.filtered(OPT_verify_additional_prefix))
+    Opts.AdditionalDiagnosticVerifierPostfixes.push_back(A->getValue());
 
   Opts.UseColor |=
       Args.hasFlag(OPT_color_diagnostics,

lib/Frontend/DiagnosticVerifier.cpp

@@ -26,13 +26,81 @@
 
 using namespace swift;
 
+namespace {
+
+struct ExpectedCheckMatchStartParser {
+  StringRef MatchStart;
+  const char *ClassificationStartLoc = nullptr;
+  std::optional<DiagnosticKind> ExpectedClassification;
+
+  ExpectedCheckMatchStartParser(StringRef MatchStart)
+      : MatchStart(MatchStart) {}
+
+  bool tryParseClassification() {
+    if (MatchStart.startswith("note")) {
+      ClassificationStartLoc = MatchStart.data();
+      ExpectedClassification = DiagnosticKind::Note;
+      MatchStart = MatchStart.substr(strlen("note"));
+      return true;
+    }
+
+    if (MatchStart.startswith("warning")) {
+      ClassificationStartLoc = MatchStart.data();
+      ExpectedClassification = DiagnosticKind::Warning;
+      MatchStart = MatchStart.substr(strlen("warning"));
+      return true;
+    }
+
+    if (MatchStart.startswith("error")) {
+      ClassificationStartLoc = MatchStart.data();
+      ExpectedClassification = DiagnosticKind::Error;
+      MatchStart = MatchStart.substr(strlen("error"));
+      return true;
+    }
+
+    if (MatchStart.startswith("remark")) {
+      ClassificationStartLoc = MatchStart.data();
+      ExpectedClassification = DiagnosticKind::Remark;
+      MatchStart = MatchStart.substr(strlen("remark"));
+      return true;
+    }
+
+    return false;
+  }
+
+  bool parse(ArrayRef<std::string> prefixes) {
+    // First try to parse as if we did not have a prefix. We always parse at
+    // least expected-*.
+    if (tryParseClassification())
+      return true;
+
+    // Otherwise, walk our prefixes until we find one that matches and attempt
+    // to check for a note, warning, error, or remark.
+    //
+    // TODO: We could make this more flexible, but this should work in the
+    // short term.
+    for (auto &p : prefixes) {
+      if (MatchStart.starts_with(p)) {
+        MatchStart = MatchStart.substr(p.size());
+        return tryParseClassification();
+      }
+    }
+
+    return false;
+  }
+};
+
+} // anonymous namespace
+
 namespace swift {
+
 struct ExpectedFixIt {
   const char *StartLoc, *EndLoc; // The loc of the {{ and }}'s.
   LineColumnRange Range;
 
   std::string Text;
 };
+
 } // end namespace swift
 
 const LineColumnRange &
@@ -494,23 +562,21 @@ DiagnosticVerifier::Result DiagnosticVerifier::verifyFile(unsigned BufferID) {
     StringRef MatchStart = InputFile.substr(Match);
     const char *DiagnosticLoc = MatchStart.data();
     MatchStart = MatchStart.substr(strlen("expected-"));
-    const char *ClassificationStartLoc = MatchStart.data();
 
-    DiagnosticKind ExpectedClassification;
-    if (MatchStart.startswith("note")) {
-      ExpectedClassification = DiagnosticKind::Note;
-      MatchStart = MatchStart.substr(strlen("note"));
-    } else if (MatchStart.startswith("warning")) {
-      ExpectedClassification = DiagnosticKind::Warning;
-      MatchStart = MatchStart.substr(strlen("warning"));
-    } else if (MatchStart.startswith("error")) {
-      ExpectedClassification = DiagnosticKind::Error;
-      MatchStart = MatchStart.substr(strlen("error"));
-    } else if (MatchStart.startswith("remark")) {
-      ExpectedClassification = DiagnosticKind::Remark;
-      MatchStart = MatchStart.substr(strlen("remark"));
-    } else
-      continue;
+    const char *ClassificationStartLoc = nullptr;
+    std::optional<DiagnosticKind> ExpectedClassification;
+    {
+      ExpectedCheckMatchStartParser parser(MatchStart);
+      // If we fail to parse... continue.
+      if (!parser.parse(AcceptedExpectedPostfixes)) {
+        continue;
+      }
+      MatchStart = parser.MatchStart;
+      ClassificationStartLoc = parser.ClassificationStartLoc;
+      ExpectedClassification = parser.ExpectedClassification;
+    }
+    assert(ClassificationStartLoc);
+    assert(bool(ExpectedClassification));
 
     // Skip any whitespace before the {{.
     MatchStart = MatchStart.substr(MatchStart.find_first_not_of(" \t"));
@@ -525,7 +591,7 @@ DiagnosticVerifier::Result DiagnosticVerifier::verifyFile(unsigned BufferID) {
 
     ExpectedDiagnosticInfo Expected(DiagnosticLoc, ClassificationStartLoc,
                                     /*ClassificationEndLoc=*/MatchStart.data(),
-                                    ExpectedClassification);
+                                    *ExpectedClassification);
     int LineOffset = 0;
 
     if (TextStartIdx > 0 && MatchStart[0] == '@') {

lib/Frontend/Frontend.cpp

@@ -363,7 +363,8 @@ bool CompilerInstance::setupDiagnosticVerifierIfNeeded() {
     DiagVerifier = std::make_unique<DiagnosticVerifier>(
         SourceMgr, InputSourceCodeBufferIDs,
         diagOpts.VerifyMode == DiagnosticOptions::VerifyAndApplyFixes,
-        diagOpts.VerifyIgnoreUnknown);
+        diagOpts.VerifyIgnoreUnknown,
+        diagOpts.AdditionalDiagnosticVerifierPostfixes);
     for (const auto &filename : diagOpts.AdditionalVerifierFiles) {
       auto result = getFileSystem().getBufferForFile(filename);
       if (!result) {

test/Frontend/diagnostic_verifier_multiple_prefix_negative.swift

@@ -0,0 +1,40 @@
+// RUN: not %target-swift-frontend %s -verify -emit-sil -o /dev/null 2>&1 | %FileCheck %s
+// RUN: not %target-swift-frontend %s -verify -DFIRST -verify-additional-prefix first- -DFIRST -emit-sil -o /dev/null 2>&1 | %FileCheck -check-prefix=FIRST %s
+// RUN: not %target-swift-frontend %s -verify -DSECOND -verify-additional-prefix second- -DSECOND  -emit-sil -o /dev/null 2>&1 | %FileCheck -check-prefix=SECOND %s
+// RUN: not %target-swift-frontend %s -verify -DBOTH -verify-additional-prefix first-  -verify-additional-prefix second- -DFIRST -DSECOND -emit-sil -o /dev/null 2>&1 | %FileCheck -check-prefix=BOTH %s
+
+func test() {
+  // CHECK-NOT: initialization of immutable value 'normal'
+  // FIRST-NOT: initialization of immutable value 'normal'
+  // SECOND-NOT: initialization of immutable value 'normal'
+  // BOTH-NOT: initialization of immutable value 'normal'
+  let normal = 5 // expected-warning {{initialization of immutable value 'normal'}}
+
+  // CHECK: initialization of immutable value 'first'
+  // FIRST-NOT: initialization of immutable value 'first'
+  // SECOND: initialization of immutable value 'first'
+  // BOTH-NOT: initialization of immutable value 'first'
+  let first = 5 // expected-first-warning {{initialization of immutable value 'first'}}
+
+  // CHECK: initialization of immutable value 'second'
+  // FIRST: initialization of immutable value 'second'
+  // SECOND-NOT: initialization of immutable value 'second'
+  // BOTH-NOT: initialization of immutable value 'second'
+  let second = 5 // expected-second-warning {{initialization of immutable value 'second'}}
+
+  // CHECK: initialization of immutable value 'third'
+  // FIRST: initialization of immutable value 'third'
+  // SECOND: initialization of immutable value 'third'
+  // BOTH: initialization of immutable value 'third'
+  let third = 5
+
+  // In this case, BOTH is accepting both first and second... so because we have
+  // a first/second warning on the next line, we fail since we consume the
+  // warning for the first diagnostic.
+  #if BOTH
+  // BOTH: expected warning not produced
+  let fourth = 5
+  // expected-second-warning @-1 {{initialization of immutable value 'fourth'}}
+  // expected-first-warning @-2 {{initialization of immutable value 'fourth'}}
+  #endif
+}

test/Frontend/diagnostic_verifier_multiple_prefix_positive.swift

@@ -0,0 +1,16 @@
+// RUN: %target-swift-frontend %s -verify -emit-sil -o /dev/null
+// RUN: %target-swift-frontend %s -verify -verify-additional-prefix first- -DFIRST -emit-sil -o /dev/null
+// RUN: %target-swift-frontend %s -verify -verify-additional-prefix second- -DSECOND  -emit-sil -o /dev/null
+// RUN: %target-swift-frontend %s -verify -verify-additional-prefix first-  -verify-additional-prefix second- -DFIRST -DSECOND -emit-sil -o /dev/null
+
+func test() {
+  let x = 5 // expected-warning {{}}
+
+  #if FIRST
+  let y = 5 // expected-first-warning {{}}
+  #endif
+
+  #if SECOND
+  let z = 5 // expected-second-warning {{}}
+  #endif
+}