@rshirohara/rekurke-parse
Version:
rekurke plugin to add support for parsing kakuyomu novel format.
95 lines (94 loc) • 2.87 kB
JavaScript
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": {
const children = [];
let joinedText = resetTextNode();
for (const item of node.children) {
if (item.type !== "text") {
if (joinedText.value !== "") {
children.push(joinedText);
joinedText = resetTextNode();
}
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,
},
},
};
}