/
entityProcessor.ts
40 lines (35 loc) · 1.52 KB
/
entityProcessor.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import { addBlock } from '../../modelApi/common/addBlock';
import { addSegment } from '../../modelApi/common/addSegment';
import { createEntity } from '../../modelApi/creators/createEntity';
import { isBlockElement } from '../utils/isBlockElement';
import { parseFormat } from '../utils/parseFormat';
import { stackFormat } from '../utils/stackFormat';
import type { ElementProcessor } from 'roosterjs-content-model-types';
/**
* Content Model Element Processor for entity
* @param group The parent block group
* @param parent Parent DOM node to process
* @param context DOM to Content Model context
*/
export const entityProcessor: ElementProcessor<HTMLElement> = (group, element, context) => {
const isBlockEntity =
isBlockElement(element) ||
(element.style.display == 'inline-block' && element.style.width == '100%');
stackFormat(
context,
{ segment: isBlockEntity ? 'empty' : undefined, paragraph: 'empty' },
() => {
const entityModel = createEntity(element, true /*isReadonly*/, context.segmentFormat);
parseFormat(element, context.formatParsers.entity, entityModel.entityFormat, context);
if (context.isInSelection) {
entityModel.isSelected = true;
}
if (isBlockEntity) {
addBlock(group, entityModel);
} else {
const paragraph = addSegment(group, entityModel);
context.domIndexer?.onSegment(element, paragraph, [entityModel]);
}
}
);
};