From 4c0ea4afb211ef7eca0869eb0c3fcf9a55629c9e Mon Sep 17 00:00:00 2001
From: RandomityGuy <31925790+RandomityGuy@users.noreply.github.com>
Date: Thu, 16 Feb 2023 00:19:11 +0530
Subject: [PATCH] js load custom levels in replays

---
 src/gui/MainMenuGui.hx | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/src/gui/MainMenuGui.hx b/src/gui/MainMenuGui.hx
index 3cac97fa..e15fc268 100644
--- a/src/gui/MainMenuGui.hx
+++ b/src/gui/MainMenuGui.hx
@@ -8,6 +8,7 @@ import src.ResourceLoader;
 import src.Settings;
 import src.Util;
 import src.Replay;
+import src.Marbleland;
 
 class MainMenuGui extends GuiImage {
 	public function new() {
@@ -114,11 +115,17 @@ class MainMenuGui extends GuiImage {
 						#end
 						if (MissionList.missions == null)
 							MissionList.buildMissionList();
-						var playMis = MissionList.missions.get(repmis);
-						if (playMis != null) {
-							cast(this.parent, Canvas).marbleGame.watchMissionReplay(playMis, replay);
+						var mi = replay.customId == 0 ? MissionList.missions.get(repmis) : Marbleland.missions.get(replay.customId);
+						if (mi.isClaMission) {
+							mi.download(() -> {
+								MarbleGame.instance.watchMissionReplay(mi, replay);
+							});
 						} else {
-							cast(this.parent, Canvas).pushDialog(new MessageBoxOkDlg("Cannot load replay."));
+							if (mi != null) {
+								cast(this.parent, Canvas).marbleGame.watchMissionReplay(mi, replay);
+							} else {
+								cast(this.parent, Canvas).pushDialog(new MessageBoxOkDlg("Cannot load replay."));
+							}
 						}
 					}
 				});