From 99785101391a7ab04518b5d1452ad30970ff7bc4 Mon Sep 17 00:00:00 2001
From: Rahul <rajamohanty12345@gmail.com>
Date: Sun, 3 Nov 2024 15:27:24 +0530
Subject: [PATCH] fix: removed redundant codes from useBlockHighlight

---
 src/core/components/canvas/static/Canvas.tsx | 63 ++++++--------------
 src/core/hooks/useBlockHighlight.ts          |  3 +-
 2 files changed, 21 insertions(+), 45 deletions(-)

diff --git a/src/core/components/canvas/static/Canvas.tsx b/src/core/components/canvas/static/Canvas.tsx
index 23f17d1c..3dd8ca79 100644
--- a/src/core/components/canvas/static/Canvas.tsx
+++ b/src/core/components/canvas/static/Canvas.tsx
@@ -1,6 +1,6 @@
 import React, { useEffect } from "react";
 import { useFrame } from "../../../frame";
-import { useSelectedBlockIds, useSelectedStylingBlocks, useUpdateBlocksProps } from "../../../hooks";
+import { useSelectedBlockIds, useSelectedStylingBlocks, useUpdateBlocksProps, useBlockHighlight } from "../../../hooks";
 import { first, isEmpty, omit, throttle } from "lodash-es";
 import { Quill } from "react-quill";
 import { useAtom } from "jotai";
@@ -47,6 +47,7 @@ const useHandleCanvasDblClick = () => {
   const INLINE_EDITABLE_BLOCKS = ["Heading", "Paragraph", "Text", "Link", "Span", "Button"];
   const updateContent = useUpdateBlocksProps();
   const [editingBlockId, setEditingBlockId] = useAtom(inlineEditingActiveAtom);
+  const { clearHighlight } = useBlockHighlight();
 
   return (e) => {
     if (editingBlockId) return;
@@ -74,10 +75,7 @@ const useHandleCanvasDblClick = () => {
       newBlock.removeEventListener("blur", blurListener, true);
       destroyQuill(quill);
       setEditingBlockId("");
-      if (lastHighlighted) {
-        lastHighlighted.removeAttribute("data-highlighted");
-        lastHighlighted = null;
-      }
+      clearHighlight();
     }
     newBlock.addEventListener("blur", blurListener, true);
 
@@ -99,7 +97,7 @@ const useHandleCanvasClick = () => {
   const [ids, setIds] = useSelectedBlockIds();
   const [editingBlockId] = useAtom(inlineEditingActiveAtom);
   const [treeRef] = useAtom(treeRefAtom);
-
+  const { clearHighlight } = useBlockHighlight();
   return (e: any) => {
     if (editingBlockId) return;
     e.stopPropagation();
@@ -107,10 +105,7 @@ const useHandleCanvasClick = () => {
     if (chaiBlock?.getAttribute("data-block-id") && chaiBlock?.getAttribute("data-block-id") === "container") {
       setIds([]);
       setStyleBlockIds([]);
-      if (lastHighlighted) {
-        lastHighlighted.removeAttribute("data-highlighted");
-        lastHighlighted = null;
-      }
+      clearHighlight();
       return;
     }
 
@@ -133,45 +128,26 @@ const useHandleCanvasClick = () => {
       setStyleBlockIds([]);
       setIds(blockId === "canvas" ? [] : [blockId]);
     }
-    if (lastHighlighted) {
-      lastHighlighted.removeAttribute("data-highlighted");
-      lastHighlighted = null;
-    }
+
+    clearHighlight();
   };
 };
 
-let lastHighlighted: HTMLElement | null = null;
-
-const handleMouseMove = throttle((e: any) => {
-  if (lastHighlighted) {
-    lastHighlighted.removeAttribute("data-highlighted");
-  }
-
-  const chaiBlock = getTargetedBlock(e.target);
-  if (chaiBlock) {
-    chaiBlock.setAttribute("data-highlighted", "true");
-    lastHighlighted = chaiBlock;
-  } else {
-    lastHighlighted = null;
-  }
-}, 16);
-
 const useHandleMouseMove = () => {
   const [editingBlockId] = useAtom(inlineEditingActiveAtom);
-  return (e: any) => {
-    if (editingBlockId) return;
-    handleMouseMove(e);
-  };
-};
+  const { highlightBlock } = useBlockHighlight();
 
-const clearHighlight = () => {
-  if (lastHighlighted) {
-    lastHighlighted.removeAttribute("data-highlighted");
-    lastHighlighted = null;
-  }
+  return throttle((e: any) => {
+    if (editingBlockId) return;
+    const chaiBlock = getTargetedBlock(e.target);
+    if (chaiBlock) {
+      highlightBlock(chaiBlock.getAttribute("data-block-id"));
+    }
+  }, 16);
 };
 
 const useHandleMouseLeave = () => {
+  const { clearHighlight } = useBlockHighlight();
   return clearHighlight;
 };
 
@@ -179,13 +155,12 @@ export const Canvas = ({ children }: { children: React.ReactNode }) => {
   const { document } = useFrame();
   const [ids] = useSelectedBlockIds();
   const [styleIds, setSelectedStylingBlocks] = useSelectedStylingBlocks();
+  const { clearHighlight } = useBlockHighlight();
 
   // Add cleanup effect
   useEffect(() => {
-    return () => {
-      clearHighlight();
-    };
-  }, []);
+    return clearHighlight;
+  }, [clearHighlight]);
 
   useEffect(() => {
     setTimeout(() => {
diff --git a/src/core/hooks/useBlockHighlight.ts b/src/core/hooks/useBlockHighlight.ts
index d12b63e3..b2fd6db6 100644
--- a/src/core/hooks/useBlockHighlight.ts
+++ b/src/core/hooks/useBlockHighlight.ts
@@ -1,6 +1,7 @@
 import { useAtom } from "jotai";
 import { canvasIframeAtom } from "../atoms/ui";
 
+//module-level variable to track the last highlighted block
 let lastHighlighted: HTMLElement | null = null;
 
 export const useBlockHighlight = () => {
@@ -29,5 +30,5 @@ export const useBlockHighlight = () => {
     }
   };
 
-  return { highlightBlock, clearHighlight };
+  return { highlightBlock, clearHighlight, lastHighlighted };
 };