ABI checker: include SPI group names in ABI descriptor files

Author: nkcsgexi

Diff for: include/swift/APIDigester/ModuleAnalyzerNodes.h

@@ -346,6 +346,7 @@ class SDKNodeDecl: public SDKNode {
   StringRef Location;
   StringRef ModuleName;
   std::vector<DeclAttrKind> DeclAttributes;
+  std::vector<StringRef> SPIGroups;
   bool IsImplicit;
   bool IsStatic;
   bool IsDeprecated;
@@ -373,6 +374,7 @@ class SDKNodeDecl: public SDKNode {
   StringRef getModuleName() const {return ModuleName;}
   StringRef getHeaderName() const;
   ArrayRef<DeclAttrKind> getDeclAttributes() const;
+  ArrayRef<StringRef> getSPIGroups() const { return SPIGroups; }
   bool hasAttributeChange(const SDKNodeDecl &Another) const;
   swift::ReferenceOwnership getReferenceOwnership() const {
     return swift::ReferenceOwnership(ReferenceOwnership);

Diff for: include/swift/IDE/DigesterEnums.def

@@ -162,6 +162,7 @@ KEY_STRING(InitKind, init_kind)
 
 KEY_STRING_ARR(SuperclassNames, superclassNames)
 KEY_STRING_ARR(ToolArgs, tool_arguments)
+KEY_STRING_ARR(SPIGroups, spi_group_names)
 
 KEY_UINT(SelfIndex, selfIndex)
 KEY_UINT(FixedBinaryOrder, fixedbinaryorder)

Diff for: lib/APIDigester/ModuleAnalyzerNodes.cpp

@@ -114,7 +114,9 @@ SDKNodeDecl::SDKNodeDecl(SDKNodeInitInfo Info, SDKNodeKind Kind)
       : SDKNode(Info, Kind), DKind(Info.DKind), Usr(Info.Usr),
         MangledName(Info.MangledName), Loc(Info.Loc),
         Location(Info.Location), ModuleName(Info.ModuleName),
-        DeclAttributes(Info.DeclAttrs), IsImplicit(Info.IsImplicit),
+        DeclAttributes(Info.DeclAttrs),
+        SPIGroups(Info.SPIGroups),
+        IsImplicit(Info.IsImplicit),
         IsStatic(Info.IsStatic), IsDeprecated(Info.IsDeprecated),
         IsProtocolReq(Info.IsProtocolReq),
         IsOverriding(Info.IsOverriding),
@@ -1436,7 +1438,12 @@ SDKNodeInitInfo::SDKNodeInitInfo(SDKContext &Ctx, Decl *D):
       IsDeprecated(D->getAttrs().getDeprecated(D->getASTContext())),
       IsABIPlaceholder(isABIPlaceholderRecursive(D)),
       IsFromExtension(isDeclaredInExtension(D)),
-      DeclAttrs(collectDeclAttributes(D)) {}
+      DeclAttrs(collectDeclAttributes(D)) {
+  // Keep track of SPI group names
+  for (auto id: D->getSPIGroups()) {
+    SPIGroups.push_back(id.str());
+  }
+}
 
 SDKNodeInitInfo::SDKNodeInitInfo(SDKContext &Ctx, OperatorDecl *OD):
     SDKNodeInitInfo(Ctx, cast<Decl>(OD)) {
@@ -2083,6 +2090,7 @@ void SDKNodeDecl::jsonize(json::Output &out) {
     out.mapRequired(getKeyContent(Ctx, KeyKind::KK_ownership).data(), Raw);
   }
   output(out, KeyKind::KK_isFromExtension, IsFromExtension);
+  out.mapOptional(getKeyContent(Ctx, KeyKind::KK_spi_group_names).data(), SPIGroups);
 }
 
 void SDKNodeDeclAbstractFunc::jsonize(json::Output &out) {

Diff for: test/api-digester/Inputs/cake.swift

@@ -148,6 +148,8 @@ public func silgenNamedFunc() {}
 
 @available(OSX 10.7, *)
 @_originallyDefinedIn(module: "Bread", OSX 10.9)
+@_spi(top_secret_1)
+@_spi(top_secret_2)
 public class SinkingClass {
   public init() {}
 }

Diff for: test/api-digester/Outputs/cake-abi.json

@@ -1656,6 +1656,10 @@
             "usr": "s:4cake12SinkingClassCACycfc",
             "mangledName": "$s5Bread12SinkingClassCACycfc",
             "moduleName": "cake",
+            "spi_group_names": [
+              "top_secret_2",
+              "top_secret_1"
+            ],
             "init_kind": "Designated"
           }
         ],
@@ -1665,8 +1669,14 @@
         "moduleName": "cake",
         "intro_Macosx": "10.7",
         "declAttributes": [
+          "SPIAccessControl",
+          "SPIAccessControl",
           "OriginallyDefinedIn",
           "Available"
+        ],
+        "spi_group_names": [
+          "top_secret_2",
+          "top_secret_1"
         ]
       },
       {

Diff for: test/api-digester/Outputs/cake.json

@@ -1517,6 +1517,10 @@
             "usr": "s:4cake12SinkingClassCACycfc",
             "mangledName": "$s5Bread12SinkingClassCACycfc",
             "moduleName": "cake",
+            "spi_group_names": [
+              "top_secret_2",
+              "top_secret_1"
+            ],
             "init_kind": "Designated"
           }
         ],
@@ -1526,8 +1530,14 @@
         "moduleName": "cake",
         "intro_Macosx": "10.7",
         "declAttributes": [
+          "SPIAccessControl",
+          "SPIAccessControl",
           "OriginallyDefinedIn",
           "Available"
+        ],
+        "spi_group_names": [
+          "top_secret_2",
+          "top_secret_1"
         ]
       },
       {