Modify isNeededAccessLevelModifier result conditions (#128)

pointfreeco · Nov 14, 2023 · 73b69bd · 73b69bd
1 parent 65b7259
commit 73b69bd
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/Sources/CasePathsMacros/CasePathableMacro.swift b/Sources/CasePathsMacros/CasePathableMacro.swift
@@ -238,6 +238,7 @@ extension DeclModifierSyntax {
   var isNeededAccessLevelModifier: Bool {
     switch self.name.tokenKind {
     case .keyword(.public): return true
+    case .keyword(.package): return true
     default: return false
     }
   }

diff --git a/Tests/CasePathsMacrosTests/CasePathableMacroTests.swift b/Tests/CasePathsMacrosTests/CasePathableMacroTests.swift
@@ -165,6 +165,37 @@ final class CasePathableMacroTests: XCTestCase {
       }
       """
     }
+    assertMacro {
+      """
+      @CasePathable package enum Foo {
+        case bar(Int)
+      }
+      """
+    } expansion: {
+      """
+      package enum Foo {
+        case bar(Int)
+
+        package struct AllCasePaths {
+          package var bar: CasePaths.AnyCasePath<Foo, Int> {
+            CasePaths.AnyCasePath<Foo, Int>(
+              embed: Foo.bar,
+              extract: {
+                guard case let .bar(v0) = $0 else {
+                  return nil
+                }
+                return v0
+              }
+            )
+          }
+        }
+        package static var allCasePaths: AllCasePaths { AllCasePaths() }
+      }
+
+      extension Foo: CasePaths.CasePathable {
+      }
+      """
+    }
     assertMacro {
       """
       @CasePathable private enum Foo {