draft-regex/build/clearEmptyBlocks.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.default = clearEmptyBlocks;

var _draftJs = require('draft-js');

function getEmptyCountBefore(contentState, key) {
  var block = contentState.getBlockBefore(key);
  if (block && block.getText() === '') {
    return 1 + getEmptyCountBefore(contentState, block.getKey());
  }
  return 0;
}

function getEmptyCountAfter(contentState, key) {
  var block = contentState.getBlockAfter(key);
  if (block && block.getText() === '') {
    return 1 + getEmptyCountAfter(contentState, block.getKey());
  }
  return 0;
}

function clearEmptyBlocks(editorState) {
  var CS = editorState.getCurrentContent();
  var SS = editorState.getSelection();
  var currentKey = SS.getFocusKey();
  var currentBlock = CS.getBlockForKey(currentKey);

  var emptyBefore = getEmptyCountBefore(CS, currentKey);
  var emptyAfter = getEmptyCountAfter(CS, currentKey);

  var MAX_EMPTY_LINES = 2;

  // Be carefull, all checks are very important. And its order of checks also important.
  // 1. Check hole above current line
  // 2. Check hole below current line
  // 3. Check hole between two non-empty lines
  if (emptyBefore > MAX_EMPTY_LINES || emptyAfter > MAX_EMPTY_LINES || emptyBefore + emptyAfter >= MAX_EMPTY_LINES && currentBlock.getText() === '' && currentBlock !== CS.getLastBlock()) {
    var keyForRemove = void 0;
    var keyForFocus = void 0;
    if (currentBlock.getText() === '') {
      keyForRemove = currentKey;
      keyForFocus = CS.getKeyBefore(currentKey);
    } else {
      keyForRemove = CS.getKeyBefore(currentKey);
      keyForFocus = currentKey;
    }

    var newBlockMap = CS.getBlockMap().delete(keyForRemove);
    var newCS = new _draftJs.ContentState({ blockMap: newBlockMap });
    var newEditorState = _draftJs.EditorState.createWithContent(newCS);
    var newSS = _draftJs.SelectionState.createEmpty(keyForFocus || newCS.getFirstBlock().getKey());
    var ES = _draftJs.EditorState.forceSelection(newEditorState, newSS);

    return ES;
  }

  return editorState;
}