From 08ae834f6c30f853cbac351ab90c9aa695cbb3e4 Mon Sep 17 00:00:00 2001
From: VIliam <viliam.dzubara@student.tuke.sk>
Date: Mon, 4 Nov 2024 14:01:58 +0100
Subject: [PATCH] cv6

---
 cv6/a_station.c | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/cv6/a_station.c b/cv6/a_station.c
index 1eff1af..b018ffa 100644
--- a/cv6/a_station.c
+++ b/cv6/a_station.c
@@ -23,6 +23,24 @@ int select_track(struct station* station, const char* target){
 
 
 void add_target_capacity(struct station* station,const char* target, int capacity){
+int index = select_track(station, target);
+    struct car* current = station->tracks[index];
+    
+    while (current) {
+        if (strcmp(current->value, target) == 0) {
+            current->capacity += capacity;  // Pridá kapacitu
+            return;
+        }
+        current = current->next;
+    }
+
+    // Ak ciel neexistuje, pridá novy zaznam do zoznamu
+    struct car* new_car = (struct car*)malloc(sizeof(struct car));
+    strncpy(new_car->value, target, TARGET_SIZE - 1);
+    new_car->value[TARGET_SIZE - 1] = '\0';
+    new_car->capacity = capacity;
+    new_car->next = station->tracks[index];
+    station->tracks[index] = new_car;
 }
 
 int get_target_capacity(struct station* station,const char* target){