UITableView+register.swift

import UIKit

public protocol UITableViewHeaderFooterViewIdentifiable: UITableViewHeaderFooterView {
    static var identifier: String { get }
}

public extension UITableViewHeaderFooterViewIdentifiable where Self: UITableViewHeaderFooterView {
    static var identifier: String {
        return String(describing: self)
    }
}

public protocol UITableViewCellIdentifiable: UITableViewCell {
    static var identifier: String { get }
}

public extension UITableViewCellIdentifiable where Self: UITableViewCell {
    static var identifier: String {
         return String(describing: self)
    }
}

public extension UITableView {
    func register(_ elements: [UIView.Type]) {
        elements.forEach { element in
            switch element {
            case is UITableViewCellIdentifiable.Type:
                guard let cell = element as? UITableViewCellIdentifiable.Type else {
                    assertionFailure("failed registering element \(element)")
                    return
                }
                self.register(cell.self, forCellReuseIdentifier: cell.identifier)
                self.register(UINib(nibName: cell.identifier, bundle: nil), forCellReuseIdentifier: cell.identifier)
            case is UITableViewHeaderFooterViewIdentifiable.Type:
                guard let cell = element as? UITableViewHeaderFooterViewIdentifiable.Type else {
                    assertionFailure("failed registering element \(element)")
                    return
                }
                self.register(cell.self, forHeaderFooterViewReuseIdentifier: cell.identifier)
                self.register(UINib(nibName: cell.identifier, bundle: nil), forHeaderFooterViewReuseIdentifier: cell.identifier)
            default:
                break
            }
        }
    }
}