96 lines
3.3 KiB
C#
96 lines
3.3 KiB
C#
using Content.Shared._CP14.MagicSpell.Components;
|
|
using Content.Shared._CP14.MagicSpell.Events;
|
|
using Content.Shared._CP14.MagicSpellStorage.Components;
|
|
using Content.Shared.Actions;
|
|
using Content.Shared.Damage;
|
|
using Content.Shared.Hands.EntitySystems;
|
|
using Content.Shared.Mind;
|
|
using Robust.Shared.Network;
|
|
|
|
namespace Content.Shared._CP14.MagicSpellStorage;
|
|
|
|
/// <summary>
|
|
/// this part of the system is responsible for storing spells in items, and the methods players use to obtain them.
|
|
/// </summary>
|
|
public sealed partial class CP14SpellStorageSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly ActionContainerSystem _actionContainer = default!;
|
|
[Dependency] private readonly SharedActionsSystem _actions = default!;
|
|
[Dependency] private readonly SharedMindSystem _mind = default!;
|
|
[Dependency] private readonly SharedHandsSystem _hands = default!;
|
|
[Dependency] private readonly INetManager _net = default!;
|
|
[Dependency] private readonly DamageableSystem _damageable = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
InitializeAccess();
|
|
|
|
SubscribeLocalEvent<CP14SpellStorageComponent, MapInitEvent>(OnMagicStorageInit);
|
|
SubscribeLocalEvent<CP14SpellStorageComponent, ComponentShutdown>(OnMagicStorageShutdown);
|
|
|
|
SubscribeLocalEvent<CP14SpellStorageUseDamageComponent, CP14SpellFromSpellStorageUsedEvent>(OnSpellUsed);
|
|
}
|
|
|
|
private void OnSpellUsed(Entity<CP14SpellStorageUseDamageComponent> ent, ref CP14SpellFromSpellStorageUsedEvent args)
|
|
{
|
|
_damageable.TryChangeDamage(ent, ent.Comp.DamagePerMana * args.Manacost);
|
|
}
|
|
|
|
/// <summary>
|
|
/// When we initialize, we create action entities, and add them to this item.
|
|
/// </summary>
|
|
private void OnMagicStorageInit(Entity<CP14SpellStorageComponent> storage, ref MapInitEvent args)
|
|
{
|
|
if (_net.IsClient)
|
|
return;
|
|
|
|
foreach (var spell in storage.Comp.Spells)
|
|
{
|
|
var spellEnt = _actionContainer.AddAction(storage, spell);
|
|
if (spellEnt is null)
|
|
continue;
|
|
|
|
var provided = EntityManager.EnsureComponent<CP14MagicEffectComponent>(spellEnt.Value);
|
|
provided.SpellStorage = storage;
|
|
|
|
storage.Comp.SpellEntities.Add(spellEnt.Value);
|
|
}
|
|
|
|
if (storage.Comp.GrantAccessToSelf)
|
|
{
|
|
if (!_mind.TryGetMind(storage.Owner, out var mind, out _))
|
|
_actions.GrantActions(storage.Owner, storage.Comp.SpellEntities, storage.Owner);
|
|
else
|
|
{
|
|
foreach (var spell in storage.Comp.SpellEntities)
|
|
{
|
|
_actionContainer.AddAction(mind, spell);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private void OnMagicStorageShutdown(Entity<CP14SpellStorageComponent> mStorage, ref ComponentShutdown args)
|
|
{
|
|
if (_net.IsClient)
|
|
return;
|
|
|
|
foreach (var spell in mStorage.Comp.SpellEntities)
|
|
{
|
|
QueueDel(spell);
|
|
}
|
|
}
|
|
|
|
private bool TryGrantAccess(Entity<CP14SpellStorageComponent> storage, EntityUid user)
|
|
{
|
|
if (!_mind.TryGetMind(user, out var mindId, out var mind))
|
|
return false;
|
|
|
|
if (mind.OwnedEntity is null)
|
|
return false;
|
|
|
|
_actions.GrantActions(user, storage.Comp.SpellEntities, storage.Owner);
|
|
return true;
|
|
}
|
|
}
|