Added an option to override the decorator generator

yjs · Apr 5, 2020 · 5f17f92 · 5f17f92
1 parent f975837
commit 5f17f92
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/plugins/cursor-plugin.js b/src/plugins/cursor-plugin.js
@@ -73,17 +73,17 @@ export const createDecorations = (state, awareness) => {
  * @param {Awareness} awareness
  * @return {any}
  */
-export const yCursorPlugin = awareness => new Plugin({
+export const yCursorPlugin = (awareness, decorator = createDecorations) => new Plugin({
   key: yCursorPluginKey,
   state: {
     init (_, state) {
-      return createDecorations(state, awareness)
+      return decorator(state, awareness)
     },
     apply (tr, prevState, oldState, newState) {
       const ystate = ySyncPluginKey.getState(newState)
       const yCursorState = tr.getMeta(yCursorPluginKey)
       if ((ystate && ystate.isChangeOrigin) || (yCursorState && yCursorState.awarenessUpdated)) {
-        return createDecorations(newState, awareness)
+        return decorator(newState, awareness)
       }
       return prevState.map(tr.mapping, tr.doc)
     }