UITableView+UITableHeaderFooterView.swift

import UIKit

extension UITableView {

     /// Layout `UITableView.headerView` with autolayout.
    func layoutTableHeaderView() {
        guard let headerView = tableHeaderView else { return }
        layout(headerView)
        self.tableHeaderView = headerView
    }

    /// Layout `UITableView.footerView` with autolayout.
    func layoutTableFooterView() {
        guard let footerView = tableFooterView else { return }
        layout(footerView)
        self.tableFooterView = footerView
    }

    private func layout(_ view: UIView) {
        view.translatesAutoresizingMaskIntoConstraints = false

        let width = view.bounds.size.width
        let temporaryWidthConstraint = view.widthAnchor.constraint(equalToConstant: width)

        view.addConstraint(temporaryWidthConstraint)

        view.setNeedsLayout()
        view.layoutIfNeeded()

        let footerSize = view.systemLayoutSizeFitting(UIView.layoutFittingCompressedSize)
        let height = footerSize.height
        var frame = view.frame

        frame.size.height = height
        view.frame = frame

        view.removeConstraint(temporaryWidthConstraint)
        view.translatesAutoresizingMaskIntoConstraints = true
    }
}