DashboardCollection.swift

import UIKit

final class DashboardCollection: UICollectionView {

    var didSelectIndicator: ((Behavior) -> Void)?
    
    fileprivate var customDatasource: DashboardCollectionDataSource?
    fileprivate var customDelegate: DashboardCollectionDelegate?
    
    convenience init() {
        self.init(frame: .zero, collectionViewLayout: UICollectionViewFlowLayout())
    }
    
    override init(frame: CGRect, collectionViewLayout layout: UICollectionViewLayout) {
        super.init(frame: frame, collectionViewLayout: layout)
        customDelegate = DashboardCollectionDelegate(self)
        customDatasource = DashboardCollectionDataSource(items: [], collectionView: self, delegate: customDelegate!)
        self.backgroundColor = UIColor.white
        
    }
    
    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
}

extension DashboardCollection {
    func updateItems(_ items: [Tray]) {
        customDatasource?.updateItems(items)
    }
}


extension DashboardCollection: DashboardDelegate {
    func didSelectIndicator(at index: IndexPath) {
        if let indicator = customDatasource?.items[index.section].behaviors?[index.row] {
            didSelectIndicator?(indicator)
        }
    }
}