Expanded pluginfind capabilities. (#272)

Author: bsneed

Sources/Segment/Plugins.swift

@@ -156,6 +156,14 @@ extension DestinationPlugin {
         timeline.remove(plugin: plugin)
     }
 
+    public func find<T: Plugin>(pluginType: T.Type) -> T? {
+        return timeline.find(pluginType: pluginType)
+    }
+    
+    public func findAll<T: Plugin>(pluginType: T.Type) -> [T]? {
+        return timeline.findAll(pluginType: pluginType)
+    }
+
 }
 
 extension Analytics {
@@ -212,6 +220,10 @@ extension Analytics {
         return timeline.find(pluginType: pluginType)
     }
 
+    public func findAll<T: Plugin>(pluginType: T.Type) -> [T]? {
+        return timeline.findAll(pluginType: pluginType)
+    }
+    
     public func find(key: String) -> DestinationPlugin? {
         return timeline.find(key: key)
     }

Sources/Segment/Timeline.swift

@@ -136,6 +136,18 @@ extension Timeline {
         return found.first as? T
     }
 
+    internal func findAll<T: Plugin>(pluginType: T.Type) -> [T]? {
+        var found = [Plugin]()
+        for type in PluginType.allCases {
+            if let mediator = plugins[type] {
+                found.append(contentsOf: mediator.plugins.filter { (plugin) -> Bool in
+                    return plugin is T
+                })
+            }
+        }
+        return found as? [T]
+    }
+    
     internal func find(key: String) -> DestinationPlugin? {
         var found = [Plugin]()
         if let mediator = plugins[.destination] {

Tests/Segment-Tests/Analytics_Tests.swift

@@ -754,4 +754,29 @@ final class Analytics_Tests: XCTestCase {
         // it's running in sync mode.
         XCTAssertEqual(analytics.pendingUploads!.count, 0)
     }
+    
+    func testFindAll() {
+        let analytics = Analytics(configuration: Configuration(writeKey: "testFindAll")
+            .flushInterval(9999)
+            .flushAt(9999)
+            .operatingMode(.synchronous))
+        
+        analytics.add(plugin: ZiggyPlugin())
+        analytics.add(plugin: ZiggyPlugin())
+        analytics.add(plugin: ZiggyPlugin())
+
+        let myDestination = MyDestination()
+        myDestination.add(plugin: GooberPlugin())
+        myDestination.add(plugin: GooberPlugin())
+        
+        analytics.add(plugin: myDestination)
+        
+        waitUntilStarted(analytics: analytics)
+        
+        let ziggysFound = analytics.findAll(pluginType: ZiggyPlugin.self)
+        let goobersFound = myDestination.findAll(pluginType: GooberPlugin.self)
+        
+        XCTAssertEqual(ziggysFound!.count, 3)
+        XCTAssertEqual(goobersFound!.count, 2)
+    }
 }