ReusableCells.swift

//
// More type-aware convenience functions for registering and dequeueing cells
//

import UIKit

fileprivate extension UICollectionViewCell {
    
    static var defaultReuseIdentifier: String {
        return String(describing: self)
    }
    
}

extension UICollectionView {

    func register<Cell: UICollectionViewCell>(_ cellType: Cell.Type) {
        register(cellType, forCellWithReuseIdentifier: cellType.defaultReuseIdentifier)
    }

    func dequeue<Cell: UICollectionViewCell>(_ cellType: Cell.Type, for indexPath: IndexPath) -> Cell? {
        return dequeueReusableCell(withReuseIdentifier: cellType.defaultReuseIdentifier, for: indexPath) as? Cell
    }

}

fileprivate extension UITableViewCell {

    static var defaultReuseIdentifier: String {
        return String(describing: self)
    }

}

extension UITableView {

    func register<Cell: UITableViewCell>(_ cellType: Cell.Type) {
        register(cellType, forCellReuseIdentifier: cellType.defaultReuseIdentifier)
    }

    func dequeue<Cell: UITableViewCell>(_ cellType: Cell.Type, for indexPath: IndexPath) -> Cell? {
        return dequeueReusableCell(withIdentifier: cellType.defaultReuseIdentifier, for: indexPath) as? Cell
    }

}