fix: next bullet index out of bounds

1 files changed, 10 insertions, 2 deletions

diff --git a/logic/controller.lua b/logic/controller.lua
index 066564e..302d764 100644
--- a/logic/controller.lua
+++ b/logic/controller.lua
@@ -9,9 +9,17 @@ end
 
 local function playerChangeNextBullet(key)
     if key == "q" then
-        G.player.nextBullet = G.player.nextBullet -1
+        if G.player.nextBullet == 1 then
+            G.player.nextBullet = #G.player.hearts
+        else
+            G.player.nextBullet = G.player.nextBullet -1
+        end
     elseif key == "e" then
-        G.player.nextBullet = G.player.nextBullet +1
+        if G.player.nextBullet == #G.player.hearts then
+            G.player.nextBullet = 1
+        else
+            G.player.nextBullet = G.player.nextBullet +1
+        end
     end
 
     -- shortcut: press number key to move to a specific index