add: class inherithance on automockable

CHANGELOG.md

@@ -5,6 +5,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+# [0.2.6] - 2025-01-03
+
+### Added
+
+- Automackable handle inherited classes
+
 # [0.2.5] - 2024-11-15
 
 ### Fixed

Sources/SageSwiftKit/AutoMockableMacros.swift

@@ -3,4 +3,7 @@
 // All Rights Reserved.
 
 @attached(peer, names: suffixed(Mock))
-public macro AutoMockable(accessLevel: String? = nil) = #externalMacro(module: "SageSwiftKitMacros", type: "AutoMockable")
+public macro AutoMockable(
+    accessLevel: String? = nil,
+    classInheritance: Bool = false
+) = #externalMacro(module: "SageSwiftKitMacros", type: "AutoMockable")

Sources/SageSwiftKitClient/main.swift

@@ -37,9 +37,21 @@ struct PlayingObject {
 }
 
 
-@AutoMockable()
-public protocol DateSelectorViewModel {
+//@NodePrinter
+public class TmpClass {
+    var a: String
+    var b: String
+
+    init(a: String, b: String) {
+        self.a = a
+        self.b = b
+    }
+}
+
+@AutoMockable(classInheritance: true)
+public protocol DateSelectorViewModel: TmpClass {
     var onTap: (String, Int?) -> Void { get set }
     var onString: String { get set }
     var onDouble: Double? { get set }
 }
+

Sources/SageSwiftKitMacros/MockableMacros/AutoMockableBuilder.swift → Sources/SageSwiftKitMacros/AutoMockableMacros/AutoMockableBuilder.swift

@@ -25,6 +25,12 @@ public enum AutoMockable: PeerMacro {
             .adapter
             .expression(cast: StringLiteralExprSyntax.self)?.representedLiteralValue ?? "internal"
 
+        let classInheritance = node
+            .adapter
+            .findArgument(id: "classInheritance")?
+            .adapter
+            .expression(cast: BooleanLiteralExprSyntax.self)?.literal.text ?? "false"
+
         let procotolName = protocolSyntax.name.text
 
         guard let members = declaration.as(ProtocolDeclSyntax.self)?.memberBlock.members else {
@@ -51,39 +57,49 @@ public enum AutoMockable: PeerMacro {
                     name: .identifier("\(procotolName)Mock"),
                     inheritanceClause: .init(
                         inheritedTypes: .init(itemsBuilder: {
+                            if classInheritance == "true" {
+                                if let inherited = protocolSyntax.inheritanceClause {
+                                    inherited.inheritedTypes
+                                }
+                            }
                             InheritedTypeSyntax(
                                 type: IdentifierTypeSyntax(
                                     name: .identifier(procotolName)
                                 )
                             )
 
-                            if let inherited = protocolSyntax.inheritanceClause {
-                                inherited.inheritedTypes
+                            if classInheritance == "false" {
+                                if let inherited = protocolSyntax.inheritanceClause {
+                                    inherited.inheritedTypes
+                                }
                             }
                         })
                     ),
                     memberBlock: MemberBlockSyntax(
                         members: try MemberBlockItemListSyntax(itemsBuilder: {
                             // Init
-                            InitializerDeclSyntax(
-                                modifiers: .init(itemsBuilder: {
-                                    DeclModifierSyntax(name: accessLevel.tokenSyntax)
-                                }),
-                                signature: .init(
-                                    parameterClause: .init(
-                                        parameters: .init(
-                                            itemsBuilder: {}
+
+                            if classInheritance == "false" {
+                                InitializerDeclSyntax(
+                                    modifiers: .init(itemsBuilder: {
+                                        DeclModifierSyntax(name: accessLevel.tokenSyntax)
+                                    }),
+                                    signature: .init(
+                                        parameterClause: .init(
+                                            parameters: .init(
+                                                itemsBuilder: {}
+                                            )
+                                        )
+                                    ),
+                                    body: .init(
+                                        statements: .init(
+                                            itemsBuilder: {
+
+                                            }
                                         )
-                                    )
-                                ),
-                                body: .init(
-                                    statements: .init(
-                                        itemsBuilder: {
-
-                                        }
                                     )
                                 )
-                            )
+                            }
 
                             // Classes that has mock data for each function
                             for funcData in functionsToMock {