@atlaskit/editor-plugin-code-block/dist/cjs/pm-plugins/codeBlockCopySelectionPlugin.js

Code block plugin for @atlaskit/editor-core

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.codeBlockCopySelectionPlugin = codeBlockCopySelectionPlugin;
exports.copySelectionPluginKey = void 0;
exports.provideVisualFeedbackForCopyButton = provideVisualFeedbackForCopyButton;
exports.removeVisualFeedbackForCopyButton = removeVisualFeedbackForCopyButton;
var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
var _copyButton = require("@atlaskit/editor-common/copy-button");
var _safePlugin = require("@atlaskit/editor-common/safe-plugin");
var _state = require("@atlaskit/editor-prosemirror/state");
var _view = require("@atlaskit/editor-prosemirror/view");
var copySelectionPluginKey = exports.copySelectionPluginKey = new _state.PluginKey('codeBlockCopySelectionPlugin');
function getSelectionDecorationStartAndEnd(_ref) {
  var state = _ref.state,
    transaction = _ref.transaction;
  var codeBlockNode = (0, _copyButton.getSelectedNodeOrNodeParentByNodeType)({
    nodeType: state.schema.nodes.codeBlock,
    selection: transaction.selection
  });
  if (!codeBlockNode) {
    return {
      decorationStartAndEnd: undefined,
      codeBlockNode: undefined
    };
  }
  var decorationStartAndEnd = [codeBlockNode.start, codeBlockNode.start + codeBlockNode.node.nodeSize];
  return {
    decorationStartAndEnd: decorationStartAndEnd,
    codeBlockNode: codeBlockNode.node
  };
}
function codeBlockCopySelectionPlugin() {
  return new _safePlugin.SafePlugin({
    key: copySelectionPluginKey,
    state: {
      init: function init() {
        return {
          decorationStartAndEnd: undefined
        };
      },
      apply: function apply(transaction, currentCodeBlockCopySelectionPluginState, _oldState, newState) {
        switch (transaction.getMeta(copySelectionPluginKey)) {
          case 'show-selection':
            {
              return getSelectionDecorationStartAndEnd({
                state: newState,
                transaction: transaction
              });
            }
          case 'remove-selection':
            return {
              decorationStartAndEnd: undefined
            };
          default:
            // The contents of the code block can change while the selection is being shown
            // (either from collab edits -- or from the user continuing to type while hovering
            // the mouse over the copy button).
            // This ensures the selection is updated in these cases.
            if (currentCodeBlockCopySelectionPluginState.decorationStartAndEnd !== undefined) {
              return getSelectionDecorationStartAndEnd({
                state: newState,
                transaction: transaction
              });
            }
            return currentCodeBlockCopySelectionPluginState;
        }
      }
    },
    props: {
      decorations: function decorations(state) {
        if (copySelectionPluginKey.getState(state).decorationStartAndEnd) {
          var _copySelectionPluginK = (0, _slicedToArray2.default)(copySelectionPluginKey.getState(state).decorationStartAndEnd, 2),
            start = _copySelectionPluginK[0],
            end = _copySelectionPluginK[1];
          return _view.DecorationSet.create(state.doc, [_view.Decoration.inline(start, end, {
            class: 'ProseMirror-fake-text-selection'
          })]);
        }
        return _view.DecorationSet.empty;
      }
    }
  });
}
function provideVisualFeedbackForCopyButton(state, dispatch) {
  var tr = state.tr;
  tr.setMeta(copySelectionPluginKey, 'show-selection');

  // note: dispatch should always be defined when called from the
  // floating toolbar. Howver the Command type which the floating toolbar
  // uses suggests it's optional.
  // Using the type here to protect against future refactors of the
  // floating toolbar
  if (dispatch) {
    dispatch(tr);
  }
  return true;
}
function removeVisualFeedbackForCopyButton(state, dispatch) {
  var tr = state.tr;
  tr.setMeta(copySelectionPluginKey, 'remove-selection');

  // note: dispatch should always be defined when called from the
  // floating toolbar. Howver the Command type which the floating toolbar
  // uses suggests it's optional.
  // Using the type here to protect against future refactors of the
  // floating toolbar
  if (dispatch) {
    dispatch(tr);
  }
  return true;
}