MyCollectionViewController #1

MyCollectionViewController1.swift

import UIKit

struct CollectionItemModel {
    let title: String?
    let image: UIImage?
}

/// First set `model` and then present on screen.
final class MyCollectionViewController: UICollectionViewController {
    var models: [CollectionItemModel]?
    override func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
        let model = models?[indexPath.row]
        if let image = model?.image {
            let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "image", for: indexPath)
            cell.largeContentImage = image
            return cell
        } else if let title = model?.title {
            let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "text", for: indexPath)
            cell.largeContentTitle = title
            return cell
        }
        return UICollectionViewCell()
    }
}


func createMyCollectionViewController() -> MyCollectionViewController {
    let viewController = MyCollectionViewController()
    let models = [
        CollectionItemModel(title: nil, image: UIImage(named: "what")),
        CollectionItemModel(title: "Text", image: nil)
    ]
    viewController.models = models
    return viewController
}