feat: support heading links

close #48
markmap · Dec 18, 2024 · 4b91ae7 · 4b91ae7
1 parent 62ca778
commit 4b91ae7
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/src/app.ts b/src/app.ts
@@ -46,7 +46,10 @@ document.addEventListener('click', (e) => {
   const el = (e.target as HTMLElement)?.closest('a');
   if (el) {
     const href = el.getAttribute('href');
-    if (!href.includes('://')) {
+    if (href.startsWith('#')) {
+      const node = findHeading(href.slice(1));
+      if (node) highlightNode(node);
+    } else if (!href.includes('://')) {
       vscode.postMessage({
         type: 'openFile',
         data: href,
@@ -98,6 +101,21 @@ function clickHandler(type: string) {
   };
 }
 
+function findHeading(id: string) {
+  function dfs(node: INode) {
+    if (!/^h\d$/.test(node.payload.tag as string)) return false;
+    const normalizedId = node.content.trim().replace(/\W/g, '-').toLowerCase();
+    if (normalizedId === id) {
+      target = node;
+      return true;
+    }
+    return node.children?.some(dfs);
+  }
+  let target: INode | undefined;
+  dfs(root);
+  return target;
+}
+
 function findActiveNode(line: number) {
   function dfs(node: INode) {
     const [start, end] =