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

int main() {
    struct station* s = create_station();

    add_target_capacity(s, "Kosice", 120);
    add_target_capacity(s, "Presov", 60);
    add_target_capacity(s, "Kosice", 30);
    add_target_capacity(s, "Bratislava", 200);
    add_target_capacity(s, "Zilina", 90);

    printf("Počet staníc: %d\n", count_targets(s));
    printf("Celkova kapacita: %d\n", count_capacity(s));

    printf("Presov ma kapacitu: %d\n", get_target_capacity(s, "Presov"));
    printf("Kosice ma kapacitu: %d\n", get_target_capacity(s, "Kosice"));
    printf("Bratislava ma kapacitu: %d\n", get_target_capacity(s, "Bratislava"));
    printf("Zvolen ma kapacitu: %d\n", get_target_capacity(s, "Zvolen"));

    destroy_station(s);
    return 0;
}