civ14/Content.Server/Electrocution/ElectrocuteCommand.cs

using Content.Server.Administration;
using Content.Shared.Administration;
using Content.Shared.StatusEffect;
using Robust.Shared.Console;

namespace Content.Server.Electrocution
{
    [AdminCommand(AdminFlags.Fun)]
    public sealed class ElectrocuteCommand : IConsoleCommand
    {
        [Dependency] private readonly IEntityManager _entManager = default!;

        public string Command => "electrocute";
        public string Description => Loc.GetString("electrocute-command-description");
        public string Help => $"{Command} <uid> <seconds> <damage>";

        [ValidatePrototypeId<StatusEffectPrototype>]
        public const string ElectrocutionStatusEffect = "Electrocution";

        public void Execute(IConsoleShell shell, string argStr, string[] args)
        {
            if (args.Length < 1)
            {
                // TODO: Localize this.
                shell.WriteError("Not enough arguments!");
                return;
            }

            if (!NetEntity.TryParse(args[0], out var uidNet) ||
                !_entManager.TryGetEntity(uidNet, out var uid) ||
                !_entManager.EntityExists(uid))
            {
                shell.WriteError($"Invalid entity specified!");
                return;
            }

            if (!_entManager.EntitySysManager.GetEntitySystem<StatusEffectsSystem>().CanApplyEffect(uid.Value, ElectrocutionStatusEffect))
            {
                shell.WriteError(Loc.GetString("electrocute-command-entity-cannot-be-electrocuted"));
                return;
            }

            if (args.Length < 2 || !int.TryParse(args[1], out var seconds))
            {
                seconds = 10;
            }

            if (args.Length < 3 || !int.TryParse(args[2], out var damage))
            {
                damage = 10;
            }

            _entManager.EntitySysManager.GetEntitySystem<ElectrocutionSystem>()
                .TryDoElectrocution(uid.Value, null, damage, TimeSpan.FromSeconds(seconds), refresh: true, ignoreInsulation: true);
        }
    }
}