From 91fdd02ecba536a3a64d99821524f34596c3d708 Mon Sep 17 00:00:00 2001
From: Justin <52604018+hiimjustin000@users.noreply.github.com>
Date: Tue, 1 Oct 2024 14:16:20 -0400
Subject: [PATCH] gauntletnode id fixes (#105)

* gauntletnode id fixes

* awesome
---
 src/GauntletNode.cpp | 70 ++++++++++++++++++++++++++------------------
 1 file changed, 41 insertions(+), 29 deletions(-)

diff --git a/src/GauntletNode.cpp b/src/GauntletNode.cpp
index 75d7be3..43f5403 100644
--- a/src/GauntletNode.cpp
+++ b/src/GauntletNode.cpp
@@ -3,6 +3,8 @@
 #include <Geode/modify/GauntletNode.hpp>
 #include <Geode/utils/NodeIDs.hpp>
 
+#include "IDCheck.hpp"
+
 using namespace geode::prelude;
 using namespace geode::node_ids;
 
@@ -37,7 +39,7 @@ struct GauntletNodeIDs : Modify<GauntletNodeIDs, GauntletNode> {
             this,
             0,
             "gauntlet-info-node",
-            "empty-node",
+            "reward-node",
             "background",
             "gauntlet-name-label",
             "gauntlet-label",
@@ -51,7 +53,7 @@ struct GauntletNodeIDs : Modify<GauntletNodeIDs, GauntletNode> {
             this,
             0,
             "gauntlet-info-node",
-            "empty-node",
+            "reward-node",
             "background",
             "gauntlet-name-label",
             "gauntlet-label",
@@ -60,32 +62,42 @@ struct GauntletNodeIDs : Modify<GauntletNodeIDs, GauntletNode> {
         );
     }
 
-    if (auto gauntletInfoNode = this->getChildByID("gauntlet-info-node")) {
-        if (
-            self->m_fields->m_gauntlet->hasCompletedMapPack() &&
-            GSM->isSpecialChestUnlocked(GSM->getGauntletRewardKey(self->m_fields->m_gauntlet->m_packID))
-        ) {
-            setIDs(
-                gauntletInfoNode,
-                0,
-                "gauntlet-sprite",
-                "gauntlet-shadow-sprite",
-                "gauntlet-progress-label",
-                "gauntlet-progress-shadow-label"
-            );
-        } else {
-            setIDs(
-                gauntletInfoNode,
-                0,
-                "gauntlet-sprite",
-                "gauntlet-shadow-sprite",
-                "gauntlet-progress-label",
-                "gauntlet-progress-shadow-label",
-                "reward-label",
-                "reward-shadow-label",
-                "chest-sprite",
-                "chest-shadow-sprite"
-            );
-        }
+    if (
+        self->m_fields->m_gauntlet->hasCompletedMapPack() &&
+        GSM->isSpecialChestUnlocked(GSM->getGauntletRewardKey(self->m_fields->m_gauntlet->m_packID))
+    ) {
+        setIDs(
+            m_gauntletInfoNode,
+            0,
+            "gauntlet-sprite",
+            "gauntlet-shadow-sprite",
+            "gauntlet-progress-label",
+            "gauntlet-progress-shadow-label"
+        );
+    } else {
+        setIDs(
+            m_gauntletInfoNode,
+            0,
+            "gauntlet-sprite",
+            "gauntlet-shadow-sprite",
+            "gauntlet-progress-label",
+            "gauntlet-progress-shadow-label",
+            "reward-label",
+            "reward-shadow-label",
+            "chest-sprite",
+            "chest-shadow-sprite"
+        );
+
+        setIDs(
+            m_rewardNode,
+            0,
+            "reward-label",
+            "reward-shadow-label",
+            "chest-sprite",
+            "chest-shadow-sprite"
+        );
     }
+
+    verifyIDSafe(m_gauntletInfoNode, "gauntlet-info-node");
+    verifyIDSafe(m_rewardNode, "reward-node");
 }