fix: allow adding Swift Plugins to ObjC Amplitude (#264)

amplitude · Mar 5, 2025 · d916d93 · d916d93
1 parent b6fd248
commit d916d93
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/Sources/Amplitude/ObjC/ObjCAmplitude.swift b/Sources/Amplitude/ObjC/ObjCAmplitude.swift
@@ -147,10 +147,17 @@ public class ObjCAmplitude: NSObject {
 
     @objc(add:)
     @discardableResult
-    public func add(plugin: ObjCPlugin) -> ObjCAmplitude {
-        let wrapper = ObjCPluginWrapper(amplitude: self, wrapped: plugin)
-        plugins.append(wrapper)
-        amplitude.add(plugin: wrapper)
+    public func add(plugin: AnyObject) -> ObjCAmplitude {
+        switch plugin {
+        case let swiftPlugin as Plugin:
+            amplitude.add(plugin: swiftPlugin)
+        case let objcPlugin as ObjCPlugin:
+            let wrapper = ObjCPluginWrapper(amplitude: self, wrapped: objcPlugin)
+            plugins.append(wrapper)
+            amplitude.add(plugin: wrapper)
+        default:
+            fatalError("Attempted to add a plugin that is not an instance of Plugin or ObjCPlugin")
+        }
         return self
     }