OldThink/Content.Server/_White/Cult/Runes/Systems/CultSystem.SoulShard.cs

using Content.Server.Mind;
using Content.Server.Roles;
using Content.Server._White.Cult.Runes.Comps;
using Content.Shared.Humanoid;
using Content.Shared.Interaction;
using Content.Shared.Mind.Components;
using Content.Shared.Mobs;
using Content.Shared.Mobs.Components;
using Content.Shared.Roles;
using Content.Shared._White.Cult;
using Content.Shared._White.Cult.Items;
using CultistComponent = Content.Shared._White.Cult.Components.CultistComponent;

namespace Content.Server._White.Cult.Runes.Systems;

public partial class CultSystem
{
    [Dependency] private readonly SharedPointLightSystem _lightSystem = default!;
    [Dependency] private readonly MindSystem _mindSystem = default!;
    [Dependency] private readonly MetaDataSystem _metaDataSystem = default!;
    [Dependency] private readonly SharedRoleSystem _roleSystem = default!;


    public void InitializeSoulShard()
    {
        SubscribeLocalEvent<SoulShardComponent, AfterInteractEvent>(OnShardInteractUse);
        SubscribeLocalEvent<SoulShardComponent, MindAddedMessage>(OnShardMindAdded);
        SubscribeLocalEvent<SoulShardComponent, MindRemovedMessage>(OnShardMindRemoved);
    }

    private void OnShardInteractUse(EntityUid uid, SoulShardComponent component, AfterInteractEvent args)
    {
        var target = args.Target;

        if (!HasComp<CultistComponent>(args.User)) return;

        if (!TryComp<MobStateComponent>(target, out var state) || state.CurrentState != MobState.Dead) return;

        if (!TryComp<MindContainerComponent>(target, out var mindComponent) || !mindComponent.Mind.HasValue || !TryComp<HumanoidAppearanceComponent>(target, out _)) return;

        _mindSystem.TransferTo(mindComponent.Mind.Value, uid);

        var targetName = MetaData(target.Value).EntityName;

        _metaDataSystem.SetEntityName(uid, Loc.GetString("soul-shard-description", ("soul", targetName)));
        _metaDataSystem.SetEntityDescription(uid, Loc.GetString("soul-shard-description", ("soul", targetName)));
    }

    private void OnShardMindAdded(EntityUid uid, SoulShardComponent component, MindAddedMessage args)
    {
        if (!TryComp<MindContainerComponent>(uid, out var mindContainer) || !mindContainer.HasMind)
        {
            return;
        }

        if (_roleSystem.MindHasRole<TraitorRoleComponent>(mindContainer.Mind.Value))
        {
            _roleSystem.MindRemoveRole<TraitorRoleComponent>(mindContainer.Mind.Value);
        }

        _appearanceSystem.SetData(uid, SoulShardVisualState.State, true);
        _lightSystem.SetEnabled(uid, true);
    }

    private void OnShardMindRemoved(EntityUid uid, SoulShardComponent component, MindRemovedMessage args)
    {
        _appearanceSystem.SetData(uid, SoulShardVisualState.State, false);
        _lightSystem.SetEnabled(uid, false);
    }
}