yaml-unist-parser/lib/transforms/document-body.js

A YAML parser that produces output compatible with unist

import { createDocumentBody } from "../factories/document-body.js";
import { getLast } from "../utils/get-last.js";
import { getMatchIndex } from "../utils/get-match-index.js";
import { getPointText } from "../utils/get-point-text.js";
export function transformDocumentBody(document, context, headEndMarkerPoint) {
    const cstNode = document.cstNode;
    const { comments, endComments, documentTrailingComment, documentHeadTrailingComment, } = categorizeNodes(cstNode, context, headEndMarkerPoint);
    const content = context.transformNode(document.contents);
    const { position, documentEndPoint } = getPosition(cstNode, content, context);
    context.comments.push(...comments, ...endComments);
    return {
        documentBody: createDocumentBody(position, content, endComments),
        documentEndPoint,
        documentTrailingComment,
        documentHeadTrailingComment,
    };
}
function categorizeNodes(document, context, headEndMarkerPoint) {
    const comments = [];
    const endComments = [];
    const documentTrailingComments = [];
    const documentHeadTrailingComments = [];
    let hasContent = false;
    for (let i = document.contents.length - 1; i >= 0; i--) {
        const node = document.contents[i];
        if (node.type === "COMMENT") {
            const comment = context.transformNode(node);
            if (headEndMarkerPoint &&
                headEndMarkerPoint.line === comment.position.start.line) {
                documentHeadTrailingComments.unshift(comment);
            }
            else if (hasContent) {
                comments.unshift(comment);
            }
            else if (comment.position.start.offset >= document.valueRange.origEnd) {
                documentTrailingComments.unshift(comment);
            }
            else {
                comments.unshift(comment);
            }
        }
        else {
            hasContent = true;
        }
    }
    // istanbul ignore next
    if (documentTrailingComments.length > 1) {
        throw new Error(`Unexpected multiple document trailing comments at ${getPointText(documentTrailingComments[1].position.start)}`);
    }
    // istanbul ignore next
    if (documentHeadTrailingComments.length > 1) {
        throw new Error(`Unexpected multiple documentHead trailing comments at ${getPointText(documentHeadTrailingComments[1].position.start)}`);
    }
    return {
        comments,
        endComments,
        documentTrailingComment: getLast(documentTrailingComments) || null,
        documentHeadTrailingComment: getLast(documentHeadTrailingComments) || null,
    };
}
function getPosition(document, content, context) {
    const markerIndex = getMatchIndex(context.text.slice(document.valueRange.origEnd), /^\.\.\./);
    let origEnd = markerIndex === -1
        ? document.valueRange.origEnd
        : Math.max(0, document.valueRange.origEnd - 1);
    // CRLF fix
    if (context.text[origEnd - 1] === "\r") {
        origEnd--;
    }
    const position = context.transformRange({
        origStart: content !== null ? content.position.start.offset : origEnd,
        origEnd,
    });
    const documentEndPoint = markerIndex === -1
        ? position.end
        : context.transformOffset(document.valueRange.origEnd + 3);
    return { position, documentEndPoint };
}