[ASTGen] Generate '@_spi_available' attribute

Author: rintaro

include/swift/AST/ASTBridging.h

@@ -877,15 +877,16 @@ enum ENUM_EXTENSIBILITY_ATTR(closed) BridgedAvailableAttrKind {
 
 SWIFT_NAME("BridgedAvailableAttr.createParsed(_:atLoc:range:domainIdentifier:"
            "domainLoc:kind:message:renamed:introduced:introducedRange:"
-           "deprecated:deprecatedRange:obsoleted:obsoletedRange:)")
+           "deprecated:deprecatedRange:obsoleted:obsoletedRange:isSPI:)")
 BridgedAvailableAttr BridgedAvailableAttr_createParsed(
     BridgedASTContext cContext, BridgedSourceLoc cAtLoc,
     BridgedSourceRange cRange, BridgedIdentifier cDomainIdentifier,
     BridgedSourceLoc cDomainLoc, BridgedAvailableAttrKind cKind,
     BridgedStringRef cMessage, BridgedStringRef cRenamed,
     BridgedVersionTuple cIntroduced, BridgedSourceRange cIntroducedRange,
     BridgedVersionTuple cDeprecated, BridgedSourceRange cDeprecatedRange,
-    BridgedVersionTuple cObsoleted, BridgedSourceRange cObsoletedRange);
+    BridgedVersionTuple cObsoleted, BridgedSourceRange cObsoletedRange,
+    bool isSPI);
 
 SWIFT_NAME("BridgedAvailableAttr.createUnavailableInEmbedded(_:atLoc:range:)")
 BridgedAvailableAttr

lib/AST/Bridging/DeclAttributeBridging.cpp

@@ -143,7 +143,8 @@ BridgedAvailableAttr BridgedAvailableAttr_createParsed(
     BridgedStringRef cMessage, BridgedStringRef cRenamed,
     BridgedVersionTuple cIntroduced, BridgedSourceRange cIntroducedRange,
     BridgedVersionTuple cDeprecated, BridgedSourceRange cDeprecatedRange,
-    BridgedVersionTuple cObsoleted, BridgedSourceRange cObsoletedRange) {
+    BridgedVersionTuple cObsoleted, BridgedSourceRange cObsoletedRange,
+    bool isSPI) {
 
   return new (cContext.unbridged())
       AvailableAttr(cAtLoc.unbridged(), cRange.unbridged(),
@@ -153,7 +154,7 @@ BridgedAvailableAttr BridgedAvailableAttr_createParsed(
                     cDeprecated.unbridged(), cDeprecatedRange.unbridged(),
                     cObsoleted.unbridged(), cObsoletedRange.unbridged(),
                     /*Implicit=*/false,
-                    /*IsSPI=*/false);
+                    /*IsSPI=*/isSPI);
 }
 
 BridgedAvailableAttr

lib/ASTGen/Sources/ASTGen/Availability.swift

@@ -23,9 +23,12 @@ extension ASTGenVisitor {
   func generateAvailableAttr(
     atLoc: BridgedSourceLoc,
     range: BridgedSourceRange,
+    attrName: SyntaxText,
     args: AvailabilityArgumentListSyntax
   ) -> [BridgedAvailableAttr] {
 
+    let isSPI = attrName == "_spi_available"
+
     // Check if this is "shorthand" syntax.
     if let firstArg = args.first?.argument {
       // We need to check availability macros specified by '-define-availability'.
@@ -40,15 +43,14 @@ extension ASTGenVisitor {
         isShorthand = false
       }
       if isShorthand {
-        return self.generateAvailableAttrShorthand(atLoc: atLoc, range: range, args: args)
+        return self.generateAvailableAttrShorthand(atLoc: atLoc, range: range, args: args, isSPI: isSPI)
       }
     }
 
     // E.g.
     //   @available(macOS, introduced: 10.12, deprecated: 11.2)
     //   @available(*, unavailable, message: "out of service")
-    let attr = self.generateAvailableAttrExtended(atLoc: atLoc, range: range, args: args)
-
+    let attr = self.generateAvailableAttrExtended(atLoc: atLoc, range: range, args: args, isSPI: isSPI)
     if let attr {
       return [attr]
     } else {
@@ -66,7 +68,8 @@ extension ASTGenVisitor {
   func generateAvailableAttrShorthand(
     atLoc: BridgedSourceLoc,
     range: BridgedSourceRange,
-    args: AvailabilityArgumentListSyntax
+    args: AvailabilityArgumentListSyntax,
+    isSPI: Bool
   ) -> [BridgedAvailableAttr] {
     let specs = self.generateAvailabilitySpecList(args: args, context: .availableAttr)
 
@@ -100,7 +103,8 @@ extension ASTGenVisitor {
         deprecated: BridgedVersionTuple(),
         deprecatedRange: BridgedSourceRange(),
         obsoleted: BridgedVersionTuple(),
-        obsoletedRange: BridgedSourceRange()
+        obsoletedRange: BridgedSourceRange(),
+        isSPI: isSPI
       )
       attr.setIsGroupMember()
       if containsWildCard {
@@ -119,7 +123,8 @@ extension ASTGenVisitor {
   func generateAvailableAttrExtended(
     atLoc: BridgedSourceLoc,
     range: BridgedSourceRange,
-    args: AvailabilityArgumentListSyntax
+    args: AvailabilityArgumentListSyntax,
+    isSPI: Bool
   ) -> BridgedAvailableAttr? {
     var args = args[...]
 
@@ -262,7 +267,8 @@ extension ASTGenVisitor {
       deprecated: deprecated?.version.bridged ?? BridgedVersionTuple(),
       deprecatedRange: deprecated?.range ?? BridgedSourceRange(),
       obsoleted: obsoleted?.version.bridged ?? BridgedVersionTuple(),
-      obsoletedRange: obsoleted?.range ?? BridgedSourceRange()
+      obsoletedRange: obsoleted?.range ?? BridgedSourceRange(),
+      isSPI: isSPI
     )
   }
 

lib/ASTGen/Sources/ASTGen/DeclAttrs.swift

@@ -126,7 +126,7 @@ extension ASTGenVisitor {
       case .allowFeatureSuppression:
         return handle(self.generateAllowFeatureSuppressionAttr(attribute: node, attrName: attrName)?.asDeclAttribute)
       case .available:
-        return self.generateAvailableAttr(attribute: node).forEach { handle($0.asDeclAttribute) }
+        return self.generateAvailableAttr(attribute: node, attrName: attrName).forEach { handle($0.asDeclAttribute) }
       case .backDeployed:
         return self.generateBackDeployedAttr(attribute: node).forEach { handle($0.asDeclAttribute) }
       case .cDecl:
@@ -180,7 +180,7 @@ extension ASTGenVisitor {
       case .silGenName:
         return handle(self.generateSILGenNameAttr(attribute: node)?.asDeclAttribute)
       case .specialize:
-        return handle(self.generateSpecializeAttr(attribute: node)?.asDeclAttribute)
+        return handle(self.generateSpecializeAttr(attribute: node, attrName: attrName)?.asDeclAttribute)
       case .spiAccessControl:
         return handle(self.generateSPIAccessControlAttr(attribute: node)?.asDeclAttribute)
       case .storageRestrictions:
@@ -501,7 +501,7 @@ extension ASTGenVisitor {
   ///   @available(macOS 10.12, iOS 13, *)
   ///   @available(macOS, introduced: 10.12)
   ///   ```
-  func generateAvailableAttr(attribute node: AttributeSyntax) -> [BridgedAvailableAttr] {
+  func generateAvailableAttr(attribute node: AttributeSyntax, attrName: SyntaxText) -> [BridgedAvailableAttr] {
     guard let args = node.arguments else {
       self.diagnose(.expectedArgumentsInAttribute(node))
       return []
@@ -514,6 +514,7 @@ extension ASTGenVisitor {
     return self.generateAvailableAttr(
       atLoc: self.generateSourceLoc(node.atSign),
       range: self.generateAttrSourceRange(node),
+      attrName: attrName,
       args: args
     )
   }
@@ -1881,7 +1882,7 @@ extension ASTGenVisitor {
   ///   ```
   ///   @_specialize(exporeted: true, T == Int)
   ///   ```
-  func generateSpecializeAttr(attribute node: AttributeSyntax) -> BridgedSpecializeAttr? {
+  func generateSpecializeAttr(attribute node: AttributeSyntax, attrName: SyntaxText) -> BridgedSpecializeAttr? {
     guard
       var args = node.arguments?.as(SpecializeAttributeArgumentListSyntax.self)?[...]
     else {
@@ -1912,6 +1913,7 @@ extension ASTGenVisitor {
             start: arg.availabilityArguments.firstToken(viewMode: .all)!,
             end: arg.semicolon
           ),
+          attrName: attrName,
           args: arg.availabilityArguments
         )
       case .labeledSpecializeArgument(let arg):

test/ASTGen/availability.swift

@@ -62,3 +62,8 @@ func testPoundIf() {
     // pass
   }
 }
+
+public class ClassWithMembers {
+  @_spi_available(macOS 10.15, *)
+  public func spiFunc() {}
+}