draft-regex/build/replaceTextRegex.js

'use strict';

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

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

var _rules = require('./rules');

function _toConsumableArray(arr) { if (Array.isArray(arr)) { for (var i = 0, arr2 = Array(arr.length); i < arr.length; i++) { arr2[i] = arr[i]; } return arr2; } else { return Array.from(arr); } }

var optionRules = {
  extraSpaces: [{ reg: new RegExp(/\s+/g), shift: ' ' }, // "Лишние пробелы"
  { reg: new RegExp(/(^\s*)|(\s*)$/g), shift: '' }],
  spacesAfterPunctuationMarks: [{ reg: new RegExp(/(,|\.)(\S)/g), shift: '$1 $2' }]
};

function createSelectionState(key) {
  var start = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 0;
  var end = arguments[2];

  return new _draftJs.SelectionState({
    anchorKey: key,
    anchorOffset: start,
    focusKey: key,
    focusOffset: end || start
  });
}

function createContentBlock(block, options) {
  var _ref = options || {},
      text = _ref.text,
      key = _ref.key,
      type = _ref.type,
      characterList = _ref.characterList;

  return new _draftJs.ContentBlock({
    text: text || block.getText(),
    key: key || block.getKey(),
    type: type || block.getType(),
    characterList: characterList || block.getCharacterList()
  });
}

function prepareOptionableRules(options) {
  var acc = [];
  Object.keys(options).forEach(function (key) {
    acc.push.apply(acc, _toConsumableArray(optionRules[key]));
    return acc;
  });
  if (acc.length > 0) return acc;
  return [];
}

function replaceTextRegex(editorState) {
  var rulesArray = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : _rules.typoRules;
  var options = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : { extraSpaces: true };

  var CS = editorState.getCurrentContent();
  var CSMap = CS.getBlockMap();
  var CSText = CS.getPlainText();

  var optionableRules = [];
  if (options) {
    var preparedRules = prepareOptionableRules(options);
    optionableRules.push.apply(optionableRules, _toConsumableArray(preparedRules));
  }
  var rules = options ? rulesArray.concat(optionableRules) : rulesArray;

  var rule = void 0;
  var match = rules.some(function (typoRule) {
    rule = typoRule;
    return typoRule && typoRule.reg.exec(CSText);
  });

  if (match) {
    var _ref2 = rule || {},
        _reg = _ref2.reg,
        _shift = _ref2.shift;

    var SS = editorState.getSelection();
    var _key = SS.getAnchorKey();
    var anchorOffset = SS.getAnchorOffset();
    var currentContentBlock = CS.getBlockForKey(_key);
    var text = currentContentBlock.getText().replace(_reg, _shift);
    var newBlock = createContentBlock(currentContentBlock, {
      text: text,
      key: _key
    });

    var diffInLengths = currentContentBlock.getLength() - newBlock.getLength();
    var newSS = createSelectionState(_key, anchorOffset - diffInLengths);

    var newBlockMap = CSMap.set(_key, newBlock);
    var newCS = new _draftJs.ContentState({ blockMap: newBlockMap });
    var ES = _draftJs.EditorState.createWithContent(newCS);
    var forceES = _draftJs.EditorState.forceSelection(ES, newSS);
    var forceESText = forceES.getCurrentContent().getPlainText();

    if (forceESText !== CSText) {
      return forceES;
    }
    return editorState;
  }
  return editorState;
}