knotes/v3
knotes/v3
1
0
Fork 0
Code Issues 34 Pull requests Projects Releases Packages 1 Wiki Activity Actions

Make headers work properly when typed normally. Fixes #1
All checks were successful
/ Push Docker image to local registry (push) Successful in 3m8s
Details

Browse source
This commit is contained in:
kalle 2025-04-16 17:22:07 +02:00
parent db9cd4f91e
commit d4f2f301d8
1 changed files with 59 additions and 54 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

113
frontend/src/editor/plugins/header_plugin.tsx
View file

@ -1,8 +1,9 @@
import { useLexicalComposerContext } from "@lexical/react/LexicalComposerContext";
import { $createParagraphNode, $isParagraphNode, $isTextNode, SELECTION_CHANGE_COMMAND, TextNode } from "lexical";
import { $createParagraphNode, $isParagraphNode, $isTextNode, SELECTION_CHANGE_COMMAND, TextNode, type LexicalEditor, type LexicalNode } from "lexical";
import { useEffect } from "react";
import { $createHeaderMarkerNode, $createHeaderNode, $isHeaderMarkerNode, $isHeaderNode, HeaderMarkerNode, HeaderNode } from "../nodes/header_node";
import { mergeRegister } from "@lexical/utils";
import { findBlockNode } from "../editor_utils";
const HEADER_REGEX = /^#+ /;
@ -11,68 +12,72 @@ export function HeaderPlugin() {
useEffect(() => mergeRegister(
// Create a header node if the text node matches the HEADER_REGEX
editor.registerNodeTransform(TextNode, (textNode) => {
const prevNode = textNode.getPreviousSibling();
if (prevNode) return;
const paragraphNode = textNode.getParent();
if (!$isParagraphNode(paragraphNode)) return;
const content = textNode.getTextContent();
const regexMatch = content.match(HEADER_REGEX);
if (!regexMatch) return;
const children = paragraphNode.getChildren();
const firstTextNode = children[0];
if (!$isTextNode(firstTextNode)) return;
const markerLength = regexMatch[0].length;
const textNodes = firstTextNode.splitText(markerLength);
const headerMarkerContent = textNodes[0];
if (!headerMarkerContent) return;
const headerNode = $createHeaderNode(markerLength - 1);
const headerMarkerNode = $createHeaderMarkerNode();
headerMarkerNode.append(headerMarkerContent);
headerNode.append(headerMarkerNode);
headerNode.append(...textNodes.slice(1));
headerNode.append(...children.slice(1));
paragraphNode.replace(headerNode, true);
editor.dispatchCommand(SELECTION_CHANGE_COMMAND, undefined);
}),
// Remove headers if they don't match the HEADER_REGEX
editor.registerNodeTransform(HeaderNode, (headerNode) => {
const content = headerNode.getTextContent();
if (content.match(HEADER_REGEX)) return;
headerNode.replace($createParagraphNode(), true);
}),
// Remove header markers if they don't match the HEADER_REGEX
editor.registerNodeTransform(TextNode, createHeaderTransform(editor)),
editor.registerNodeTransform(TextNode, ensureHeaderValidTransform),
editor.registerNodeTransform(HeaderNode, ensureHeaderValidTransform),
// Remove header markers if they aren't in a header
editor.registerNodeTransform(HeaderMarkerNode, (node) => {
const headerNode = node.getParent();
const content = node.getTextContent();
if ($isHeaderNode(headerNode) &&
content.match(HEADER_REGEX) &&
content.length - 1 == headerNode.getLevel()) {
if ($isHeaderNode(headerNode)) {
return;
}
node.getChildren().reverse().forEach(child => node.insertAfter(child));
node.remove();
}),
// Remove header nodes without a header marker
editor.registerNodeTransform(HeaderNode, (node) => {
const children = node.getChildren();
const headerMarker = children[0];
if ($isHeaderMarkerNode(headerMarker)) return;
node.replace($createParagraphNode(), true);
}),
));
return null;
}
const createHeaderTransform = (editor: LexicalEditor) => (textNode: TextNode) => {
const prevNode = textNode.getPreviousSibling();
if (prevNode) return;
const paragraphNode = textNode.getParent();
if (!$isParagraphNode(paragraphNode)) return;
const content = textNode.getTextContent();
const regexMatch = content.match(HEADER_REGEX);
if (!regexMatch) return;
const children = paragraphNode.getChildren();
const firstTextNode = children[0];
if (!$isTextNode(firstTextNode)) return;
const markerLength = regexMatch[0].length;
const textNodes = firstTextNode.splitText(markerLength);
const headerMarkerContent = textNodes[0];
if (!headerMarkerContent) return;
const headerNode = $createHeaderNode(markerLength - 1);
const headerMarkerNode = $createHeaderMarkerNode();
headerMarkerNode.append(headerMarkerContent);
headerNode.append(headerMarkerNode);
headerNode.append(...textNodes.slice(1));
headerNode.append(...children.slice(1));
paragraphNode.replace(headerNode, true);
editor.dispatchCommand(SELECTION_CHANGE_COMMAND, undefined);
}
const ensureHeaderValidTransform = (node: LexicalNode) => {
const headerNode = findBlockNode(node);
if (!$isHeaderNode(headerNode)) {
return;
}
const markerNode = headerNode.getFirstChild();
const markerContent = markerNode?.getTextContent();
if ($isHeaderMarkerNode(markerNode)
&& markerContent?.match(HEADER_REGEX)
&& markerContent.length == headerNode.getLevel() + 1) {
return
}
headerNode.replace($createParagraphNode(), true);
}