如何在draft-js上做多行加亮？

const contentState = editorState.getCurrentContent();
const selectionState = editorState.getSelection();

// 创建一个样式对象
const style = {
  backgroundColor: 'yellow',
};

// 将样式应用到选中的文本块
const newContentState = Modifier.applyInlineStyle(
  contentState,
  selectionState,
  'HIGHLIGHT',
);

// 更新编辑器状态
const newEditorState = EditorState.push(
  editorState,
  newContentState,
  'change-inline-style',
);

// 设置新的选中状态
const newSelectionState = newContentState.getSelectionAfter();
const newEditorStateWithSelection = EditorState.forceSelection(
  newEditorState,
  newSelectionState,
);

// 更新编辑器状态
setEditorState(newEditorStateWithSelection);

const handleKeyDown = (e) => {
  // 按下Shift+Enter键时触发加亮操作
  if (e.shiftKey && e.key === 'Enter') {
    e.preventDefault();
    highlightText();
  }
};

<Editor
  editorState={editorState}
  onChange={setEditorState}
  onKeyDown={handleKeyDown}
/>

const highlightText = () => {
  const selectionState = editorState.getSelection();

  // 获取选中的文本块
  const startKey = selectionState.getStartKey();
  const endKey = selectionState.getEndKey();
  const contentState = editorState.getCurrentContent();
  const blockMap = contentState.getBlockMap();
  const selectedBlocks = blockMap
    .skipUntil((_, key) => key === startKey)
    .takeUntil((_, key) => key === endKey)
    .concat(blockMap.get(endKey));

  // 应用样式到选中的文本块
  let newContentState = contentState;
  selectedBlocks.forEach((block) => {
    const blockKey = block.getKey();
    const blockSelectionState = selectionState.merge({
      anchorKey: blockKey,
      focusKey: blockKey,
    });
    newContentState = Modifier.applyInlineStyle(
      newContentState,
      blockSelectionState,
      'HIGHLIGHT',
    );
  });

  // 更新编辑器状态
  const newEditorState = EditorState.push(
    editorState,
    newContentState,
    'change-inline-style',
  );
  setEditorState(newEditorState);
};