Check cooldown for IShock

OpenShock · Apr 24, 2024 · 716eae4 · 716eae4
1 parent ba28093
commit 716eae4
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/ShockOsc/Services/ShockOsc.cs b/ShockOsc/Services/ShockOsc.cs
@@ -293,7 +293,6 @@ private async Task ReceiveLogic()
         switch (action)
         {
             case "IShock":
-                // TODO: check Cooldowns
                 if (value is not true) return;
                 if (_underscoreConfig.KillSwitch)
                 {
@@ -308,6 +307,21 @@ private async Task ReceiveLogic()
                     await LogIgnoredAfk();
                     return;
                 }
+
+                var cooldownTime = _configManager.Config.Behaviour.CooldownTime;
+                if(programGroup.ConfigGroup is { OverrideCooldownTime: true }) 
+                    cooldownTime = programGroup.ConfigGroup.CooldownTime;
+
+                var isActiveOrOnCooldown =
+                    programGroup.LastExecuted.AddMilliseconds(cooldownTime)
+                        .AddMilliseconds(programGroup.LastDuration) > DateTime.UtcNow;
+
+                if (isActiveOrOnCooldown)
+                {
+                    programGroup.TriggerMethod = TriggerMethod.None;
+                    _logger.LogInformation("Ignoring IShock, group {Group} is on cooldown", programGroup.Name);
+                    return;
+                }
 
                 OsTask.Run(() => InstantShock(programGroup, GetDuration(programGroup), GetIntensity(programGroup)));