add CocoaProxy

Example/InfiniteLayout.xcodeproj/project.pbxproj

@@ -291,10 +291,12 @@
 			);
 			inputPaths = (
 				"${SRCROOT}/Pods/Target Support Files/Pods-InfiniteLayout_Example/Pods-InfiniteLayout_Example-frameworks.sh",
+				"${BUILT_PRODUCTS_DIR}/CocoaProxy/CocoaProxy.framework",
 				"${BUILT_PRODUCTS_DIR}/InfiniteLayout/InfiniteLayout.framework",
 			);
 			name = "[CP] Embed Pods Frameworks";
 			outputPaths = (
+				"${TARGET_BUILD_DIR}/${FRAMEWORKS_FOLDER_PATH}/CocoaProxy.framework",
 				"${TARGET_BUILD_DIR}/${FRAMEWORKS_FOLDER_PATH}/InfiniteLayout.framework",
 			);
 			runOnlyForDeploymentPostprocessing = 0;

Example/Podfile.lock

@@ -1,5 +1,7 @@
 PODS:
-  - InfiniteLayout (0.1.0)
+  - CocoaProxy (0.1.1)
+  - InfiniteLayout (0.1.3):
+    - CocoaProxy (~> 0.1)
 
 DEPENDENCIES:
   - InfiniteLayout (from `../`)
@@ -9,7 +11,8 @@ EXTERNAL SOURCES:
     :path: ../
 
 SPEC CHECKSUMS:
-  InfiniteLayout: 82a21b8255623e2d72174f3082bef9e576140621
+  CocoaProxy: 35ab81e24325b33834cffe45a3d1fd48ca67ef3a
+  InfiniteLayout: 0ab39bf2a4d9ce5c473ebccfc0d3b7278de18ac9
 
 PODFILE CHECKSUM: 3a658536624f41ec07c5a7a2c55407b9b8f48528
 

InfiniteLayout.podspec

@@ -8,7 +8,7 @@
 
 Pod::Spec.new do |s|
   s.name             = 'InfiniteLayout'
-  s.version          = '0.1.3'
+  s.version          = '0.1.4'
   s.summary          = 'Horizontal and Vertical infinite scrolling feature for UICollectionView'
 
 # This description is used to generate tags and improve search results.
@@ -40,5 +40,5 @@ Horizontal and Vertical infinite scrolling feature for UICollectionView with Pag
 
   # s.public_header_files = 'Pod/Classes/**/*.h'
   # s.frameworks = 'UIKit', 'MapKit'
-  # s.dependency 'AFNetworking', '~> 2.3'
+  s.dependency 'CocoaProxy', '~> 0.1'
 end

InfiniteLayout/Classes/InfiniteCollectionView.swift

@@ -9,10 +9,8 @@ import UIKit
 
 open class InfiniteCollectionView: UICollectionView {
 
-    lazy var delegateProxy = InfiniteCollectionViewDelegateProxy(self, exceptions: ["scrollViewDidScroll:",
-                                                                                    "scrollViewWillEndDragging:withVelocity:targetContentOffset:"])
-    lazy var dataSourceProxy = InfiniteCollectionViewDataSourceProxy(self, exceptions: ["numberOfSectionsInCollectionView:",
-                                                                                        "collectionView:numberOfItemsInSection:"])
+    lazy var delegateProxy = InfiniteCollectionViewDelegateProxy(collectionView: self)
+    lazy var dataSourceProxy = InfiniteCollectionViewDataSourceProxy(collectionView: self)
 
     @IBInspectable var isItemPagingEnabled: Bool = false
     @IBInspectable var velocityMultiplier: CGFloat = 500 {

InfiniteLayout/Classes/InfiniteCollectionViewProxy.swift

@@ -0,0 +1,85 @@
+//
+//  Proxy.swift
+//  InfiniteLayout
+//
+//  Created by Arnaud Dorgans on 20/12/2017.
+//
+
+import UIKit
+import CocoaProxy
+
+class InfiniteCollectionViewProxy<T: NSObjectProtocol>: CocoaProxy {
+
+    var collectionView: InfiniteCollectionView! {
+        get { return self.proxies.first as? InfiniteCollectionView }
+        set {
+            if !self.proxies.isEmpty {
+                self.proxies.removeFirst()
+            }
+            self.proxies.insert(newValue, at: 0)
+        }
+    }
+
+    var delegate: T? {
+        get {
+            guard self.proxies.count > 1 else {
+                return nil
+            }
+            return self.proxies.last as? T
+        } set {
+            while self.proxies.count > 1 {
+                self.proxies.removeLast()
+            }
+            guard let delegate = newValue else {
+                return
+            }
+            self.proxies.append(delegate)
+        }
+    }
+
+    override func proxies(for aSelector: Selector) -> [NSObjectProtocol] {
+         return super.proxies(for: aSelector).reversed()
+    }
+
+    init(collectionView: InfiniteCollectionView) {
+        super.init(proxies: [])
+        self.collectionView = collectionView
+    }
+}
+
+class InfiniteCollectionViewDelegateProxy: InfiniteCollectionViewProxy<UICollectionViewDelegate>, UICollectionViewDelegate {
+
+    override func proxies(for aSelector: Selector) -> [NSObjectProtocol] {
+        return super.proxies(for: aSelector)
+            .first { proxy in
+                guard !(aSelector == #selector(UIScrollViewDelegate.scrollViewDidScroll(_:)) ||
+                    aSelector == #selector(UIScrollViewDelegate.scrollViewWillEndDragging(_:withVelocity:targetContentOffset:))) else {
+                        return proxy is InfiniteCollectionView
+                }
+                return true
+            }.flatMap { [$0] } ?? []
+    }
+}
+
+class InfiniteCollectionViewDataSourceProxy: InfiniteCollectionViewProxy<UICollectionViewDataSource>, UICollectionViewDataSource {
+
+    override func proxies(for aSelector: Selector) -> [NSObjectProtocol] {
+        return super.proxies(for: aSelector)
+            .first { proxy in
+                guard !(aSelector == #selector(UICollectionViewDataSource.numberOfSections(in:)) ||
+                    aSelector == #selector(UICollectionViewDataSource.collectionView(_:numberOfItemsInSection:))) else {
+                        return proxy is InfiniteCollectionView
+                }
+                return true
+            }.flatMap { [$0] } ?? []
+    }
+
+    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
+        return self.collectionView.collectionView(collectionView, numberOfItemsInSection: section)
+    }
+
+    func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
+        return self.delegate?.collectionView(collectionView, cellForItemAt: indexPath) ??
+            self.collectionView.collectionView(collectionView, cellForItemAt: indexPath)
+    }
+}