Sharedization of entityEffects and crystals slots system
This commit is contained in:
@@ -11,128 +11,4 @@ namespace Content.Server._CP14.MagicEnergy;
|
||||
|
||||
public sealed class CP14MagicEnergyCrystalSlotSystem : SharedCP14MagicEnergyCrystalSlotSystem
|
||||
{
|
||||
[Dependency] private readonly ItemSlotsSystem _itemSlots = default!;
|
||||
[Dependency] private readonly CP14MagicEnergySystem _magicEnergy = default!;
|
||||
[Dependency] private readonly SharedPopupSystem _popup = default!;
|
||||
[Dependency] private readonly SharedAppearanceSystem _appearance = default!;
|
||||
[Dependency] private readonly SharedContainerSystem _container = default!;
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
|
||||
SubscribeLocalEvent<CP14MagicEnergyCrystalComponent, CP14MagicEnergyLevelChangeEvent>(OnEnergyChanged);
|
||||
|
||||
SubscribeLocalEvent<CP14MagicEnergyCrystalSlotComponent, ExaminedEvent>(OnExamined);
|
||||
SubscribeLocalEvent<CP14MagicEnergyCrystalSlotComponent, CP14SlotCrystalChangedEvent>(OnCrystalChanged);
|
||||
}
|
||||
|
||||
private void OnCrystalChanged(Entity<CP14MagicEnergyCrystalSlotComponent> ent, ref CP14SlotCrystalChangedEvent args)
|
||||
{
|
||||
var realPowered = TryGetEnergyCrystalFromSlot((ent, ent), out var energyComp);
|
||||
if (energyComp is not null)
|
||||
realPowered = energyComp.Value.Comp.Energy > 0;
|
||||
|
||||
if (ent.Comp.Powered == realPowered)
|
||||
return;
|
||||
|
||||
ent.Comp.Powered = realPowered;
|
||||
_appearance.SetData(ent, CP14MagicSlotVisuals.Powered, realPowered);
|
||||
|
||||
RaiseLocalEvent(ent, new CP14SlotCrystalPowerChangedEvent(realPowered));
|
||||
}
|
||||
|
||||
private void OnEnergyChanged(Entity<CP14MagicEnergyCrystalComponent> crystal, ref CP14MagicEnergyLevelChangeEvent args)
|
||||
{
|
||||
if (!_container.TryGetContainingContainer((crystal.Owner, null, null), out var container))
|
||||
return;
|
||||
|
||||
if (!TryComp(container.Owner, out CP14MagicEnergyCrystalSlotComponent? slot))
|
||||
return;
|
||||
|
||||
if (!_itemSlots.TryGetSlot(container.Owner, slot.SlotId, out var itemSlot))
|
||||
return;
|
||||
|
||||
if (itemSlot.Item != crystal)
|
||||
return;
|
||||
|
||||
RaiseLocalEvent(container.Owner, new CP14SlotCrystalChangedEvent(false));
|
||||
}
|
||||
|
||||
private void OnExamined(Entity<CP14MagicEnergyCrystalSlotComponent> ent, ref ExaminedEvent args)
|
||||
{
|
||||
if (!TryGetEnergyCrystalFromSlot((ent, ent), out var energyEnt))
|
||||
return;
|
||||
|
||||
if (!args.IsInDetailsRange)
|
||||
return;
|
||||
|
||||
// TODO: scan energy ability
|
||||
// var scanEvent = new CP14MagicEnergyScanEvent();
|
||||
// RaiseLocalEvent(args.Examiner, scanEvent);
|
||||
//
|
||||
// if (!scanEvent.CanScan)
|
||||
// return;
|
||||
|
||||
args.PushMarkup(_magicEnergy.GetEnergyExaminedText((energyEnt.Value, energyEnt)));
|
||||
}
|
||||
|
||||
public bool TryGetEnergyCrystalFromSlot(Entity<CP14MagicEnergyCrystalSlotComponent?> ent,
|
||||
[NotNullWhen(true)] out Entity<CP14MagicEnergyContainerComponent>? energyEnt)
|
||||
{
|
||||
energyEnt = null;
|
||||
|
||||
if (!Resolve(ent, ref ent.Comp, false))
|
||||
return false;
|
||||
|
||||
if (!_itemSlots.TryGetSlot(ent, ent.Comp.SlotId, out var slot))
|
||||
return false;
|
||||
|
||||
if (slot.Item is null)
|
||||
return false;
|
||||
|
||||
if (!TryComp<CP14MagicEnergyContainerComponent>(slot.Item, out var energyComp))
|
||||
return false;
|
||||
|
||||
energyEnt = (slot.Item.Value, energyComp);
|
||||
return true;
|
||||
}
|
||||
|
||||
public bool HasEnergy(Entity<CP14MagicEnergyCrystalSlotComponent?> ent,
|
||||
FixedPoint2 energy,
|
||||
EntityUid? user = null)
|
||||
{
|
||||
if (!TryGetEnergyCrystalFromSlot(ent, out var energyEnt))
|
||||
{
|
||||
if (user is not null)
|
||||
_popup.PopupEntity(Loc.GetString("cp14-magic-energy-no-crystal"), ent,user.Value);
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
if (energyEnt.Value.Comp.Energy >= energy)
|
||||
return true;
|
||||
|
||||
if (user is not null)
|
||||
_popup.PopupEntity(Loc.GetString("cp14-magic-energy-insufficient"), ent, user.Value);
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
public bool TryChangeEnergy(Entity<CP14MagicEnergyCrystalSlotComponent?> ent,
|
||||
FixedPoint2 energy,
|
||||
EntityUid? user = null,
|
||||
bool safe = false)
|
||||
{
|
||||
if (!TryGetEnergyCrystalFromSlot(ent, out var energyEnt))
|
||||
{
|
||||
if (user is not null)
|
||||
_popup.PopupEntity(Loc.GetString("cp14-magic-energy-no-crystal"), ent, user.Value);
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
_magicEnergy.ChangeEnergy((energyEnt.Value, energyEnt.Value), energy, out _, out _, safe);
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user