Reconnect to last world automatically

Author: austinvaness

PluginLoader/LoaderTools.cs

@@ -21,13 +21,16 @@
 using VRage;
 using VRage.Audio;
 using Sandbox.Game.Gui;
+using System.Collections.Generic;
 
 namespace avaness.PluginLoader
 {
     public static class LoaderTools
     {
         public static string PluginsDir => Path.GetFullPath(Path.Combine(MyFileSystem.ExePath, "Plugins"));
 
+        public const string AutoRejoinArg = "-auto_join_world";
+
         public static DialogResult ShowMessageBox(string msg, MessageBoxButtons buttons = MessageBoxButtons.OK, MessageBoxIcon icon = MessageBoxIcon.None, MessageBoxDefaultButton defaultButton = MessageBoxDefaultButton.Button1)
         {
             if (Application.OpenForms.Count > 0)
@@ -58,9 +61,15 @@ public static DialogResult ShowMessageBox(string msg, MessageBoxButtons buttons
         public static void AskToRestart()
         {
             if (MyGuiScreenGamePlay.Static != null)
-                AskSave(delegate { UnloadAndRestart(); });
+                AskSave(delegate {
+                    Unload();
+                    Restart();
+                });
             else
-                UnloadAndRestart();
+            {
+                Unload();
+                Restart();
+            }
         }
 
         /// <summary>
@@ -127,21 +136,44 @@ void UnloadAndExitAfterScreenshotWasTaken(object sender, EventArgs e)
             }
         }
 
-
-        public static void UnloadAndRestart()
+        public static void Unload()
         {
             LogFile.Dispose();
             MySessionLoader.Unload();
             MySandboxGame.Config.ControllerDefaultOnStart = MyInput.Static.IsJoystickLastUsed;
             MySandboxGame.Config.Save();
             MyScreenManager.CloseAllScreensNowExcept(null);
             MyPlugins.Unload();
-            Restart();
         }
 
-        public static void Restart()
+        public static void Restart(bool autoRejoin = false)
+        {
+            Start(autoRejoin);
+            Process.GetCurrentProcess().Kill();
+        }
+
+        private static void Start(bool autoRejoin)
         {
-            Application.Restart();
+            // Regular app case 
+            StringBuilder sb = new StringBuilder();
+            IEnumerable<string> args = Environment.GetCommandLineArgs().Skip(1).Where(x => x != AutoRejoinArg);
+            if(autoRejoin)
+                args = args.Append(AutoRejoinArg);
+            foreach (string arg in args)
+            {
+                if (sb.Length > 0)
+                    sb.Append(' ');
+                sb.Append('"');
+                sb.Append(arg);
+                sb.Append('"');
+            }
+
+            ProcessStartInfo currentStartInfo = Process.GetCurrentProcess().StartInfo;
+            currentStartInfo.FileName = Application.ExecutablePath;
+            if (sb.Length > 0)
+                currentStartInfo.Arguments = sb.ToString();
+
+            Process.Start(currentStartInfo);
             Process.GetCurrentProcess().Kill();
         }
 

PluginLoader/Patch/Patch_CreateMenu.cs

@@ -6,6 +6,7 @@
 using VRage.Game;
 using VRage.Utils;
 using VRageMath;
+using System;
 
 // ReSharper disable InconsistentNaming
 
@@ -14,7 +15,9 @@ namespace avaness.PluginLoader.Patch
     [HarmonyPatch(typeof(MyGuiScreenMainMenu), "CreateMainMenu")]
     public static class Patch_CreateMainMenu
     {
-        public static void Postfix(MyGuiScreenMainMenu __instance, Vector2 leftButtonPositionOrigin, ref Vector2 lastButtonPosition)
+        private static bool usedAutoRejoin = false;
+
+        public static void Postfix(MyGuiScreenMainMenu __instance, Vector2 leftButtonPositionOrigin, ref Vector2 lastButtonPosition, MyGuiControlButton ___m_continueButton)
         {
             MyGuiControlButton lastBtn = null;
             foreach (var control in __instance.Controls)
@@ -45,6 +48,12 @@ public static void Postfix(MyGuiScreenMainMenu __instance, Vector2 leftButtonPos
                 BorderColor = Vector4.Zero
             };
             __instance.Controls.Add(openBtn);
+
+            if (___m_continueButton != null && ___m_continueButton.Visible && !usedAutoRejoin && Environment.GetCommandLineArgs().Contains(LoaderTools.AutoRejoinArg))
+            {
+                ___m_continueButton.PressButton();
+                usedAutoRejoin = true;
+            }
         }
     }
 
@@ -54,7 +63,7 @@ public static class Patch_CreateInGameMenu
     {
         public static void Postfix(MyGuiScreenMainMenu __instance, Vector2 leftButtonPositionOrigin, ref Vector2 lastButtonPosition)
         {
-            Patch_CreateMainMenu.Postfix(__instance, leftButtonPositionOrigin, ref lastButtonPosition);
+            Patch_CreateMainMenu.Postfix(__instance, leftButtonPositionOrigin, ref lastButtonPosition, null);
         }
     }
 }

PluginLoader/Patch/Patch_IngameShortcuts.cs

@@ -62,7 +62,10 @@ public static void ShowRestartMenu()
         private static void OnRestartMessageClosed(MyGuiScreenMessageBox.ResultEnum result)
         {
             if(result == MyGuiScreenMessageBox.ResultEnum.YES)
-                LoaderTools.UnloadAndRestart();
+            {
+                LoaderTools.Unload();
+                LoaderTools.Restart(true);
+            }
         }
     }
 }

PluginLoader/Properties/AssemblyInfo.cs

@@ -31,5 +31,5 @@
 // You can specify all the values or you can default the Build and Revision Numbers
 // by using the '*' as shown below:
 // [assembly: AssemblyVersion("1.0.*")]
-[assembly: AssemblyVersion("1.12.2.0")]
-[assembly: AssemblyFileVersion("1.12.2.0")]
+[assembly: AssemblyVersion("1.12.3.0")]
+[assembly: AssemblyFileVersion("1.12.3.0")]