fix: refactor code fix

Author: falsaadehh

packages/super-editor/src/extensions/comment/comments-plugin.js

@@ -1,5 +1,4 @@
-import { Plugin, PluginKey } from 'prosemirror-state';
-import { TextSelection } from 'prosemirror-state';
+import { Plugin, PluginKey, TextSelection } from 'prosemirror-state';
 import { Extension } from '@core/Extension.js';
 import { Decoration, DecorationSet } from 'prosemirror-view';
 import { removeCommentsById, getHighlightColor } from './comments-helpers.js';
@@ -237,28 +236,21 @@ export const CommentsPlugin = Extension.create({
               shouldUpdate = true;
             }
 
-            if (prevActiveThreadId !== currentActiveThreadId) {
-              shouldUpdate = true;
-              prevActiveThreadId = currentActiveThreadId;
-            }
-
             // Check if document changed
             if (prevDoc && !prevDoc.eq(doc)) shouldUpdate = true;
 
-            // Check if active thread changed - this is the key fix
+            // Check if active thread changed
             if (prevActiveThreadId !== currentActiveThreadId) {
               shouldUpdate = true;
               prevActiveThreadId = currentActiveThreadId;
             }
 
             if (!shouldUpdate) return;
-
             prevDoc = doc;
             shouldUpdate = false;
 
             const decorations = [];
             const allCommentPositions = {};
-
             doc.descendants((node, pos) => {
               const { marks = [] } = node;
               const commentMarks = marks.filter((mark) => mark.type.name === CommentMarkName);
@@ -289,11 +281,6 @@ export const CommentsPlugin = Extension.create({
                   editor,
                 });
 
-                // Debug logging to verify color calculation
-                if (currentActiveThreadId === threadId) {
-                  console.log('Active comment color:', { threadId, color, isInternal, currentActiveThreadId });
-                }
-
                 const deco = Decoration.inline(pos, pos + node.nodeSize, {
                   style: `background-color: ${color};`,
                   'data-thread-id': threadId,
@@ -321,7 +308,7 @@ export const CommentsPlugin = Extension.create({
                   currentBounds,
                   node,
                 });
-                // Add decoration for tracked changes
+                // Add decoration for tracked changes when activated
                 const isActiveTrackedChange = currentActiveThreadId === id;
                 if (isActiveTrackedChange) {
                   const trackedChangeDeco = Decoration.inline(pos, pos + node.nodeSize, {
@@ -674,13 +661,11 @@ function findRangeById(doc, id) {
   let from = null,
     to = null;
   doc.descendants((node, pos) => {
-    // Check for tracked change marks
     const trackedMark = node.marks.find((m) => TRACK_CHANGE_MARKS.includes(m.type.name) && m.attrs.id === id);
     if (trackedMark) {
       if (from === null || pos < from) from = pos;
       if (to === null || pos + node.nodeSize > to) to = pos + node.nodeSize;
     }
-    // Check for comment marks
     const commentMark = node.marks.find(
       (m) => m.type.name === CommentMarkName && (m.attrs.commentId === id || m.attrs.importedId === id),
     );