#include <stdio.h>
#include <stdlib.h>
#include "a_station.h"
#include <string.h>

int main() {
    // Vytvorenie stanice
    struct station* moja_stanica = create_station();
    if (moja_stanica == NULL) {
        printf("Chyba pri vytváraní stanice.\n");
        return 1;
    }

    // Načítanie údajov zo súboru
    load_from_file(moja_stanica);

    // Cyklus pre zadávanie nových záznamov
    char target[TARGET_SIZE];
    int capacity;
    while (1) {
        printf("Zadajte názov stanice (alebo 'exit' pre ukončenie): ");
        scanf("%s", target);
        if (strcmp(target, "exit") == 0) {
            break; // Ukončiť cyklus
        }

        printf("Zadajte počet cestujúcich: ");
        scanf("%d", &capacity);
        add_target_capacity(moja_stanica, target, capacity);
    }

    // Výstup informácií o staniciach
    printf("Uložené údaje:\n");
    for (int i = 0; i < moja_stanica->track_count; i++) {
        struct car* current = moja_stanica->tracks[i];
        while (current != NULL) {
            printf("Stanica: %s, Cestujúcich: %d\n", current->value, current->capacity);
            current = current->next;
        }
    }

    // Uloženie údajov do súboru pred ukončením
    save_to_file(moja_stanica);

    // Uvoľnenie pamäte
    destroy_station(moja_stanica);

    return 0;
}