Notes/src/public/app/widgets/find_in_text.js

export default class FindInText {
    constructor(parent) {
        /** @property {FindWidget} */
        this.parent = parent;
    }

    async getTextEditor() {
        return this.parent.noteContext.getTextEditor();
    }

    async getInitialSearchTerm() {
        const textEditor = await this.getTextEditor();

        const selection = textEditor.model.document.selection;
        const range = selection.getFirstRange();

        for (const item of range.getItems()) {
            // Fill in the findbox with the current selection if
            // any
            return item.data;
        }
    }

    async performFind(searchTerm, matchCase, wholeWord) {
        // Do this even if the searchTerm is empty so the markers are cleared and
        // the counters updated
        const textEditor = await this.getTextEditor();
        const model = textEditor.model;
        let findResult = null;
        let totalFound = 0;
        let currentFound = -1;

        // Clear
        const findAndReplaceEditing = textEditor.plugins.get('FindAndReplaceEditing');
        findAndReplaceEditing.state.clear(model);
        findAndReplaceEditing.stop();
        if (searchTerm !== "") {
            // Parameters are callback/text, options.matchCase=false, options.wholeWords=false
            // See https://github.com/ckeditor/ckeditor5/blob/b95e2faf817262ac0e1e21993d9c0bde3f1be594/packages/ckeditor5-find-and-replace/src/findcommand.js#L44
            // XXX Need to use the callback version for regexp
            // searchTerm = escapeRegExp(searchTerm);
            // let re = new RegExp(searchTerm, 'gi');
            // let m = text.match(re);
            // totalFound = m ? m.length : 0;
            const options = { "matchCase" : matchCase, "wholeWords" : wholeWord };
            findResult = textEditor.execute('find', searchTerm, options);
            totalFound = findResult.results.length;
            // Find the result beyond the cursor
            const cursorPos = model.document.selection.getLastPosition();
            for (let i = 0; i < findResult.results.length; ++i) {
                const marker = findResult.results.get(i).marker;
                const fromPos = marker.getStart();
                if (fromPos.compareWith(cursorPos) !== "before") {
                    currentFound = i;
                    break;
                }
            }
        }

        this.findResult = findResult;

        // Calculate curfound if not already, highlight it as
        // selected
        if (totalFound > 0) {
            currentFound = Math.max(0, currentFound);
            // XXX Do this accessing the private data?
            // See
            // https://github.com/ckeditor/ckeditor5/blob/b95e2faf817262ac0e1e21993d9c0bde3f1be594/packages/ckeditor5-find-and-replace/src/findnextcommand.js
            for (let i = 0 ; i < currentFound; ++i) {
                textEditor.execute('findNext', searchTerm);
            }
        }

        return {
            totalFound,
            currentFound: currentFound + 1
        };
    }

    async findNext(direction, currentFound, nextFound) {
        const textEditor = await this.getTextEditor();

        // There are no parameters for findNext/findPrev
        // See https://github.com/ckeditor/ckeditor5/blob/b95e2faf817262ac0e1e21993d9c0bde3f1be594/packages/ckeditor5-find-and-replace/src/findnextcommand.js#L57
        // curFound wrap around above assumes findNext and
        // findPrevious wraparound, which is what they do
        if (direction > 0) {
            textEditor.execute('findNext');
        } else {
            textEditor.execute('findPrevious');
        }
    }

    async cleanup(totalFound, currentFound) {
        if (totalFound > 0) {
            const textEditor = await this.getTextEditor();
            // Clear the markers and set the caret to the
            // current occurrence
            const model = textEditor.model;
            const range = this.findResult.results.get(currentFound).marker.getRange();
            // From
            // https://github.com/ckeditor/ckeditor5/blob/b95e2faf817262ac0e1e21993d9c0bde3f1be594/packages/ckeditor5-find-and-replace/src/findandreplace.js#L92
            // XXX Roll our own since already done for codeEditor and
            //     will probably allow more refactoring?
            let findAndReplaceEditing = textEditor.plugins.get('FindAndReplaceEditing');
            findAndReplaceEditing.state.clear(model);
            findAndReplaceEditing.stop();
            model.change(writer => {
                writer.setSelection(range, 0);
            });
            textEditor.editing.view.scrollToTheSelection();
        }

        this.findResult = null;
    }

    async close() {
        const textEditor = await this.getTextEditor();
        textEditor.focus();
    }
}