Add context in configure

Author: onmyway133

Sources/Micro/DataSource.swift

@@ -9,7 +9,7 @@ import UIKit
 
 open class DataSource: NSObject {
     private var cellRegister: Set<String> = Set()
-    private var finalState: State = .init()
+    internal var trueState: State = .init()
     public weak var collectionView: UICollectionView?
 
     public init(collectionView: UICollectionView) {
@@ -18,7 +18,7 @@ open class DataSource: NSObject {
 
     public var state: State {
         get {
-            finalState
+            trueState
         }
         set {
             reload(newState: newValue)
@@ -34,7 +34,6 @@ open class DataSource: NSObject {
         let request = Reloader.Request(
             dataSource: self,
             collectionView: collectionView,
-            oldState: finalState,
             newState: newState,
             isAnimated: isAnimated,
             completion: completion ?? { _ in }

Sources/Micro/ForEach.swift

@@ -14,10 +14,10 @@ public func forEach<Model: DiffAware>(
 ) -> State {
     let onReload: Reloader.Reload = { request in
         let reload: () -> Void = {
-            let oldModels: [Model] = request.oldState.models.compactMap({ $0 as? Model })
+            let oldModels: [Model] = request.dataSource.trueState.models.compactMap({ $0 as? Model })
 
             if oldModels.isEmpty {
-                request.dataSource.finalState = request.newState
+                request.dataSource.trueState = request.newState
                 request.collectionView.reloadData()
                 request.completion(true)
                 return
@@ -28,7 +28,7 @@ public func forEach<Model: DiffAware>(
                 changes: changes,
                 section: 0,
                 updateData: {
-                    request.dataSource.finalState = request.newState
+                    request.dataSource.trueState = request.newState
                 },
                 completion: request.completion
             )

Sources/Micro/Item.swift

@@ -14,11 +14,11 @@ public protocol ObserverOwner {
 public struct Item<Cell: UICollectionViewCell>: ObserverOwner {
     public let observer = Observer()
 
-    public init(_ closure: @escaping (Cell) -> Void) {
+    public init(_ closure: @escaping (Context, Cell) -> Void) {
         observer.onConfigure = { context, dataSource in
             dataSource.registerIfNeeded(collectionView: context.collectionView, cellType: Cell.self)
             if let cell: Cell = context.collectionView.dequeue(for: context.indexPath) {
-                closure(cell)
+                closure(context, cell)
                 return cell
             } else {
                 return nil

Sources/Micro/Reloader.swift

@@ -11,7 +11,6 @@ public struct Reloader {
     public struct Request {
         public let dataSource: DataSource
         public let collectionView: UICollectionView
-        public let oldState: State
         public let newState: State
         public let isAnimated: Bool
         public let completion: Completion

Tests/MicroTests/MicroTests.swift

@@ -30,7 +30,7 @@ final class MicroTests: XCTestCase {
         ]
 
         dataSource.state = forEach(models: persons) { person in
-            Item<PersonCell>() { cell in
+            Item<PersonCell>() { context, cell in
                 cell.nameLabel.text = person.name
             }
             .onSelect { _ in