@rshirohara/repixe-parse/dist/lib/postprocessor.js

repixe plugin to add support for parsing pixiv novel format.

export const postprocess = (node) => {
    const processors = [joinTexts, removeLastBreaks];
    let result = { ...node };
    for (const processor of processors) {
        result = processor(result);
    }
    return result;
};
function removeLastBreaks(node) {
    switch (node.type) {
        case "root": {
            return {
                ...node,
                children: [...node.children].map((node) => removeLastBreaks(node)),
            };
        }
        case "paragraph": {
            if (node.children.length > 0 &&
                node.children[node.children.length - 1].type === "break") {
                return { ...node, children: node.children.slice(0, -1) };
            }
            return node;
        }
        default: {
            return node;
        }
    }
}
function joinTexts(node) {
    switch (node.type) {
        case "root": {
            return {
                ...node,
                children: [...node.children].map((node) => joinTexts(node)),
            };
        }
        case "paragraph":
        case "heading":
        case "link": {
            let target = [...node.children];
            if (node.type === "paragraph") {
                target = [...node.children].map((node) => joinTexts(node));
            }
            const children = [];
            let joinedText = resetTextNode();
            for (const item of target) {
                if (item.type !== "text") {
                    if (joinedText.value !== "") {
                        children.push(joinedText);
                        joinedText = resetTextNode();
                    }
                    // @ts-expect-error: `children` が `node.children` であることは実行時に保証される
                    children.push(item);
                    continue;
                }
                joinedText = generateJoinedText(joinedText, item);
            }
            if (joinedText.value !== "") {
                children.push(joinedText);
            }
            return { ...node, children };
        }
        default: {
            return node;
        }
    }
}
function resetTextNode() {
    return {
        type: "text",
        value: "",
        position: {
            start: { line: 0, column: 0, offset: 0 },
            end: { line: 0, column: 0, offset: 0 },
        },
    };
}
function generateJoinedText(prev, curr) {
    return {
        type: "text",
        value: prev.value + curr.value,
        position: {
            start: {
                line: prev.value === ""
                    ? (curr.position?.start.line ?? 0)
                    : (prev.position?.start.line ?? 0),
                column: prev.value === ""
                    ? (curr.position?.start.column ?? 0)
                    : (prev.position?.start.column ?? 0),
                offset: prev.value === ""
                    ? (curr.position?.start.offset ?? 0)
                    : (prev.position?.start.offset ?? 0),
            },
            end: {
                line: curr.position?.end.line ?? 0,
                column: curr.position?.end.column ?? 0,
                offset: curr.position?.end.offset ?? 0,
            },
        },
    };
}