All files / roosterjs-content-model-dom/lib/modelToDom/handlers handleSegment.ts

100% Statements 18/18
100% Branches 13/13
100% Functions 1/1
100% Lines 17/17

Press n or j to go to the next uncovered block, b, p or k for the previous block.

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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55                1x             1173x     1173x 151x         1173x   895x 895x     104x 104x     44x 44x     1x 1x     4x 4x         1173x 153x          
import type {
    ContentModelSegment,
    ContentModelSegmentHandler,
} from 'roosterjs-content-model-types';
 
/**
 * @internal
 */
export const handleSegment: ContentModelSegmentHandler<ContentModelSegment> = (
    doc,
    parent,
    segment,
    context,
    segmentNodes
) => {
    const regularSelection = context.regularSelection;
 
    // If start position is not set yet, and current segment is in selection, set start position
    if (segment.isSelected && !regularSelection.start) {
        regularSelection.start = {
            ...regularSelection.current,
        };
    }
 
    switch (segment.segmentType) {
        case 'Text':
            context.modelHandlers.text(doc, parent, segment, context, segmentNodes);
            break;
 
        case 'Br':
            context.modelHandlers.br(doc, parent, segment, context, segmentNodes);
            break;
 
        case 'Image':
            context.modelHandlers.image(doc, parent, segment, context, segmentNodes);
            break;
 
        case 'General':
            context.modelHandlers.generalSegment(doc, parent, segment, context, segmentNodes);
            break;
 
        case 'Entity':
            context.modelHandlers.entitySegment(doc, parent, segment, context, segmentNodes);
            break;
    }
 
    // If end position is not set, or it is not finalized, and current segment is still in selection, set end position
    // If there is other selection, we will overwrite regularSelection.end when we process that segment
    if (segment.isSelected && regularSelection.start) {
        regularSelection.end = {
            ...regularSelection.current,
        };
    }
};