fix(useCooldownTimer): derive cooldown from last message (#1879)

Author: arnautov-anton

src/components/MessageInput/hooks/useCooldownTimer.tsx

@@ -1,4 +1,4 @@
-import React, { useEffect, useState } from 'react';
+import React, { useEffect, useMemo, useState } from 'react';
 
 import { useChatContext } from '../../../context/ChatContext';
 import { useChannelStateContext } from '../../../context/ChannelStateContext';
@@ -17,30 +17,36 @@ export const useCooldownTimer = <
   StreamChatGenerics extends DefaultStreamChatGenerics = DefaultStreamChatGenerics
 >(): CooldownTimerState => {
   const { latestMessageDatesByChannels } = useChatContext<StreamChatGenerics>('useCooldownTimer');
-  const { channel } = useChannelStateContext<StreamChatGenerics>('useCooldownTimer');
+  const { channel, messages = [] } = useChannelStateContext<StreamChatGenerics>('useCooldownTimer');
+  const { client } = useChatContext<StreamChatGenerics>('useCooldownTimer');
+  const [cooldownRemaining, setCooldownRemaining] = useState<number>();
 
   const { cooldown: cooldownInterval, own_capabilities } = (channel.data ||
     {}) as ChannelResponse<StreamChatGenerics>;
 
-  const [cooldownRemaining, setCooldownRemaining] = useState<number>();
-
   const skipCooldown = !own_capabilities?.includes('slow-mode');
 
+  const ownLatestMessageDate = useMemo(
+    () =>
+      latestMessageDatesByChannels[channel.cid] ??
+      [...messages]
+        .sort((a, b) => (b.created_at as Date)?.getTime() - (a.created_at as Date)?.getTime())
+        .find((v) => v.user?.id === client.user?.id)?.created_at,
+    [messages, client.user?.id, latestMessageDatesByChannels, channel.cid],
+  ) as Date;
+
   useEffect(() => {
-    const latestMessageDate = latestMessageDatesByChannels[channel.cid];
-    if (!cooldownInterval || !latestMessageDate) {
-      return;
-    }
+    if (skipCooldown || !cooldownInterval || !ownLatestMessageDate) return;
+
     const remainingCooldown = Math.round(
-      cooldownInterval - (new Date().getTime() - latestMessageDate.getTime()) / 1000,
+      cooldownInterval - (new Date().getTime() - ownLatestMessageDate.getTime()) / 1000,
     );
-    if (remainingCooldown > 0 && !skipCooldown) {
-      setCooldownRemaining(remainingCooldown);
-    }
-  }, [channel.id, cooldownInterval, latestMessageDatesByChannels[channel.cid]]);
+
+    if (remainingCooldown > 0) setCooldownRemaining(remainingCooldown);
+  }, [cooldownInterval, ownLatestMessageDate, skipCooldown]);
 
   return {
-    cooldownInterval: cooldownInterval || 0,
+    cooldownInterval: cooldownInterval ?? 0,
     cooldownRemaining,
     setCooldownRemaining,
   };