Работа с Footer и Header в UICollectionView (Work with CollectionView footer, header) - Swift 5

var footerHide = false
// Расширение, так удобнее
extension CategoriesCollectionViewController: UICollectionViewDelegateFlowLayout {

// Процедура для работы с Header и Footer View
override func collectionView(_ collectionView: UICollectionView, viewForSupplementaryElementOfKind kind: String, at indexPath: IndexPath) -> UICollectionReusableView {

// Работа с Footer
if (kind == UICollectionView.elementKindSectionFooter) {
let footerView = collectionView.dequeueReusableSupplementaryView(ofKind: kind, withReuseIdentifier: "CartFooterCollectionReusableView", for: indexPath)

return footerView

// Работа с Header
} else if (kind == UICollectionView.elementKindSectionHeader) {
let headerView = collectionView.dequeueReusableSupplementaryView(ofKind: kind, withReuseIdentifier: "CartHeaderCollectionReusableView", for: indexPath)

return headerView
fatalError()
}
// SizeForFooter - Работа с размером Footer
func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, referenceSizeForFooterInSection section: Int) -> CGSize {
if footerHide == true {
return CGSize.zero
}
return CGSize(width: collectionView.frame.width, height: 55)
}