From 244372803bff10fbd2dd5658378b68efeed6466a Mon Sep 17 00:00:00 2001
From: unintendedfraud <briceunintendedfraud@fastmail.com>
Date: Fri, 5 Apr 2024 22:17:41 +0200
Subject: [PATCH] some fix

---
 handlers/message_created.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/handlers/message_created.go b/handlers/message_created.go
index 2b05647..da8d931 100644
--- a/handlers/message_created.go
+++ b/handlers/message_created.go
@@ -143,14 +143,18 @@ func (h Handler) updateUserHistoryCount(userID string) bool {
 		return true
 	}
 
+	if !u.bannedUntilAt.IsZero() {
+		u.bannedUntilAt = time.Time{}
+	}
+
 	if u.date.Add(SPAM_PERIOD).After(now) {
 		u.count = 1
 		u.date = now
-
 		return false
 	}
 
 	if u.count > 5 {
+		u.bannedUntilAt = now.Add(2 * time.Minute)
 		return true
 	}