@atlaskit/editor-plugin-max-content-size/dist/es2019/maxContentSizePlugin.js

max-content-size plugin for @atlaskit/editor-core

import { SafePlugin } from '@atlaskit/editor-common/safe-plugin';
import { PluginKey } from '@atlaskit/editor-prosemirror/state';
import { isExperimentEnabled } from '@atlaskit/platform-feature-experiments/is-experiment-enabled';
export const pluginKey = new PluginKey('maxContentSizePlugin');
export function createPlugin(dispatch, maxContentSize) {
  if (!maxContentSize) {
    return;
  }
  let maxContentSizeReached = false;
  return new SafePlugin({
    state: {
      init: () => ({
        maxContentSizeReached: false
      }),
      apply(tr, state) {
        const result = tr.doc && tr.doc.nodeSize > maxContentSize - 1;
        return {
          maxContentSizeReached: result
        };
      }
    },
    key: pluginKey,
    filterTransaction(tr) {
      const result = tr.doc && tr.doc.nodeSize > maxContentSize;
      if (result || result !== maxContentSizeReached) {
        dispatch(pluginKey, {
          maxContentSizeReached: result
        });
      }
      maxContentSizeReached = result;
      if (isExperimentEnabled('platform_editor_max_content_size_allow_delete')) {
        // A document can load already over the limit, so only block transactions that grow it
        // further — otherwise there is no way to edit it back under the limit.
        return !result || tr.doc.nodeSize <= tr.before.nodeSize;
      }
      return !result;
    }
  });
}
const maxContentSizePlugin = ({
  config: maxContentSize,
  api
}) => {
  return {
    name: 'maxContentSize',
    getSharedState(editorState) {
      if (!editorState) {
        return undefined;
      }
      return pluginKey.getState(editorState);
    },
    pmPlugins() {
      return [{
        name: 'maxContentSize',
        plugin: ({
          dispatch
        }) => createPlugin(dispatch, maxContentSize)
      }];
    }
  };
};
export default maxContentSizePlugin;