GitHawk/Classes/Section Controllers/SegmentedControl/SegmentedControlSectionController.swift

//
//  SegmentedControlSectionController.swift
//  Freetime
//
//  Created by Ryan Nystrom on 5/14/17.
//  Copyright © 2017 Ryan Nystrom. All rights reserved.
//

import UIKit
import IGListKit

protocol SegmentedControlSectionControllerDelegate: class {
    func didChangeSelection(sectionController: SegmentedControlSectionController, model: SegmentedControlModel)
}

final class SegmentedControlSectionController: ListGenericSectionController<SegmentedControlModel>, SegmentedControlCellDelegate {

    private weak var delegate: SegmentedControlSectionControllerDelegate? = nil
    private let height: CGFloat

    init(delegate: SegmentedControlSectionControllerDelegate?, height: CGFloat) {
        self.delegate = delegate
        self.height = height
        super.init()
    }

    override func sizeForItem(at index: Int) -> CGSize {
        guard let context = collectionContext else { fatalError("Collection context must be set") }
        // 28 is the default height of UISegmentedControl
        let height = 28 + 2*Styles.Sizes.rowSpacing
        return CGSize(width: context.containerSize.width, height: height)
    }

    override func cellForItem(at index: Int) -> UICollectionViewCell {
        guard let object = object,
            let cell = collectionContext?.dequeueReusableCell(of: SegmentedControlCell.self, for: self, at: index) as? SegmentedControlCell
            else { fatalError("Collection context must be set, missing object, or cell incorrect type") }
        cell.configure(items: object.items, selectedIndex: object.selectedIndex)
        cell.delegate = self
        return cell
    }

    // MARK: SegmentedControlCellDelegate

    func didChangeSelection(cell: SegmentedControlCell, index: Int) {
        guard let object = object else { return }
        object.selectedIndex = index
        delegate?.didChangeSelection(sectionController: self, model: object)
    }
    
}