import { EditorView, Decoration, MatchDecorator, ViewPlugin, ViewUpdate } from "@codemirror/view";
import { RangeSet, RangeSetBuilder } from "@codemirror/state";

const searchMatchDecoration = Decoration.mark({ class: "cm-searchMatch" });

interface Match {
    from: number;
    to: number;
}

export class SearchHighlighter {
    matches: RangeSet<Decoration>;
    currentFound: number;
    totalFound: number;
    matcher?: MatchDecorator;
    private parsedMatches: Match[];

    constructor(public view: EditorView) {
        this.parsedMatches = [];
        this.currentFound = 0;
        this.totalFound = 0;
        this.matches = (new RangeSetBuilder<Decoration>()).finish();
    }

    searchFor(searchTerm: string, matchCase: boolean, wholeWord: boolean) {
        // Escape the search term for use in RegExp
        const escapedTerm = searchTerm.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
        const wordBoundary = wholeWord ? "\\b" : "";
        const flags = matchCase ? "g" : "gi";
        const regex = new RegExp(`${wordBoundary}${escapedTerm}${wordBoundary}`, flags);

        this.matcher = new MatchDecorator({
            regexp: regex,
            decoration: searchMatchDecoration,
        });
        this.updateSearchData(this.view);
        this.scrollToMatchNearestSelection();
    }

    updateSearchData(view: EditorView) {
        if (!this.matcher) {
            return;
        }

        const matches = this.matcher.createDeco(view);
        const cursor = matches.iter();
        while (cursor.value) {
            this.parsedMatches.push({
                from: cursor.from,
                to: cursor.to
            });
            cursor.next();
        }

        this.matches = matches;
        this.totalFound = this.parsedMatches.length;
    }

    update(update: ViewUpdate) {
        if (update.docChanged || update.viewportChanged) {
            this.updateSearchData(update.view);
        }
    }

    scrollToMatch(matchIndex: number) {
        if (this.parsedMatches.length <= matchIndex) {
            return;
        }

        const match = this.parsedMatches[matchIndex];
        this.currentFound = matchIndex + 1;
        this.view.dispatch({
            effects: EditorView.scrollIntoView(match.from, { y: "center" }),
            scrollIntoView: true
        });
    }

    scrollToMatchNearestSelection() {
        const cursorPos = this.view.state.selection.main.head;
        let index = 0;
        for (const match of this.parsedMatches) {
            if (match.from >= cursorPos) {
                this.scrollToMatch(index);
                return;
            }

            index++;
        }
    }

    destroy() {
        // Do nothing.
    }

    static deco = (v: SearchHighlighter) => v.matches;
}

export function createSearchHighlighter() {
    return ViewPlugin.fromClass(SearchHighlighter, {
        decorations: v => v.matches,
        provide: (plugin) => plugin
    });
}

export const searchMatchHighlightTheme = EditorView.baseTheme({
    ".cm-searchMatch": {
        backgroundColor: "rgba(255, 255, 0, 0.4)",
        borderRadius: "2px"
    }
});