fix: consistent table click handlers

Fixes #230
mdx-editor · Dec 11, 2023 · edfc236 · edfc236
1 parent d695747
commit edfc236
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/plugins/table/TableEditor.tsx b/src/plugins/table/TableEditor.tsx
@@ -47,12 +47,11 @@ export const TableEditor: React.FC<TableEditorProps> = ({ mdastNode, parentEdito
   const [activeCell, setActiveCell] = React.useState<[number, number] | null>(null)
   const [iconComponentFor, readOnly] = corePluginHooks.useEmitterValues('iconComponentFor', 'readOnly')
   const getCellKey = React.useMemo(() => {
-    const cellKeyMap = new WeakMap<Mdast.TableCell, string>()
-    return (cell: Mdast.TableCell) => {
-      if (!cellKeyMap.has(cell)) {
-        cellKeyMap.set(cell, uuidv4())
+    return (cell: Mdast.TableCell & { __cacheKey?: string }) => {
+      if (!cell.__cacheKey) {
+        cell.__cacheKey = uuidv4()
       }
-      return cellKeyMap.get(cell)!
+      return cell.__cacheKey
     }
   }, [])