From 753fc7632418dd7066062488ef045cf7dc4902c6 Mon Sep 17 00:00:00 2001
From: Chrissi2812 <cf@pixel-kraft.de>
Date: Tue, 28 May 2019 13:51:37 +0200
Subject: [PATCH] fix search highlight on doc change

---
 packages/tiptap-extensions/src/extensions/Search.js | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/packages/tiptap-extensions/src/extensions/Search.js b/packages/tiptap-extensions/src/extensions/Search.js
index fad8cdfa..3279bbc5 100644
--- a/packages/tiptap-extensions/src/extensions/Search.js
+++ b/packages/tiptap-extensions/src/extensions/Search.js
@@ -95,9 +95,15 @@ export default class Search extends Extension {
       new Plugin({
         state: {
           init: (_, { doc }) => this.createDeco(doc),
-          apply: (tr, old) => (
-            (tr.docChanged || this.options.searching) ? this.createDeco(tr.doc) : old
-          ),
+          apply: (tr, old) => {
+            if (this.options.searching) {
+              return this.createDeco(tr.doc)
+            }
+            if (tr.docChanged) {
+              return old.map(tr.mapping, tr.doc)
+            }
+            return old
+          },
         },
         props: {
           decorations(state) { return this.getState(state) },