-

Author: NSFatalError

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

@@ -11,13 +11,24 @@ import SwiftSyntaxMacros
 extension IfConfigDeclSyntax {
 
     public var availability: Self? {
-        guard clauses.contains(where: { $0.availability != nil }) else {
-            return nil
+        var elements = [IfConfigClauseSyntax]()
+        var isEmpty = true
+
+        for clause in clauses {
+            if let availability = clause.availability {
+                elements.append(availability)
+                isEmpty = false
+            } else {
+                elements.append(clause.with(\.elements, .attributes([])))
+            }
         }
-        let elements = clauses.compactMap { clause in
-            clause.availability ?? clause.with(\.elements, .attributes([]))
+
+        guard !isEmpty else {
+            return nil
         }
-        return with(\.clauses, IfConfigClauseListSyntax(elements))
+
+        let clauses = IfConfigClauseListSyntax(elements)
+        return with(\.clauses, clauses)
     }
 }
 

Tests/PrincipleMacrosTests/Syntax/Extensions/AvailabilityTests.swift

@@ -0,0 +1,59 @@
+//
+//  AvailabilityTests.swift
+//  PrincipleMacros
+//
+//  Created by Kamil Strzelecki on 22/11/2025.
+//  Copyright © 2025 Kamil Strzelecki. All rights reserved.
+//
+
+@testable import PrincipleMacros
+import Testing
+
+internal struct AvailabilityTests {
+
+    @Test
+    func withoutAvailability() throws {
+        let decl: DeclSyntax = """
+        @MainActor @Observable
+        class MyClass {}
+        """
+
+        let classDecl = try #require(decl.as(ClassDeclSyntax.self))
+        #expect(classDecl.availability == nil)
+    }
+
+    @Test
+    func withAvailability() throws {
+        let decl: DeclSyntax = """
+        @MainActor @Observable
+        @available(iOS 26, *)
+        class MyClass {}
+        """
+
+        let classDecl = try #require(decl.as(ClassDeclSyntax.self))
+        #expect(classDecl.availability?.trimmedDescription == "@available(iOS 26, *)")
+    }
+
+    @Test
+    func withIfConfig() throws {
+        let decl: DeclSyntax = """
+        #if os(macOS)
+        @MainActor 
+        #else
+        @Observable
+        @available(iOS 26, *)
+        #endif
+        class MyClass {}
+        """
+
+        let expectation = """
+        #if os(macOS)
+        #else
+        @available(iOS 26, *)
+        #endif
+        """
+
+        let classDecl = try #require(decl.as(ClassDeclSyntax.self))
+        #expect(classDecl.availability?.trimmedDescription == expectation)
+    }
+}