Dequeuing.swift

import UIKit

extension UICollectionView {
    func register<CellClass: UICollectionReusableView>(_ cellClass: CellClass.Type, ofKind kind: String? = nil) {
        let cellId = String(describing: type(of: cellClass.classForCoder()))
        
        if let kind = kind {
            register(cellClass, forSupplementaryViewOfKind: kind, withReuseIdentifier: cellId)
        } else if let collectionViewCellClass = cellClass as? UICollectionViewCell.Type {
            register(collectionViewCellClass, forCellWithReuseIdentifier: cellId)
        }
    }
    
    func dequeue<CellClass: UICollectionReusableView>(_ cellClass: CellClass.Type, ofKind kind: String? = nil, for indexPath: IndexPath) -> CellClass {
        let cellId = String(describing: type(of: cellClass.classForCoder()))

        if let kind = kind {
            return dequeueReusableSupplementaryView(ofKind: kind, withReuseIdentifier: cellId, for: indexPath) as! CellClass
        } else {
            return dequeueReusableCell(withReuseIdentifier: cellId, for: indexPath) as! CellClass
        }
    }
}

// Usage
collectionView.register(CustomCell.self)

let typedCell = collectionView.dequeue(CustomCell.self, for: indexPath)