Add support for package access modifier to DependencyClient macro (#209)

pointfreeco · Apr 30, 2024 · 7b139d8 · 7b139d8
1 parent 9620f73
commit 7b139d8
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 0 deletions.
diff --git a/Sources/DependenciesMacrosPlugin/DependencyClientMacro.swift b/Sources/DependenciesMacrosPlugin/DependencyClientMacro.swift
@@ -213,6 +213,7 @@ public enum DependencyClientMacro: MemberAttributeMacro, MemberMacro {
 private enum Access: Comparable {
   case `private`
   case `internal`
+  case `package`
   case `public`
 
   init?(modifiers: DeclModifierListSyntax) {
@@ -224,6 +225,9 @@ private enum Access: Comparable {
       case .keyword(.internal):
         self = .internal
         return
+      case .keyword(.package):
+        self = .package
+        return
       case .keyword(.public):
         self = .public
         return
@@ -240,6 +244,8 @@ private enum Access: Comparable {
       return .keyword(.private)
     case .internal:
       return nil
+    case .package:
+      return .keyword(.package)
     case .public:
       return .keyword(.public)
     }

diff --git a/Tests/DependenciesMacrosPluginTests/DependencyClientMacroTests.swift b/Tests/DependenciesMacrosPluginTests/DependencyClientMacroTests.swift
@@ -419,6 +419,33 @@ final class DependencyClientMacroTests: BaseTestCase {
     }
   }
 
+  func testPackage() {
+    assertMacro {
+      """
+      @DependencyClient
+      package struct Client {
+        package var endpoint: () -> Void
+      }
+      """
+    } expansion: {
+      """
+      package struct Client {
+        @DependencyEndpoint
+        package var endpoint: () -> Void
+
+        package init(
+          endpoint: @escaping () -> Void
+        ) {
+          self.endpoint = endpoint
+        }
+
+        package init() {
+        }
+      }
+      """
+    }
+  }
+
   func testSendable() {
     assertMacro {
       """