IGListKit/Example/IGListKitExamples/ViewControllers/MixedDataViewController.swift
Jake Lin 0ecd0ddd9e Update example code to make it more Swifty
Summary:
In this PR, we updated example code to make it more Swifty by removing `CGRect` for using `.zero` only.

- [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/CONTRIBUTING.md)
Closes https://github.com/Instagram/IGListKit/pull/111

Differential Revision: D4065463

Pulled By: rnystrom

fbshipit-source-id: 089258466e320c58afc9d547f78a481ce9458e49
2016-10-22 20:14:15 -07:00

91 lines
3.9 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 MixedDataViewController: UIViewController, IGListAdapterDataSource {
lazy var adapter: IGListAdapter = {
return IGListAdapter(updater: IGListAdapterUpdater(), viewController: self, workingRangeSize: 0)
}()
let collectionView = IGListCollectionView(frame: .zero, collectionViewLayout: UICollectionViewFlowLayout())
let data = [
"Maecenas faucibus mollis interdum. Duis mollis, est non commodo luctus, nisi erat porttitor ligula, eget lacinia odio sem nec elit.",
GridItem(color: UIColor(red: 237/255.0, green: 73/255.0, blue: 86/255.0, alpha: 1), itemCount: 6),
User(pk: 2, name: "Ryan Olson", handle: "ryanolsonk"),
"Praesent commodo cursus magna, vel scelerisque nisl consectetur et.",
User(pk: 4, name: "Oliver Rickard", handle: "ocrickard"),
GridItem(color: UIColor(red: 56/255.0, green: 151/255.0, blue: 240/255.0, alpha: 1), itemCount: 5),
"Nullam quis risus eget urna mollis ornare vel eu leo. Praesent commodo cursus magna, vel scelerisque nisl consectetur et.",
User(pk: 3, name: "Jesse Squires", handle: "jesse_squires"),
GridItem(color: UIColor(red: 112/255.0, green: 192/255.0, blue: 80/255.0, alpha: 1), itemCount: 3),
"Fusce dapibus, tellus ac cursus commodo, tortor mauris condimentum nibh, ut fermentum massa justo sit amet risus.",
GridItem(color: UIColor(red: 163/255.0, green: 42/255.0, blue: 186/255.0, alpha: 1), itemCount: 7),
User(pk: 1, name: "Ryan Nystrom", handle: "_ryannystrom"),
] as [Any]
let segments: [(String, Any.Type?)] = [
("All", nil),
("Colors", GridItem.self),
("Text", String.self),
("Users", User.self)
]
var selectedClass: Any.Type?
override func viewDidLoad() {
super.viewDidLoad()
let control = UISegmentedControl(items: segments.map { return $0.0 } )
control.selectedSegmentIndex = 0
control.addTarget(self, action: #selector(MixedDataViewController.onControl(_:)), for: .valueChanged)
navigationItem.titleView = control
view.addSubview(collectionView)
adapter.collectionView = collectionView
adapter.dataSource = self
}
override func viewDidLayoutSubviews() {
super.viewDidLayoutSubviews()
collectionView.frame = view.bounds
}
func onControl(_ control: UISegmentedControl) {
selectedClass = segments[control.selectedSegmentIndex].1
adapter.performUpdates(animated: true, completion: nil)
}
//MARK: IGListAdapterDataSource
func objects(for listAdapter: IGListAdapter) -> [IGListDiffable] {
guard selectedClass != nil else {
return data.map { $0 as! IGListDiffable }
}
return data.filter { type(of: $0) == selectedClass! }
.map { $0 as! IGListDiffable }
}
func listAdapter(_ listAdapter: IGListAdapter, sectionControllerFor object: Any) -> IGListSectionController {
switch object {
case is String: return ExpandableSectionController()
case is GridItem: return GridSectionController()
default: return UserSectionController()
}
}
func emptyView(for listAdapter: IGListAdapter) -> UIView? { return nil }
}