如何使用swift展开和折叠UICollectionViewCell？

class CustomCollectionViewCell: UICollectionViewCell {
    var isExpanded: Bool = false
    
    // 添加展开和折叠的逻辑
}

class ViewController: UIViewController, UICollectionViewDataSource, UICollectionViewDelegateFlowLayout {
    // 创建UICollectionView，并设置数据源和代理
    
    func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
        let cell = collectionView.cellForItem(at: indexPath) as? CustomCollectionViewCell
        
        if cell?.isExpanded == true {
            // 返回展开状态下的cell高度
        } else {
            // 返回折叠状态下的cell高度
        }
    }
    
    func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
        let cell = collectionView.cellForItem(at: indexPath) as? CustomCollectionViewCell
        
        // 修改cell的展开状态属性
        // 调用collectionView的reloadItems(at:)方法刷新布局
    }
}