From 3eb101b838f478cf5f6828670d3f01a827371244 Mon Sep 17 00:00:00 2001
From: Maher Santina <maher.santina90@gmail.com>
Date: Wed, 12 Sep 2018 08:46:57 +1000
Subject: [PATCH] Fixed issue where tableViewCell and collectionViewCell are
 nil (#16)

* Added variables and explanation

* Fixed cells always nil
---
 MSAutoView/Classes/MSCollectionViewCell.swift | 14 +++++---------
 MSAutoView/Classes/MSTableViewCell.swift      | 14 +++++---------
 2 files changed, 10 insertions(+), 18 deletions(-)

diff --git a/MSAutoView/Classes/MSCollectionViewCell.swift b/MSAutoView/Classes/MSCollectionViewCell.swift
index 8a09fd7..2def479 100644
--- a/MSAutoView/Classes/MSCollectionViewCell.swift
+++ b/MSAutoView/Classes/MSCollectionViewCell.swift
@@ -7,7 +7,7 @@
 
 import UIKit
 
-open class MSCollectionViewCell<T: UIView>: UICollectionViewCell {
+open class MSCollectionViewCell<T: MSAutoView>: UICollectionViewCell {
     public var mainView = T()
     
     override init(frame: CGRect) {
@@ -24,25 +24,21 @@ open class MSCollectionViewCell<T: UIView>: UICollectionViewCell {
         mainView.removeFromSuperview()
         contentView.addSubviewWithConstraints(mainView)
         backgroundColor = mainView.backgroundColor
+        mainView.collectionViewCell = self
     }
 }
 
 public protocol CollectionViewCellContainable {
-    associatedtype CollectionViewCellContainedViewType: UIView
+    associatedtype CollectionViewCellContainedViewType: MSAutoView
     
-    var collectionViewCell: MSCollectionViewCell<CollectionViewCellContainedViewType> { get }
     static var collectionViewCell: MSCollectionViewCell<CollectionViewCellContainedViewType>.Type { get }
 }
 
-extension CollectionViewCellContainable where Self: UIView {
-    
-    public var collectionViewCell: MSCollectionViewCell<Self> {
-        return MSCollectionViewCell()
-    }
+extension CollectionViewCellContainable where Self: MSAutoView {
     
     public static var collectionViewCell: MSCollectionViewCell<Self>.Type {
         return MSCollectionViewCell<Self>.self
     }
 }
 
-extension UIView: CollectionViewCellContainable { }
+extension MSAutoView: CollectionViewCellContainable { }
diff --git a/MSAutoView/Classes/MSTableViewCell.swift b/MSAutoView/Classes/MSTableViewCell.swift
index 5007da8..edf7a8c 100644
--- a/MSAutoView/Classes/MSTableViewCell.swift
+++ b/MSAutoView/Classes/MSTableViewCell.swift
@@ -7,7 +7,7 @@
 
 import UIKit
 
-open class MSTableViewCell<T: UIView>: UITableViewCell {
+open class MSTableViewCell<T: MSAutoView>: UITableViewCell {
     
     public var mainView = T()
     
@@ -25,26 +25,22 @@ open class MSTableViewCell<T: UIView>: UITableViewCell {
         mainView.removeFromSuperview()
         contentView.addSubviewWithConstraints(mainView)
         backgroundColor = mainView.backgroundColor
+        mainView.tableViewCell = self
     }
 }
 
 
 public protocol TableViewCellContainable {
-    associatedtype TableViewCellContainedViewType: UIView
+    associatedtype TableViewCellContainedViewType: MSAutoView
     
-    var tableViewCell: MSTableViewCell<TableViewCellContainedViewType> { get }
     static var tableViewCell: MSTableViewCell<TableViewCellContainedViewType>.Type { get }
 }
 
-extension TableViewCellContainable where Self: UIView {
-
-    public var tableViewCell: MSTableViewCell<Self> {
-        return MSTableViewCell()
-    }
+extension TableViewCellContainable where Self: MSAutoView {
     
     public static var tableViewCell: MSTableViewCell<Self>.Type {
         return MSTableViewCell<Self>.self
     }
 }
 
-extension UIView: TableViewCellContainable { }
+extension MSAutoView: TableViewCellContainable { }