Add observer owner

Author: onmyway133

Sources/Micro/DataSource.swift

@@ -9,7 +9,7 @@ import UIKit
 
 open class DataSource: NSObject {
     private var cellRegister: Set<String> = Set()
-    public var finalState: State = .init()
+    private var finalState: State = .init()
     public weak var collectionView: UICollectionView?
 
     public init(collectionView: UICollectionView) {

Sources/Micro/ForEach.swift

@@ -8,16 +8,16 @@
 import UIKit
 import DeepDiff
 
-public func forEach<Model: DiffAware, Cell: UICollectionViewCell>(
+public func forEach<Model: DiffAware>(
     models: [Model],
-    transform: (Model) -> Item<Cell>
+    transform: (Model) -> ObserverOwner
 ) -> State {
     let onReload: Reloader.Reload = { request in
         let reload: () -> Void = {
             let oldModels: [Model] = request.oldState.models.compactMap({ $0 as? Model })
 
             if oldModels.isEmpty {
-                request.dataSource.state = request.newState
+                request.dataSource.finalState = request.newState
                 request.collectionView.reloadData()
                 request.completion(true)
                 return
@@ -43,12 +43,12 @@ public func forEach<Model: DiffAware, Cell: UICollectionViewCell>(
         }
     }
 
-    let items: [Item<Cell>] = models.map(transform)
-    let observers: [Observer] = items.map({ $0.observer })
+    let owners: [ObserverOwner] = models.map(transform)
+    let observers: [Observer] = owners.map({ $0.observer })
 
     return State(
         models: models,
-        items: items,
+        owners: owners,
         observers: observers,
         reloader: Reloader(onReload: onReload)
     )

Sources/Micro/Item.swift

@@ -7,7 +7,11 @@
 
 import UIKit
 
-public struct Item<Cell: UICollectionViewCell> {
+public protocol ObserverOwner {
+    var observer: Observer { get }
+}
+
+public struct Item<Cell: UICollectionViewCell>: ObserverOwner {
     public let observer = Observer()
 
     public init(_ closure: @escaping (Cell) -> Void) {

Sources/Micro/State.swift

@@ -10,15 +10,15 @@ import DeepDiff
 
 public struct State {
     let models: [Any]
-    let items: [Any]
+    let owners: [ObserverOwner]
     let observers: [Observer]
     let reloader: Reloader
 }
 
 extension State {
     public init() {
         self.models = []
-        self.items = []
+        self.owners = []
         self.observers = []
         self.reloader = Reloader(onReload: { _ in })
     }