From b5f27a18984562722397f32480dc89e7946f854f Mon Sep 17 00:00:00 2001
From: "Jeet Mandaliya github: sereneinserenade" <jeet.mandaliya7@gmail.com>
Date: Sun, 28 Jan 2024 00:55:47 +0530
Subject: [PATCH] feat: nextSearchResult & previousSearchResul

---
 demos/vue/src/components/Tiptap.vue | 20 ++++++--
 src/searchAndReplace.ts             | 77 ++++++++++++++++++++---------
 2 files changed, 71 insertions(+), 26 deletions(-)

diff --git a/demos/vue/src/components/Tiptap.vue b/demos/vue/src/components/Tiptap.vue
index e1fbde7..f84c731 100644
--- a/demos/vue/src/components/Tiptap.vue
+++ b/demos/vue/src/components/Tiptap.vue
@@ -78,6 +78,13 @@
         >
           Clear
         </button>
+        <button
+          @click="previous"
+          type="button"
+          class="relative inline-flex items-center px-4 py-2 -ml-px text-sm font-medium text-gray-700 bg-white border border-gray-300 hover:bg-gray-50 focus:z-10 focus:border-indigo-500 focus:outline-none focus:ring-1 focus:ring-indigo-500"
+        >
+          Previous
+        </button>
         <button
           @click="next"
           type="button"
@@ -101,7 +108,7 @@
         </button>
 
         <div class="block text-sm font-medium text-gray-700 py-2 px-4">
-          Results: {{ editor?.storage?.searchAndReplace?.results.length }}
+          Results: {{ editor?.storage?.searchAndReplace?.resultIndex + 1 }} / {{ editor?.storage?.searchAndReplace?.results.length }}
         </div>
       </span>
     </div>
@@ -115,8 +122,8 @@ import StarterKit from "@tiptap/starter-kit";
 import Link from "@tiptap/extension-link";
 import { onMounted, ref, watch } from "vue";
 
-import SearchAndReplace from "@sereneinserenade/tiptap-search-and-replace";
-// import { SearchAndReplace } from '../../../../src/searchAndReplace'
+// import SearchAndReplace from "@sereneinserenade/tiptap-search-and-replace";
+import { SearchAndReplace } from '../../../../src/searchAndReplace'
 
 import GithubIcon from "../assets/github.svg";
 
@@ -191,7 +198,12 @@ const replace = () => {
 };
 
 const next = () => {
-  editor.value?.commands.next();
+  editor.value?.commands.nextSearchResult();
+  goToSelection();
+};
+
+const previous = () => {
+  editor.value?.commands.previousSearchResult();
   goToSelection();
 };
 
diff --git a/src/searchAndReplace.ts b/src/searchAndReplace.ts
index d571bc4..33a2d2c 100644
--- a/src/searchAndReplace.ts
+++ b/src/searchAndReplace.ts
@@ -1,6 +1,6 @@
 // MIT License
 
-// Copyright (c) 2023 Jeet Mandaliya (Github Username: sereneinserenade)
+// Copyright (c) 2023 - 2024 Jeet Mandaliya (Github Username: sereneinserenade)
 
 // Permission is hereby granted, free of charge, to any person obtaining a copy
 // of this software and associated documentation files (the "Software"), to deal
@@ -22,7 +22,12 @@
 
 import { Extension, Range, type Dispatch } from "@tiptap/core";
 import { Decoration, DecorationSet } from "@tiptap/pm/view";
-import { Plugin, PluginKey, type EditorState, type Transaction } from "@tiptap/pm/state";
+import {
+  Plugin,
+  PluginKey,
+  type EditorState,
+  type Transaction,
+} from "@tiptap/pm/state";
 import { Node as PMNode } from "@tiptap/pm/model";
 
 declare module "@tiptap/core" {
@@ -47,7 +52,11 @@ declare module "@tiptap/core" {
       /**
        * @description Find next instance of search result.
        */
-      next: () => ReturnType;
+      nextSearchResult: () => ReturnType;
+      /**
+       * @description Find previous instance of search result.
+       */
+      previousSearchResult: () => ReturnType;
       /**
        * @description Replace first instance of search result with given replace term.
        */
@@ -88,12 +97,17 @@ function processSearches(
   resultIndex: number,
 ): ProcessedSearches {
   const decorations: Decoration[] = [];
-  let textNodesWithPosition: TextNodesWithPosition[] = [];
   const results: Range[] = [];
+
+  let textNodesWithPosition: TextNodesWithPosition[] = [];
   let index = 0;
 
-  if (!searchTerm)
-    return { decorationsToReturn: DecorationSet.empty, results: [] };
+  if (!searchTerm) {
+    return {
+      decorationsToReturn: DecorationSet.empty,
+      results: [],
+    };
+  }
 
   doc?.descendants((node, pos) => {
     if (node.isText) {
@@ -150,21 +164,20 @@ function processSearches(
     decorationsToReturn: DecorationSet.create(doc, decorations),
     results,
   };
-};
+}
 
 const replace = (
   replaceTerm: string,
   results: Range[],
-  index: number,
   { state, dispatch }: { state: EditorState; dispatch: Dispatch },
 ) => {
-  const firstResult = results[index]
+  const firstResult = results[0];
 
-  if (!firstResult) return ``
+  if (!firstResult) return;
 
-  const { from, to } = results[index]
+  const { from, to } = results[0];
 
-  dispatch?.(state.tr.insertText(replaceTerm, from, to))
+  if (dispatch) dispatch(state.tr.insertText(replaceTerm, from, to));
 };
 
 const rebaseNextResult = (
@@ -196,13 +209,14 @@ const replaceAll = (
   results: Range[],
   { tr, dispatch }: { tr: Transaction; dispatch: Dispatch },
 ) => {
-  let offset = 0
-  let resultsCopy = results.slice()
+  let offset = 0;
+
+  let resultsCopy = results.slice();
 
   if (!resultsCopy.length) return;
 
   for (let i = 0; i < resultsCopy.length; i += 1) {
-    const { from, to } = resultsCopy[i]
+    const { from, to } = resultsCopy[i];
 
     tr.insertText(replaceTerm, from, to);
 
@@ -219,7 +233,7 @@ const replaceAll = (
     resultsCopy = rebaseNextResultResponse[1];
   }
 
-  dispatch?.(tr)
+  dispatch(tr);
 };
 
 export const searchAndReplacePluginKey = new PluginKey(
@@ -298,13 +312,32 @@ export const SearchAndReplace = Extension.create<
 
           return false;
         },
-      next:
+      nextSearchResult:
         () =>
         ({ editor }) => {
           const { results, resultIndex } = editor.storage.searchAndReplace;
 
-          if (results[resultIndex + 1]) {
-            editor.storage.searchAndReplace.resultIndex = resultIndex + 1;
+          const nextIndex = resultIndex + 1;
+
+          if (results[nextIndex]) {
+            editor.storage.searchAndReplace.resultIndex = nextIndex;
+          } else {
+            editor.storage.searchAndReplace.resultIndex = 0;
+          }
+
+          return false;
+        },
+      previousSearchResult:
+        () =>
+        ({ editor }) => {
+          const { results, resultIndex } = editor.storage.searchAndReplace;
+
+          const prevIndex = resultIndex - 1;
+
+          if (results[prevIndex]) {
+            editor.storage.searchAndReplace.resultIndex = prevIndex;
+          } else {
+            editor.storage.searchAndReplace.resultIndex = results.length - 1;
           }
 
           return false;
@@ -312,9 +345,9 @@ export const SearchAndReplace = Extension.create<
       replace:
         () =>
         ({ editor, state, dispatch }) => {
-          const { replaceTerm, results, resultIndex } = editor.storage.searchAndReplace;
+          const { replaceTerm, results } = editor.storage.searchAndReplace;
 
-          replace(replaceTerm, results, resultIndex, { state, dispatch });
+          replace(replaceTerm, results, { state, dispatch });
 
           return false;
         },
@@ -377,7 +410,7 @@ export const SearchAndReplace = Extension.create<
               doc,
               getRegex(searchTerm, disableRegex, caseSensitive),
               searchResultClass,
-              resultIndex
+              resultIndex,
             );
 
             editor.storage.searchAndReplace.results = results;