-
Notifications
You must be signed in to change notification settings - Fork 52
/
HitsCollectionViewDataSource.swift
46 lines (32 loc) · 1.27 KB
/
HitsCollectionViewDataSource.swift
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
//
// HitsCollectionViewDataSource.swift
// InstantSearch
//
// Created by Vladislav Fitc on 02/08/2019.
//
import Foundation
import UIKit
open class HitsCollectionViewDataSource<DataSource: HitsSource>: NSObject, UICollectionViewDataSource {
public var cellConfigurator: CollectionViewCellConfigurator<DataSource.Record>
public var templateCellProvider: () -> UICollectionViewCell
public weak var hitsSource: DataSource?
public init(cellConfigurator: @escaping CollectionViewCellConfigurator<DataSource.Record>) {
self.cellConfigurator = cellConfigurator
self.templateCellProvider = { return .init() }
}
open func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
guard let hitsSource = hitsSource else {
fatalError("Missing hits source")
}
return hitsSource.numberOfHits()
}
open func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
guard let hitsSource = hitsSource else {
fatalError("Missing hits source")
}
guard let hit = hitsSource.hit(atIndex: indexPath.row) else {
return templateCellProvider()
}
return cellConfigurator(collectionView, hit, indexPath)
}
}