// Adjust size of table header view
extension UITableViewController {
    var tableHeaderView: UIView {
        return self.customHeaderView
    }

    var tableFooterView: UIView {
        return self.customFooterView
    }

    override func viewDidLayoutSubviews() {
        super.viewDidLayoutSubviews()

        self.adjustSizeOfTableHeaderView()
        self.adjustSizeOfTableFooterView()
    }

    func adjustSizeOfTableHeaderView() {
        self.tableHeaderView.layoutIfNeeded()
        let size = self.tableHeaderView.systemLayoutSizeFitting(UILayoutFittingCompressedSize)
        self.tableView.tableHeaderView?.frame.size = size
        self.tableView.tableHeaderView = self.tableHeaderView
    }

    func adjustSizeOfTableFooterView() {
        self.tableFooterView.layoutIfNeeded()
        let size = self.tableFooterView.systemLayoutSizeFitting(UILayoutFittingCompressedSize)
        self.tableView.tableFooterView?.frame.size = size
        self.tableView.tableFooterView = self.tableFooterView
    }
}