perf(rehype): stop using hast-util-select and simplify the implementa…

…tion

package.json

@@ -26,7 +26,6 @@
   "license": "MIT",
   "homepage": "https://github.com/inkdropapp/html2markdown",
   "dependencies": {
-    "hast-util-select": "^5.0.5",
     "mdast-util-gfm": "^3.0.0",
     "rehype-parse": "^8.0.4",
     "rehype-remark": "^9.1.2",

src/index.ts

@@ -5,7 +5,7 @@ import { comment as toMdastComment } from './to-mdast-comment'
 import { unified } from 'unified'
 import rehypeParse from 'rehype-parse'
 import rehype2remark from 'rehype-remark'
-import { rehypeInsert } from './rehype-insert'
+import { rehypeInsertBaseURI } from './rehype-insert-base-uri'
 import stringify, { Options as RemarkStringifyOptions } from 'remark-stringify'
 import { squeezeLinks } from 'remark-squeeze-links'
 import { gfmToMarkdown } from 'mdast-util-gfm'
@@ -24,45 +24,34 @@ function getConverter(opts?: Options) {
     baseURI = null
   } = opts || {}
 
-  const remark = unified()
+  let remark = unified()
     .data('toMarkdownExtensions', [gfmToMarkdown()])
     .use(rehypeParse)
-    .use(rehypeInsert, {
-      insertions: baseURI
-        ? [
-            {
-              selector: 'head',
-              insert: {
-                type: 'element',
-                tagName: 'base',
-                properties: {
-                  href: baseURI
-                },
-                children: []
-              }
-            }
-          ]
-        : []
-    })
-    .use(rehype2remark, {
-      handlers: {
-        pre: toMdastCodeBlock,
-        comment: toMdastComment,
-        ...(toMdastOptions.handlers || {})
-      },
-      ...toMdastOptions
-    })
-  // @ts-ignore
-  return remark.use(squeezeLinks).use(stringify, {
-    listItemIndent: 'one',
-    bullet: '*',
-    fences: true,
-    handlers: {
-      break: hardBreak,
-      ...(stringifyOptions.handlers || {})
-    },
-    ...stringifyOptions
-  })
+  if (baseURI) remark = remark.use(rehypeInsertBaseURI, baseURI || '/')
+
+  return (
+    remark
+      .use(rehype2remark, {
+        handlers: {
+          pre: toMdastCodeBlock,
+          comment: toMdastComment,
+          ...(toMdastOptions.handlers || {})
+        },
+        ...toMdastOptions
+      })
+      .use(squeezeLinks)
+      // @ts-ignore
+      .use(stringify, {
+        listItemIndent: 'one',
+        bullet: '*',
+        fences: true,
+        handlers: {
+          break: hardBreak,
+          ...(stringifyOptions.handlers || {})
+        },
+        ...stringifyOptions
+      })
+  )
 }
 
 export function html2Markdown(html: string, opts?: Options): string {

src/rehype-insert-base-uri.ts

@@ -0,0 +1,21 @@
+import type { Plugin } from 'unified'
+import type { Root } from 'hast'
+import { visit } from 'unist-util-visit'
+
+export const rehypeInsertBaseURI: Plugin<[string], Root, Root> = baseURI => {
+  return function rehypeInsertTransformer(tree) {
+    visit(tree, { type: 'element', tagName: 'head' }, headNode => {
+      headNode.children = [
+        ...headNode.children,
+        {
+          type: 'element',
+          tagName: 'base',
+          properties: {
+            href: baseURI
+          },
+          children: []
+        }
+      ]
+    })
+  }
+}