@atlaskit/editor-plugin-selection-extension/dist/es2019/pm-plugins/main.js

editor-plugin-selection-extension plugin for @atlaskit/editor-core

import { SafePlugin } from '@atlaskit/editor-common/safe-plugin';
import { PluginKey } from '@atlaskit/editor-prosemirror/state';
import { ReplaceAroundStep, ReplaceStep } from '@atlaskit/editor-prosemirror/transform';
import { SelectionExtensionActionTypes } from '../types';
export const selectionExtensionPluginKey = new PluginKey('selectionExtensionPlugin');
export const createPlugin = () => {
  return new SafePlugin({
    key: selectionExtensionPluginKey,
    state: {
      init: () => {
        return {
          activeExtension: undefined
        };
      },
      apply: (tr, pluginState) => {
        const meta = tr.getMeta(selectionExtensionPluginKey);
        switch (meta === null || meta === void 0 ? void 0 : meta.type) {
          case SelectionExtensionActionTypes.SET_ACTIVE_EXTENSION:
            return {
              ...pluginState,
              activeExtension: meta.extension
            };
          case SelectionExtensionActionTypes.CLEAR_ACTIVE_EXTENSION:
            return {
              ...pluginState,
              activeExtension: undefined
            };
          case SelectionExtensionActionTypes.SET_SELECTED_NODE:
            return {
              ...pluginState,
              selectedNode: meta.selectedNode,
              nodePos: meta.nodePos,
              startTrackChanges: true,
              docChangedAfterClick: false // Reset the flag when starting to track changes
            };
          case SelectionExtensionActionTypes.START_TRACK_CHANGES:
            return {
              ...pluginState,
              startTrackChanges: meta.startTrackChanges
            };
        }
        const docChangedAfterClick = pluginState.startTrackChanges && tr.steps.some(step => step instanceof ReplaceStep || step instanceof ReplaceAroundStep);
        if (docChangedAfterClick) {
          return {
            ...pluginState,
            docChangedAfterClick: true,
            startTrackChanges: false // Reset the flag to stop tracking after the document has changed
          };
        }
        // clear activeExtension if the selection has changed and not empty
        if (tr.selectionSet && !tr.selection.empty) {
          return {
            ...pluginState,
            activeExtension: undefined // Clear active extension on selection change
          };
        }
        return pluginState;
      }
    }
  });
};