using System.Linq; using Content.Shared.Drone; using Content.Server.Drone.Components; using Content.Shared.Actions; using Content.Server.Light.Components; using Content.Shared.MobState; using Content.Shared.Interaction.Events; using Content.Shared.Interaction.Components; using Content.Shared.Examine; using Content.Shared.Tag; using Content.Shared.Throwing; using Content.Shared.Item; using Content.Shared.Emoting; using Content.Shared.Body.Components; using Content.Server.Popups; using Content.Server.Mind.Components; using Content.Server.Ghost.Components; using Content.Server.Ghost.Roles.Components; using Content.Server.Hands.Components; using Content.Server.UserInterface; using Robust.Shared.Player; using Content.Shared.Hands.EntitySystems; using Robust.Shared.Timing; namespace Content.Server.Drone { public sealed class DroneSystem : SharedDroneSystem { [Dependency] private readonly PopupSystem _popupSystem = default!; [Dependency] private readonly TagSystem _tagSystem = default!; [Dependency] private readonly EntityLookupSystem _lookup = default!; [Dependency] private readonly SharedHandsSystem _handsSystem = default!; [Dependency] private readonly SharedActionsSystem _actionsSystem = default!; [Dependency] private readonly IGameTiming _gameTiming = default!; public override void Initialize() { base.Initialize(); SubscribeLocalEvent(OnInteractionAttempt); SubscribeLocalEvent(OnActivateUIAttempt); SubscribeLocalEvent(OnMobStateChanged); SubscribeLocalEvent(OnExamined); SubscribeLocalEvent(OnMindAdded); SubscribeLocalEvent(OnMindRemoved); SubscribeLocalEvent(OnEmoteAttempt); SubscribeLocalEvent(OnThrowAttempt); } private void OnInteractionAttempt(EntityUid uid, DroneComponent component, InteractionAttemptEvent args) { if (!component.ApplyLaws) return; if (args.Target != null && !HasComp(args.Target) && NonDronesInRange(uid, component)) args.Cancel(); if (HasComp(args.Target) && !HasComp(args.Target)) { if (!_tagSystem.HasAnyTag(args.Target.Value, "DroneUsable", "Trash")) args.Cancel(); } } private void OnActivateUIAttempt(EntityUid uid, DroneComponent component, UserOpenActivatableUIAttemptEvent args) { if (!component.ApplyLaws) return; if (!_tagSystem.HasTag(args.Target, "DroneUsable")) { args.Cancel(); } } private void OnExamined(EntityUid uid, DroneComponent component, ExaminedEvent args) { if (args.IsInDetailsRange) { if (TryComp(uid, out var mind) && mind.HasMind) { args.PushMarkup(Loc.GetString("drone-active")); } else { args.PushMarkup(Loc.GetString("drone-dormant")); } } } private void OnMobStateChanged(EntityUid uid, DroneComponent drone, MobStateChangedEvent args) { if (args.Component.IsDead()) { var body = Comp(uid); //There's no way something can have a mobstate but not a body... foreach (var item in drone.ToolUids.Select((value, i) => ( value, i ))) { if (_tagSystem.HasTag(item.value, "Drone")) { RemComp(item.value); } else { EntityManager.DeleteEntity(item.value); } } body.Gib(); EntityManager.DeleteEntity(uid); } } private void OnMindAdded(EntityUid uid, DroneComponent drone, MindAddedMessage args) { UpdateDroneAppearance(uid, DroneStatus.On); _popupSystem.PopupEntity(Loc.GetString("drone-activated"), uid, Filter.Pvs(uid)); if (drone.AlreadyAwoken == false) { var spawnCoord = Transform(uid).Coordinates; if (drone.Tools.Count == 0) return; if (TryComp(uid, out var hands) && hands.Count >= drone.Tools.Count) { foreach (var entry in drone.Tools) { var item = EntityManager.SpawnEntity(entry.PrototypeId, spawnCoord); AddComp(item); if (!_handsSystem.TryPickupAnyHand(uid, item, checkActionBlocker: false)) { QueueDel(item); Logger.Error($"Drone ({ToPrettyString(uid)}) failed to pick up innate item ({ToPrettyString(item)})"); continue; } drone.ToolUids.Add(item); } } if (TryComp(uid, out var actions) && TryComp(uid, out var flashlight)) { _actionsSystem.AddAction(uid, flashlight.ToggleAction, null, actions); } drone.AlreadyAwoken = true; } } private void OnMindRemoved(EntityUid uid, DroneComponent drone, MindRemovedMessage args) { UpdateDroneAppearance(uid, DroneStatus.Off); EnsureComp(uid); } private void OnEmoteAttempt(EntityUid uid, DroneComponent component, EmoteAttemptEvent args) { // No. args.Cancel(); } private void OnThrowAttempt(EntityUid uid, DroneComponent drone, ThrowAttemptEvent args) { args.Cancel(); } private void UpdateDroneAppearance(EntityUid uid, DroneStatus status) { if (TryComp(uid, out var appearance)) { appearance.SetData(DroneVisuals.Status, status); } } private bool NonDronesInRange(EntityUid uid, DroneComponent component) { var xform = Comp(uid); foreach (var entity in _lookup.GetEntitiesInRange(xform.MapID, xform.WorldPosition, component.InteractionBlockRange)) { if (HasComp(entity) && !HasComp(entity) && !HasComp(entity)) { if (_gameTiming.IsFirstTimePredicted) _popupSystem.PopupEntity(Loc.GetString("drone-too-close", ("being", entity)), uid, Filter.Entities(uid)); return true; } } return false; } } }