All files / roosterjs-content-model-plugins/lib/edit/deleteSteps deleteWordSelection.ts

82.18% Statements 83/101
73.33% Branches 55/75
100% Functions 4/4
81.11% Lines 73/90

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 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 1931x                         1x 1x 1x 1x 1x 1x 1x                   2x 9x       9x 9x 9x   9x 9x   9x 54x           54x   9x         9x 9x     1x 1x 1x           1x     39x 1x 1x 38x 33x   5x   39x     3x 2x   1x   3x     2x 1x 1x       1x   2x                       9x 9x 9x   14x 13x   13x 13x   13x 54x 41x   54x 54x 54x 54x   54x 46x   46x   46x 46x     46x   46x 41x   41x 14x     5x   5x 2x     5x       5x                                               1x           1x         1x  
import {
    isPunctuation,
    isSpace,
    isWhiteSpacePreserved,
    mutateBlock,
    normalizeText,
} from 'roosterjs-content-model-dom';
import type {
    DeleteSelectionContext,
    DeleteSelectionStep,
    ShallowMutableContentModelParagraph,
} from 'roosterjs-content-model-types';
 
const enum DeleteWordState {
    Start,
    Punctuation,
    Text,
    NonText,
    Space,
    End,
}
 
interface CharInfo {
    text: boolean;
    space: boolean;
    punctuation: boolean;
}
 
function getDeleteWordSelection(direction: 'forward' | 'backward'): DeleteSelectionStep {
    return context => {
        Iif (context.deleteResult != 'notDeleted') {
            return;
        }
 
        const { marker, paragraph } = context.insertPoint;
        const startIndex = paragraph.segments.indexOf(marker);
        const deleteNext = direction == 'forward';
 
        const iterator = iterateSegments(mutateBlock(paragraph), startIndex, deleteNext, context);
        let curr = iterator.next();
 
        for (let state = DeleteWordState.Start; state != DeleteWordState.End && !curr.done; ) {
            const { punctuation, space, text } = curr.value;
 
            // This is a state machine of how to delete a whole word together with space and punctuations.
            // For a full state machine chart, see
            // Forward delete: https://github.com/microsoft/roosterjs/blob/master/assets/design-charts/ForwardDeleteWord.png
            // Backward delete: https://github.com/microsoft/roosterjs/blob/master/assets/design-charts/BackwardDeleteWord.png
            switch (state) {
                case DeleteWordState.Start:
                    state = space
                        ? DeleteWordState.Space
                        : punctuation
                        ? DeleteWordState.Punctuation
                        : DeleteWordState.Text;
                    curr = iterator.next(true /*delete*/);
                    break;
 
                case DeleteWordState.Punctuation:
                    Eif (deleteNext && space) {
                        state = DeleteWordState.NonText;
                        curr = iterator.next(true /*delete*/);
                    } else if (punctuation) {
                        curr = iterator.next(true /*delete*/);
                    } else {
                        state = DeleteWordState.End;
                    }
                    break;
 
                case DeleteWordState.Text:
                    if (deleteNext && space) {
                        state = DeleteWordState.NonText;
                        curr = iterator.next(true /*delete*/);
                    } else if (text) {
                        curr = iterator.next(true /*delete*/);
                    } else {
                        state = DeleteWordState.End;
                    }
                    break;
 
                case DeleteWordState.NonText:
                    if (punctuation || !space) {
                        state = DeleteWordState.End;
                    } else {
                        curr = iterator.next(true /*delete*/);
                    }
                    break;
 
                case DeleteWordState.Space:
                    if (space) {
                        curr = iterator.next(true /*delete*/);
                    } else Iif (punctuation) {
                        state = deleteNext ? DeleteWordState.NonText : DeleteWordState.Punctuation;
                        curr = iterator.next(true /*delete*/);
                    } else {
                        state = deleteNext ? DeleteWordState.End : DeleteWordState.Text;
                    }
                    break;
            }
        }
    };
}
 
function* iterateSegments(
    paragraph: ShallowMutableContentModelParagraph,
    markerIndex: number,
    forward: boolean,
    context: DeleteSelectionContext
): Generator<CharInfo, null, boolean> {
    const step = forward ? 1 : -1;
    const segments = paragraph.segments;
    const preserveWhiteSpace = isWhiteSpacePreserved(paragraph.format.whiteSpace);
 
    for (let i = markerIndex + step; i >= 0 && i < segments.length; i += step) {
        const segment = segments[i];
 
        switch (segment.segmentType) {
            case 'Text':
                for (
                    let j = forward ? 0 : segment.text.length - 1;
                    Ij >= 0 && j < segment.text.length;
                    j += step
                ) {
                    const c = segment.text[j];
                    const punctuation = isPunctuation(c);
                    const space = isSpace(c);
                    const text = !punctuation && !space;
 
                    Eif (yield { punctuation, space, text }) {
                        let newText = segment.text;
 
                        newText = newText.substring(0, j) + newText.substring(j + 1);
 
                        Eif (!preserveWhiteSpace) {
                            newText = normalizeText(newText, forward);
                        }
 
                        context.deleteResult = 'range';
 
                        if (newText) {
                            segment.text = newText;
 
                            if (step > 0) {
                                j -= step;
                            }
                        } else {
                            segments.splice(i, 1);
 
                            if (step > 0) {
                                i -= step;
                            }
 
                            break;
                        }
                    }
                }
                break;
 
            case 'Image':
                if (
                    yield { punctuation: true, space: false, text: false } // Treat image as punctuation since they have the same behavior.
                ) {
                    segments.splice(i, 1);
 
                    if (step > 0) {
                        i -= step;
                    }
 
                    context.deleteResult = 'range';
                }
                break;
 
            case 'SelectionMarker':
                break;
 
            default:
                return null;
        }
    }
 
    return null;
}
 
/**
 * @internal
 */
export const forwardDeleteWordSelection = getDeleteWordSelection('forward');
 
/**
 * @internal
 */
export const backwardDeleteWordSelection = getDeleteWordSelection('backward');