-

Author: NSFatalError

Sources/PrincipleMacros/Syntax/Extensions/IfConfigDeclSyntax+EnclosingIfConfig.swift

@@ -95,15 +95,22 @@ extension DeclSyntaxProtocol {
     public func applyingEnclosingIfConfig(
         to members: MemberBlockItemListSyntax
     ) -> IfConfigDeclSyntax? {
-        guard var parent = parent?.parent?.as(MemberBlockItemListSyntax.self) else {
+        guard var ancestor = parent?.parent?.parent?.as(IfConfigClauseSyntax.self) else {
             return nil
         }
 
-        parent.replaceSubrange(
-            parent.startIndex ..< parent.endIndex,
-            with: members
-        )
+        ancestor = ancestor.with(\.elements, .decls(members.withLeadingNewline))
+        return ancestor.enclosingIfConfig
+    }
+
+    public func applyingEnclosingIfConfig(
+        to statements: CodeBlockItemListSyntax
+    ) -> IfConfigDeclSyntax? {
+        guard var ancestor = parent?.parent?.parent?.as(IfConfigClauseSyntax.self) else {
+            return nil
+        }
 
-        return parent.withLeadingNewline.enclosingIfConfig
+        ancestor = ancestor.with(\.elements, .statements(statements.withLeadingNewline))
+        return ancestor.enclosingIfConfig
     }
 }

Tests/PrincipleMacrosTests/Syntax/Extensions/IfConfigDeclSyntaxTests.swift

@@ -163,6 +163,40 @@ internal enum IfConfigDeclSyntaxTests {
             #expect(ifConfig?.description == expectation)
         }
 
+        @Test
+        func applyToNewStatements() throws {
+            let decl: DeclSyntax = """
+            class MyClass {
+                #if DEBUG
+                var other = "hello"
+                #else
+                    #if os(macOS)
+                    var test = 123
+                    #endif
+                #endif
+            }
+            """
+
+            let newStatements: CodeBlockItemListSyntax = """
+            var replacement = "Hello"
+            func test() {}
+            """
+
+            let expectation = """
+            #if DEBUG
+            #else 
+            #if os(macOS)
+            var replacement = "Hello"
+            func test() {}
+            #endif
+            #endif
+            """
+
+            let property = try parseLastProperty(in: decl)
+            let ifConfig = property.underlying.applyingEnclosingIfConfig(to: newStatements)
+            #expect(ifConfig?.description == expectation)
+        }
+
         // swiftlint:enable empty_line_after_type_declaration
     }
 }