94 lines
3.0 KiB
C#
94 lines
3.0 KiB
C#
using System.Diagnostics.CodeAnalysis;
|
|
using Content.Server.GameTicking;
|
|
using Content.Server.GameTicking.Rules;
|
|
using Content.Server.GameTicking.Rules.Components;
|
|
using Content.Server.Paper;
|
|
using Content.Server.Traitor.Components;
|
|
using Robust.Shared.Random;
|
|
using Robust.Shared.Utility;
|
|
using System.Linq;
|
|
|
|
namespace Content.Server.Traitor.Systems;
|
|
|
|
public sealed class TraitorCodePaperSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly GameTicker _gameTicker = default!;
|
|
[Dependency] private readonly TraitorRuleSystem _traitorRuleSystem = default!;
|
|
[Dependency] private readonly IRobustRandom _random = default!;
|
|
[Dependency] private readonly PaperSystem _paper = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
SubscribeLocalEvent<TraitorCodePaperComponent, MapInitEvent>(OnMapInit);
|
|
}
|
|
|
|
private void OnMapInit(EntityUid uid, TraitorCodePaperComponent component, MapInitEvent args)
|
|
{
|
|
SetupPaper(uid, component);
|
|
}
|
|
|
|
private void SetupPaper(EntityUid uid, TraitorCodePaperComponent? component = null)
|
|
{
|
|
if (!Resolve(uid, ref component))
|
|
return;
|
|
|
|
if (HasComp<PaperComponent>(uid))
|
|
{
|
|
if (TryGetTraitorCode(out var paperContent, component))
|
|
{
|
|
_paper.SetContent(uid, paperContent);
|
|
}
|
|
}
|
|
}
|
|
|
|
private bool TryGetTraitorCode([NotNullWhen(true)] out string? traitorCode, TraitorCodePaperComponent component)
|
|
{
|
|
traitorCode = null;
|
|
|
|
var codesMessage = new FormattedMessage();
|
|
List<string> codeList = new();
|
|
// Find the first nuke that matches the passed location.
|
|
if (_gameTicker.IsGameRuleAdded<TraitorRuleComponent>())
|
|
{
|
|
var ruleEnts = _gameTicker.GetAddedGameRules();
|
|
foreach (var ruleEnt in ruleEnts)
|
|
{
|
|
if (TryComp(ruleEnt, out TraitorRuleComponent? traitorComp))
|
|
{
|
|
codeList.AddRange(traitorComp.Codewords.ToList());
|
|
}
|
|
}
|
|
}
|
|
if (codeList.Count == 0)
|
|
{
|
|
if (component.FakeCodewords)
|
|
codeList = _traitorRuleSystem.GenerateTraitorCodewords(new TraitorRuleComponent()).ToList();
|
|
else
|
|
codeList = [Loc.GetString("traitor-codes-none")];
|
|
}
|
|
|
|
_random.Shuffle(codeList);
|
|
|
|
int i = 0;
|
|
foreach (var code in codeList)
|
|
{
|
|
i++;
|
|
if (i > component.CodewordAmount && !component.CodewordShowAll)
|
|
break;
|
|
|
|
codesMessage.PushNewline();
|
|
codesMessage.AddMarkup(code);
|
|
}
|
|
|
|
if (!codesMessage.IsEmpty)
|
|
{
|
|
if (i == 1)
|
|
traitorCode = Loc.GetString("traitor-codes-message-singular") + codesMessage;
|
|
else
|
|
traitorCode = Loc.GetString("traitor-codes-message-plural") + codesMessage;
|
|
}
|
|
return !codesMessage.IsEmpty;
|
|
}
|
|
}
|