Ampel.swift

//
//  Ampel.swift
//  Ampel
//
//  Created by Vincent Friedrich on 19.03.19.
//  Copyright © 2019 neoxapps. All rights reserved.
//

import Foundation

enum AmpelColor {
    case red, yellow, green
}

protocol AmpelDelegate {
    func ampelColorDidChange(to color: AmpelColor)
}

class Ampel {
    
    var delegate: AmpelDelegate?
    var color: AmpelColor = .red
    
    func next() {
        switch color {
        case .red:
            self.color = .yellow
        case .yellow:
            self.color = .green
        case .green:
            self.color = .red
        }
        
        delegate?.ampelColorDidChange(to: self.color)
    }
}