|
- /**
- * Copyright (c) Facebook, Inc. and its affiliates.
- *
- * This source code is licensed under the MIT license found in the
- * LICENSE file in the root directory of this source tree.
- *
- * @format
- *
- * @emails oncall+draft_js
- */
- 'use strict';
-
- var CharacterMetadata = require("./CharacterMetadata");
-
- var _require = require("immutable"),
- Map = _require.Map;
-
- var ContentStateInlineStyle = {
- add: function add(contentState, selectionState, inlineStyle) {
- return modifyInlineStyle(contentState, selectionState, inlineStyle, true);
- },
- remove: function remove(contentState, selectionState, inlineStyle) {
- return modifyInlineStyle(contentState, selectionState, inlineStyle, false);
- }
- };
-
- function modifyInlineStyle(contentState, selectionState, inlineStyle, addOrRemove) {
- var blockMap = contentState.getBlockMap();
- var startKey = selectionState.getStartKey();
- var startOffset = selectionState.getStartOffset();
- var endKey = selectionState.getEndKey();
- var endOffset = selectionState.getEndOffset();
- var newBlocks = blockMap.skipUntil(function (_, k) {
- return k === startKey;
- }).takeUntil(function (_, k) {
- return k === endKey;
- }).concat(Map([[endKey, blockMap.get(endKey)]])).map(function (block, blockKey) {
- var sliceStart;
- var sliceEnd;
-
- if (startKey === endKey) {
- sliceStart = startOffset;
- sliceEnd = endOffset;
- } else {
- sliceStart = blockKey === startKey ? startOffset : 0;
- sliceEnd = blockKey === endKey ? endOffset : block.getLength();
- }
-
- var chars = block.getCharacterList();
- var current;
-
- while (sliceStart < sliceEnd) {
- current = chars.get(sliceStart);
- chars = chars.set(sliceStart, addOrRemove ? CharacterMetadata.applyStyle(current, inlineStyle) : CharacterMetadata.removeStyle(current, inlineStyle));
- sliceStart++;
- }
-
- return block.set('characterList', chars);
- });
- return contentState.merge({
- blockMap: blockMap.merge(newBlocks),
- selectionBefore: selectionState,
- selectionAfter: selectionState
- });
- }
-
- module.exports = ContentStateInlineStyle;
|