UICollectionView+Cell.swift

//
//  UICollectionView+Register.swift
//  Sucrose
//
//  Created by Pedro Carrasco on 14/02/2019.
//  Copyright © 2019 Pedro Carrasco. All rights reserved.
//

import UIKit

public extension UICollectionView {
    
    enum SupplementaryViewKind {
        case header
        case footer
        
        var value: String {
            switch self {
            case .header: return UICollectionView.elementKindSectionHeader
            case .footer: return UICollectionView.elementKindSectionFooter
            }
        }
    }

    func register<T: UICollectionViewCell>(_: T.Type) {
        register(T.self, forCellWithReuseIdentifier: T.name)
    }
    
    func register<T: UICollectionReusableView>(_: T.Type, as kind: SupplementaryViewKind) {
        register(T.self, forSupplementaryViewOfKind: kind.value, withReuseIdentifier: T.name)
    }

    func dequeue<T: UICollectionViewCell>(_ type: T.Type, in indexPath: IndexPath) -> T {
        guard let cell = dequeueReusableCell(withReuseIdentifier: type.name, for: indexPath) as? T else {
            fatalError("Unable to dequeue \(T.name)")
        }
    
        return cell
    }
    
    func dequeue<T: UICollectionReusableView>(_ type: T.Type, in indexPath: IndexPath, as kind: SupplementaryViewKind) -> T {

        guard let view = dequeueReusableSupplementaryView(ofKind: kind.value, withReuseIdentifier: T.name, for: indexPath) as? T else {
            fatalError("Unable to dequeue \(T.name)")
        }
        
        return view
    }
}