LateJoinSettingRoleEventArgs & LateJoinSetRoleEventArgs

Author: marchellcx

LabExtended/Events/ExRoundEvents.cs

@@ -40,6 +40,21 @@ public static class ExRoundEvents
         /// </summary>
         public static event Action<AssigningRolesEventArgs>? AssigningRoles;
 
+        /// <summary>
+        /// Gets called after player's round-start roles are assigned.
+        /// </summary>
+        public static event Action<AssignedRolesEventArgs>? AssignedRoles;
+
+        /// <summary>
+        /// Gets called before a player's role is set by late-join.
+        /// </summary>
+        public static event Action<LateJoinSettingRoleEventArgs>? LateJoinSettingRole;
+
+        /// <summary>
+        /// Gets called after a player's role is set by late-join.
+        /// </summary>
+        public static event Action<LateJoinSetRoleEventArgs>? LateJoinSetRole;
+
         /// <summary>
         /// Invokes the <see cref="Ended"/> event.
         /// </summary>
@@ -76,5 +91,26 @@ public static void OnWaitingForPlayers()
         /// <param name="args">The event's arguments.</param>
         public static bool OnAssigningRoles(AssigningRolesEventArgs args)
             => AssigningRoles.InvokeBooleanEvent(args);
+
+        /// <summary>
+        /// Invokes the <see cref="AssignedRoles"/> event.
+        /// </summary>
+        /// <param name="args">The event's arguments.</param>
+        public static void OnAssignedRoles(AssignedRolesEventArgs args)
+            => AssignedRoles.InvokeEvent(args);
+
+        /// <summary>
+        /// Invokes the <see cref="LateJoinSettingRole"/> event.
+        /// </summary>
+        /// <param name="args">The event's arguments.</param>
+        public static bool OnLateJoinSettingRole(LateJoinSettingRoleEventArgs args)
+            => LateJoinSettingRole.InvokeBooleanEvent(args);
+
+        /// <summary>
+        /// Invokes the <see cref="LateJoinSetRole"/> event.
+        /// </summary>
+        /// <param name="args">The event's arguments.</param>
+        public static void OnLateJoinSetRole(LateJoinSetRoleEventArgs args)
+            => LateJoinSetRole.InvokeEvent(args);
     }
 }

LabExtended/Events/Round/LateJoinSetRoleEventArgs.cs

@@ -0,0 +1,41 @@
+using LabExtended.API;
+
+using PlayerRoles;
+
+namespace LabExtended.Events.Round
+{
+    /// <summary>
+    /// Gets called after a player's role is set by late join.
+    /// </summary>
+    public class LateJoinSetRoleEventArgs : EventArgs
+    {
+        /// <summary>
+        /// Gets the player who joined late.
+        /// </summary>
+        public ExPlayer Player { get; }
+
+        /// <summary>
+        /// Gets or sets the role the player will be spawned as.
+        /// </summary>
+        public RoleTypeId Role { get; }
+
+        /// <summary>
+        /// Gets or sets the role's spawn flags.
+        /// </summary>
+        public RoleSpawnFlags Flags { get; }
+
+        /// <summary>
+        /// Initializes a new instance of the LateJoinSetRoleEventArgs class with the specified player, role, and
+        /// spawn flags.
+        /// </summary>
+        /// <param name="player">The player for whom the role is being set during a late join event. Cannot be null.</param>
+        /// <param name="role">The role to assign to the player.</param>
+        /// <param name="flags">The flags that control how the role is spawned for the player.</param>
+        public LateJoinSetRoleEventArgs(ExPlayer player, RoleTypeId role, RoleSpawnFlags flags)
+        {
+            Player = player;
+            Role = role;
+            Flags = flags;
+        }
+    }
+}

LabExtended/Events/Round/LateJoinSettingRoleEventArgs.cs

@@ -0,0 +1,41 @@
+using LabExtended.API;
+
+using PlayerRoles;
+
+namespace LabExtended.Events.Round
+{
+    /// <summary>
+    /// Gets called before a player's role is set by late join.
+    /// </summary>
+    public class LateJoinSettingRoleEventArgs : BooleanEventArgs
+    {
+        /// <summary>
+        /// Gets the player who joined late.
+        /// </summary>
+        public ExPlayer Player { get; }
+
+        /// <summary>
+        /// Gets or sets the role the player will be spawned as.
+        /// </summary>
+        public RoleTypeId Role { get; set; }
+
+        /// <summary>
+        /// Gets or sets the role's spawn flags.
+        /// </summary>
+        public RoleSpawnFlags Flags { get; set; }
+
+        /// <summary>
+        /// Initializes a new instance of the LateJoinSettingRoleEventArgs class with the specified player, role, and
+        /// spawn flags.
+        /// </summary>
+        /// <param name="player">The player for whom the role is being set during a late join event. Cannot be null.</param>
+        /// <param name="role">The role to assign to the player.</param>
+        /// <param name="flags">The flags that control how the role is spawned for the player.</param>
+        public LateJoinSettingRoleEventArgs(ExPlayer player, RoleTypeId role, RoleSpawnFlags flags)
+        {
+            Player = player;
+            Role = role;
+            Flags = flags;
+        }
+    }
+}

LabExtended/Patches/Events/Round/LateJoinSettingRolePatches.cs

@@ -0,0 +1,53 @@
+using CentralAuth;
+
+using GameCore;
+
+using HarmonyLib;
+
+using LabExtended.API;
+
+using LabExtended.Events;
+using LabExtended.Events.Round;
+
+using PlayerRoles;
+using PlayerRoles.RoleAssign;
+
+namespace LabExtended.Patches.Events.Round
+{
+    /// <summary>
+    /// Implements the <see cref="ExRoundEvents.LateJoinSettingRole"/> and <see cref="ExRoundEvents.LateJoinSetRole"/> events.
+    /// </summary>
+    public static class LateJoinSettingRolePatches
+    {
+        [HarmonyPatch(typeof(RoleAssigner), nameof(RoleAssigner.CheckLateJoin))]
+        private static bool Prefix(ReferenceHub hub, ClientInstanceMode cim)
+        {
+            if (!ExPlayer.TryGet(hub, out var player)
+                || !RoleAssigner.CheckPlayer(hub)
+                || !RoleAssigner._spawned)
+                return false;
+
+            var lateJoinTime = ConfigFile.ServerConfig.GetFloat("late_join_time", 0f);
+            var lateJoinRole = RoleTypeId.None;
+
+            if (!RoleAssigner.AlreadySpawnedPlayers.Add(player.UserId)
+                || RoleAssigner.LateJoinTimer.Elapsed.TotalSeconds > lateJoinTime)
+                lateJoinRole = RoleTypeId.Spectator;
+            else
+                lateJoinRole = HumanSpawner.NextHumanRoleToSpawn;
+
+            var settingEventArgs = new LateJoinSettingRoleEventArgs(player, lateJoinRole, RoleSpawnFlags.All);
+
+            if (!ExRoundEvents.OnLateJoinSettingRole(settingEventArgs)
+                || settingEventArgs.Role is RoleTypeId.None)
+                return false;
+
+            player.Role.LateJoinRole = settingEventArgs.Role;
+            player.Role.Set(settingEventArgs.Role, RoleChangeReason.LateJoin, settingEventArgs.Flags);
+
+            ExRoundEvents.OnLateJoinSetRole(new LateJoinSetRoleEventArgs(player, settingEventArgs.Role, settingEventArgs.Flags));
+
+            return false;
+        }
+    }
+}