Verb predict (#5638)

2021-12-16 23:42:02 +13:00
parent 2e141347ed
commit 7e49b22a74
40 changed files with 551 additions and 395 deletions
--- a/Content.Shared/PDA/SharedPDASystem.cs
+++ b/Content.Shared/PDA/SharedPDASystem.cs
@@ -0,0 +1,65 @@
+using Content.Shared.Access.Components;
+using Content.Shared.Containers.ItemSlots;
+using Robust.Shared.Containers;
+using Robust.Shared.GameObjects;
+using Robust.Shared.IoC;
+
+namespace Content.Shared.PDA
+{
+    public abstract class SharedPDASystem : EntitySystem
+    {
+        [Dependency] protected readonly ItemSlotsSystem ItemSlotsSystem = default!;
+
+        public override void Initialize()
+        {
+            base.Initialize();
+
+            SubscribeLocalEvent<PDAComponent, ComponentInit>(OnComponentInit);
+            SubscribeLocalEvent<PDAComponent, ComponentRemove>(OnComponentRemove);
+
+            SubscribeLocalEvent<PDAComponent, EntInsertedIntoContainerMessage>(OnItemInserted);
+            SubscribeLocalEvent<PDAComponent, EntRemovedFromContainerMessage>(OnItemRemoved);
+        }
+
+        protected virtual void OnComponentInit(EntityUid uid, PDAComponent pda, ComponentInit args)
+        {
+            if (pda.IdCard != null)
+                pda.IdSlot.StartingItem = pda.IdCard;
+
+            ItemSlotsSystem.AddItemSlot(uid, $"{pda.Name}-id", pda.IdSlot);
+            ItemSlotsSystem.AddItemSlot(uid, $"{pda.Name}-pen", pda.PenSlot);
+
+            UpdatePDAAppearance(pda);
+        }
+
+        private void OnComponentRemove(EntityUid uid, PDAComponent pda, ComponentRemove args)
+        {
+            ItemSlotsSystem.RemoveItemSlot(uid, pda.IdSlot);
+            ItemSlotsSystem.RemoveItemSlot(uid, pda.PenSlot);
+        }
+
+        protected virtual void OnItemInserted(EntityUid uid, PDAComponent pda, EntInsertedIntoContainerMessage args)
+        {
+            if (!pda.Initialized) return;
+
+            if (args.Container.ID == pda.IdSlot.ID)
+                pda.ContainedID = CompOrNull<IdCardComponent>(args.Entity);
+
+            UpdatePDAAppearance(pda);
+        }
+
+        protected virtual void OnItemRemoved(EntityUid uid, PDAComponent pda, EntRemovedFromContainerMessage args)
+        {
+            if (args.Container.ID == pda.IdSlot.ID)
+                pda.ContainedID = null;
+
+            UpdatePDAAppearance(pda);
+        }
+
+        private void UpdatePDAAppearance(PDAComponent pda)
+        {
+            if (TryComp(pda.Owner, out AppearanceComponent ? appearance))
+                appearance.SetData(PDAVisuals.IDCardInserted, pda.ContainedID != null);
+        }
+    }
+}