mirror of
https://github.com/ReactiveX/RxSwift.git
synced 2024-10-05 06:27:29 +03:00
235 lines
8.6 KiB
Swift
235 lines
8.6 KiB
Swift
//
|
|
// UICollectionView+Rx.swift
|
|
// RxCocoa
|
|
//
|
|
// Created by Krunoslav Zaher on 4/2/15.
|
|
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
|
|
//
|
|
|
|
#if os(iOS) || os(tvOS)
|
|
|
|
import Foundation
|
|
#if !RX_NO_MODULE
|
|
import RxSwift
|
|
#endif
|
|
import UIKit
|
|
|
|
// Items
|
|
|
|
extension UICollectionView {
|
|
|
|
/**
|
|
Binds sequences of elements to collection view items.
|
|
|
|
- parameter source: Observable sequence of items.
|
|
- parameter cellFactory: Transform between sequence elements and view cells.
|
|
- returns: Disposable object that can be used to unbind.
|
|
*/
|
|
public func rx_itemsWithCellFactory<S: SequenceType, O: ObservableType where O.E == S>
|
|
(source: O)
|
|
-> (cellFactory: (UICollectionView, Int, S.Generator.Element) -> UICollectionViewCell)
|
|
-> Disposable {
|
|
return { cellFactory in
|
|
let dataSource = RxCollectionViewReactiveArrayDataSourceSequenceWrapper<S>(cellFactory: cellFactory)
|
|
return self.rx_itemsWithDataSource(dataSource)(source: source)
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
Binds sequences of elements to collection view items.
|
|
|
|
- parameter cellIdentifier: Identifier used to dequeue cells.
|
|
- parameter source: Observable sequence of items.
|
|
- parameter configureCell: Transform between sequence elements and view cells.
|
|
- parameter cellType: Type of table view cell.
|
|
- returns: Disposable object that can be used to unbind.
|
|
*/
|
|
public func rx_itemsWithCellIdentifier<S: SequenceType, Cell: UICollectionViewCell, O : ObservableType where O.E == S>
|
|
(cellIdentifier: String, cellType: Cell.Type = Cell.self)
|
|
-> (source: O)
|
|
-> (configureCell: (Int, S.Generator.Element, Cell) -> Void)
|
|
-> Disposable {
|
|
return { source in
|
|
return { configureCell in
|
|
let dataSource = RxCollectionViewReactiveArrayDataSourceSequenceWrapper<S> { (cv, i, item) in
|
|
let indexPath = NSIndexPath(forItem: i, inSection: 0)
|
|
let cell = cv.dequeueReusableCellWithReuseIdentifier(cellIdentifier, forIndexPath: indexPath) as! Cell
|
|
configureCell(i, item, cell)
|
|
return cell
|
|
}
|
|
|
|
return self.rx_itemsWithDataSource(dataSource)(source: source)
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
Binds sequences of elements to collection view items using a custom reactive data used to perform the transformation.
|
|
|
|
- parameter dataSource: Data source used to transform elements to view cells.
|
|
- parameter source: Observable sequence of items.
|
|
- returns: Disposable object that can be used to unbind.
|
|
*/
|
|
public func rx_itemsWithDataSource<DataSource: protocol<RxCollectionViewDataSourceType, UICollectionViewDataSource>, S: SequenceType, O: ObservableType where DataSource.Element == S, O.E == S>
|
|
(dataSource: DataSource)
|
|
-> (source: O)
|
|
-> Disposable {
|
|
return { source in
|
|
return source.subscribeProxyDataSourceForObject(self, dataSource: dataSource, retainDataSource: false) { [weak self] (_: RxCollectionViewDataSourceProxy, event) -> Void in
|
|
guard let collectionView = self else {
|
|
return
|
|
}
|
|
dataSource.collectionView(collectionView, observedEvent: event)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
extension UICollectionView {
|
|
|
|
/**
|
|
Factory method that enables subclasses to implement their own `rx_delegate`.
|
|
|
|
- returns: Instance of delegate proxy that wraps `delegate`.
|
|
*/
|
|
public override func rx_createDelegateProxy() -> RxScrollViewDelegateProxy {
|
|
return RxCollectionViewDelegateProxy(parentObject: self)
|
|
}
|
|
|
|
/**
|
|
Factory method that enables subclasses to implement their own `rx_dataSource`.
|
|
|
|
- returns: Instance of delegate proxy that wraps `dataSource`.
|
|
*/
|
|
public func rx_createDataSourceProxy() -> RxCollectionViewDataSourceProxy {
|
|
return RxCollectionViewDataSourceProxy(parentObject: self)
|
|
}
|
|
|
|
/**
|
|
Reactive wrapper for `dataSource`.
|
|
|
|
For more information take a look at `DelegateProxyType` protocol documentation.
|
|
*/
|
|
public var rx_dataSource: DelegateProxy {
|
|
return proxyForObject(RxCollectionViewDataSourceProxy.self, self)
|
|
}
|
|
|
|
/**
|
|
Installs data source as forwarding delegate on `rx_dataSource`.
|
|
|
|
It enables using normal delegate mechanism with reactive delegate mechanism.
|
|
|
|
- parameter dataSource: Data source object.
|
|
- returns: Disposable object that can be used to unbind the data source.
|
|
*/
|
|
public func rx_setDataSource(dataSource: UICollectionViewDataSource)
|
|
-> Disposable {
|
|
let proxy = proxyForObject(RxCollectionViewDataSourceProxy.self, self)
|
|
return installDelegate(proxy, delegate: dataSource, retainDelegate: false, onProxyForObject: self)
|
|
}
|
|
|
|
/**
|
|
Reactive wrapper for `delegate` message `collectionView:didSelectItemAtIndexPath:`.
|
|
*/
|
|
public var rx_itemSelected: ControlEvent<NSIndexPath> {
|
|
let source = rx_delegate.observe(#selector(UICollectionViewDelegate.collectionView(_:didSelectItemAtIndexPath:)))
|
|
.map { a in
|
|
return a[1] as! NSIndexPath
|
|
}
|
|
|
|
return ControlEvent(events: source)
|
|
}
|
|
|
|
/**
|
|
Reactive wrapper for `delegate` message `collectionView:didSelectItemAtIndexPath:`.
|
|
*/
|
|
public var rx_itemDeselected: ControlEvent<NSIndexPath> {
|
|
let source = rx_delegate.observe(#selector(UICollectionViewDelegate.collectionView(_:didDeselectItemAtIndexPath:)))
|
|
.map { a in
|
|
return a[1] as! NSIndexPath
|
|
}
|
|
|
|
return ControlEvent(events: source)
|
|
}
|
|
|
|
/**
|
|
Reactive wrapper for `delegate` message `collectionView:didSelectItemAtIndexPath:`.
|
|
|
|
It can be only used when one of the `rx_itemsWith*` methods is used to bind observable sequence,
|
|
or any other data source conforming to `SectionedViewDataSourceType` protocol.
|
|
|
|
```
|
|
collectionView.rx_modelSelected(MyModel.self)
|
|
.map { ...
|
|
```
|
|
*/
|
|
public func rx_modelSelected<T>(modelType: T.Type) -> ControlEvent<T> {
|
|
let source: Observable<T> = rx_itemSelected.flatMap { [weak self] indexPath -> Observable<T> in
|
|
guard let view = self else {
|
|
return Observable.empty()
|
|
}
|
|
|
|
return Observable.just(try view.rx_modelAtIndexPath(indexPath))
|
|
}
|
|
|
|
return ControlEvent(events: source)
|
|
}
|
|
|
|
/**
|
|
Reactive wrapper for `delegate` message `collectionView:didSelectItemAtIndexPath:`.
|
|
|
|
It can be only used when one of the `rx_itemsWith*` methods is used to bind observable sequence,
|
|
or any other data source conforming to `SectionedViewDataSourceType` protocol.
|
|
|
|
```
|
|
collectionView.rx_modelDeselected(MyModel.self)
|
|
.map { ...
|
|
```
|
|
*/
|
|
public func rx_modelDeselected<T>(modelType: T.Type) -> ControlEvent<T> {
|
|
let source: Observable<T> = rx_itemDeselected.flatMap { [weak self] indexPath -> Observable<T> in
|
|
guard let view = self else {
|
|
return Observable.empty()
|
|
}
|
|
|
|
return Observable.just(try view.rx_modelAtIndexPath(indexPath))
|
|
}
|
|
|
|
return ControlEvent(events: source)
|
|
}
|
|
|
|
/**
|
|
Syncronous helper method for retrieving a model at indexPath through a reactive data source
|
|
*/
|
|
public func rx_modelAtIndexPath<T>(indexPath: NSIndexPath) throws -> T {
|
|
let dataSource: SectionedViewDataSourceType = castOrFatalError(self.rx_dataSource.forwardToDelegate(), message: "This method only works in case one of the `rx_itemsWith*` methods was used.")
|
|
|
|
let element = try dataSource.modelAtIndexPath(indexPath)
|
|
|
|
return element as! T
|
|
}
|
|
}
|
|
#endif
|
|
|
|
#if os(tvOS)
|
|
|
|
extension UICollectionView {
|
|
|
|
/**
|
|
Reactive wrapper for `delegate` message `collectionView:didUpdateFocusInContext:withAnimationCoordinator:`.
|
|
*/
|
|
public var rx_didUpdateFocusInContextWithAnimationCoordinator: ControlEvent<(context: UIFocusUpdateContext, animationCoordinator: UIFocusAnimationCoordinator)> {
|
|
|
|
let source = rx_delegate.observe(#selector(UICollectionViewDelegate.collectionView(_:didUpdateFocusInContext:withAnimationCoordinator:)))
|
|
.map { a -> (context: UIFocusUpdateContext, animationCoordinator: UIFocusAnimationCoordinator) in
|
|
let context = a[1] as! UIFocusUpdateContext
|
|
let animationCoordinator = a[2] as! UIFocusAnimationCoordinator
|
|
return (context: context, animationCoordinator: animationCoordinator)
|
|
}
|
|
|
|
return ControlEvent(events: source)
|
|
}
|
|
}
|
|
#endif
|