Fix challenge prompt for Mantis Lords

HKMP/Game/Client/Entity/Component/ChallengePromptComponent.cs

@@ -0,0 +1,55 @@
+using Hkmp.Networking.Client;
+using Hkmp.Networking.Packet.Data;
+using Hkmp.Util;
+using UnityEngine;
+
+namespace Hkmp.Game.Client.Entity.Component;
+
+/// <inheritdoc />
+/// This component manages the challenge prompt that appears for Mantis Lords.
+internal class ChallengePromptComponent : EntityComponent {
+    /// <summary>
+    /// The game object that handles the challenge prompt pop-up.
+    /// </summary>
+    private readonly GameObject _promptObj;
+
+    public ChallengePromptComponent(
+        NetClient netClient,
+        ushort entityId,
+        HostClientPair<GameObject> gameObject
+    ) : base(netClient, entityId, gameObject) {
+        var hostObj = gameObject.Host;
+        var parent = hostObj.transform.parent;
+        _promptObj = parent.Find("Challenge Prompt").gameObject;
+
+        var promptFsm = _promptObj.LocateMyFSM("Challenge Start");
+        promptFsm.InsertMethod("Take Control", 6, () => {
+            var data = new EntityNetworkData {
+                Type = EntityComponentType.ChallengePrompt
+            };
+            data.Packet.Write(true);
+
+            SendData(data);
+        });
+    }
+
+    /// <inheritdoc />
+    public override void InitializeHost() {
+    }
+
+    /// <inheritdoc />
+    public override void Update(EntityNetworkData data) {
+        var destroyPrompt = data.Packet.ReadBool();
+        if (!destroyPrompt) {
+            return;
+        }
+
+        if (_promptObj != null) {
+            Object.Destroy(_promptObj);
+        }
+    }
+
+    /// <inheritdoc />
+    public override void Destroy() {
+    }
+}

HKMP/Game/Client/Entity/Component/ComponentFactory.cs

@@ -63,6 +63,8 @@ HostClientPair<GameObject> objects
                     Client = spriteRendererClient,
                     Host = spriteRendererHost
                 });
+            case EntityComponentType.ChallengePrompt:
+                return new ChallengePromptComponent(netClient, entityId, objects);
             default:
                 throw new ArgumentOutOfRangeException(nameof(type), type, $"Could not instantiate entity component for type: {type}");
         }

HKMP/Game/Client/Entity/Component/EntityComponent.cs

@@ -69,7 +69,7 @@ protected void SendData(EntityNetworkData data) {
 /// Enum for data types.
 /// </summary>
 [JsonConverter(typeof(StringEnumConverter))]
-internal enum EntityComponentType : byte {
+internal enum EntityComponentType : ushort {
     Fsm = 0,
     Death,
     Invincibility,
@@ -85,4 +85,5 @@ internal enum EntityComponentType : byte {
     ChildrenActivation,
     SpawnJar,
     SpriteRenderer,
-}
+    ChallengePrompt
+}

HKMP/Networking/Packet/Data/EntityUpdate.cs

@@ -545,7 +545,7 @@ public EntityNetworkData() {
 
     /// <inheritdoc cref="IPacketData.WriteData" />
     public void WriteData(IPacket packet) {
-        packet.Write((byte) Type);
+        packet.Write((ushort) Type);
 
         var data = Packet.ToArray();
 
@@ -563,7 +563,7 @@ public void WriteData(IPacket packet) {
 
     /// <inheritdoc cref="IPacketData.ReadData" />
     public void ReadData(IPacket packet) {
-        Type = (EntityComponentType) packet.ReadByte();
+        Type = (EntityComponentType) packet.ReadUShort();
 
         var length = packet.ReadUShort();
         var data = new byte[length];

HKMP/Resource/entity-registry.json

@@ -395,6 +395,9 @@
                 "type": "MantisLord",
                 "fsm_name": "Mantis Lord"
             }
+        ],
+        "components": [
+            "ChallengePrompt"
         ]
     },
     {