#include "a_train.h"

int main() {
    struct car* train = NULL;  // Začiatok spojkového zoznamu (vlak)
    char buffer[SIZE];
    int passengers;

    // Načítanie vstupných údajov od používateľa
    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 (údaj sa ďalej nepoužíva)
        scanf("%d", &passengers);
        getchar();  // Odstráni zvyšný znak '\n'

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

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

    // Odstránenie všetkých vozňov so zadanou stanicou
    train = clear_train(train, buffer);

    // Výpis výsledného vlaku
    printf("\nVýsledný vlak bez stanice %s bude:\n", buffer);
    print_train(train);

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

    return 0;
}