From 5eb6b8c2b5c106d55c0307740806adc51754050e Mon Sep 17 00:00:00 2001
From: jzitnik-dev <email@jzitnik.dev>
Date: Tue, 27 May 2025 12:40:09 +0200
Subject: [PATCH] fix: Selling items

---
 src/main/java/cz/jull/tui/Menu.java | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/main/java/cz/jull/tui/Menu.java b/src/main/java/cz/jull/tui/Menu.java
index 15bf13f..be3e666 100644
--- a/src/main/java/cz/jull/tui/Menu.java
+++ b/src/main/java/cz/jull/tui/Menu.java
@@ -63,7 +63,7 @@ public class Menu {
                 List<Item> buildings = game.getBuildings();
                 Map<Item, Integer> grouped = group(buildings);
                 List<Item> keys = grouped.keySet().stream().toList();
-                List<String> mapped = new ArrayList<>(grouped.keySet().stream().map(item -> grouped.get(item) + "x " + item).toList());
+                List<String> mapped = new ArrayList<>(keys.stream().map(item -> grouped.get(item) + "x " + item).toList());
                 mapped.add("Go back");
 
                 int itemIndex = Cli.selectOptionIndex(mapped);
@@ -101,11 +101,18 @@ public class Menu {
 
                         List<Item> inventory = game.getPlayer().getInventory();
                         Map<Item, Integer> grouped = group(inventory);
-                        List<String> mapped = new ArrayList<>(grouped.keySet().stream().map(item -> grouped.get(item) + "x " + item).toList());
+                        List<Item> keys = grouped.keySet().stream().toList();
+                        List<String> mapped = new ArrayList<>(keys.stream().map(item -> grouped.get(item) + "x " + item).toList());
                         mapped.add("Go back");
 
                         int itemIndex = Cli.selectOptionIndex(mapped);
-                        Item item = game.getMarket().getAvailableItems().get(itemIndex);
+
+                        if (itemIndex == mapped.size() - 1) {
+                            start(day);
+                            return;
+                        }
+
+                        Item item = keys.get(itemIndex);
                         game.getMarket().sellItem(game, item);
                     }
                     case 2 -> start(day);