add(SharedDoAfterSystem): Add a IsRunning "Utility" function (space-w…

…izards#30512)
Errant-4 · Jul 31, 2024 · 5bc9c04 · 5bc9c04
1 parent 778bfe3
commit 5bc9c04
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/Content.Shared/DoAfter/SharedDoAfterSystem.cs b/Content.Shared/DoAfter/SharedDoAfterSystem.cs
@@ -393,5 +393,18 @@ public DoAfterStatus GetStatus(EntityUid entity, ushort id, DoAfterComponent? co
         // networking whether a do-after has raised its events or not.
         return DoAfterStatus.Finished;
     }
+
+    public bool IsRunning(DoAfterId? id, DoAfterComponent? comp = null)
+    {
+        if (id == null)
+            return false;
+
+        return GetStatus(id.Value.Uid, id.Value.Index, comp) == DoAfterStatus.Running;
+    }
+
+    public bool IsRunning(EntityUid entity, ushort id, DoAfterComponent? comp = null)
+    {
+        return GetStatus(entity, id, comp) == DoAfterStatus.Running;
+    }
     #endregion
 }