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

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

    // Pridanie nových záznamov
    add_target_capacity(moja_stanica, "Bratislava", 50);
    add_target_capacity(moja_stanica, "Košice", 30);
    add_target_capacity(moja_stanica, "Prešov", 20);

    // Získanie informácií o mestách
    printf("Počet cestujúcich do Bratislava: %d\n", get_target_capacity(moja_stanica, "Bratislava"));
    printf("Počet cestujúcich do Košice: %d\n", get_target_capacity(moja_stanica, "Košice"));
    printf("Počet cestujúcich do Prešov: %d\n", get_target_capacity(moja_stanica, "Prešov"));

    // Výpočet celkového počtu cieľových staníc a cestujúcich
    printf("Celkový počet cieľových staníc: %d\n", count_targets(moja_stanica));
    printf("Celkový počet cestujúcich: %d\n", count_capacity(moja_stanica));

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

    return 0;
}