Delta-v/Content.Shared/PowerCell/PowerCellSystem.Draw.cs

using Content.Shared.PowerCell.Components;
using JetBrains.Annotations;

namespace Content.Shared.PowerCell;

public sealed partial class PowerCellSystem
{
    /// <summary>
    /// Enables or disables the power cell draw.
    /// </summary>
    [PublicAPI]
    public void SetDrawEnabled(Entity<PowerCellDrawComponent?> ent, bool enabled)
    {
        if (Resolve(ent, ref ent.Comp, false) && ent.Comp.Enabled != enabled)
        {
            ent.Comp.Enabled = enabled;
            Dirty(ent, ent.Comp);
        }

        // BEGIN DeltaV - Allow PowerDraw to prevent recharge
        if (TryGetBatteryFromSlot(ent.Owner, out var battery) && battery is { } existingBattery)
        {
            _battery.RefreshChargeRate(existingBattery.AsNullable());

            if (ent.Comp != null && ent.Comp.PauseRecharging)
                _battery.TrySetChargeCooldown(existingBattery.Owner);
        }
        // END DeltaV
    }


    /// <summary>
    /// Returns whether the entity has a slotted battery and <see cref="PowerCellDrawComponent.UseCharge"/> charge.
    /// </summary>
    /// <param name="ent">The device with the power cell slot.</param>
    /// <param name="user">Show a popup to this user with the relevant details if specified.</param>
    /// <param name="user">Whether to predict the popup or not.</param>
    [PublicAPI]
    public bool HasActivatableCharge(Entity<PowerCellDrawComponent?, PowerCellSlotComponent?> ent, EntityUid? user = null, bool predicted = false)
    {
        // Default to true if we don't have the components.
        if (!Resolve(ent, ref ent.Comp1, ref ent.Comp2, false))
            return true;

        return HasCharge((ent, ent.Comp2), ent.Comp1.UseCharge, user, predicted);
    }

    /// <summary>
    /// Tries to use the <see cref="PowerCellDrawComponent.UseCharge"/> for this entity.
    /// </summary>
    /// <param name="ent">The device with the power cell slot.</param>
    /// <param name="user">Show a popup to this user with the relevant details if specified.</param>
    /// <param name="user">Whether to predict the popup or not.</param>
    [PublicAPI]
    public bool TryUseActivatableCharge(Entity<PowerCellDrawComponent?, PowerCellSlotComponent?> ent, EntityUid? user = null, bool predicted = false)
    {
        // Default to true if we don't have the components.
        if (!Resolve(ent, ref ent.Comp1, ref ent.Comp2, false))
            return true;

        if (TryUseCharge((ent, ent.Comp2), ent.Comp1.UseCharge, user, predicted))
            return true;

        return false;
    }

    /// <summary>
    /// Whether the power cell has any power at all for the draw rate.
    /// </summary>
    /// <param name="ent">The device with the power cell slot.</param>
    /// <param name="user">Show a popup to this user with the relevant details if specified.</param>
    /// <param name="user">Whether to predict the popup or not.</param>
    [PublicAPI]
    public bool HasDrawCharge(Entity<PowerCellDrawComponent?, PowerCellSlotComponent?> ent, EntityUid? user = null, bool predicted = false)
    {
        // Default to true if we don't have the components.
        if (!Resolve(ent, ref ent.Comp1, ref ent.Comp2, false))
            return true;

        // 1 second of charge at the required draw rate.
        return HasCharge((ent, ent.Comp2), ent.Comp1.DrawRate, user, predicted);
    }
}