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

A YAML parser that produces output compatible with unist

import { createDocumentHead } from "../factories/document-head.js";
import { getMatchIndex } from "../utils/get-match-index.js";
export function transformDocumentHead(document, context) {
    const cstNode = document.cstNode;
    const { directives, comments, endComments } = categorizeNodes(cstNode, context);
    const { position, endMarkerPoint } = getPosition(cstNode, directives, context);
    context.comments.push(...comments, ...endComments);
    const createDocumentHeadWithTrailingComment = (trailingComment) => {
        if (trailingComment) {
            context.comments.push(trailingComment);
        }
        return createDocumentHead(position, directives, endComments, trailingComment);
    };
    return {
        createDocumentHeadWithTrailingComment,
        documentHeadEndMarkerPoint: endMarkerPoint,
    };
}
function categorizeNodes(document, context) {
    const directives = [];
    const comments = [];
    const endComments = [];
    let hasDirective = false;
    for (let i = document.directives.length - 1; i >= 0; i--) {
        const node = context.transformNode(document.directives[i]);
        if (node.type === "comment") {
            if (hasDirective) {
                comments.unshift(node);
            }
            else {
                endComments.unshift(node);
            }
        }
        else {
            hasDirective = true;
            directives.unshift(node);
        }
    }
    return { directives, comments, endComments };
}
function getPosition(document, directives, context) {
    let endMarkerIndex = getMatchIndex(context.text.slice(0, document.valueRange.origStart), /---\s*$/);
    // end marker should start with the first character on the line
    if (endMarkerIndex > 0 && !/[\r\n]/.test(context.text[endMarkerIndex - 1])) {
        endMarkerIndex = -1;
    }
    const range = endMarkerIndex === -1
        ? {
            origStart: document.valueRange.origStart,
            origEnd: document.valueRange.origStart,
        }
        : {
            origStart: endMarkerIndex,
            origEnd: endMarkerIndex + 3,
        };
    if (directives.length !== 0) {
        range.origStart = directives[0].position.start.offset;
    }
    return {
        position: context.transformRange(range),
        endMarkerPoint: endMarkerIndex === -1 ? null : context.transformOffset(endMarkerIndex),
    };
}