From 8d157865de6b17874e2f335d38ceef76f7dce1cb Mon Sep 17 00:00:00 2001
From: Marvin Liu <marvin@amplitude.com>
Date: Mon, 19 Dec 2022 22:50:39 -0800
Subject: [PATCH] feat: update destination plugin to class

---
 .../Plugins/AmplitudeDestinationPlugin.swift  | 17 ++++------
 .../Amplitude/Plugins/DestinationPlugin.swift | 34 +++++++++++++++++--
 2 files changed, 38 insertions(+), 13 deletions(-)

diff --git a/Sources/Amplitude/Plugins/AmplitudeDestinationPlugin.swift b/Sources/Amplitude/Plugins/AmplitudeDestinationPlugin.swift
index 3cda8acc..96718fc1 100644
--- a/Sources/Amplitude/Plugins/AmplitudeDestinationPlugin.swift
+++ b/Sources/Amplitude/Plugins/AmplitudeDestinationPlugin.swift
@@ -6,9 +6,6 @@
 //
 
 public class AmplitudeDestinationPlugin: DestinationPlugin {
-    public let timeline = Timeline()
-    public var amplitude: Amplitude?
-    public let type: PluginType = .destination
     private var pipeline: EventPipeline?
 
     internal func enqueue(event: BaseEvent?) {
@@ -21,31 +18,31 @@ public class AmplitudeDestinationPlugin: DestinationPlugin {
         }
     }
 
-    public func track(event: BaseEvent) -> BaseEvent? {
+    public override func track(event: BaseEvent) -> BaseEvent? {
         enqueue(event: event)
         return event
     }
 
-    public func identify(event: IdentifyEvent) -> IdentifyEvent? {
+    public override func identify(event: IdentifyEvent) -> IdentifyEvent? {
         enqueue(event: event)
         return event
     }
 
-    public func groupIdentify(event: GroupIdentifyEvent) -> GroupIdentifyEvent? {
+    public override func groupIdentify(event: GroupIdentifyEvent) -> GroupIdentifyEvent? {
         enqueue(event: event)
         return event
     }
 
-    public func revenue(event: RevenueEvent) -> RevenueEvent? {
+    public override func revenue(event: RevenueEvent) -> RevenueEvent? {
         enqueue(event: event)
         return event
     }
 
-    public func flush() {
+    public override func flush() {
         pipeline?.flush()
     }
 
-    public func setup(amplitude: Amplitude) {
+    public override func setup(amplitude: Amplitude) {
         self.amplitude = amplitude
         pipeline = EventPipeline(amplitude: amplitude)
         pipeline?.start()
@@ -53,7 +50,7 @@ public class AmplitudeDestinationPlugin: DestinationPlugin {
         add(plugin: IdentityEventSender())
     }
 
-    public func execute(event: BaseEvent?) -> BaseEvent? {
+    public override func execute(event: BaseEvent?) -> BaseEvent? {
         return event
     }
 }
diff --git a/Sources/Amplitude/Plugins/DestinationPlugin.swift b/Sources/Amplitude/Plugins/DestinationPlugin.swift
index 39b95a4c..be7c81dd 100644
--- a/Sources/Amplitude/Plugins/DestinationPlugin.swift
+++ b/Sources/Amplitude/Plugins/DestinationPlugin.swift
@@ -5,12 +5,40 @@
 //  Created by Hao Yu on 11/15/22.
 //
 
-public protocol DestinationPlugin: EventPlugin {
-    var timeline: Timeline { get }
+public class DestinationPlugin: EventPlugin {
+    public let type: PluginType = .destination
+    public var amplitude: Amplitude?
+    private var timeline = Timeline()
+
+    public func track(event: BaseEvent) -> BaseEvent? {
+        return event
+    }
+
+    public func identify(event: IdentifyEvent) -> IdentifyEvent? {
+        return event
+    }
+
+    public func groupIdentify(event: GroupIdentifyEvent) -> GroupIdentifyEvent? {
+        return event
+    }
+
+    public func revenue(event: RevenueEvent) -> RevenueEvent? {
+        return event
+    }
+
+    public func flush() {
+    }
+
+    public func execute(event: BaseEvent?) -> BaseEvent? {
+        return event
+    }
+
+    public func setup(amplitude: Amplitude) {
+        self.amplitude = amplitude
+    }
 }
 
 extension DestinationPlugin {
-
     var enabled: Bool {
         return true
     }