Sema: Inherit SPI groups when synthesizing modify accessor.

Resolves rdar://108069565

Author: tshortli

lib/Sema/TypeCheckStorage.cpp

@@ -2324,6 +2324,17 @@ createCoroutineAccessorPrototype(AbstractStorageDecl *storage,
   AvailabilityInference::applyInferredAvailableAttrs(accessor,
                                                      asAvailableAs, ctx);
 
+  // A modify coroutine should have the same SPI visibility as the setter.
+  if (kind == AccessorKind::Modify) {
+    if (FuncDecl *setter = storage->getParsedAccessor(AccessorKind::Set)) {
+      if (!setter->getSPIGroups().empty()) {
+        auto spiAttr = SPIAccessControlAttr::create(
+            ctx, SourceLoc(), SourceRange(), setter->getSPIGroups());
+        accessor->getAttrs().add(spiAttr);
+      }
+    }
+  }
+
   finishImplicitAccessor(accessor, ctx);
 
   return accessor;

test/SPI/private_swiftinterface.swift

@@ -139,6 +139,22 @@ public class SomeClass {
 }
 
 public struct PublicStruct {
+  @_spi(S) public var spiVar: SomeClass
+  // CHECK-PRIVATE: @_spi(S) public var spiVar: {{.*}}.SomeClass
+  // CHECK-PUBLIC-NOT: spiVar
+
+  public var publicVarWithSPISet: SomeClass {
+    get { SomeClass() }
+    @_spi(S) set {}
+  }
+  // CHECK-PRIVATE: public var publicVarWithSPISet: {{.*}}.SomeClass {
+  // CHECK-PRIVATE-NEXT:   get
+  // CHECK-PRIVATE-NEXT:   @_spi(S) set
+  // CHECK-PRIVATE-NEXT: }
+  // CHECK-PUBLIC: public var publicVarWithSPISet: {{.*}}.SomeClass {
+  // CHECK-PUBLIC-NEXT:   get
+  // CHECK-PUBLIC-NEXT: }
+
   @_spi(S) @Wrapper public var spiWrappedSimple: SomeClass
   // CHECK-PRIVATE: @_spi(S) @{{.*}}.Wrapper public var spiWrappedSimple: {{.*}}.SomeClass
   // CHECK-PUBLIC-NOT: spiWrappedSimple

test/api-digester/ignore-spi-by-given-group-name.swift

@@ -2,21 +2,39 @@
 
 // RUN: %target-swift-frontend -emit-module -o %t/Foo.swiftmodule -emit-abi-descriptor-path %t/abi-before.json %s -enable-library-evolution -DBASELINE -emit-tbd-path %t/abi-before.tbd
 // RUN: %target-swift-frontend -emit-module -o %t/Foo.swiftmodule -emit-abi-descriptor-path %t/abi-after.json %s -enable-library-evolution -emit-tbd-path %t/abi-after.tbd
-// RUN: %api-digester -diagnose-sdk --input-paths %t/abi-before.json -input-paths %t/abi-after.json -abi -o %t/result.txt
-// RUN: %FileCheck %s -check-prefix CHECK_INCLUDE_SPI < %t/result.txt
+// RUN: %api-digester -diagnose-sdk --input-paths %t/abi-before.json -input-paths %t/abi-after.json -abi -o %t/include-result.txt
+// RUN: %FileCheck %s -check-prefix CHECK_INCLUDE_SPI < %t/include-result.txt
 
-// RUN: %api-digester -diagnose-sdk --input-paths %t/abi-before.json -input-paths %t/abi-after.json -abi -o %t/result.txt -ignore-spi-group secret
-// RUN: %FileCheck -check-prefix CHECK_EXCLUDE_SPI %s < %t/result.txt
+// RUN: %api-digester -diagnose-sdk --input-paths %t/abi-before.json -input-paths %t/abi-after.json -abi -o %t/exclude-result.txt -ignore-spi-group secret
+// RUN: %FileCheck -check-prefix CHECK_EXCLUDE_SPI %s < %t/exclude-result.txt
 
 #if BASELINE
 
+public struct Struct {
+  public var x: Int {
+    get { 0 }
+    @_spi(secret) set {}
+  }
+}
+
 @_spi(secret)
 public func foo() {}
 
 #else
 
+public struct Struct {
+  public var x: Int {
+    get { 0 }
+  }
+}
 
 #endif
 
+// CHECK_INCLUDE_SPI: Accessor Struct.x.Modify() has been removed
+// CHECK_EXCLUDE_SPI-NOT: Struct.x.Modify()
+
+// CHECK_INCLUDE_SPI: Accessor Struct.x.Set() has been removed
+// CHECK_EXCLUDE_SPI-NOT: Struct.x.Set()
+
 // CHECK_INCLUDE_SPI: Func foo() has been removed
 // CHECK_EXCLUDE_SPI-NOT: foo()