From 3e2f384b802a5c2af21bf60273e4042b742d044c Mon Sep 17 00:00:00 2001
From: kudlajz <kudlajzzz@gmail.com>
Date: Tue, 17 Sep 2024 16:58:34 +0200
Subject: [PATCH] Refetch coverage when update event happens

---
 .../extensions/coverage/components/CoverageElement.tsx | 10 ++++++++++
 packages/slate-editor/src/modules/events/types.ts      |  3 +++
 2 files changed, 13 insertions(+)

diff --git a/packages/slate-editor/src/extensions/coverage/components/CoverageElement.tsx b/packages/slate-editor/src/extensions/coverage/components/CoverageElement.tsx
index cc8ad0720..9ba51fd07 100644
--- a/packages/slate-editor/src/extensions/coverage/components/CoverageElement.tsx
+++ b/packages/slate-editor/src/extensions/coverage/components/CoverageElement.tsx
@@ -60,6 +60,16 @@ export function CoverageElement({
         }
     }
 
+    function handleCoverageUpdated({ coverage }: { coverage: Pick<CoverageEntry, 'id'>}) {
+        if (element.coverage.id === coverage.id) {
+            loadCoverage();
+        }
+    }
+
+    useEffect(() => {
+        EventsEditor.addEventListener(editor, 'coverage-updated', handleCoverageUpdated);
+    }, []);
+
     return (
         <EditorBlock
             {...attributes}
diff --git a/packages/slate-editor/src/modules/events/types.ts b/packages/slate-editor/src/modules/events/types.ts
index 7c8125998..c7bada18e 100644
--- a/packages/slate-editor/src/modules/events/types.ts
+++ b/packages/slate-editor/src/modules/events/types.ts
@@ -74,6 +74,9 @@ export type EditorEventMap = {
     'coverage-placeholder-submitted': {
         coverage: Pick<CoverageEntry, 'id'>;
     };
+    'coverage-updated': {
+        coverage: Pick<CoverageEntry, 'id'>;
+    },
     'coverage-removed': {
         uuid: string;
     };