fish-station/Content.Server/Vampire/VampireSystem.cs
2024-12-12 20:48:50 +02:00

380 lines
No EOL
16 KiB
C#

using Content.Server.Administration.Logs;
using Content.Server.Atmos.Rotting;
using Content.Server.Beam;
using Content.Server.Body.Systems;
using Content.Server.Chat.Systems;
using Content.Server.Interaction;
using Content.Server.Nutrition.EntitySystems;
using Content.Server.Polymorph.Systems;
using Content.Server.Storage.EntitySystems;
using Content.Server.Mind;
using Content.Shared.Actions;
using Content.Shared.Body.Systems;
using Content.Shared.Buckle;
using Content.Shared.Bed.Sleep;
using Content.Shared.Chemistry.EntitySystems;
using Content.Shared.Construction.Components;
using Content.Shared.Damage;
using Content.Shared.DoAfter;
using Content.Shared.Examine;
using Content.Shared.FixedPoint;
using Content.Shared.Hands.EntitySystems;
using Content.Shared.Humanoid;
using Content.Shared.Interaction;
using Content.Shared.Interaction.Events;
using Content.Shared.Maps;
using Content.Shared.Mobs;
using Content.Shared.Mobs.Systems;
using Content.Shared.Popups;
using Content.Shared.Prayer;
using Content.Shared.StatusEffect;
using Content.Shared.Stunnable;
using Content.Shared.Vampire;
using Content.Shared.Vampire.Components;
using Robust.Server.GameObjects;
using Robust.Shared.Player;
using Robust.Shared.GameStates;
using Robust.Shared.Audio.Systems;
using Robust.Shared.Containers;
using Robust.Shared.Map;
using Robust.Shared.Prototypes;
using System.Linq;
namespace Content.Server.Vampire;
public sealed partial class VampireSystem : EntitySystem
{
[Dependency] private readonly MindSystem _mind = default!;
[Dependency] private readonly IAdminLogManager _admin = default!;
[Dependency] private readonly FoodSystem _food = default!;
[Dependency] private readonly EntityStorageSystem _entityStorage = default!;
[Dependency] private readonly BloodstreamSystem _blood = default!;
[Dependency] private readonly RottingSystem _rotting = default!;
[Dependency] private readonly StomachSystem _stomach = default!;
[Dependency] private readonly PolymorphSystem _polymorph = default!;
[Dependency] private readonly ChatSystem _chat = default!;
[Dependency] private readonly BeamSystem _beam = default!;
[Dependency] private readonly SharedInteractionSystem _interaction = default!;
[Dependency] private readonly SharedDoAfterSystem _doAfter = default!;
[Dependency] private readonly SharedAudioSystem _audio = default!;
[Dependency] private readonly SharedTransformSystem _transform = default!;
[Dependency] private readonly SharedPopupSystem _popup = default!;
[Dependency] private readonly IMapManager _mapMan = default!;
[Dependency] private readonly SharedMapSystem _mapSystem = default!;
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
[Dependency] private readonly SharedActionsSystem _action = default!;
[Dependency] private readonly ActionContainerSystem _actionContainer = default!;
[Dependency] private readonly SharedBodySystem _body = default!;
[Dependency] private readonly SharedSolutionContainerSystem _solution = default!;
[Dependency] private readonly SharedStunSystem _stun = default!;
[Dependency] private readonly StatusEffectsSystem _statusEffects = default!;
[Dependency] private readonly DamageableSystem _damageableSystem = default!;
[Dependency] private readonly MobThresholdSystem _mobThreshold = default!;
[Dependency] private readonly MobStateSystem _mobState = default!;
[Dependency] private readonly EntityLookupSystem _entityLookup = default!;
[Dependency] private readonly SharedHandsSystem _hands = default!;
[Dependency] private readonly MetabolizerSystem _metabolism = default!;
[Dependency] private readonly UserInterfaceSystem _uiSystem = default!;
[Dependency] private readonly SharedVampireSystem _vampire = default!;
public override void Initialize()
{
base.Initialize();
SubscribeLocalEvent<VampireComponent, ComponentStartup>(OnComponentStartup);
//SubscribeLocalEvent<VampireComponent, VampireSelfPowerEvent>(OnUseSelfPower);
//SubscribeLocalEvent<VampireComponent, VampireTargetedPowerEvent>(OnUseTargetedPower);
SubscribeLocalEvent<VampireComponent, ExaminedEvent>(OnExamined);
SubscribeLocalEvent<VampireComponent, VampireBloodChangedEvent>(OnVampireBloodChangedEvent);
SubscribeLocalEvent<VampireComponent, AfterAutoHandleStateEvent>(GetState);
SubscribeLocalEvent<VampireComponent, VampireMutationPrototypeSelectedMessage>(OnMutationSelected);
InitializePowers();
InitializeObjectives();
}
/// <summary>
/// Handles healing and damaging in space
/// </summary>
public override void Update(float frameTime)
{
base.Update(frameTime);
var stealthQuery = EntityQueryEnumerator<VampireComponent, VampireSealthComponent>();
while (stealthQuery.MoveNext(out var uid, out var vampire, out var stealth))
{
if (vampire == null || stealth == null)
continue;
if (stealth.NextStealthTick <= 0)
{
stealth.NextStealthTick = 1;
if (!SubtractBloodEssence((uid, vampire), stealth.Upkeep))
RemCompDeferred<VampireSealthComponent>(uid);
}
stealth.NextStealthTick -= frameTime;
}
var healingQuery = EntityQueryEnumerator<VampireComponent, VampireHealingComponent>();
while (healingQuery.MoveNext(out var uid, out _, out var healing))
{
if (healing == null)
continue;
if (healing.NextHealTick <= 0)
{
healing.NextHealTick = 1;
DoCoffinHeal(uid, healing);
}
healing.NextHealTick -= frameTime;
}
/*var query = EntityQueryEnumerator<VampireComponent>();
while (query.MoveNext(out var uid, out var vampireComponent))
{
var vampire = (uid, vampireComponent);
if (IsInSpace(uid))
{
if (vampireComponent.NextSpaceDamageTick <= 0)
{
vampireComponent.NextSpaceDamageTick = 1;
DoSpaceDamage(vampire);
}
vampireComponent.NextSpaceDamageTick -= frameTime;
}
}*/
}
private void OnComponentStartup(EntityUid uid, VampireComponent component, ComponentStartup args)
{
//MakeVampire(uid);
}
private void OnExamined(EntityUid uid, VampireComponent component, ExaminedEvent args)
{
if (HasComp<VampireFangsExtendedComponent>(uid) && args.IsInDetailsRange && !_food.IsMouthBlocked(uid))
args.AddMarkup($"{Loc.GetString("vampire-fangs-extended-examine")}{Environment.NewLine}");
}
private bool AddBloodEssence(Entity<VampireComponent> vampire, FixedPoint2 quantity)
{
if (quantity < 0)
return false;
vampire.Comp.TotalBloodDrank += quantity.Float();
vampire.Comp.Balance[VampireComponent.CurrencyProto] += quantity;
UpdateBloodDisplay(vampire);
var ev = new VampireBloodChangedEvent();
RaiseLocalEvent(vampire, ev);
return true;
}
private bool SubtractBloodEssence(Entity<VampireComponent> vampire, FixedPoint2 quantity)
{
if (quantity < 0)
return false;
if (vampire.Comp.Balance[VampireComponent.CurrencyProto] < quantity)
return false;
vampire.Comp.Balance[VampireComponent.CurrencyProto] -= quantity;
UpdateBloodDisplay(vampire);
var ev = new VampireBloodChangedEvent();
RaiseLocalEvent(vampire, ev);
return true;
}
/// <summary>
/// Use the charges display on SummonHeirloom to show the remaining blood essence
/// </summary>
/// <param name="vampire"></param>
public void UpdateBloodDisplay(EntityUid vampire)
{
if (!TryComp<VampireComponent>(vampire, out var comp))
return;
//Sanity check, you never know who is going to touch this code
if (!comp.Balance.TryGetValue(VampireComponent.CurrencyProto, out var balance))
return;
var chargeDisplay = (int) Math.Round((decimal) balance);
var mutationsAction = GetPowerEntity(comp, VampireComponent.MutationsActionPrototype);
if (mutationsAction == null)
return;
_action.SetCharges(mutationsAction, chargeDisplay);
}
private void OnVampireBloodChangedEvent(EntityUid uid, VampireComponent component, VampireBloodChangedEvent args)
{
var bloodEssence = _vampire.GetBloodEssence(uid);
AbilityInfo entity = default;
if (TryComp<VampireAlertComponent>(uid, out var alertComp))
_vampire.SetAlertBloodAmount(alertComp,_vampire.GetBloodEssence(uid).Int());
if (component.actionEntities.TryGetValue("ActionVampireCloakOfDarkness", out entity) && !HasComp<VampireSealthComponent>(uid) && _vampire.GetBloodEssence(uid) < FixedPoint2.New(300))
component.actionEntities.Remove("ActionVampireCloakOfDarkness");
UpdateAbilities(uid, component , VampireComponent.MutationsActionPrototype, null , bloodEssence >= FixedPoint2.New(50) && !HasComp<VampireSealthComponent>(uid));
//Hemomancer
// Blood Steal
UpdateAbilities(uid, component , "ActionVampireBloodSteal", "BloodSteal" , bloodEssence >= FixedPoint2.New(200) && component.CurrentMutation == VampireMutationsType.Hemomancer);
// Screech
UpdateAbilities(uid, component , "ActionVampireScreech", "Screech" , bloodEssence >= FixedPoint2.New(300) && component.CurrentMutation == VampireMutationsType.Hemomancer);
//Umbrae
//Glare
UpdateAbilities(uid, component , "ActionVampireGlare", "Glare" , bloodEssence >= FixedPoint2.New(200) && component.CurrentMutation == VampireMutationsType.Umbrae);
//CloakOfDarkness
UpdateAbilities(uid, component , "ActionVampireCloakOfDarkness", "CloakOfDarkness" , bloodEssence >= FixedPoint2.New(300) && component.CurrentMutation == VampireMutationsType.Umbrae);
//Gargantua
/*
if (_vampire.GetBloodEssence(uid) >= FixedPoint2.New(200) && !_actionEntities.TryGetValue("ActionVampireUnnaturalStrength", out entity) && component.CurrentMutation == VampireMutationsType.Gargantua)
{
var vampire = new Entity<VampireComponent>(uid, component);
UnnaturalStrength(vampire);
_actionEntities["ActionVampireUnnaturalStrength"] = vampire;
}
if (_vampire.GetBloodEssence(uid) >= FixedPoint2.New(300) && !_actionEntities.TryGetValue("ActionVampireSupernaturalStrength", out entity) && component.CurrentMutation == VampireMutationsType.Gargantua)
{
var vampire = new Entity<VampireComponent>(uid, component);
SupernaturalStrength(vampire);
_actionEntities["ActionVampireSupernaturalStrength"] = vampire;
}
*/
//Bestia
UpdateAbilities(uid, component , "ActionVampireBatform", "PolymorphBat" , bloodEssence >= FixedPoint2.New(200) && component.CurrentMutation == VampireMutationsType.Bestia);
UpdateAbilities(uid, component , "ActionVampireMouseform", "PolymorphMouse" , bloodEssence >= FixedPoint2.New(300) && component.CurrentMutation == VampireMutationsType.Bestia);
}
private void UpdateAbilities(EntityUid uid, VampireComponent component, string actionId, string? powerId, bool addAction)
{
EntityUid? actionEntity = null;
if (addAction)
{
if (!component.actionEntities.ContainsKey(actionId))
{
_action.AddAction(uid, ref actionEntity, actionId);
if (actionEntity != null)
{
component.actionEntities[actionId] = new AbilityInfo(uid, actionEntity.Value);
if (powerId != null && !component.UnlockedPowers.ContainsKey(powerId))
component.UnlockedPowers.Add(powerId, actionEntity.Value);
}
}
}
else
{
if (component.actionEntities.TryGetValue(actionId, out var abilityInfo) && abilityInfo.Owner == uid)
{
if (TryComp(uid, out ActionsComponent? comp))
{
_action.RemoveAction(uid, abilityInfo.Action, comp);
component.actionEntities.Remove(actionId);
if (powerId != null && component.UnlockedPowers.ContainsKey(powerId))
component.UnlockedPowers.Remove(powerId);
}
}
}
}
private void DoSpaceDamage(Entity<VampireComponent> vampire)
{
_damageableSystem.TryChangeDamage(vampire, VampireComponent.SpaceDamage, true, origin: vampire);
_popup.PopupEntity(Loc.GetString("vampire-startlight-burning"), vampire, vampire, PopupType.LargeCaution);
}
private bool IsInSpace(EntityUid vampireUid)
{
var vampireTransform = Transform(vampireUid);
var vampirePosition = _transform.GetMapCoordinates(vampireTransform);
if (!_mapMan.TryFindGridAt(vampirePosition, out _, out var grid))
return true;
if (!_mapSystem.TryGetTileRef(vampireUid, grid, vampireTransform.Coordinates, out var tileRef))
return true;
return tileRef.Tile.IsEmpty || tileRef.IsSpace();
}
private bool IsNearPrayable(EntityUid vampireUid)
{
var mapCoords = _transform.GetMapCoordinates(vampireUid);
var nearbyPrayables = _entityLookup.GetEntitiesInRange<PrayableComponent>(mapCoords, 5);
foreach (var prayable in nearbyPrayables)
{
if (Transform(prayable).Anchored)
return true;
}
return false;
}
private void OnMutationSelected(EntityUid uid, VampireComponent component, VampireMutationPrototypeSelectedMessage args)
{
if (component.CurrentMutation == args.SelectedId)
return;
ChangeMutation(uid, args.SelectedId, component);
}
private void ChangeMutation(EntityUid uid, VampireMutationsType newMutation, VampireComponent component)
{
var vampire = new Entity<VampireComponent>(uid, component);
if (SubtractBloodEssence(vampire, FixedPoint2.New(50)))
{
component.CurrentMutation = newMutation;
UpdateUi(uid, component);
var ev = new VampireBloodChangedEvent();
RaiseLocalEvent(uid, ev);
TryOpenUi(uid, component.Owner, component);
}
}
private void GetState(EntityUid uid, VampireComponent component, ref AfterAutoHandleStateEvent args)
{
args.State = new VampireMutationComponentState
{
SelectedMutation = component.CurrentMutation
};
}
private void TryOpenUi(EntityUid uid, EntityUid user, VampireComponent? component = null)
{
if (!Resolve(uid, ref component))
return;
if (!TryComp(user, out ActorComponent? actor))
return;
_uiSystem.TryToggleUi(uid, VampireMutationUiKey.Key, actor.PlayerSession);
}
public void UpdateUi(EntityUid uid, VampireComponent? component = null)
{
if (!Resolve(uid, ref component))
return;
var state = new VampireMutationBoundUserInterfaceState(component.VampireMutations, component.CurrentMutation);
_uiSystem.SetUiState(uid, VampireMutationUiKey.Key, state);
}
}