using Content.Server.Light.Components; using Content.Server.Storage.Components; using Content.Shared.ActionBlocker; using Content.Shared.Interaction; using Content.Shared.Interaction.Events; using Content.Shared.Light; using Content.Shared.Popups; using JetBrains.Annotations; using Robust.Shared.Audio; using Robust.Shared.Containers; using Robust.Shared.GameObjects; using Robust.Shared.IoC; using Robust.Shared.Localization; using Robust.Shared.Player; using System; using System.Linq; namespace Content.Server.Light.EntitySystems { [UsedImplicitly] public class LightReplacerSystem : EntitySystem { [Dependency] private readonly ActionBlockerSystem _blocker = default!; [Dependency] private readonly PoweredLightSystem _poweredLight = default!; [Dependency] private readonly SharedPopupSystem _popupSystem = default!; public override void Initialize() { base.Initialize(); SubscribeLocalEvent(OnInit); SubscribeLocalEvent(HandleInteract); SubscribeLocalEvent(HandleAfterInteract); } private void OnInit(EntityUid uid, LightReplacerComponent replacer, ComponentInit args) { replacer.InsertedBulbs = ContainerHelpers.EnsureContainer(replacer.Owner, "light_replacer_storage"); } private void HandleAfterInteract(EntityUid uid, LightReplacerComponent component, AfterInteractEvent eventArgs) { if (eventArgs.Handled) return; // standard interaction checks if (!_blocker.CanUse(eventArgs.User)) return; if (!eventArgs.CanReach) return; // behaviour will depends on target type if (eventArgs.Target != null) { var targetUid = eventArgs.Target.Uid; // replace broken light in fixture? if (EntityManager.TryGetComponent(targetUid, out PoweredLightComponent? fixture)) eventArgs.Handled = TryReplaceBulb(uid, targetUid, eventArgs.User.Uid, component, fixture); // add new bulb to light replacer container? else if (EntityManager.TryGetComponent(targetUid, out LightBulbComponent? bulb)) eventArgs.Handled = TryInsertBulb(uid, targetUid, eventArgs.User.Uid, true, component, bulb); } } private void HandleInteract(EntityUid uid, LightReplacerComponent component, InteractUsingEvent eventArgs) { if (eventArgs.Handled) return; // standard interaction checks if (!_blocker.CanInteract(eventArgs.User)) return; if (eventArgs.Used != null) { var usedUid = eventArgs.Used.Uid; // want to insert a new light bulb? if (EntityManager.TryGetComponent(usedUid, out LightBulbComponent ? bulb)) eventArgs.Handled = TryInsertBulb(uid, usedUid, eventArgs.User.Uid, true, component, bulb); // add bulbs from storage? else if (EntityManager.TryGetComponent(usedUid, out ServerStorageComponent? storage)) eventArgs.Handled = TryInsertBulbsFromStorage(uid, usedUid, eventArgs.User.Uid, component, storage); } } /// /// Try to replace a light bulb in /// using light replacer. Light fixture should have . /// /// True if successfully replaced light, false otherwise public bool TryReplaceBulb(EntityUid replacerUid, EntityUid fixtureUid, EntityUid? userUid = null, LightReplacerComponent? replacer = null, PoweredLightComponent? fixture = null) { if (!Resolve(replacerUid, ref replacer)) return false; if (!Resolve(fixtureUid, ref fixture)) return false; // check if light bulb is broken or missing var fixtureBulbUid = _poweredLight.GetBulb(fixture.Owner.Uid, fixture); if (fixtureBulbUid != null) { if (!EntityManager.TryGetComponent(fixtureBulbUid.Value, out LightBulbComponent? fixtureBulb)) return false; if (fixtureBulb.State == LightBulbState.Normal) return false; } // try get first inserted bulb of the same type as targeted light fixtutre var bulb = replacer.InsertedBulbs.ContainedEntities.FirstOrDefault( (e) => e.GetComponentOrNull()?.Type == fixture.BulbType); // found bulb in inserted storage if (bulb != null) { // try to remove it var hasRemoved = replacer.InsertedBulbs.Remove(bulb); if (!hasRemoved) return false; } // try to create new instance of bulb from LightReplacerEntity else { var bulbEnt = replacer.Contents.FirstOrDefault((e) => e.Type == fixture.BulbType && e.Amount > 0); // found right bulb, let's spawn it if (bulbEnt != null) { bulb = EntityManager.SpawnEntity(bulbEnt.PrototypeName, replacer.Owner.Transform.Coordinates); bulbEnt.Amount--; } // not found any light bulbs else { if (userUid != null) { var msg = Loc.GetString("comp-light-replacer-missing-light", ("light-replacer", replacer.Owner)); _popupSystem.PopupEntity(msg, replacerUid, Filter.Entities(userUid.Value)); } return false; } } // insert it into fixture var wasReplaced = _poweredLight.ReplaceBulb(fixtureUid, bulb.Uid, fixture); if (wasReplaced) { SoundSystem.Play(Filter.Pvs(replacerUid), replacer.Sound.GetSound(), replacerUid, AudioParams.Default.WithVolume(-4f)); } return wasReplaced; } /// /// Try to insert a new bulb inside light replacer /// /// True if successfully inserted light, false otherwise public bool TryInsertBulb(EntityUid replacerUid, EntityUid bulbUid, EntityUid? userUid = null, bool showTooltip = false, LightReplacerComponent? replacer = null, LightBulbComponent? bulb = null) { if (!Resolve(replacerUid, ref replacer)) return false; if (!Resolve(bulbUid, ref bulb)) return false; // only normal (non-broken) bulbs can be inserted inside light replacer if (bulb.State != LightBulbState.Normal) { if (showTooltip && userUid != null) { var msg = Loc.GetString("comp-light-replacer-insert-broken-light"); _popupSystem.PopupEntity(msg, replacerUid, Filter.Entities(userUid.Value)); } return false; } // try insert light and show message var hasInsert = replacer.InsertedBulbs.Insert(bulb.Owner); if (hasInsert && showTooltip && userUid != null) { var msg = Loc.GetString("comp-light-replacer-insert-light", ("light-replacer", replacer.Owner), ("bulb", bulb.Owner)); _popupSystem.PopupEntity(msg, replacerUid, Filter.Entities(userUid.Value)); } return hasInsert; } /// /// Try to insert all light bulbs from storage (for example light tubes box) /// /// /// Returns true if storage contained at least one light bulb /// which was successfully inserted inside light replacer /// public bool TryInsertBulbsFromStorage(EntityUid replacerUid, EntityUid storageUid, EntityUid? userUid = null, LightReplacerComponent? replacer = null, ServerStorageComponent? storage = null) { if (!Resolve(replacerUid, ref replacer)) return false; if (!Resolve(storageUid, ref storage)) return false; if (storage.StoredEntities == null) return false; var insertedBulbs = 0; var storagedEnts = storage.StoredEntities.ToArray(); foreach (var ent in storagedEnts) { if (EntityManager.TryGetComponent(ent.Uid, out LightBulbComponent? bulb)) { if (TryInsertBulb(replacerUid, ent.Uid, userUid, false, replacer, bulb)) insertedBulbs++; } } // show some message if success if (insertedBulbs > 0 && userUid != null) { var msg = Loc.GetString("comp-light-replacer-refill-from-storage", ("light-replacer", storage.Owner)); _popupSystem.PopupEntity(msg, replacerUid, Filter.Entities(userUid.Value)); } return insertedBulbs > 0; } } }