Jesus-QC · January 5, 2024 02:01 · Jan 5, 2024
diff --git a/JanitorSubclass.cs b/JanitorSubclass.cs
@@ -0,0 +1,75 @@
+using System.Collections.Generic;
+using System.Globalization;
+using CursedMod.Events.Arguments.Facility.Hazards;
+using CursedMod.Events.Handlers;
+using CursedMod.Features.Enums;
+using Mirror;
+using UnityEngine;
+
+namespace OriginsSL.Modules.Subclasses.DefinedClasses.ClassD;
+
+public class JanitorSubclass : SubclassBase
+{
+    private const float TantrumCleaningTime = 3f;
+
+    public override string CodeName => "janitor";
+    public override string Name => "<color=#c0b2e6>J<lowercase>anitor</lowercase></color>";
+    public override string Description => _counter > 0 ? $"cleaning tantrum {(_counter/TantrumCleaningTime*100).ToString(CultureInfo.InvariantCulture).Substring(0, 2)}%" : "cleans up tantrums by standing on them, also immune to sinkholes";
+
+    public override float SpawnChance => 0.7f;
+    public override List<ItemType> AdditiveInventory { get; } = [ItemType.KeycardJanitor];
+
+    public override bool KeepAfterEscaping => true;
+
+    // Counter for the tantrum cleaning
+    private float _counter;
+
+    public class JanitorSubclassHandler : ISubclassEventsHandler
+    {
+        public void OnLoaded()
+        {
+            CursedHazardsEventHandler.EnteringHazard += OnPlayerEnteringHazard;
+            CursedHazardsEventHandler.StayingOnHazard += OnPlayerStayingOnHazard;
+            CursedHazardsEventHandler.ExitingHazard += OnPlayerExitingHazard;
+        }
+
+        private static void OnPlayerEnteringHazard(PlayerEnteringHazardEventArgs args)
+        {
+            if (args.Hazard.HazardType != EnvironmentalHazardType.Sinkhole)
+                return;
+
+            if (!args.Player.TryGetSubclass(out ISubclass subclass) || subclass is not JanitorSubclass)
+                return;
+
+            args.IsAllowed = false;
+        }
+
+        private static void OnPlayerStayingOnHazard(PlayerStayingOnHazardEventArgs args)
+        {
+            if (args.Hazard.HazardType != EnvironmentalHazardType.Tantrum)
+                return;
+
+            if (!args.Player.TryGetSubclass(out ISubclass subclass) || subclass is not JanitorSubclass janitorSubclass)
+                return;
+
+            janitorSubclass._counter += Time.deltaTime;
+
+            if (janitorSubclass._counter < TantrumCleaningTime)
+                return;
+
+            janitorSubclass._counter = 0;
+            NetworkServer.Destroy(args.Hazard.EnvironmentalHazard.gameObject);
+        }
+
+        private static void OnPlayerExitingHazard(PlayerExitingHazardEventArgs args)
+        {
+            if (args.Hazard.HazardType != EnvironmentalHazardType.Tantrum)
+                return;
+
+            if (!args.Player.TryGetSubclass(out ISubclass subclass) || subclass is not JanitorSubclass janitorSubclass)
+                return;
+
+            janitorSubclass._counter = 0;
+        }
+    }
+}