#include "a_train.h"

int main(void) {
    struct car* train = NULL;
    char buffer[SIZE];
    char numbuf[SIZE];

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

    while (1) {
        // Názov stanice
        if (!fgets(buffer, SIZE, stdin)) break;
        if (buffer[0] == '\n') break;
        buffer[strcspn(buffer, "\n")] = '\0';

        // Počet cestujúcich (iba pre čítanie, neukladá sa)
        if (!fgets(numbuf, SIZE, stdin)) break;

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

    printf("\nZadajte stanicu, ktorá sa má vyradiť:\n");
    if (fgets(buffer, SIZE, stdin)) {
        buffer[strcspn(buffer, "\n")] = '\0';
        train = clear_train(train, buffer);
        printf("\nVýsledný vlak bez stanice %s bude:\n", buffer);
    }

    print_train(train);
    cancel_train(train);
    return 0;
}