From 9cca67c0f51600e21141d6718df24711fd69e89b Mon Sep 17 00:00:00 2001
From: Srdan Rasic <srdan.rasic@gmail.com>
Date: Sun, 29 Nov 2015 20:55:40 +0100
Subject: [PATCH] Handle potential table/collection view binding issue

---
 ReactiveUIKit.podspec                |  4 ++--
 ReactiveUIKit/Info.plist             |  2 +-
 ReactiveUIKit/UICollectionView.swift | 13 ++++++++-----
 ReactiveUIKit/UITableView.swift      | 15 +++++++++------
 4 files changed, 20 insertions(+), 14 deletions(-)

diff --git a/ReactiveUIKit.podspec b/ReactiveUIKit.podspec
index a79e5b6..2f3091a 100644
--- a/ReactiveUIKit.podspec
+++ b/ReactiveUIKit.podspec
@@ -1,11 +1,11 @@
 Pod::Spec.new do |s|
   s.name             = "ReactiveUIKit"
-  s.version          = "1.0.6"
+  s.version          = "1.0.8"
   s.summary          = "Reactive extensions for UIKit framework."
   s.homepage         = "https://github.com/ReactiveKit/ReactiveUIKit"
   s.license          = 'MIT'
   s.author           = { "Srdan Rasic" => "srdan.rasic@gmail.com" }
-  s.source           = { :git => "https://github.com/ReactiveKit/ReactiveUIKit.git", :tag => "v1.0.6" }
+  s.source           = { :git => "https://github.com/ReactiveKit/ReactiveUIKit.git", :tag => "v1.0.8" }
 
   s.ios.deployment_target       = '8.0'
   s.tvos.deployment_target      = '9.0'
diff --git a/ReactiveUIKit/Info.plist b/ReactiveUIKit/Info.plist
index 61718cd..a70ac6a 100644
--- a/ReactiveUIKit/Info.plist
+++ b/ReactiveUIKit/Info.plist
@@ -15,7 +15,7 @@
 	<key>CFBundlePackageType</key>
 	<string>FMWK</string>
 	<key>CFBundleShortVersionString</key>
-	<string>1.0.6</string>
+	<string>1.0.8</string>
 	<key>CFBundleSignature</key>
 	<string>????</string>
 	<key>CFBundleVersion</key>
diff --git a/ReactiveUIKit/UICollectionView.swift b/ReactiveUIKit/UICollectionView.swift
index 37cf17e..79c1ce8 100644
--- a/ReactiveUIKit/UICollectionView.swift
+++ b/ReactiveUIKit/UICollectionView.swift
@@ -54,7 +54,8 @@ extension ObservableCollectionType where Collection.Index == Int {
 
 public class RKCollectionViewDataSource<C: ObservableCollectionType where C.Collection.Index == Int>: NSObject, UICollectionViewDataSource {
   
-  private let collection: C
+  private let observableCollection: C
+  private var sourceCollection: C.Collection
   private weak var collectionView: UICollectionView!
   private let createCell: (NSIndexPath, C.Collection, UICollectionView) -> UICollectionViewCell
   private weak var proxyDataSource: RKCollectionViewProxyDataSource?
@@ -64,15 +65,17 @@ public class RKCollectionViewDataSource<C: ObservableCollectionType where C.Coll
     self.collectionView = collectionView
     self.createCell = createCell
     self.proxyDataSource = proxyDataSource
-    self.collection = collection
+    self.observableCollection = collection
+    self.sourceCollection = collection.collection
     self.animated = animated
     super.init()
     
     collectionView.dataSource = self
     collectionView.reloadData()
     
-    collection.observe(on: ImmediateOnMainExecutionContext) { [weak self] event in
+    observableCollection.observe(on: ImmediateOnMainExecutionContext) { [weak self] event in
       if let uSelf = self {
+        uSelf.sourceCollection = event.collection
         if animated {
           uSelf.collectionView.performBatchUpdates({
             RKCollectionViewDataSource.applyRowUnitChangeSet(event, collectionView: uSelf.collectionView, sectionIndex: 0, dataSource: uSelf.proxyDataSource)
@@ -109,11 +112,11 @@ public class RKCollectionViewDataSource<C: ObservableCollectionType where C.Coll
   }
   
   @objc public func collectionView(collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
-    return collection.collection.count
+    return sourceCollection.count
   }
   
   @objc public func collectionView(collectionView: UICollectionView, cellForItemAtIndexPath indexPath: NSIndexPath) -> UICollectionViewCell {
-    return createCell(indexPath, collection.collection, collectionView)
+    return createCell(indexPath, sourceCollection, collectionView)
   }
   
   @objc public func collectionView(collectionView: UICollectionView, viewForSupplementaryElementOfKind kind: String, atIndexPath indexPath: NSIndexPath) -> UICollectionReusableView {
diff --git a/ReactiveUIKit/UITableView.swift b/ReactiveUIKit/UITableView.swift
index 135de64..4ef3e0b 100644
--- a/ReactiveUIKit/UITableView.swift
+++ b/ReactiveUIKit/UITableView.swift
@@ -65,7 +65,8 @@ extension ObservableCollectionType where Collection.Index == Int {
 
 public class RKTableViewDataSource<C: ObservableCollectionType where C.Collection.Index == Int>: NSObject, UITableViewDataSource {
   
-  private let collection: C
+  private let observableCollection: C
+  private var sourceCollection: C.Collection
   private weak var tableView: UITableView!
   private let createCell: (NSIndexPath, C.Collection, UITableView) -> UITableViewCell
   private weak var proxyDataSource: RKTableViewProxyDataSource?
@@ -75,15 +76,17 @@ public class RKTableViewDataSource<C: ObservableCollectionType where C.Collectio
     self.tableView = tableView
     self.createCell = createCell
     self.proxyDataSource = proxyDataSource
-    self.collection = collection
+    self.observableCollection = collection
+    self.sourceCollection = collection.collection
     self.animated = animated
     super.init()
-    
+
     tableView.dataSource = self
     tableView.reloadData()
     
-    collection.observe(on: ImmediateOnMainExecutionContext) { [weak self] event in
+    observableCollection.observe(on: ImmediateOnMainExecutionContext) { [weak self] event in
       if let uSelf = self {
+        uSelf.sourceCollection = event.collection
         if animated {
           uSelf.tableView.beginUpdates()
           RKTableViewDataSource.applyRowUnitChangeSet(event, tableView: uSelf.tableView, sectionIndex: 0, dataSource: uSelf.proxyDataSource)
@@ -120,11 +123,11 @@ public class RKTableViewDataSource<C: ObservableCollectionType where C.Collectio
   }
   
   @objc public func tableView(tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
-    return collection.collection.count
+    return sourceCollection.count
   }
   
   @objc public func tableView(tableView: UITableView, cellForRowAtIndexPath indexPath: NSIndexPath) -> UITableViewCell {
-    return createCell(indexPath, collection.collection, tableView)
+    return createCell(indexPath, sourceCollection, tableView)
   }
   
   @objc public func tableView(tableView: UITableView, titleForHeaderInSection section: Int) -> String? {