diff --git a/Modules/DisableDevice.cs b/Modules/DisableDevice.cs index 136a8aef7..cb709cb5f 100644 --- a/Modules/DisableDevice.cs +++ b/Modules/DisableDevice.cs @@ -6,6 +6,7 @@ using HarmonyLib; using Hazel; using Il2CppInterop.Runtime.InteropTypes.Arrays; using UnityEngine; +using static EHR.Modules.LevelImposterCompatibility; namespace EHR; @@ -50,16 +51,20 @@ internal static class DisableDevice int adminId = 0; int cameraId = 0; + int binoId = 0; int vitalId = 0; - foreach (var admin in Object.FindObjectsOfType(true)) + foreach (var admin in AllAdminTables) DevicePos[$"LIAdmin{adminId++}"] = admin.transform.position; - foreach (var console in Object.FindObjectsOfType(true)) - { - if (console.name == "Surv_Panel") DevicePos[$"LICamera{cameraId++}"] = console.transform.position; - else if (console.name.Contains("Vital")) DevicePos[$"LIVital{vitalId++}"] = console.transform.position; - } + foreach (var camera in AllCameraPanels) + DevicePos[$"LICamera{cameraId++}"] = camera.transform.position; + + foreach (var bino in AllBinoculars) + DevicePos[$"LIBinocular{binoId++}"] = bino.transform.position; + + foreach (var vital in AllVitals) + DevicePos[$"LIVital{vitalId++}"] = vital.transform.position; } private static bool DoDisable => Options.DisableDevices.GetBool(); diff --git a/Roles/Standard/Crewmate/Investigate/Sentry.cs b/Roles/Standard/Crewmate/Investigate/Sentry.cs index 66b3fdf3a..6c15e4ec3 100644 --- a/Roles/Standard/Crewmate/Investigate/Sentry.cs +++ b/Roles/Standard/Crewmate/Investigate/Sentry.cs @@ -106,8 +106,8 @@ internal class Sentry : RoleBase { AdditionalDevicesStrings.None => false, AdditionalDevicesStrings.DoorLog => x.Key.Contains("DoorLog"), - AdditionalDevicesStrings.Binoculars => x.Key.Contains("Camera") && x.Key.Contains("Fungle"), - AdditionalDevicesStrings.DoorLogAndBinoculars => x.Key.Contains("DoorLog") || (x.Key.Contains("Camera") && x.Key.Contains("Fungle")), + AdditionalDevicesStrings.Binoculars => x.Key.Contains("Camera") && x.Key.Contains("Fungle") || x.Key.Contains("Binocular"), + AdditionalDevicesStrings.DoorLogAndBinoculars => x.Key.Contains("DoorLog") || (x.Key.Contains("Camera") && x.Key.Contains("Fungle")) || x.Key.Contains("Binocular"), _ => false }; }