a06ea58249
* add AntagObjectives from GenericAntag * add AntagRandomObjectives that traitor and thief can use * make ObjectivesSystem use initial character name which AntagSelection passes * make thief and traitor use AntagRandomObjectives * remove now unused locale * make sleeper agents rule use baseTraitorRule * restore dragon rule oop * bandaid for genericantag * real * typo --------- Co-authored-by: deltanedas <@deltanedas:kde.org>
53 lines
1.9 KiB
C#
53 lines
1.9 KiB
C#
using Content.Server.Antag.Components;
|
|
using Content.Server.Objectives;
|
|
using Content.Shared.Mind;
|
|
using Content.Shared.Objectives.Components;
|
|
using Content.Shared.Objectives.Systems;
|
|
using Robust.Shared.Random;
|
|
|
|
namespace Content.Server.Antag;
|
|
|
|
/// <summary>
|
|
/// Adds fixed objectives to an antag made with <c>AntagRandomObjectivesComponent</c>.
|
|
/// </summary>
|
|
public sealed class AntagRandomObjectivesSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly IRobustRandom _random = default!;
|
|
[Dependency] private readonly SharedMindSystem _mind = default!;
|
|
[Dependency] private readonly ObjectivesSystem _objectives = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<AntagRandomObjectivesComponent, AfterAntagEntitySelectedEvent>(OnAntagSelected);
|
|
}
|
|
|
|
private void OnAntagSelected(Entity<AntagRandomObjectivesComponent> ent, ref AfterAntagEntitySelectedEvent args)
|
|
{
|
|
if (!_mind.TryGetMind(args.Session, out var mindId, out var mind))
|
|
{
|
|
Log.Error($"Antag {ToPrettyString(args.EntityUid):player} was selected by {ToPrettyString(ent):rule} but had no mind attached!");
|
|
return;
|
|
}
|
|
|
|
var difficulty = 0f;
|
|
foreach (var set in ent.Comp.Sets)
|
|
{
|
|
if (!_random.Prob(set.Prob))
|
|
continue;
|
|
|
|
for (var pick = 0; pick < set.MaxPicks && ent.Comp.MaxDifficulty > difficulty; pick++)
|
|
{
|
|
if (_objectives.GetRandomObjective(mindId, mind, set.Groups) is not {} objective)
|
|
continue;
|
|
|
|
_mind.AddObjective(mindId, mind, objective);
|
|
var adding = Comp<ObjectiveComponent>(objective).Difficulty;
|
|
difficulty += adding;
|
|
Log.Debug($"Added objective {ToPrettyString(objective):objective} to {ToPrettyString(args.EntityUid):player} with {adding} difficulty");
|
|
}
|
|
}
|
|
}
|
|
}
|