[Distributed] Resolved should use the task-local way of copying access control

Author: ktoso

lib/Macros/Sources/SwiftMacros/DistributedResolvableMacro.swift

@@ -45,7 +45,7 @@ extension DistributedResolvableMacro {
       return []
     }
 
-    let accessModifiers: String = proto.accessModifiersString
+    let accessModifiers = proto.accessControlModifiers
 
     let requirementStubs =
       proto.memberBlock.members // requirements
@@ -71,7 +71,7 @@ extension DistributedResolvableMacro {
     return [extensionDecl.cast(ExtensionDeclSyntax.self)]
   }
 
-  static func stubMethodDecl(access: String, _ requirement: MemberBlockItemListSyntax.Element) -> String {
+  static func stubMethodDecl(access: DeclModifierListSyntax, _ requirement: MemberBlockItemListSyntax.Element) -> String {
     // do we need to stub a computed variable?
     if let variable = requirement.decl.as(VariableDeclSyntax.self) {
       var accessorStubs: [String] = []
@@ -142,7 +142,7 @@ extension DistributedResolvableMacro {
                  """, id: .invalidApplication)
     }
 
-    let accessModifiers = proto.accessModifiersString
+    let accessModifiers = proto.accessControlModifiers
 
     for req in proto.genericWhereClause?.requirements ?? [] {
       switch req.requirement {
@@ -198,8 +198,10 @@ extension DistributedResolvableMacro {
     }
   }
 
-  private static func typealiasActorSystem(access: String, _ proto: ProtocolDeclSyntax, _ type: TypeSyntax) -> DeclSyntax {
-    "\(raw: access)typealias ActorSystem = \(type)"
+  private static func typealiasActorSystem(access: DeclModifierListSyntax,
+                                           _ proto: ProtocolDeclSyntax,
+                                           _ type: TypeSyntax) -> DeclSyntax {
+    "\(access)typealias ActorSystem = \(type)"
   }
 }
 

lib/Macros/Sources/SwiftMacros/SyntaxExtensions.swift

@@ -0,0 +1,33 @@
+//===----------------------------------------------------------------------===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2022-2024 Apple Inc. and the Swift project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See https://swift.org/LICENSE.txt for license information
+// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
+//
+//===----------------------------------------------------------------------===//
+// Common Syntax extensions used by standard library macros.                  //
+//===----------------------------------------------------------------------===//
+
+import SwiftSyntax
+import SwiftSyntaxMacros
+import SwiftDiagnostics
+
+extension DeclGroupSyntax {
+  internal var accessControlModifiers: DeclModifierListSyntax {
+      modifiers.filter { modifier in
+      modifier.isAccessControl
+    }
+  }
+}
+
+extension VariableDeclSyntax {
+  internal var accessControlModifiers: DeclModifierListSyntax {
+    modifiers.filter { modifier in
+      modifier.isAccessControl
+    }
+  }
+}

test/Concurrency/Macros/task_local_macro_expansion.swift

@@ -0,0 +1,39 @@
+// REQUIRES: swift_swift_parser, asserts
+//
+// UNSUPPORTED: back_deploy_concurrency
+// REQUIRES: concurrency
+//
+// RUN: %empty-directory(%t)
+// RUN: %empty-directory(%t-scratch)
+
+// RUN: %target-swift-frontend -disable-availability-checking -typecheck -verify -plugin-path %swift-plugin-dir -I %t -dump-macro-expansions %s -dump-macro-expansions 2>&1 | %FileCheck %s
+
+struct Nein {
+  @TaskLocal
+  public static var example: String = "hello"
+}
+
+// CHECK: public static let $example: TaskLocal<String> = TaskLocal(wrappedValue: "hello")
+//
+// CHECK:  {
+// CHECK:    get {
+// CHECK:      $example.get()
+// CHECK:    }
+// CHECK:  }
+
+struct Available {
+  @available(OSX 10.9, *)
+  struct AvailableValue {}
+
+  @TaskLocal
+  @available(OSX 10.9, *)
+  private static var example: AvailableValue?
+}
+
+// CHECK: private static let $example: TaskLocal<AvailableValue?> = TaskLocal(wrappedValue: nil)
+//
+// CHECK:  {
+// CHECK:    get {
+// CHECK:      $example.get()
+// CHECK:    }
+// CHECK:  }