Propagate @ViewAction macro availability (#2785)

pointfreeco · Feb 8, 2024 · e10f42e · e10f42e
1 parent d9a3cfb
commit e10f42e
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 1 deletion.
diff --git a/Sources/ComposableArchitectureMacros/ViewActionMacro.swift b/Sources/ComposableArchitectureMacros/ViewActionMacro.swift
@@ -68,7 +68,8 @@ public struct ViewActionMacro: ExtensionMacro {
 
     let ext: DeclSyntax =
       """
-      extension \(type.trimmed): ComposableArchitecture.ViewActionSending {}
+      \(declaration.attributes.availability)extension \(type.trimmed): \
+      ComposableArchitecture.ViewActionSending {}
       """
     return [ext.cast(ExtensionDeclSyntax.self)]
   }

diff --git a/Tests/ComposableArchitectureMacrosTests/ViewActionMacroTests.swift b/Tests/ComposableArchitectureMacrosTests/ViewActionMacroTests.swift
@@ -438,5 +438,33 @@
         """
       }
     }
+
+    func testAvailability() {
+      assertMacro {
+        """
+        @available(iOS 17, macOS 14, tvOS 17, visionOS 1, watchOS 10, *)
+        @ViewAction(for: Feature.self)
+        struct FeatureView: View {
+          @State var store: StoreOf<Feature>
+          var body: some View {
+            EmptyView()
+          }
+        }
+        """
+      } expansion: {
+        """
+        @available(iOS 17, macOS 14, tvOS 17, visionOS 1, watchOS 10, *)
+        struct FeatureView: View {
+          @State var store: StoreOf<Feature>
+          var body: some View {
+            EmptyView()
+          }
+        }
+
+        @available(iOS 17, macOS 14, tvOS 17, visionOS 1, watchOS 10, *) extension FeatureView: ComposableArchitecture.ViewActionSending {
+        }
+        """
+      }
+    }
   }
 #endif