Workaround for 1.6.0 PartDatabase rebuild

Author: sarbian

Source/ModuleManager/Fix16.cs

@@ -0,0 +1,81 @@
+using System.Collections;
+
+namespace ModuleManager
+{
+    class Fix16 : LoadingSystem
+    {
+        private void Awake()
+        {
+            if (Instance != null)
+            {
+                DestroyImmediate(this);
+                return;
+            }
+
+            Instance = this;
+            DontDestroyOnLoad(gameObject);
+        }
+
+        private static Fix16 Instance { get; set; }
+
+        private bool ready;
+
+        private int count;
+
+        private int current;
+
+        private const int yieldStep = 20;
+
+        public override bool IsReady()
+        {
+            return ready;
+        }
+
+        public override string ProgressTitle()
+        {
+            return "Fix 1.6.0 " + current + "/" + count;
+        }
+
+        public override float ProgressFraction()
+        {
+            return (float) current / count;
+        }
+
+        public override void StartLoad()
+        {
+            ready = false;
+            
+            count = PartLoader.LoadedPartsList.Count;
+
+            StartCoroutine(DoFix());
+        }
+
+        private IEnumerator DoFix()
+        {
+            int yieldCounter = 0;
+            for (current = 0; current < count; current++)
+            {
+                AvailablePart avp = PartLoader.LoadedPartsList[current];
+                if (avp.partPrefab.dragModel == Part.DragModel.CUBE && !avp.partPrefab.DragCubes.Procedural &&
+                    !avp.partPrefab.DragCubes.None && avp.partPrefab.DragCubes.Cubes.Count == 0)
+                {
+                    DragCubeSystem.Instance.LoadDragCubes(avp.partPrefab);
+                }
+
+                if (yieldCounter++ >= yieldStep)
+                {
+                    yieldCounter = 0;
+                    yield return null;
+                }
+            }
+
+            ready = true;
+            yield return null;
+        }
+
+        public override float LoadWeight()
+        {
+            return 0.1f;
+        }
+    }
+}

Source/ModuleManager/ModuleManager.cs

@@ -118,6 +118,13 @@ internal void Awake()
 
                 int gameDatabaseIndex = list.FindIndex(s => s is GameDatabase);
                 list.Insert(gameDatabaseIndex + 1, loader);
+
+                // Workaround for 1.6.0 Editor bug after a PartDatabase rebuild.
+                if (Versioning.version_major == 1 && Versioning.version_minor == 6 && Versioning.Revision == 0)
+                {
+                    Fix16 fix16 = aGameObject.AddComponent<Fix16>();
+                    list.Add(fix16);
+                }
             }
 
             bool foolsDay = (DateTime.Now.Month == 4 && DateTime.Now.Day == 1);

Source/ModuleManager/ModuleManager.csproj

@@ -32,6 +32,7 @@
     <ConsolePause>False</ConsolePause>
   </PropertyGroup>
   <ItemGroup>
+    <Compile Include="Fix16.cs" />
     <Compile Include="Cats\CatAnimator.cs" />
     <Compile Include="Cats\CatManager.cs" />
     <Compile Include="Cats\CatMover.cs" />