Add back to editor button

OpenRA · Sep 9, 2022 · b4e39cd · b4e39cd
1 parent 45835e4
commit b4e39cd
Show file tree

Hide file tree

Showing 5 changed files with 58 additions and 3 deletions.
diff --git a/OpenRA.Mods.Common/Widgets/Logic/Ingame/IngameMenuLogic.cs b/OpenRA.Mods.Common/Widgets/Logic/Ingame/IngameMenuLogic.cs
@@ -37,6 +37,8 @@ public class IngameMenuLogic : ChromeLogic
 		bool leaving;
 		bool hideMenu;
 
+		static bool lastGameEditor = false;
+
 		[TranslationReference]
 		static readonly string Leave = "leave";
 
@@ -145,6 +147,18 @@ public class IngameMenuLogic : ChromeLogic
 		[TranslationReference]
 		static readonly string PlayMapWarningCancel = "play-map-warning-cancel";
 
+		[TranslationReference]
+		static readonly string ExitToMapEditorTitle = "exit-to-map-editor-title";
+
+		[TranslationReference]
+		static readonly string ExitToMapEditorPrompt = "exit-to-map-editor-prompt";
+
+		[TranslationReference]
+		static readonly string ExitToMapEditorAccept = "exit-to-map-editor-accept";
+
+		[TranslationReference]
+		static readonly string ExitToMapEditorCancel = "exit-to-map-editor-cancel";
+
 		[ObjectCreator.UseCtor]
 		public IngameMenuLogic(Widget widget, ModData modData, World world, Action onExit, WorldRenderer worldRenderer,
 			IngameInfoPanel initialPanel, Dictionary<string, MiniYaml> logicArgs)
@@ -157,6 +171,7 @@ public class IngameMenuLogic : ChromeLogic
 			var buttonHandlers = new Dictionary<string, Action>
 			{
 				{ "ABORT_MISSION", CreateAbortMissionButton },
+				{ "BACK_TO_EDITOR", CreateBackToEditorButton },
 				{ "RESTART", CreateRestartButton },
 				{ "SURRENDER", CreateSurrenderButton },
 				{ "LOAD_GAME", CreateLoadGameButton },
@@ -248,6 +263,7 @@ void OnQuit()
 				exitDelay += 40 * mpe.Info.FadeLength;
 			}
 
+			lastGameEditor = false;
 			Game.RunAfterDelay(exitDelay, () =>
 			{
 				if (!Game.IsCurrentWorld(world))
@@ -518,6 +534,8 @@ void CreatePlayMapButton()
 
 					ExitEditor(actionManager, () =>
 					{
+						lastGameEditor = true;
+
 						Ui.CloseWindow();
 						Ui.ResetTooltips();
 						if (map.Visibility == MapVisibility.Lobby)
@@ -545,6 +563,38 @@ void CreatePlayMapButton()
 				};
 		}
 
+		void CreateBackToEditorButton()
+		{
+			if (world.Type != WorldType.Regular || !lastGameEditor)
+				return;
+
+			AddButton("BACK_TO_EDITOR", "Back To Editor")
+				.OnClick = () =>
+				{
+					hideMenu = true;
+					Action onConfirm = () =>
+					{
+						lastGameEditor = false;
+						var map = modData.MapCache.GetUpdatedMap(world.Map.Uid);
+						if (map == null)
+							Game.LoadShellMap();
+						else
+						{
+							DiscordService.UpdateStatus(DiscordState.InMapEditor);
+							Game.LoadEditor(map);
+						}
+					};
+
+					ConfirmationDialogs.ButtonPrompt(modData,
+						title: ExitToMapEditorTitle,
+						text: ExitToMapEditorPrompt,
+						onConfirm: onConfirm,
+						onCancel: ShowMenu,
+						confirmText: ExitToMapEditorAccept,
+						cancelText: ExitToMapEditorCancel);
+				};
+		}
+
 		void CreateExitEditorButton()
 		{
 			if (world.Type != WorldType.Editor)

diff --git a/mods/cnc/chrome/ingame-menu.yaml b/mods/cnc/chrome/ingame-menu.yaml
@@ -2,7 +2,7 @@ Container@INGAME_MENU:
 	Width: WINDOW_RIGHT
 	Height: WINDOW_BOTTOM
 	Logic: IngameMenuLogic
-		Buttons: EXIT_EDITOR, PLAY_MAP, SAVE_MAP, ABORT_MISSION, SURRENDER, RESTART, LOAD_GAME, SAVE_GAME, MUSIC, SETTINGS, RESUME
+		Buttons: EXIT_EDITOR, PLAY_MAP, SAVE_MAP, ABORT_MISSION, BACK_TO_EDITOR, SURRENDER, RESTART, LOAD_GAME, SAVE_GAME, MUSIC, SETTINGS, RESUME
 		ButtonStride: 130, 0
 	Children:
 		Image@EVA:

diff --git a/mods/common/chrome/ingame-menu.yaml b/mods/common/chrome/ingame-menu.yaml
@@ -2,7 +2,7 @@ Container@INGAME_MENU:
 	Width: WINDOW_RIGHT
 	Height: WINDOW_BOTTOM
 	Logic: IngameMenuLogic
-		Buttons: RESUME, LOAD_GAME, SAVE_GAME, SETTINGS, MUSIC, SURRENDER, RESTART, ABORT_MISSION, SAVE_MAP, PLAY_MAP, EXIT_EDITOR
+		Buttons: RESUME, LOAD_GAME, SAVE_GAME, SETTINGS, MUSIC, SURRENDER, RESTART, BACK_TO_EDITOR, ABORT_MISSION, SAVE_MAP, PLAY_MAP, EXIT_EDITOR
 		ButtonStride: 0, 40
 	Children:
 		Background@BORDER:

diff --git a/mods/common/languages/en.ftl b/mods/common/languages/en.ftl
@@ -206,6 +206,11 @@ play-map-warning-prompt = The map may have been deleted or has
 play-map-warning-accept = Okay
 play-map-warning-cancel = Okay
 
+exit-to-map-editor-title = Leave Mission
+exit-to-map-editor-prompt = Leave this game and return to the editor?
+exit-to-map-editor-accept = Back To Editor
+exit-to-map-editor-cancel = Stay
+
 ## IngamePowerBarLogic
 ## IngamePowerCounterLogic
 power-usage = Power Usage

diff --git a/mods/d2k/chrome/ingame-menu.yaml b/mods/d2k/chrome/ingame-menu.yaml
@@ -2,7 +2,7 @@ Container@INGAME_MENU:
 	Width: WINDOW_RIGHT
 	Height: WINDOW_BOTTOM
 	Logic: IngameMenuLogic
-		Buttons: RESUME, LOAD_GAME, SAVE_GAME, SETTINGS, MUSIC, SURRENDER, RESTART, ABORT_MISSION, SAVE_MAP, PLAY_MAP, EXIT_EDITOR
+		Buttons: RESUME, LOAD_GAME, SAVE_GAME, SETTINGS, MUSIC, SURRENDER, RESTART, BACK_TO_EDITOR, ABORT_MISSION, SAVE_MAP, PLAY_MAP, EXIT_EDITOR
 		ButtonStride: 0, 40
 	Children:
 		Label@VERSION_LABEL: