TestApp

MenuTableViewCellModel.swift

//
//  MenuTableViewCellModel.swift
//  TestApp
//
//  Created by Jeff Kereakoglow on 2/26/18.
//  Copyright © 2018 Alexis Digital. All rights reserved.
//

import Foundation
import UIKit.UIViewController

struct MenuTableViewCellModel {
    let title: String
    let viewController: () -> UIViewController
}

MenuTableViewController.swift

//
//  MenuTableViewController.swift
//  TestApp
//
//  Created by Jeff Kereakoglow on 2/26/18.
//  Copyright © 2018 Alexis Digital. All rights reserved.
//

import UIKit

final class MenuTableViewController: UITableViewController {
    var viewModel: [MenuTableViewCellModel]?

    private let cellReuseIdentifier = "TableViewCell"

    init() {
        super.init(style: .grouped)
    }

    required init?(coder aDecoder: NSCoder) {
        super.init(coder: aDecoder)
    }

    override func viewDidLoad() {
        super.viewDidLoad()

        title = "Test App"

        tableView.register(UITableViewCell.self, forCellReuseIdentifier: cellReuseIdentifier)
    }
}

// MARK: - UITableViewDataSource
extension MenuTableViewController {
    override func tableView(_ tableView: UITableView, titleForHeaderInSection section: Int)
        -> String? {
            return "Screens"
    }

    override func numberOfSections(in tableView: UITableView) -> Int {
        return viewModel == nil ? 0 : 1
    }

    override func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        return viewModel?.count ?? 0
    }

    override func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath)
        -> UITableViewCell {
            let cell = tableView.dequeueReusableCell(
                withIdentifier: cellReuseIdentifier, for: indexPath
            )

            cell.textLabel?.text = viewModel?[indexPath.row].title ?? ""

            return cell
    }
}

// MARK: - UITableViewDelegate
extension MenuTableViewController {
    override func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) ->
        CGFloat {
            return 56
    }

    override func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
        // Invoke the handler for the selected cell
        guard let viewController = viewModel?[indexPath.row].viewController else { return }

        show(viewController(), sender: self)
    }
}

MenuTableViewModelBuilder.swift

//
//  MenuTableViewModelBuilder.swift
//  TestApp
//
//  Created by Jeff Kereakoglow on 2/26/18.
//  Copyright © 2018 Alexis Digital. All rights reserved.
//

@testable import Core
final class MenuTableViewModelBuilder {
    var viewModel: [MenuTableViewCellModel] {
        return [
            MenuTableViewCellModel(title: "Edit Message Screen") {
                return EditMessageViewController()
            },
            MenuTableViewCellModel(title: "Welcome Screen") {
                return WelcomeViewController()
            },
            MenuTableViewCellModel(title: "Log In Screen") {
                return LogInViewController()
            }
        ]
    }
}