swift.swift

extension UIViewController {
  func resizeHeaderView(tableView:UITableView) {
    guard let headerView = tableView.tableHeaderView else {
      return
    }

    headerView.setNeedsLayout()
    headerView.layoutIfNeeded()
    tableView.sectionHeaderHeight = UITableView.automaticDimension


    let size = headerView.systemLayoutSizeFitting(UIView.layoutFittingCompressedSize)
    if headerView.frame.size.height != size.height {
      headerView.frame.size.height = size.height

      tableView.tableHeaderView = headerView
    }
  }
}