From d7810c4cd0d0dbad6c2e5b231c4a749a60640ced Mon Sep 17 00:00:00 2001
From: Dragory <2606411+Dragory@users.noreply.github.com>
Date: Sun, 26 Sep 2021 00:25:03 +0300
Subject: [PATCH] Add missing register/clear for expiring tempbans in !ban

---
 backend/src/plugins/ModActions/commands/BanCmd.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/backend/src/plugins/ModActions/commands/BanCmd.ts b/backend/src/plugins/ModActions/commands/BanCmd.ts
index 40a352c7..0371a3df 100644
--- a/backend/src/plugins/ModActions/commands/BanCmd.ts
+++ b/backend/src/plugins/ModActions/commands/BanCmd.ts
@@ -15,6 +15,7 @@ import { isBanned } from "../functions/isBanned";
 import { readContactMethodsFromArgs } from "../functions/readContactMethodsFromArgs";
 import { modActionsCmd } from "../types";
 import { LogsPlugin } from "../../Logs/LogsPlugin";
+import { clearExpiringTempban, registerExpiringTempban } from "../../../data/loops/expiringTempbansLoop";
 
 const opts = {
   mod: ct.member({ option: true }),
@@ -92,11 +93,14 @@ export const BanCmd = modActionsCmd({
           // Update or add new tempban / remove old tempban
           if (time && time > 0) {
             if (existingTempban) {
-              pluginData.state.tempbans.updateExpiryTime(user.id, time, mod.id);
+              await pluginData.state.tempbans.updateExpiryTime(user.id, time, mod.id);
             } else {
-              pluginData.state.tempbans.addTempban(user.id, time, mod.id);
+              await pluginData.state.tempbans.addTempban(user.id, time, mod.id);
             }
+            const tempban = (await pluginData.state.tempbans.findExistingTempbanForUserId(user.id))!;
+            registerExpiringTempban(tempban);
           } else if (existingTempban) {
+            clearExpiringTempban(existingTempban);
             pluginData.state.tempbans.clear(user.id);
           }