From 2df32639fe7b73aaf061667e880575ac1e4fd8ef Mon Sep 17 00:00:00 2001
From: Guillaume Claret <dev@clarus.me>
Date: Thu, 4 Jul 2019 12:32:14 +0200
Subject: [PATCH] Add highlight of the search results

---
 app/server/static/components/annotator.vue    | 45 ++++++++++++++++++-
 .../static/components/sequence_labeling.vue   |  1 +
 2 files changed, 45 insertions(+), 1 deletion(-)

diff --git a/app/server/static/components/annotator.vue b/app/server/static/components/annotator.vue
index a447a9f3..06502628 100644
--- a/app/server/static/components/annotator.vue
+++ b/app/server/static/components/annotator.vue
@@ -8,7 +8,12 @@
         color: id2label[r.label].text_color, \
         backgroundColor: id2label[r.label].background_color \
       }"
-    ) {{ textPart(r) }}
+    )
+      span
+        span(
+          v-for="highlightedChunk in getSearchHighlightedChunks(textPart(r))"
+          v-bind:class="highlightedChunk.highlight && 'has-background-warning'"
+        ) {{ highlightedChunk.content }}
       button.delete.is-small(v-if="id2label[r.label].text_color", v-on:click="removeLabel(r)")
 </template>
 
@@ -19,6 +24,10 @@ export default {
       type: Array, // [{id: Integer, color: String, text: String}]
       default: () => [],
     },
+    searchQuery: {
+      type: String,
+      default: '',
+    },
     text: {
       type: String,
       default: '',
@@ -84,6 +93,40 @@ export default {
   },
 
   methods: {
+    getSearchHighlightedChunks(text) {
+      if (this.searchQuery) {
+        const chunks = [];
+        let currentText = text;
+        let nextIndex;
+
+        do {
+          nextIndex = currentText.toLowerCase().indexOf(this.searchQuery.toLowerCase());
+
+          if (nextIndex !== -1) {
+            chunks.push({
+              content: currentText.substring(0, nextIndex),
+              highlight: false,
+            });
+            chunks.push({
+              content: currentText.substring(nextIndex, nextIndex + this.searchQuery.length),
+              highlight: true,
+            });
+            nextIndex += this.searchQuery.length;
+            currentText = currentText.substring(nextIndex);
+          } else {
+            chunks.push({
+              content: currentText.substring(nextIndex),
+              highlight: false,
+            });
+          }
+        } while (nextIndex !== -1);
+
+        return chunks.filter(({ content }) => content);
+      }
+
+      return [{ content: text, highlight: false }];
+    },
+
     getChunkClass(chunk) {
       if (!chunk.id) {
         return {};
diff --git a/app/server/static/components/sequence_labeling.vue b/app/server/static/components/sequence_labeling.vue
index f1112e01..a463a014 100644
--- a/app/server/static/components/sequence_labeling.vue
+++ b/app/server/static/components/sequence_labeling.vue
@@ -25,6 +25,7 @@ block annotation-area
         annotator(
           v-bind:labels="labels"
           v-bind:entity-positions="annotations[pageNumber]"
+          v-bind:search-query="searchQuery"
           v-bind:text="docs[pageNumber].text"
           v-on:remove-label="removeLabel"
           v-on:add-label="addLabel"