fix: only add methods to the proper class (#21)

Author: markemer

Sources/CapacitorPluginSyntaxTools/AddPluginToClass.swift

@@ -11,9 +11,19 @@ class AddPluginToClass: SyntaxRewriter {
     }
 
     override func visit(_ node: ClassDeclSyntax) -> DeclSyntax {
-        var newNode = capacitorPluginSyntax.addBridgedPluginConformance(node)
-        newNode.memberBlock.members.insert(contentsOf: capacitorPluginSyntax.createMemberBlock(),
-                                           at: node.memberBlock.members.startIndex)
-        return DeclSyntax(newNode)
+        guard let inheritedTypes = node.inheritanceClause?.inheritedTypes else {
+            return DeclSyntax(node)
+        }
+
+        if let inheritedType = inheritedTypes.first(where: { $0.isNamed("CAPPlugin") }) {
+            var newNode = capacitorPluginSyntax.addBridgedPluginConformance(node)
+            newNode.memberBlock.members.insert(contentsOf: capacitorPluginSyntax.createMemberBlock(),
+                                               at: node.memberBlock.members.startIndex)
+            return DeclSyntax(newNode)
+        } else {
+            return DeclSyntax(node)
+        }
     }
+
 }
+

Sources/CapacitorPluginSyntaxTools/SwiftSyntaxExtensions.swift

@@ -0,0 +1,13 @@
+import Foundation
+import SwiftSyntax
+import SwiftParser
+
+extension InheritedTypeSyntax {
+    func isNamed(_ searchName: String) -> Bool {
+        if let foundName = self.type.as(IdentifierTypeSyntax.self)?.name.text {
+            return foundName == searchName
+        } else {
+            return false
+        }
+    }
+}

Sources/CapacitorPluginTools/CapacitorPluginPackage.swift

@@ -22,6 +22,14 @@ public enum CapacitorPluginError: Error {
     }
 }
 
+struct StandardError: TextOutputStream, Sendable {
+    private static let handle = FileHandle.standardError
+
+    public func write(_ string: String) {
+        Self.handle.write(Data(string.utf8))
+    }
+}
+
 public class CapacitorPluginPackage {
     public let pluginDirectoryName: String
     public let basePathURL: URL