sizeHeaderFooterToFit

sizeHeaderFooterToFit.swift

private func sizeHeaderFooterToFit() {
      if let _ = self.tableView.tableHeaderView {
          let headerView = self.tableView.tableHeaderView!
          headerView.setNeedsLayout()
          headerView.layoutIfNeeded()
          let heightHeader = headerView.systemLayoutSizeFittingSize(UILayoutFittingCompressedSize).height
          var frameHeader = headerView.frame
          frameHeader.size.height = heightHeader
          headerView.frame = frameHeader
          self.tableView.tableHeaderView = headerView
      }

      if let _ = self.tableView.tableFooterView {
          let footerView = self.tableView.tableFooterView!
          footerView.setNeedsLayout()
          footerView.layoutIfNeeded()
          let heightFooter = footerView.systemLayoutSizeFittingSize(UILayoutFittingCompressedSize).height
          var frameFooter = footerView.frame
          frameFooter.size.height = heightFooter
          footerView.frame = frameFooter
          self.tableView.tableFooterView = footerView
      }
  }