Merge pull request #65 from colinhumber/host-blacklist

Added ability to blacklist hosts from being recorded

Sources/CustomHTTPProtocol.swift

@@ -9,7 +9,8 @@
 import Foundation
 
 public class CustomHTTPProtocol: URLProtocol {
-
+    static var blacklistedHosts = [String]()
+
     struct Constants {
         static let RequestHandledKey = "URLProtocolRequestHandled"
     }
@@ -27,6 +28,8 @@ public class CustomHTTPProtocol: URLProtocol {
     }
 
     override public class func canInit(with request: URLRequest) -> Bool {
+        guard CustomHTTPProtocol.shouldHandleRequest(request) else { return false }
+
         if CustomHTTPProtocol.property(forKey: Constants.RequestHandledKey, in: request) != nil {
             return false
         }
@@ -53,6 +56,7 @@ public class CustomHTTPProtocol: URLProtocol {
         if let startDate = currentRequest?.date{
             currentRequest?.duration = fabs(startDate.timeIntervalSinceNow) * 1000 //Find elapsed time and convert to milliseconds
         }
+
         Storage.shared.saveRequest(request: currentRequest)
         session?.invalidateAndCancel()
     }
@@ -70,6 +74,14 @@ public class CustomHTTPProtocol: URLProtocol {
             return data as Data
         }
     }
+
+    /// Inspects the request to see if the host has not been blacklisted and can be handled by this URL protocol.
+    /// - Parameter request: The request being processed.
+    private class func shouldHandleRequest(_ request: URLRequest) -> Bool {
+        guard let host = request.url?.host else { return false }
+
+        return CustomHTTPProtocol.blacklistedHosts.filter({ host.hasSuffix($0) }).isEmpty
+    }
 
     deinit {
         session = nil

Sources/Wormholy.swift

@@ -11,6 +11,11 @@ import UIKit
 
 public class Wormholy: NSObject
 {
+    @objc public static var blacklistedHosts: [String] {
+        get { return CustomHTTPProtocol.blacklistedHosts }
+        set { CustomHTTPProtocol.blacklistedHosts = newValue }
+    }
+
     @objc public static func swiftyLoad() {
         NotificationCenter.default.addObserver(forName: fireWormholy, object: nil, queue: nil) { (notification) in
             Wormholy.presentWormholyFlow()

WormholyDemo/ViewController.swift

@@ -8,11 +8,12 @@
 
 import UIKit
 import Foundation
-
+import Wormholy
 class ViewController: UIViewController {
 
     override func viewDidLoad() {
         super.viewDidLoad()
+
         if #available(iOS 10.0, *) {
             let timer = Timer.scheduledTimer(withTimeInterval: 8, repeats: true) { (timer) in
                 DataFetcher.sharedInstance.getPost(id: Utils.random(max: 128), completion: {