add bundle var on CollectionCellDescribable to enable using collor in…

… external frameworks
sncf-connect-tech · Mar 7, 2018 · 1f6d97f · 1f6d97f
1 parent e01385d
commit 1f6d97f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/Collor/Classes/CollectionCellDescribable.swift b/Collor/Classes/CollectionCellDescribable.swift
@@ -20,6 +20,7 @@ public protocol CollectionCellDescribable : Identifiable {
     var selectable: Bool { get }
     func getAdapter() -> CollectionAdapter
     func size(_ collectionView: UICollectionView, sectionDescriptor: CollectionSectionDescribable) -> CGSize
+    var bundle: Bundle { get }
 }
 
 extension CollectionCellDescribable {
@@ -32,3 +33,10 @@ extension CollectionCellDescribable {
         }
     }
 }
+
+public extension CollectionCellDescribable {
+    var bundle: Bundle {
+        let typeClass = type(of: self)
+        return Bundle(for: typeClass)
+    }
+}
diff --git a/Collor/Classes/CollectionDataSource.swift b/Collor/Classes/CollectionDataSource.swift
@@ -37,7 +37,7 @@ public class CollectionDataSource: NSObject, UICollectionViewDataSource {
         let cellDescriptor = collectionData.sections[indexPath.section].cells[indexPath.item]
 
         if collectionData.registeredCells.contains(cellDescriptor.identifier) == false {
-            let nib = UINib(nibName: cellDescriptor.className, bundle: nil)
+            let nib = UINib(nibName: cellDescriptor.className, bundle: cellDescriptor.bundle)
             collectionView.register(nib, forCellWithReuseIdentifier: cellDescriptor.identifier)
             collectionData.registeredCells.insert(cellDescriptor.identifier)
         }