91 lines
2.8 KiB
C#
91 lines
2.8 KiB
C#
using System.Linq;
|
|
using Content.Server.GameTicking;
|
|
using Content.Server.GameTicking.Rules.Components;
|
|
using Content.Server._White.AspectsSystem.Aspects.Components;
|
|
using Content.Server._White.AspectsSystem.Base;
|
|
using Content.Server.GameTicking.Components;
|
|
using Content.Shared.Hands.Components;
|
|
using Content.Shared.Hands.EntitySystems;
|
|
using Content.Shared.Humanoid;
|
|
using Content.Shared._White.NonPeacefulRoundEnd;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Random;
|
|
|
|
namespace Content.Server._White.AspectsSystem.Aspects;
|
|
|
|
public sealed class BattledAspect : AspectSystem<BattledAspectComponent>
|
|
{
|
|
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
|
|
[Dependency] private readonly SharedHandsSystem _handsSystem = default!;
|
|
[Dependency] private readonly IRobustRandom _robustRandom = default!;
|
|
[Dependency] private readonly ChatHelper _chatHelper = default!;
|
|
|
|
private NonPeacefulRoundItemsPrototype _nonPeacefulRoundItemsPrototype = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<PlayerSpawnCompleteEvent>(HandleLateJoin);
|
|
}
|
|
|
|
protected override void Started(EntityUid uid, BattledAspectComponent component, GameRuleComponent gameRule, GameRuleStartedEvent args)
|
|
{
|
|
base.Started(uid, component, gameRule, args);
|
|
|
|
var prototypes = _prototypeManager.EnumeratePrototypes<NonPeacefulRoundItemsPrototype>().ToList();
|
|
|
|
if (prototypes.Count == 0)
|
|
ForceEndSelf(uid, gameRule);
|
|
|
|
_nonPeacefulRoundItemsPrototype = _robustRandom.Pick(prototypes);
|
|
|
|
var query = EntityQueryEnumerator<HumanoidAppearanceComponent>();
|
|
while (query.MoveNext(out var ent, out _))
|
|
{
|
|
GiveItem(ent);
|
|
}
|
|
|
|
}
|
|
|
|
private void HandleLateJoin(PlayerSpawnCompleteEvent ev)
|
|
{
|
|
var query = EntityQueryEnumerator<BattledAspectComponent, GameRuleComponent>();
|
|
while (query.MoveNext(out var ruleEntity, out _, out var gameRule))
|
|
{
|
|
if (!GameTicker.IsGameRuleAdded(ruleEntity, gameRule))
|
|
continue;
|
|
|
|
if (!ev.LateJoin)
|
|
return;
|
|
|
|
var mob = ev.Mob;
|
|
|
|
GiveItem(mob);
|
|
_chatHelper.SendAspectDescription(mob, Loc.GetString("battled-aspect-desc"));
|
|
}
|
|
}
|
|
|
|
#region Helpers
|
|
|
|
private void GiveItem(EntityUid player)
|
|
{
|
|
var item = _robustRandom.Pick(_nonPeacefulRoundItemsPrototype.Items);
|
|
|
|
var transform = CompOrNull<TransformComponent>(player);
|
|
|
|
if (transform == null)
|
|
return;
|
|
|
|
if (!HasComp<HandsComponent>(player))
|
|
return;
|
|
|
|
var weaponEntity = EntityManager.SpawnEntity(item, transform.Coordinates);
|
|
|
|
_handsSystem.TryDrop(player);
|
|
_handsSystem.PickupOrDrop(player, weaponEntity);
|
|
}
|
|
|
|
#endregion
|
|
}
|