Zombie Rework & Polymorph Expansion (#8413)

Co-authored-by: Kara <[email protected]>
Co-authored-by: metalgearsloth <[email protected]>

Content.Server/Disease/Effects/DiseaseAddComponent.cs

@@ -0,0 +1,30 @@
+using JetBrains.Annotations;
+using Content.Shared.Disease;
+
+namespace Content.Server.Disease.Effects
+{
+    /// <summary>
+    /// Adds a component to the diseased entity
+    /// </summary>
+    [UsedImplicitly]
+    public sealed class DiseaseAddComponent : DiseaseEffect
+    {
+        /// <summary>
+        /// The component that is added at the end of build up
+        /// </summary>
+        [DataField("comp")]
+        public string? Comp = null;
+
+        public override void Effect(DiseaseEffectArgs args)
+        {
+            if (Comp == null)
+                return;
+
+            EntityUid uid = args.DiseasedEntity;
+            var newComponent = (Component) IoCManager.Resolve<IComponentFactory>().GetComponent(Comp);
+            newComponent.Owner = uid;
+            if (!args.EntityManager.HasComponent(uid, newComponent.GetType()))
+                args.EntityManager.AddComponent(uid, newComponent);
+        }
+    }
+}

Content.Server/Inventory/ServerInventorySystem.cs

@@ -5,6 +5,7 @@
 using Content.Shared.Interaction.Events;
 using Content.Shared.Inventory;
 using Content.Shared.Inventory.Events;
+using Robust.Shared.Containers;
 using InventoryComponent = Content.Shared.Inventory.InventoryComponent;
 
 namespace Content.Server.Inventory
@@ -42,5 +43,35 @@ private void OnOpenSlotStorage(OpenSlotStorageNetworkMessage ev, EntitySessionEv
                 _storageSystem.OpenStorageUI(entityUid.Value, uid, storageComponent);
             }
         }
+
+        public void TransferEntityInventories(EntityUid uid, EntityUid target)
+        {
+            if (TryGetContainerSlotEnumerator(uid, out var enumerator))
+            {
+                Dictionary<string, EntityUid?> inventoryEntities = new();
+                var slots = GetSlots(uid);
+                while (enumerator.MoveNext(out var containerSlot))
+                {
+                    //records all the entities stored in each of the target's slots
+                    foreach (var slot in slots)
+                    {
+                        if (TryGetSlotContainer(target, slot.Name, out var conslot, out var _) &&
+                            conslot.ID == containerSlot.ID)
+                        {
+                            inventoryEntities.Add(slot.Name, containerSlot.ContainedEntity);
+                        }
+                    }
+                    //drops everything in the target's inventory on the ground
+                    containerSlot.EmptyContainer();
+                }
+                /// This takes the objects we removed and stored earlier
+                /// and actually equips all of it to the new entity
+                foreach (var item in inventoryEntities)
+                {
+                    if (item.Value != null)
+                        TryEquip(target, item.Value.Value, item.Key, true);
+                }
+            }
+        }
     }
 }