From 31e56ae17fdecdcbbede3e8960354dc730e5b76c Mon Sep 17 00:00:00 2001
From: Melledy <121644117+Melledy@users.noreply.github.com>
Date: Sat, 13 Dec 2025 00:33:27 -0800
Subject: [PATCH] Fix potential count when giving up a Monolith run

---
 src/main/java/emu/nebula/game/tower/StarTowerGame.java        | 4 ++++
 .../emu/nebula/server/handlers/HandlerStarTowerGiveUpReq.java | 1 +
 2 files changed, 5 insertions(+)

diff --git a/src/main/java/emu/nebula/game/tower/StarTowerGame.java b/src/main/java/emu/nebula/game/tower/StarTowerGame.java
index c7589e1..3a40616 100644
--- a/src/main/java/emu/nebula/game/tower/StarTowerGame.java
+++ b/src/main/java/emu/nebula/game/tower/StarTowerGame.java
@@ -215,6 +215,10 @@ public class StarTowerGame {
         return this.getRarePotentialCount().get(charId);
     }
     
+    public int getTotalPotentialCount() {
+        return this.getItems().values().intStream().reduce(0, Integer::sum);
+    }
+    
     /**
      * Gets the team element, if the team has 2+ or more elements, then returns null
      */
diff --git a/src/main/java/emu/nebula/server/handlers/HandlerStarTowerGiveUpReq.java b/src/main/java/emu/nebula/server/handlers/HandlerStarTowerGiveUpReq.java
index 44c48dd..cd003fd 100644
--- a/src/main/java/emu/nebula/server/handlers/HandlerStarTowerGiveUpReq.java
+++ b/src/main/java/emu/nebula/server/handlers/HandlerStarTowerGiveUpReq.java
@@ -20,6 +20,7 @@ public class HandlerStarTowerGiveUpReq extends NetHandler {
         // Build response
         var rsp = StarTowerGiveUpResp.newInstance()
                 .setBuild(game.getBuild().toProto())
+                .setPotentialCnt(game.getTotalPotentialCount())
                 .setFloor(game.getFloorCount());
         
         rsp.getMutableChange();