cleanup and fix MH launchsite selector logic.

GER-Space · Oct 16, 2019 · a89f307 · a89f307
1 parent 04f630d
commit a89f307
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 31 deletions.
diff --git a/src/Core/LaunchSites/LaunchSiteManager.cs b/src/Core/LaunchSites/LaunchSiteManager.cs
@@ -256,12 +256,12 @@ internal static void AttachLaunchSite(StaticInstance instance, ConfigNode instan
         /// </summary>
         internal static void CloseLaunchSite(KKLaunchSite site)
         {
+            site.SetClosed();
             if (Expansions.ExpansionsLoader.IsExpansionInstalled("MakingHistory") && HighLogic.LoadedScene == GameScenes.EDITOR)
             {
                 Log.Normal("Close: " + site.LaunchSiteName);
                 AlterMHSelector();
             }
-            site.SetClosed();
         }
 
         /// <summary>
@@ -274,12 +274,14 @@ internal static void OpenLaunchSite(KKLaunchSite site)
                 site.staticInstance.TrySpawn();
                 site.staticInstance.destructible.Reset();
             }
+
+            site.staticInstance.isDestroyed = false;
+            site.SetOpen();
             if (Expansions.ExpansionsLoader.IsExpansionInstalled("MakingHistory") && HighLogic.LoadedScene == GameScenes.EDITOR)
             {
                 Log.Normal("OpenSite: " + site.LaunchSiteName);
                 AlterMHSelector();
             }
-            site.SetOpen();
         }
 
 
@@ -419,30 +421,20 @@ internal static void RegisterLaunchSite(KKLaunchSite site, bool isSquad = false)
         }
 
 
-        public static void AlterMHSelector(bool triggerRestart = true)
+        public static void AlterMHSelector()
         {
+            Log.Normal("working on launchsites");
             if (!HighLogic.CurrentGame.Parameters.Difficulty.AllowOtherLaunchSites)
             {
                 return;
             }
 
             RegisterMHLaunchSites(EditorDriver.editorFacility);
 
-            if (triggerRestart)
-            {
-                GameEvents.onEditorRestart.Fire();
-            }
-            else
-            {
-                KSP.UI.UILaunchsiteController uILaunchsiteController = Resources.FindObjectsOfTypeAll<KSP.UI.UILaunchsiteController>().FirstOrDefault();
-                if (uILaunchsiteController == null)
-                {
-                    Log.UserWarning("UILaunchsiteController not found");
-                    return;
-                }
-                uILaunchsiteController.GetType().GetMethod("resetItems", BindingFlags.Instance | BindingFlags.NonPublic).Invoke(uILaunchsiteController, null);
-            }
+            KerbalKonstructs.launchSitesAltering = true;
+            GameEvents.onEditorRestart.Fire();
 
+            KerbalKonstructs.launchSitesAltering = false;
         }
 
 

diff --git a/src/KerbalKonstructs.cs b/src/KerbalKonstructs.cs
@@ -149,6 +149,8 @@ internal bool debugMode
         internal static List<StaticInstance> deletedInstances = new List<StaticInstance>();
         internal static List<GroupCenter> deletedGroups = new List<GroupCenter>();
 
+        internal static bool launchSitesAltering = false;
+
         /// <summary>
         /// Unity GameObject Awake function
         /// </summary>
@@ -568,21 +570,13 @@ public IEnumerator WaitAndReset(GameEvents.VesselSpawnInfo info)
 
         public void OnEditorRestart()
         {
-            Log.Normal("On Editor Restart");
-            LaunchSiteManager.AlterMHSelector(false);
-        }
-
-        //public void OnGameSceneLoadRequested(GameScenes scene)
-        //{
-        //    if (scene == GameScenes.MAINMENU)
-        //    {
-        //        Log.Normal("OnGameSceneLoadRequested");
-        //        LaunchSiteManager.ResetLaunchSites();
-        //        typeof(EditorDriver).GetMethod("setupValidLaunchSites", BindingFlags.NonPublic | BindingFlags.Static).Invoke(null, null);
-
+            //Log.Normal("On Editor Restart");
 
-        //    }
-        //}
+            if (!KerbalKonstructs.launchSitesAltering)
+            {
+                LaunchSiteManager.AlterMHSelector();
+            }
+        }
 
 
         /// <summary>