From d37f15a29d15e49e9a950c20bbfec04c7ef4aa72 Mon Sep 17 00:00:00 2001
From: 1ilit <1ilit@proton.me>
Date: Thu, 23 Oct 2025 23:41:39 +0400
Subject: [PATCH] Retain table id after redo (#659)

---
 src/components/EditorHeader/ControlPanel.jsx | 2 +-
 src/context/DiagramContext.jsx               | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/components/EditorHeader/ControlPanel.jsx b/src/components/EditorHeader/ControlPanel.jsx
index 1083422..5a34c8a 100644
--- a/src/components/EditorHeader/ControlPanel.jsx
+++ b/src/components/EditorHeader/ControlPanel.jsx
@@ -326,7 +326,7 @@ export default function ControlPanel({
 
     if (a.action === Action.ADD) {
       if (a.element === ObjectType.TABLE) {
-        addTable(null, false);
+        addTable(null, false, a.id);
       } else if (a.element === ObjectType.AREA) {
         addArea(null, false);
       } else if (a.element === ObjectType.NOTE) {
diff --git a/src/context/DiagramContext.jsx b/src/context/DiagramContext.jsx
index b554cb2..804dcc9 100644
--- a/src/context/DiagramContext.jsx
+++ b/src/context/DiagramContext.jsx
@@ -16,8 +16,8 @@ export default function DiagramContextProvider({ children }) {
   const { setUndoStack, setRedoStack } = useUndoRedo();
   const { selectedElement, setSelectedElement } = useSelect();
 
-  const addTable = (data, addToHistory = true) => {
-    const id = nanoid();
+  const addTable = (data, addToHistory = true, idToUse = null) => {
+    const id = idToUse || nanoid();
     if (data) {
       setTables((prev) => {
         const temp = prev.slice();