using Content.Shared.Implants;
using Content.Shared.Implants.Components;
using Content.Shared.Roles;
using JetBrains.Annotations;
using Robust.Shared.Prototypes;
using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom.Prototype.Set;

namespace Content.Server.Jobs;

/// <summary>
/// Adds implants on spawn to the entity
/// </summary>
[UsedImplicitly]
public sealed partial class AddImplantSpecial : JobSpecial
{

    [DataField("implants", customTypeSerializer: typeof(PrototypeIdHashSetSerializer<EntityPrototype>))]
    public HashSet<String> Implants { get; private set; } = new();

    public override void AfterEquip(EntityUid mob)
    {
        var entMan = IoCManager.Resolve<IEntityManager>();
        var implantSystem = entMan.System<SharedSubdermalImplantSystem>();
        var xformQuery = entMan.GetEntityQuery<TransformComponent>();

        if (!xformQuery.TryGetComponent(mob, out var xform))
            return;

        foreach (var implantId in Implants)
        {
            var implant = entMan.SpawnEntity(implantId, xform.Coordinates);

            if (!entMan.TryGetComponent<SubdermalImplantComponent>(implant, out var implantComp))
                return;

            implantSystem.ForceImplant(mob, implant, implantComp);
        }
    }
}