32 lines
1.1 KiB
C#
32 lines
1.1 KiB
C#
using Content.Server._DV.Objectives.Components;
|
|
using Content.Server.Objectives.Components;
|
|
using Content.Server.Objectives.Systems;
|
|
using Content.Server.Roles;
|
|
using Content.Shared.Objectives.Components;
|
|
using Content.Shared.Roles;
|
|
|
|
namespace Content.Server._DV.Objectives.Systems;
|
|
|
|
/// <summary>
|
|
/// Handles picking a random traitor for the kill fellow traitor objective.
|
|
/// </summary>
|
|
public sealed class PickRandomTraitorSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly PickObjectiveTargetSystem _pickTarget = default!;
|
|
[Dependency] private readonly SharedRoleSystem _role = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<PickRandomTraitorComponent, ObjectiveAssignedEvent>(OnRandomTraitorAssigned);
|
|
}
|
|
|
|
private void OnRandomTraitorAssigned(Entity<PickRandomTraitorComponent> ent, ref ObjectiveAssignedEvent args)
|
|
{
|
|
_pickTarget.AssignRandomTarget(ent, ref args, mindId =>
|
|
_role.MindHasRole<TraitorRoleComponent>(mindId),
|
|
fallbackToAny: false); // bruh
|
|
}
|
|
}
|