Autosize UITableView header and footer views

UITableView.swift

extension UITableView {
    func autoSizeHeaderAndFooter() {
        let width = self.bounds.width
        let autoSize = { (view: UIView) in
            view.translatesAutoresizingMaskIntoConstraints = false

            let widthConstraint = NSLayoutConstraint(item: view,
                                                     attribute: NSLayoutAttribute.width,
                                                     relatedBy: NSLayoutRelation.equal,
                                                     toItem: nil,
                                                     attribute: NSLayoutAttribute.notAnAttribute,
                                                     multiplier: 1,
                                                     constant: width)

            view.addConstraint(widthConstraint)
            let height = view.systemLayoutSizeFitting(UILayoutFittingCompressedSize).height
            view.removeConstraint(widthConstraint)

            view.frame = CGRect(x: 0, y: 0, width: width, height: height)
            view.translatesAutoresizingMaskIntoConstraints = true
        }

        if let view = tableHeaderView {
            autoSize(view)
            tableHeaderView = view
        }

        if let view = tableFooterView {
            autoSize(view)
            tableFooterView = view
        }
    }
}