refactor shared attribute parsing

Author: hnrklssn

lib/ClangImporter/ImportDecl.cpp

@@ -8279,6 +8279,41 @@ static bool importAsUnsafe(ClangImporter::Implementation &impl,
   return false;
 }
 
+static bool applyImportedAttribute(ClangImporter::Implementation &Impl,
+                                   Decl *MappedDecl, StringRef attr) {
+  // Dig out a source file we can use for parsing.
+  auto &sourceFile = Impl.getClangSwiftAttrSourceFile(
+      *MappedDecl->getDeclContext()->getParentModule(), attr);
+
+  // Spin up a parser.
+  swift::Parser parser(sourceFile.getBufferID(), sourceFile,
+                       &Impl.SwiftContext.Diags, nullptr, nullptr);
+  // Prime the lexer.
+  parser.consumeTokenWithoutFeedingReceiver();
+
+  bool hadError = false;
+  if (parser.Tok.is(tok::at_sign)) {
+    SourceLoc atEndLoc = parser.Tok.getRange().getEnd();
+    SourceLoc atLoc = parser.consumeToken(tok::at_sign);
+    PatternBindingInitializer *initContext = nullptr;
+    hadError = parser
+                   .parseDeclAttribute(MappedDecl->getAttrs(), atLoc, atEndLoc,
+                                       initContext,
+                                       /*isFromClangAttribute=*/true)
+                   .isError();
+  } else {
+    SourceLoc staticLoc;
+    StaticSpellingKind staticSpelling;
+    hadError = parser
+                   .parseDeclModifierList(MappedDecl->getAttrs(), staticLoc,
+                                          staticSpelling,
+                                          /*isFromClangAttribute=*/true)
+                   .isError();
+  }
+
+  return hadError;
+}
+
 void
 ClangImporter::Implementation::importSwiftAttrAttributes(Decl *MappedDecl) {
   auto ClangDecl =
@@ -8297,7 +8332,6 @@ ClangImporter::Implementation::importSwiftAttrAttributes(Decl *MappedDecl) {
 
   std::optional<const clang::SwiftAttrAttr *> seenMainActorAttr;
   const clang::SwiftAttrAttr *seenMutabilityAttr = nullptr;
-  PatternBindingInitializer *initContext = nullptr;
 
   auto importAttrsFromDecl = [&](const clang::NamedDecl *ClangDecl) {
     //
@@ -8417,37 +8451,8 @@ ClangImporter::Implementation::importSwiftAttrAttributes(Decl *MappedDecl) {
         continue;
       }
 
-      // Dig out a source file we can use for parsing.
-      auto &sourceFile = getClangSwiftAttrSourceFile(
-          *MappedDecl->getDeclContext()->getParentModule(),
-          swiftAttr->getAttribute());
-
-      // Spin up a parser.
-      swift::Parser parser(
-          sourceFile.getBufferID(), sourceFile, &SwiftContext.Diags,
-          nullptr, nullptr);
-      // Prime the lexer.
-      parser.consumeTokenWithoutFeedingReceiver();
-
-      bool hadError = false;
-      if (parser.Tok.is(tok::at_sign)) {
-        SourceLoc atEndLoc = parser.Tok.getRange().getEnd();
-        SourceLoc atLoc = parser.consumeToken(tok::at_sign);
-        hadError = parser
-                       .parseDeclAttribute(MappedDecl->getAttrs(), atLoc,
-                                           atEndLoc, initContext,
-                                           /*isFromClangAttribute=*/true)
-                       .isError();
-      } else {
-        SourceLoc staticLoc;
-        StaticSpellingKind staticSpelling;
-        hadError = parser
-                       .parseDeclModifierList(MappedDecl->getAttrs(), staticLoc,
-                                              staticSpelling,
-                                              /*isFromClangAttribute=*/true)
-                       .isError();
-      }
-
+      bool hadError =
+          applyImportedAttribute(*this, MappedDecl, swiftAttr->getAttribute());
       if (hadError) {
         // Complain about the unhandled attribute or modifier.
         HeaderLoc attrLoc(swiftAttr->getLocation());
@@ -8602,28 +8607,7 @@ void ClangImporter::Implementation::importBoundsAttributes(
     }
   }
 
-
-  // Dig out a source file we can use for parsing.
-  auto &sourceFile = getClangSwiftAttrSourceFile(
-      *MappedDecl->getDeclContext()->getParentModule(), MacroString);
-
-  // Spin up a parser.
-  swift::Parser parser(sourceFile.getBufferID(), sourceFile,
-                       &SwiftContext.Diags, nullptr, nullptr);
-  // Prime the lexer.
-  parser.consumeTokenWithoutFeedingReceiver();
-
-  bool hadError = false;
-  assert(parser.Tok.is(tok::at_sign));
-  SourceLoc atEndLoc = parser.Tok.getRange().getEnd();
-  SourceLoc atLoc = parser.consumeToken(tok::at_sign);
-  DeclContext *DC = MappedDecl->getParent();
-  auto initContext = PatternBindingInitializer::create(DC);
-  hadError = parser
-                 .parseDeclAttribute(MappedDecl->getAttrs(), atLoc, atEndLoc,
-                                     initContext,
-                                     /*isFromClangAttribute=*/true)
-                 .isError();
+  bool hadError = applyImportedAttribute(*this, MappedDecl, MacroString);
   if (hadError) {
     HeaderLoc attrLoc(ClangDecl->getLocation());
     diagnose(attrLoc, diag::clang_pointer_bounds_unhandled,