All files / roosterjs-content-model-core/lib/command/cutCopy adjustSelectionForCopyCut.ts

100% Statements 14/14
87.5% Branches 21/24
100% Functions 3/3
100% Lines 13/13

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 351x                     1x         20x 20x 8x 7x   8x     12x 12x 10x 10x 10x     12x      
import { iterateSelections } from 'roosterjs-content-model-dom';
import type {
    ContentModelDocument,
    ContentModelParagraph,
    ContentModelSegment,
    ReadonlyTableSelectionContext,
} from 'roosterjs-content-model-types';
 
/**
 * @internal
 */
export function adjustSelectionForCopyCut(pasteModel: ContentModelDocument) {
    let selectionMarker: ContentModelSegment | undefined;
    let firstBlock: ContentModelParagraph | undefined;
    let tableContext: ReadonlyTableSelectionContext | undefined;
 
    iterateSelections(pasteModel, (_, tableCtxt, block, segments) => {
        if (selectionMarker) {
            if (tableCtxt != tableContext && firstBlock?.segments.includes(selectionMarker)) {
                firstBlock.segments.splice(firstBlock.segments.indexOf(selectionMarker), 1);
            }
            return true;
        }
 
        const marker = segments?.find(segment => segment.segmentType == 'SelectionMarker');
        if (!selectionMarker && marker) {
            tableContext = tableCtxt;
            firstBlock = block?.blockType == 'Paragraph' ? block : undefined;
            selectionMarker = marker;
        }
 
        return false;
    });
}