Addressed an issue where the revived kill button wasn’t showing for revived players, and fixed an issue where revived players couldn’t move or see death task info

Author: CallOfCreator

NewMod/Buttons/Necromancer/ReviveButton.cs

@@ -100,9 +100,6 @@ protected override void OnClick()
             var local = PlayerControl.LocalPlayer;
             var body = GetReviveTarget();
 
-            if (body == null)
-                return;
-
             SoundManager.Instance.PlaySound(NewModAsset.ReviveSound?.LoadAsset(), false, 2f);
 
             Utils.HandleRevive(

NewMod/Buttons/RevivedKillButton.cs

@@ -55,7 +55,7 @@ public override void SetOutline(bool active)
 
         public override bool CanUse()
         {
-            return CanUseRevivedKillButton();
+            return base.CanUse() && CanUseRevivedKillButton();
         }
 
         protected override void OnClick()

NewMod/Patches/Roles/RevivePatch.cs

@@ -14,6 +14,8 @@ public static bool Prefix(PlayerControl __instance)
             __instance.MyPhysics.ResetMoveState(true);
             __instance.clickKillCollider.enabled = true;
             __instance.cosmetics.SetNameMask(true);
+            __instance.cosmetics.SetPetSource(__instance);
+            __instance.moveable = true;
 
             if (__instance.AmOwner)
             {
@@ -24,4 +26,13 @@ public static bool Prefix(PlayerControl __instance)
             return false;
         }
     }
+    [HarmonyPatch(typeof(KillOverlay), nameof(KillOverlay.IsOpen), MethodType.Getter)]
+    public static class KillOverlayPatch
+    {
+        public static void Postfix(ref bool __result)
+        {
+            if (__result && PlayerControl.LocalPlayer != null && !PlayerControl.LocalPlayer.Data.IsDead)
+                __result = false;
+        }
+    }
 }

NewMod/Utilities/Utils.cs

@@ -394,6 +394,8 @@ public static IEnumerator HandleRevive(PlayerControl source, byte revivedId, Rol
 
             revived.Revive();
             revived.RemainingEmergencies = 0;
+            RoleManager.Instance.SetRole(revived, roleToSet);
+            revived.Data.Role.SpawnTaskHeader(revived);
 
             if (AmongUsClient.Instance.AmHost)
             {
@@ -829,7 +831,7 @@ public static IEnumerator StartFeignDeath(PlayerControl player)
 
             if (player.AmOwner)
             {
-                HudManager.Instance.SetHudActive(false);
+                HudManager.Instance.SetHudActive(player, player.Data.Role, false);
             }
             yield return new WaitForSeconds(0.5f);