From 0eeee250ed37b07a9de7f50223fe5aa91c7db6c7 Mon Sep 17 00:00:00 2001
From: n9k <originalcocaineninja@protonmail.com>
Date: Wed, 16 Feb 2022 10:07:17 +0000
Subject: [PATCH] Fix js removing messages

---
 anonstream/static/anonstream.js | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/anonstream/static/anonstream.js b/anonstream/static/anonstream.js
index a7bd6a8..249cab8 100644
--- a/anonstream/static/anonstream.js
+++ b/anonstream/static/anonstream.js
@@ -127,12 +127,17 @@ const on_websocket_message = (event) => {
       update_user_styles();
 
       const ids = new Set(receipt.chat.map((message) => {return message.id;}));
+      const to_delete = [];
       for (const chat_message of chat_messages.children) {
-        if (!ids.has(parseInt(chat_message.dataset.id))) {
-          console.log('removing', chat_message);
-          chat_message.remove();
+        const chat_message_id = parseInt(chat_message.dataset.id);
+        if (!ids.has(chat_message_id)) {
+          to_delete.push(chat_message);
         }
       }
+      for (const chat_message of to_delete) {
+        chat_message.remove();
+      }
+
       const last_id = Math.max(...[...chat_messages.children].map((element) => parseInt(element.dataset.id)));
       for (const message of receipt.chat) {
         if (message.id > last_id) {