summaryrefslogtreecommitdiff
path: root/packages/markdown-support/src/rehype-collect-headers.ts
diff options
context:
space:
mode:
authorGravatar Matthew Phillips <matthew@matthewphillips.info> 2021-06-04 14:19:01 -0400
committerGravatar GitHub <noreply@github.com> 2021-06-04 14:19:01 -0400
commit50e6f491ad2258689ba8735a3f98a0d99e4ba336 (patch)
treecc2c1284b3632d6447d3c9bab3b5c1aa461c35cf /packages/markdown-support/src/rehype-collect-headers.ts
parenta2594ef572c41ce05dd01ef095e8595ffcd35842 (diff)
downloadastro-50e6f491ad2258689ba8735a3f98a0d99e4ba336.tar.gz
astro-50e6f491ad2258689ba8735a3f98a0d99e4ba336.tar.zst
astro-50e6f491ad2258689ba8735a3f98a0d99e4ba336.zip
Use npm package names to load internal deps (#294)
* Use npm package names to load internal deps This is necessary so that published Astro components work. These components will be built by esinstall and therefore they cannot rely on `_astro_internal`. The fix is to use npm specifiers everywhere. * Move most of frontend to internal * Mark astro/internal/markdown.js as external * Move markdown stuff to its own package This moves the markdown stuff to its own package so that we can externalize it in the markdown component. * Add the changeset
Diffstat (limited to 'packages/markdown-support/src/rehype-collect-headers.ts')
-rw-r--r--packages/markdown-support/src/rehype-collect-headers.ts30
1 files changed, 30 insertions, 0 deletions
diff --git a/packages/markdown-support/src/rehype-collect-headers.ts b/packages/markdown-support/src/rehype-collect-headers.ts
new file mode 100644
index 000000000..edfcd29bc
--- /dev/null
+++ b/packages/markdown-support/src/rehype-collect-headers.ts
@@ -0,0 +1,30 @@
+import { visit } from 'unist-util-visit';
+import slugger from 'github-slugger';
+
+/** */
+export default function createCollectHeaders() {
+ const headers: any[] = [];
+
+ const visitor = (node: any) => {
+ if (node.type !== 'element') return;
+ const { tagName, children } = node;
+ if (tagName[0] !== 'h') return;
+ let [_, depth] = tagName.match(/h([0-6])/) ?? [];
+ if (!depth) return;
+ depth = Number.parseInt(depth);
+
+ let text = '';
+ visit(node, 'text', (child) => {
+ text += child.value;
+ });
+
+ let slug = slugger.slug(text);
+ node.properties = node.properties || {};
+ node.properties.id = slug;
+ headers.push({ depth, slug, text });
+
+ return node;
+ };
+
+ return { headers, rehypeCollectHeaders: () => (tree: any) => visit(tree, visitor) };
+}