IGListKit/Examples/Examples-iOS/IGListKitExamples/SectionControllers/GridSectionController.swift
Daniel Rosado 90cf9d5dfb Fix crash in Examples-iOS app when tapping "Mixed Data"
Summary:
Fix crash in Examples-iOS app when tapping "Mixed Data"

Make GridItem conform to IGListDiffable now that NSObject+IGListDiffable category has been removed.

- [x] All tests pass. Demo project builds and runs.
- [ ] I added tests, an experiment, or detailed why my change isn't tested.
- [x] I have reviewed the [contributing guide](https://github.com/Instagram/IGListKit/blob/master/.github/CONTRIBUTING.md)
Closes https://github.com/Instagram/IGListKit/pull/251

Differential Revision: D4224337

Pulled By: jessesquires

fbshipit-source-id: 16a437c4e5e45185481b4262ea7770f4a8559618
2016-11-22 17:14:00 -08:00

75 lines
2.1 KiB
Swift

/**
Copyright (c) 2016-present, Facebook, Inc. All rights reserved.
The examples provided by Facebook are for non-commercial testing and evaluation
purposes only. Facebook reserves all rights not expressly granted.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
FACEBOOK BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
import UIKit
import IGListKit
class GridItem: NSObject {
let color: UIColor
let itemCount: Int
init(color: UIColor, itemCount: Int) {
self.color = color
self.itemCount = itemCount
}
}
extension GridItem: IGListDiffable {
func diffIdentifier() -> NSObjectProtocol {
return self
}
func isEqual(toDiffableObject object: IGListDiffable?) -> Bool {
return self === object ? true : self.isEqual(object)
}
}
final class GridSectionController: IGListSectionController, IGListSectionType {
var object: GridItem?
override init() {
super.init()
self.minimumInteritemSpacing = 1
self.minimumLineSpacing = 1
}
func numberOfItems() -> Int {
return object?.itemCount ?? 0
}
func sizeForItem(at index: Int) -> CGSize {
let width = collectionContext?.containerSize.width ?? 0
let itemSize = floor(width / 4)
return CGSize(width: itemSize, height: itemSize)
}
func cellForItem(at index: Int) -> UICollectionViewCell {
let cell = collectionContext!.dequeueReusableCell(of: CenterLabelCell.self, for: self, at: index) as! CenterLabelCell
cell.label.text = "\(index + 1)"
cell.backgroundColor = object?.color
return cell
}
func didUpdate(to object: Any) {
self.object = object as? GridItem
}
func didSelectItem(at index: Int) {}
}