Extracted method for getting suggestion marks.

Author: VV-EE

demo/main.ts

@@ -127,23 +127,23 @@ const suggestChangesUiPlugin = new Plugin({
     const toggleButton = document.createElement("button");
     toggleButton.appendChild(document.createTextNode("Enable suggestions"));
     toggleButton.addEventListener("click", () => {
-      // eslint-disable-next-line @typescript-eslint/unbound-method
+       
       toggleSuggestChanges(view.state, view.dispatch);
       view.focus();
     });
 
     const applyAllButton = document.createElement("button");
     applyAllButton.appendChild(document.createTextNode("Apply all"));
     applyAllButton.addEventListener("click", () => {
-      // eslint-disable-next-line @typescript-eslint/unbound-method
+       
       applySuggestions(view.state, view.dispatch);
       view.focus();
     });
 
     const revertAllButton = document.createElement("button");
     revertAllButton.appendChild(document.createTextNode("Revert all"));
     revertAllButton.addEventListener("click", () => {
-      // eslint-disable-next-line @typescript-eslint/unbound-method
+       
       revertSuggestions(view.state, view.dispatch);
       view.focus();
     });
@@ -205,5 +205,5 @@ const view = new EditorView(editorEl, {
   dispatchTransaction: withSuggestChanges(),
 });
 
-// eslint-disable-next-line @typescript-eslint/unbound-method
+ 
 enableSuggestChanges(view.state, view.dispatch);

src/__tests__/replaceAroundStep.test.ts

@@ -115,7 +115,7 @@ describe("ReplaceAroundStep", () => {
             id: 1,
             type: "mark",
             // we know this mark exists
-            // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
+             
             previousValue: doc.children[0]!.marks[0]!.toJSON(),
             newValue: null,
           },

src/addNodeMarkStep.ts

@@ -3,6 +3,7 @@ import { type EditorState, type Transaction } from "prosemirror-state";
 import { type AddNodeMarkStep, type Step } from "prosemirror-transform";
 
 import { rebasePos } from "./rebasePos.js";
+import { getSuggestionMarks } from "./utils.js";
 
 /**
  * Transform an add node mark step into its equivalent tracked steps.
@@ -18,12 +19,7 @@ export function trackAddNodeMarkStep(
   prevSteps: Step[],
   suggestionId: number,
 ) {
-  const { modification } = state.schema.marks;
-  if (!modification) {
-    throw new Error(
-      `Failed to apply modifications to node: schema does not contain modification mark. Did you forget to add it?`,
-    );
-  }
+  const { modification } = getSuggestionMarks(state.schema);
 
   const rebasedPos = rebasePos(step.pos, prevSteps, trackedTransaction.steps);
   const $pos = trackedTransaction.doc.resolve(rebasedPos);

src/attrStep.ts

@@ -3,6 +3,7 @@ import { type EditorState, type Transaction } from "prosemirror-state";
 import { type AttrStep, type Step } from "prosemirror-transform";
 
 import { rebasePos } from "./rebasePos.js";
+import { getSuggestionMarks } from "./utils.js";
 
 /**
  * Transform an attr mark step into its equivalent tracked steps.
@@ -18,12 +19,7 @@ export function trackAttrStep(
   prevSteps: Step[],
   suggestionId: number,
 ) {
-  const { modification } = state.schema.marks;
-  if (!modification) {
-    throw new Error(
-      `Failed to apply modifications to node: schema does not contain modification mark. Did you forget to add it?`,
-    );
-  }
+  const { modification } = getSuggestionMarks(state.schema);
 
   const rebasedPos = rebasePos(step.pos, prevSteps, trackedTransaction.steps);
   const $pos = trackedTransaction.doc.resolve(rebasedPos);

src/commands.ts

@@ -15,6 +15,7 @@ import { type EditorView } from "prosemirror-view";
 
 import { findSuggestionMarkEnd } from "./findSuggestionMarkEnd.js";
 import { suggestChangesKey } from "./plugin.js";
+import { getSuggestionMarks } from "./utils.js";
 
 /**
  * Given a node and a transform, add a set of steps to the
@@ -86,9 +87,8 @@ function applySuggestionsToTransform(
 }
 
 function revertModifications(node: Node, pos: number, tr: Transform) {
-  const existingMods = node.marks.filter(
-    (mark) => mark.type === node.type.schema.marks["modification"],
-  );
+  const { modification } = getSuggestionMarks(node.type.schema);
+  const existingMods = node.marks.filter((mark) => mark.type === modification);
   for (const mod of existingMods) {
     if (
       mod.attrs["type"] === "attr" &&
@@ -129,12 +129,7 @@ function applyModificationsToTransform(
   dir: number,
   suggestionId?: number,
 ) {
-  const { modification } = node.type.schema.marks;
-  if (!modification) {
-    throw new Error(
-      `Failed to apply modifications to node: schema does not contain modification mark. Did you forget to add it?`,
-    );
-  }
+  const { modification } = getSuggestionMarks(node.type.schema);
 
   const modificationIsInSet =
     suggestionId === undefined
@@ -177,17 +172,7 @@ function applyModificationsToTransform(
 }
 
 function applySuggestionsToNode(node: Node) {
-  const { deletion, insertion } = node.type.schema.marks;
-  if (!deletion) {
-    throw new Error(
-      `Failed to apply tracked changes to node: schema does not contain deletion mark. Did you forget to add it?`,
-    );
-  }
-  if (!insertion) {
-    throw new Error(
-      `Failed to apply tracked changes to node: schema does not contain insertion mark. Did you forget to add it?`,
-    );
-  }
+  const { deletion, insertion } = getSuggestionMarks(node.type.schema);
 
   if (deletion.isInSet(node.marks)) {
     return null;
@@ -219,17 +204,7 @@ export function applySuggestions(
   state: EditorState,
   dispatch?: EditorView["dispatch"],
 ) {
-  const { deletion, insertion } = state.schema.marks;
-  if (!deletion) {
-    throw new Error(
-      `Failed to apply tracked changes to node: schema does not contain deletion mark. Did you forget to add it?`,
-    );
-  }
-  if (!insertion) {
-    throw new Error(
-      `Failed to apply tracked changes to node: schema does not contain insertion mark. Did you forget to add it?`,
-    );
-  }
+  const { deletion, insertion } = getSuggestionMarks(state.schema);
 
   const tr = state.tr;
   applySuggestionsToTransform(state.doc, tr, insertion, deletion);
@@ -248,17 +223,7 @@ export function applySuggestions(
  */
 export function applySuggestion(suggestionId: number): Command {
   return (state, dispatch) => {
-    const { deletion, insertion } = state.schema.marks;
-    if (!deletion) {
-      throw new Error(
-        `Failed to apply tracked changes to node: schema does not contain deletion mark. Did you forget to add it?`,
-      );
-    }
-    if (!insertion) {
-      throw new Error(
-        `Failed to apply tracked changes to node: schema does not contain insertion mark. Did you forget to add it?`,
-      );
-    }
+    const { deletion, insertion } = getSuggestionMarks(state.schema);
 
     const tr = state.tr;
     applySuggestionsToTransform(
@@ -287,17 +252,7 @@ export function revertSuggestions(
   state: EditorState,
   dispatch?: EditorView["dispatch"],
 ) {
-  const { deletion, insertion } = state.schema.marks;
-  if (!deletion) {
-    throw new Error(
-      `Failed to apply tracked changes to node: schema does not contain deletion mark. Did you forget to add it?`,
-    );
-  }
-  if (!insertion) {
-    throw new Error(
-      `Failed to apply tracked changes to node: schema does not contain insertion mark. Did you forget to add it?`,
-    );
-  }
+  const { deletion, insertion } = getSuggestionMarks(state.schema);
   const tr = state.tr;
   applySuggestionsToTransform(state.doc, tr, deletion, insertion);
   applyModificationsToTransform(tr.doc, tr, -1);
@@ -315,17 +270,7 @@ export function revertSuggestions(
  */
 export function revertSuggestion(suggestionId: number): Command {
   return (state, dispatch) => {
-    const { deletion, insertion } = state.schema.marks;
-    if (!deletion) {
-      throw new Error(
-        `Failed to apply tracked changes to node: schema does not contain deletion mark. Did you forget to add it?`,
-      );
-    }
-    if (!insertion) {
-      throw new Error(
-        `Failed to apply tracked changes to node: schema does not contain insertion mark. Did you forget to add it?`,
-      );
-    }
+    const { deletion, insertion } = getSuggestionMarks(state.schema);
 
     const tr = state.tr;
     applySuggestionsToTransform(
@@ -348,22 +293,9 @@ export function revertSuggestion(suggestionId: number): Command {
  */
 export function selectSuggestion(suggestionId: number): Command {
   return (state, dispatch) => {
-    const { deletion, insertion, modification } = state.schema.marks;
-    if (!deletion) {
-      throw new Error(
-        `Failed to apply tracked changes to node: schema does not contain deletion mark. Did you forget to add it?`,
-      );
-    }
-    if (!insertion) {
-      throw new Error(
-        `Failed to apply tracked changes to node: schema does not contain insertion mark. Did you forget to add it?`,
-      );
-    }
-    if (!modification) {
-      throw new Error(
-        `Failed to apply tracked changes to node: schema does not contain modification mark. Did you forget to add it?`,
-      );
-    }
+    const { deletion, insertion, modification } = getSuggestionMarks(
+      state.schema,
+    );
 
     let changeStart = null as number | null;
     let changeEnd = null as number | null;

src/decorations.ts

@@ -5,6 +5,7 @@ import {
   DecorationSet,
   type DecorationSource,
 } from "prosemirror-view";
+import { getSuggestionMarks } from "./utils.js";
 
 function pilcrow() {
   const span = document.createElement("span");
@@ -13,17 +14,7 @@ function pilcrow() {
 }
 
 export function getSuggestionDecorations(state: EditorState): DecorationSource {
-  const { deletion, insertion } = state.schema.marks;
-  if (!deletion) {
-    throw new Error(
-      `Failed to apply tracked changes to node: schema does not contain deletion mark. Did you forget to add it?`,
-    );
-  }
-  if (!insertion) {
-    throw new Error(
-      `Failed to apply tracked changes to node: schema does not contain insertion mark. Did you forget to add it?`,
-    );
-  }
+  const { deletion, insertion } = getSuggestionMarks(state.schema);
 
   const changeDecorations: Decoration[] = [];
   let lastParentNode: Node | null = null;

src/removeNodeMarkStep.ts

@@ -3,6 +3,7 @@ import { type EditorState, type Transaction } from "prosemirror-state";
 import { type RemoveNodeMarkStep, type Step } from "prosemirror-transform";
 
 import { rebasePos } from "./rebasePos.js";
+import { getSuggestionMarks } from "./utils.js";
 
 /**
  * Transform a remove node mark step into its equivalent tracked steps.
@@ -18,12 +19,7 @@ export function suggestRemoveNodeMarkStep(
   prevSteps: Step[],
   suggestionId: number,
 ) {
-  const { modification } = state.schema.marks;
-  if (!modification) {
-    throw new Error(
-      `Failed to apply modifications to node: schema does not contain modification mark. Did you forget to add it?`,
-    );
-  }
+  const { modification } = getSuggestionMarks(state.schema);
 
   const rebasedPos = rebasePos(step.pos, prevSteps, trackedTransaction.steps);
   const $pos = trackedTransaction.doc.resolve(rebasedPos);

src/replaceAroundStep.ts

@@ -16,6 +16,7 @@ import { applySuggestionsToSlice } from "./commands.js";
 import { rebasePos } from "./rebasePos.js";
 import { suggestRemoveNodeMarkStep } from "./removeNodeMarkStep.js";
 import { suggestReplaceStep } from "./replaceStep.js";
+import { getSuggestionMarks } from "./utils.js";
 
 /**
  * This detects and handles changes from `setNodeMarkup` so that these are tracked as a modification
@@ -37,12 +38,7 @@ function suggestSetNodeMarkup(
     step.gapFrom === step.from + 1 &&
     (step as ReplaceAroundStep & { structure: boolean }).structure
   ) {
-    const { modification } = state.schema.marks;
-    if (!modification) {
-      throw new Error(
-        `Failed to apply modifications to node: schema does not contain modification mark. Did you forget to add it?`,
-      );
-    }
+    const { modification } = getSuggestionMarks(state.schema);
 
     const newNode = step.slice.content.firstChild;
     let from = rebasePos(step.from, prevSteps, trackedTransaction.steps);

src/replaceStep.ts

@@ -8,6 +8,7 @@ import { type ReplaceStep, type Step } from "prosemirror-transform";
 
 import { findSuggestionMarkEnd } from "./findSuggestionMarkEnd.js";
 import { rebasePos } from "./rebasePos.js";
+import { getSuggestionMarks } from "./utils.js";
 
 /**
  * Transform a replace step into its equivalent tracked steps.
@@ -47,17 +48,7 @@ export function suggestReplaceStep(
   prevSteps: Step[],
   suggestionId: number,
 ) {
-  const { deletion, insertion } = state.schema.marks;
-  if (!deletion) {
-    throw new Error(
-      `Failed to apply tracked changes to node: schema does not contain deletion mark. Did you forget to add it?`,
-    );
-  }
-  if (!insertion) {
-    throw new Error(
-      `Failed to apply tracked changes to node: schema does not contain insertion mark. Did you forget to add it?`,
-    );
-  }
+  const { deletion, insertion } = getSuggestionMarks(state.schema);
 
   // Check for insertion and deletion marks directly
   // adjacent to this step's boundaries. If they exist,

src/utils.ts

@@ -0,0 +1,35 @@
+import { type MarkType, type Schema } from "prosemirror-model";
+
+export interface SuggestionMarks {
+  insertion: MarkType;
+  deletion: MarkType;
+  modification: MarkType;
+}
+
+/**
+ * Get the suggestion mark types from a schema, with proper error handling.
+ * Throws an error if any of the required marks are not found.
+ */
+export function getSuggestionMarks(schema: Schema): SuggestionMarks {
+  const { insertion, deletion, modification } = schema.marks;
+
+  if (!insertion) {
+    throw new Error(
+      "Failed to find insertion mark in schema. Did you forget to add it?",
+    );
+  }
+
+  if (!deletion) {
+    throw new Error(
+      "Failed to find deletion mark in schema. Did you forget to add it?",
+    );
+  }
+
+  if (!modification) {
+    throw new Error(
+      "Failed to find modification mark in schema. Did you forget to add it?",
+    );
+  }
+
+  return { insertion, deletion, modification };
+}