Fix crash with Platformer Saves, closes #3

Signed-off-by: Kevadroz <[email protected]>

Author: Kevadroz

src/Hooks/PlayLayer.cpp

@@ -43,6 +43,9 @@ bool ModPlayLayer::init(
 void ModPlayLayer::setupHasCompleted() {
 	PlayLayer::setupHasCompleted();
 
+	if (m_progressBar == nullptr)
+		return;
+
 	m_fields->m_pbCheckpointContainer = CCNodeRGBA::create();
 	m_fields->m_pbCheckpointContainer->setPosition(
 		ccp(0.f, m_progressBar->getContentHeight() / 2.f)
@@ -93,11 +96,6 @@ void ModPlayLayer::resetLevel() {
 
 	if (checkpoint != nullptr)
 		m_checkpointArray->removeObject(checkpoint->m_checkpoint);
-
-	// union {
-	// 	float f;
-	// 	uint32_t u;
-	// } percent = {.f = getCurrentPercent()};
 }
 
 void ModPlayLayer::loadFromCheckpoint(CheckpointObject* checkpoint) {
@@ -213,11 +211,11 @@ void ModPlayLayer::registerKeybindListeners() {
 }
 
 void ModPlayLayer::updateModUI() {
+	static_cast<ModUILayer*>(m_uiLayer)->updateSwitcher();
+	
 	if (m_fields->m_pbCheckpointContainer == nullptr)
 		return;
 
-	static_cast<ModUILayer*>(m_uiLayer)->updateSwitcher();
-
 	if (m_isPlatformer)
 		return;
 

src/Hooks/PlayLayer.hpp

@@ -4,8 +4,8 @@
 
 #include <Geode/modify/PlayLayer.hpp>
 #include <functional>
-#include <variant>
 #include <optional>
+#include <variant>
 #ifndef GEODE_IS_IOS
 #include <geode.custom-keybinds/include/Keybinds.hpp>
 #endif
@@ -78,7 +78,6 @@ class $modify(ModPlayLayer, PlayLayer) {
 
 	void togglePracticeMode(bool enabled);
 
-
 	// Custom
 	void registerKeybindListeners();
 	void updateModUI();
@@ -111,4 +110,9 @@ class $modify(ModPlayLayer, PlayLayer) {
 	void removeCurrentSaveLayer();
 	void swapSaveLayers(unsigned int left, unsigned int right);
 	void updateSaveLayerCount();
+
+	static void onModify(auto& self) {
+		if (!self.setHookPriorityPost("PlayLayer::setupHasCompleted", Priority::VeryLate))
+			log::warn("Failed to set PlayLayer::setupHasCompleted hook priority!");
+	}
 };