kotaindah55_extended-markdown-syntax/src/editor-mode/parser/ParserState.ts

import { Line, Text } from "@codemirror/state";
import { Tree, TreeCursor } from "@lezer/common";
import { StateConfig, TokenGroup } from "src/types";
import { TokenQueue } from "src/editor-mode/parser";
import { Format, LineCtx } from "src/enums";
import { NonHighlightFormats, SpaceRestrictedFormats } from "src/shared-configs";
import { SKIPPED_NODE_RE } from "src/editor-mode/parser/regexps";
import { findNode, getContextFromNode } from "src/editor-mode/parser/utils";

export class ParserState {
    doc: Text;
    tree: Tree;
    cursor: TreeCursor | null;
    line: Line;
    /** block start */
    maxLine: number;
    offset: number;
    tokens: TokenGroup;
    queue: TokenQueue;
    curCtx: LineCtx = LineCtx.NONE;
    prevCtx: LineCtx = LineCtx.NONE;
    constructor(config: StateConfig, tokens: TokenGroup) {
        this.doc = config.doc;
        this.tree = config.tree;
        this.line = this.doc.lineAt(config.offset);
        this.maxLine = config.maxLine || this.doc.lineAt(this.tree.length).number;
        this.offset = config.offset - this.line.from;
        this.tokens = tokens;
        this.cursor = this.tree.cursor();
        this.nextCursor();
        if (
            (!this.lastToken || this.lastToken.to < this.line.from) &&
            this.lineStr[0] == "!" && this.offset < 11
        ) { this.offset = 0 }
    }
    /** global offset */
    get gOffset() {
        return this.offset + this.line.from;
    }
    get linePos() {
        return this.line.number;
    }
    get lineStr() {
        return this.line.text;
    }
    get char() {
        let char = this.line.text[this.offset];
        if (!char && !this.isLastLine()) {
            return "\n";
        }
        return char ?? "";
    }
    get lastToken() {
        return this.tokens.at(-1);
    }
    get prevLine() {
        let linePos = this.linePos;
        if (linePos == 1) { return null }
        return this.doc.line(linePos - 1);
    }
    advance(n = 1) {
        let restLen = this.lineStr.length - this.offset;
        if (!restLen) {
            this.queue.resolve(SpaceRestrictedFormats);
            return false;
        }
        if (n > restLen) {
            this.offset += restLen;
        } else {
            this.offset += n;
        }
        return true;
    }
    setGOffset(gOffset: number) {
        if (gOffset > this.doc.length) {
            this.line = this.doc.line(this.doc.lines);
            this.offset = this.line.to;
        } else {
            this.line = this.doc.lineAt(gOffset);
            this.offset = gOffset - this.line.from;
        }
    }
    isSpace(side: -1 | 0 | 1 = 0) {
        let char = this.lineStr[this.offset + side];
        return char == " " || char == "\t" || !char;
    }
    seekWhitespace(maxOffset = this.line.length) {
        let offset = this.offset;
        for (let char = this.line.text[offset]; offset < maxOffset; char = this.line.text[++offset]) {
            if (char == " " || char == "\t") {
                return offset;
            }
        }
        return null;
    }
    nextLine(skipBlankLine = true) {
        if (this.linePos == this.maxLine) {
            this.queue.resolveAll();
            return null;
        }
        this.line = this.doc.line(this.linePos + 1);
        this.resolveContext();
        if (skipBlankLine) {
            if (this.isBlankLine()) {
                this.offset = this.line.length;
                this.queue.resolveAll();
                while (this.linePos != this.maxLine) {
                    this.line = this.doc.line(this.linePos + 1);
                    if (!this.isBlankLine()) { break }
                }
            }
        }
        this.offset = 0;
        return this.line;
    }
    isLastLine() {
        return this.line.number >= this.doc.lines;
    }
    isBlankLine() {
        return !this.lineStr.trimEnd();
    }
    nextCursor(enter = true) {
        if (this.cursor) {
            if (this.cursor.next(enter) && this.cursor.name != "Document") { return true }
            this.cursor = null;
        }
        return false;
    }
    cursorPos(endSide: 0 | -1 = 0): "after" | "before" | "touch" | null {
        let offset = this.gOffset;
        if (!this.cursor) { return null }
        if (offset < this.cursor.from) { return "after" }
        if (offset > this.cursor.to + endSide) { return "before" }
        return "touch";
    }
    processCursor() {
        let cursorPos = this.cursorPos(-1);
        while (cursorPos == "before") {
            this.nextCursor();
            cursorPos = this.cursorPos(-1);
        }
        if (cursorPos != "touch") { return null }
        let nodeName = this.cursor!.name;
        if (nodeName.includes("formatting-highlight")) { return "hl_delim" }
        if (nodeName.includes("table-sep")) { return "table_sep" }
        if (SKIPPED_NODE_RE.test(nodeName)) { return "skipped" }
        return null;
    }
    skipCursorRange() {
        if (!this.cursor) { return false }
        let cursorTo = this.cursor.to - this.line.from,
        whitespaceOffset = this.seekWhitespace(cursorTo);
        if (whitespaceOffset !== null) {
            this.offset = whitespaceOffset;
            this.queue.resolve(SpaceRestrictedFormats);
        }
        this.offset = cursorTo;
        return true;
    }
    getContext(line = this.line) {
        if (line.number != this.linePos) {
            let node = findNode(
                this.tree, line.from, line.from,
                (node) => node.parent?.name == "Document"
            );
            if (node) { return getContextFromNode(node) }
        } else if (this.cursorPos() == "touch") {
            let node = this.cursor!.node;
            return getContextFromNode(node);
        }
        return LineCtx.NONE;
    }
    setContext(ctx: LineCtx) {
        this.prevCtx = this.curCtx;
        this.curCtx = ctx;
    }
    resolveContext() {
        if (this.offset) { return }
        while (this.cursorPos() == "before") { this.nextCursor() }
        this.setContext(this.getContext());
        let isSkip = false,
            toBeResolved = false,
            includesHl = false,
            offset = this.line.from;
        switch (this.curCtx) {
            case LineCtx.HR_LINE:
            case LineCtx.CODEBLOCK:
            case LineCtx.TABLE_DELIM:
                isSkip = true;
                toBeResolved = true;
                break;
            case LineCtx.BLOCKQUOTE:
                if (this.prevCtx != LineCtx.BLOCKQUOTE) {
                    toBeResolved = true;
                }
                break;
            case LineCtx.LIST_HEAD:
                includesHl = true;
            // eslint-disable-next-line no-fallthrough
            case LineCtx.HEADING:
            case LineCtx.FOOTNOTE_HEAD:
                toBeResolved = true;
                break;
        }
        switch (this.prevCtx) {
            case LineCtx.HEADING:
            case LineCtx.TABLE:
                toBeResolved = true;
                offset -= 1;
        }
        toBeResolved && this.queue.resolve(NonHighlightFormats, offset);
        includesHl && this.queue.resolve([Format.HIGHLIGHT], offset);
        isSkip && this.skipCursorRange();
        this.curCtx && this.nextCursor();
    }
}