From 4f72c9564da72c97523a5e4af195e6dc3d920e25 Mon Sep 17 00:00:00 2001
From: Ramon Wenger <ramon.wenger@iterativ.ch>
Date: Tue, 16 Jan 2024 16:58:09 +0100
Subject: [PATCH] Update highlights on cache change

---
 .../content-blocks/ContentComponent.vue       | 58 +++++++++++--------
 1 file changed, 34 insertions(+), 24 deletions(-)

diff --git a/client/src/components/content-blocks/ContentComponent.vue b/client/src/components/content-blocks/ContentComponent.vue
index 559ccb8a..1c0e79d7 100644
--- a/client/src/components/content-blocks/ContentComponent.vue
+++ b/client/src/components/content-blocks/ContentComponent.vue
@@ -24,7 +24,7 @@
 </template>
 
 <script setup lang="ts">
-import { defineAsyncComponent, ref, computed, onMounted } from 'vue';
+import { defineAsyncComponent, ref, computed, onMounted, watch } from 'vue';
 import { useStore } from 'vuex';
 import { constructContentComponentBookmarkMutation } from '@/helpers/update-content-bookmark-mutation';
 import { useMutation } from '@vue/apollo-composable';
@@ -144,6 +144,21 @@ const componentClass = computed(() => {
   }
   return classes;
 });
+const paragraphs = computed(() =>
+  contentComponentDiv.value ? contentComponentDiv.value.getElementsByTagName('p') : []
+);
+
+watch(
+  () => filteredHighlights.value.length,
+  () => {
+    for (const paragraph of paragraphs.value) {
+      console.log(paragraph);
+      const instance = new Mark(paragraph);
+      instance.unmark();
+    }
+    markHighlights();
+  }
+);
 
 const addNote = (id: string) => {
   const type = Object.prototype.hasOwnProperty.call(props.parent, '__typename')
@@ -179,33 +194,28 @@ const bookmarkContent = (bookmarked: boolean) => {
       bookmarked,
     },
   };
-  console.log(newVars);
   mutateBookmarkContent(newVars);
 };
 
-onMounted(() => {
-  setTimeout(() => {
-    if (contentComponentDiv.value) {
-      const paragraphs = contentComponentDiv.value.getElementsByTagName('p');
-      for (const highlight of filteredHighlights.value) {
-        console.log(paragraphs);
-        console.log(highlight);
+const markHighlights = () => {
+  for (const highlight of filteredHighlights.value) {
+    const element = paragraphs.value[highlight.paragraphIndex];
+    const instance = new Mark(element);
+    const ranges: Mark.Range[] = [
+      {
+        start: highlight.startPosition,
+        length: highlight.selectionLength,
+      },
+    ];
+    instance.markRanges(ranges, {
+      className: `highlight highlight--${highlight.color}`,
+    });
+  }
+};
 
-        const element = paragraphs[highlight.paragraphIndex];
-        console.log(element);
-        const instance = new Mark(element);
-        console.log(instance);
-        const ranges: Mark.Range[] = [
-          {
-            start: highlight.startPosition,
-            length: highlight.startPosition,
-          },
-        ];
-        console.log(ranges);
-        instance.markRanges(ranges);
-      }
-    }
-  }, 2000);
+onMounted(() => {
+  console.log('onMounted ContentComponent called');
+  setTimeout(markHighlights, 500);
 });
 </script>