update implicit transfer

RW-NodeTree · Feb 28, 2024 · dc8d35b · dc8d35b
1 parent b5935ad
commit dc8d35b
Showing 1 changed file with 46 additions and 1 deletion.
diff --git a/RW_ModularizationWeapon/CompModularizationWeapon.cs b/RW_ModularizationWeapon/CompModularizationWeapon.cs
@@ -749,7 +749,52 @@ public static implicit operator Thing(CompModularizationWeapon node)
 
         public static implicit operator CompModularizationWeapon(Thing thing)
         {
-            return thing?.TryGetComp<CompModularizationWeapon>();
+            List<ThingComp> comps = (thing as ThingWithComps)?.AllComps;
+            if (comps == null || comps.Count < 2 || ((CompChildNodeProccesser)thing) == null) return null;
+
+            CompNoCompMarker marker = comps[1] as CompNoCompMarker;
+            if (marker != null) return null;
+            CompModularizationWeapon result = comps[1] as CompModularizationWeapon;
+            int i = 2;
+            if (result != null) return result;
+            else
+            {
+                for (; i < comps.Count; i++)
+                {
+                    result = comps[i] as CompModularizationWeapon;
+                    if (result != null) break;
+                }
+            }
+            if (result != null)
+            {
+                comps.RemoveAt(i);
+                comps.Insert(1, result);
+            }
+            else
+            {
+                i = 2;
+                for (; i < comps.Count; i++)
+                {
+                    marker = comps[i] as CompNoCompMarker;
+                    if (marker != null) break;
+                }
+                if (marker != null)
+                {
+                    comps.RemoveAt(i);
+                    comps.Insert(1, marker);
+                }
+                else
+                {
+                    marker = new CompNoCompMarker();
+                    comps.Insert(1, marker);
+                }
+            }
+            return result;
+        }
+
+        internal class CompNoCompMarker : ThingComp
+        {
+
         }
         #endregion