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

struct station* create_station(){
   struct station* station = (struct station*)calloc(1,sizeof(struct station));
   station->tracks = (struct car**)calloc(STATION_SIZE, sizeof(struct car*));
   station->track_count = STATION_SIZE;
   return station;
}

void destroy_station(struct station* station){

    for (int i = 0; i < station -> track_count; i++){
        struct car* temp = station -> tracks[i]; //docasne ulozim strukturu, nech mam zalohu smerniku
        while (temp != NULL) {
            struct car* temp2 = temp; // prehadzujem smernik aby som nevymazal to na com stojim
            temp = temp -> next;
            free (temp2); //mazem
        }
    }
    free(station); //zmazem na koniec to co som na zaciatku dostal

}

int select_track(struct station* station, const char* target){
    int hash = 0;
    while (*target) {
        hash = hash + *target; // vyratam nejake cislo "kolaje" ktore by mohlo byt unikatne pre kazdu lokaciu
        target++;
    }
    return hash % (station -> track_count);
}

void add_target_capacity(struct station* station,const char* target, int capacity){
    int track = select_track(station, target); // zistim ktora kolaj by to mala byt
    struct car* kolaj = station -> tracks[track]; // "loadnem" si kolaj aby som s nou vedel pracovat
    while (kolaj) { // hladam ci to existuje a ked najdem tak pridam kapacitu kym stojim na kolaji
        if (strcmp (kolaj -> value, target) == 0) {
            kolaj -> capacity = kolaj -> capacity + capacity;// ak najdem zhodu (strcmp == 0) tak pridam kapacitu (napojim novy vagon)
            return;
        }
        kolaj = kolaj -> next; // idem pozerat vedlajsiu kolaj
    }
    //vytvaram novy vagon
    struct car* vagon = (struct car*) malloc (sizeof (struct car)); // vyhradim nove miesto v pamati pre novy vagon,
                                                                    // inac by sa mi nezapisala nova premenna z lokalnej funkcie
    strncpy(vagon -> value, target, TARGET_SIZE - 1);
    vagon -> value[TARGET_SIZE - 1] = '\0'; //ukoncim string
    vagon -> capacity = capacity; //zadam kapacitu
    vagon -> next = station -> tracks[track]; // nastavim nasledovnika, zaclenim do databazy
    station -> tracks[track] = vagon; // pridam vagon na kolaj

}

int get_target_capacity(struct station* station,const char* target){
    return 0;
}

int count_targets(struct station* station){
    return 0;
}

int count_capacity(struct station* station){
    int capacity = 0;

    for (int i = 0; i < station -> track_count; i++;){
        struct car* vagon = station ->tracks[i]; // kontrolujem kazdu kolaj, kazdy vagon linked hash tablu
        while (vagon) {
            capacity = capacity + vagon -> capacity;
            vagon = vagon -> next; // zapisem kapacitu kazdeho vagonu na kolaji
        }
    }
    return capacity;
}