#include "a_train.h"

int main() {
    struct car* train = NULL;  // Начало списка
    char buffer[SIZE];
    char numbuf[SIZE];
    int passengers;

    printf("Zadajte zoznam cieľových staníc a počet cestujúcich.\n");
    printf("Zoznam zakončite prázdnym riadkom.\n");

    while (1) {
        // Načítame názov stanice
        if (!fgets(buffer, SIZE, stdin)) break;
        if (buffer[0] == '\n') break;  // prázdny riadok → koniec

        buffer[strcspn(buffer, "\n")] = 0;  // odstránenie '\n'

        // Načítame počet cestujúcich
        if (!fgets(numbuf, SIZE, stdin)) break;
        passengers = atoi(numbuf);  // používame atoi (údaj sa aj tak nepoužíva)

        // Pridáme vozeň
        train = add_car(train, buffer);
    }

    // Zistenie stanice, ktorú treba odstrániť
    printf("\nZadajte stanicu, ktorá sa má vyradiť:\n");
    if (fgets(buffer, SIZE, stdin)) {
        buffer[strcspn(buffer, "\n")] = 0;
    }

    // Odstránenie
    train = clear_train(train, buffer);

    // Výsledok
    printf("\nVýsledný vlak bez stanice %s bude:\n", buffer);
    print_train(train);

    // Uvoľnenie pamäte
    cancel_train(train);

    return 0;
}