Function createEditorHighlightDecorations

src/editor/EditorHighlightDecorations.ts:17–67

(options: EditorHighlightDecorationOptions)

}

export function createEditorHighlightDecorations(options: EditorHighlightDecorationOptions) {
    const { plugin, highlightService, highlightRepository } = options;
    const highlightCommentResolver = new HighlightCommentResolver(highlightRepository);

    return ViewPlugin.fromClass(class {
        decorations: DecorationSet;

        constructor(view: EditorView) {
            this.decorations = this.buildDecorations(view);
        }

        update(update: ViewUpdate): void {
            if (update.docChanged || update.viewportChanged || update.transactions.length > 0) {
                this.decorations = this.buildDecorations(update.view);
            }
        }

        private buildDecorations(view: EditorView): DecorationSet {
            const activeView = plugin.app.workspace.getActiveViewOfType(MarkdownView);
            const file = activeView?.file;

            if (!file || !highlightService.shouldProcessFile(file)) {
                return Decoration.none;
            }

            const decorations: Range<Decoration>[] = [];
            const highlights = highlightService.extractHighlights(view.state.doc.toString(), file);

            for (const highlight of highlights) {
                if (highlight.position === undefined) continue;

                const commentHighlight = highlightCommentResolver.normalizeHighlight(highlight);
                commentHighlight.comments = highlightCommentResolver.getCommentsForHighlight(file, commentHighlight, {
                    onTextChanged: (storedHighlight, currentHighlight) => {
                        emitHighlightTextChange(plugin, file, storedHighlight, currentHighlight);
                    }
                });

                const highlightEndPos = highlight.position + (highlight.originalLength ?? highlight.text.length + 4);

                if (shouldShowCommentWidget(plugin)) {
                    decorations.push(createCommentWidget(plugin, commentHighlight).range(highlightEndPos));
                }
            }

            return Decoration.set(decorations.sort((a, b) => a.from - b.from));
        }
    }, {
        decorations: value => value.decorations
    });
}

function createCommentWidget(plugin: HiNotePluginContext, highlight: HiNote): Decoration {
    return Decoration.widget({