LsA

//
//  main.swift
//  test
//
//  Created by Vratislav Holub on 24/03/2019.
//  Copyright © 2019 Vratislav Holub. All rights reserved.
//

import Foundation

var pokracovat : Bool = true
var pokracovat2 : Bool = true

var jidlo = [String]()
var produkty = [String]()


while(pokracovat){
    print("Zadej produkt: ")
    let produkt : String = readLine()!
    pokracovat2=true
    while(pokracovat2){
        print("Jedná se o jídlo? a/n")
        let jidl : String = readLine()!

        if (jidl.lowercased() == "a"){
            jidlo.append(produkt)
            pokracovat2=false
        }

        else if (jidl.lowercased() == "n"){
            produkty.append(produkt)
            pokracovat2=false
        }

        else{
            print("Chybná volba, opakuj akci")
            pokracovat2=true
        }
    }

    pokracovat2=true

    while(pokracovat2){
        print("----------------------------\nPřeješ si zadat další produkt? a/n\n----------------------------")
        var volba : String = readLine()!
        volba = volba.lowercased()

        if (volba=="a"){
            pokracovat=true
            pokracovat2=false
        }
        else if (volba=="n"){
            pokracovat=false
            pokracovat2=false
        }
        else{
            print("Chybná volba, opakuj akci")
            pokracovat2 = true
        }
    }
}

pokracovat = true
while(pokracovat){
    print("\n1. Výpis všech produktů\n2. Výpis jídla\n3. Výpis nejedlých produktů")
    var volba : String = readLine()!
    volba = volba.lowercased()

    if (volba=="1"){
        for hodnota in produkty{
            print(hodnota, terminator: " ")
            pokracovat=false
        }
        for hodnota in jidlo{
            print(hodnota, terminator: " ")
            pokracovat=false
        }
    }
    else if (volba=="2"){
        for hodnota in jidlo{
            print(hodnota, terminator: " ")
            pokracovat=false
        }
    }
    else if (volba=="3"){
        for hodnota in produkty{
            print(hodnota, terminator: " ")
            pokracovat=false
        }
    }
    else{
        print("Špatná volba, opakuj akci")
    }
}