Change types to use mdast types

Author: wooorm

index.js

@@ -1,5 +1,5 @@
 /**
- * @typedef {import('unist').Node} Node
+ * @typedef {import('mdast').Root|import('mdast').Content} Node
  * @typedef {import('mdast').Heading} Heading
  */
 
@@ -23,24 +23,19 @@ export function normalizeHeadings(tree) {
   /** @type {Heading|undefined} */
   let title
 
-  visit(
-    tree,
-    'heading',
-    /** @type {import('unist-util-visit').Visitor<Heading>} */
-    (node) => {
-      if (!first) {
-        first = node
-      }
+  visit(tree, 'heading', (node) => {
+    if (!first) {
+      first = node
+    }
 
-      if (node.depth === 1) {
-        if (title) {
-          multiple = true
-        } else {
-          title = node
-        }
+    if (node.depth === 1) {
+      if (title) {
+        multiple = true
+      } else {
+        title = node
       }
     }
-  )
+  })
 
   // If there are no titles, but there is a first heading.
   if (!title && first) {
@@ -49,16 +44,11 @@ export function normalizeHeadings(tree) {
 
   // If there are multiple titles.
   if (multiple) {
-    visit(
-      tree,
-      'heading',
-      /** @type {import('unist-util-visit').Visitor<Heading>} */
-      (node) => {
-        if (node !== title && node.depth < max) {
-          node.depth++
-        }
+    visit(tree, 'heading', (node) => {
+      if (node !== title && node.depth < max) {
+        node.depth++
       }
-    )
+    })
   }
 
   return tree

package.json

@@ -38,8 +38,7 @@
   ],
   "dependencies": {
     "@types/mdast": "^3.0.0",
-    "@types/unist": "^2.0.0",
-    "unist-util-visit": "^3.0.0"
+    "unist-util-visit": "^4.0.0"
   },
   "devDependencies": {
     "@types/tape": "^4.0.0",

test/index.js

@@ -1,3 +1,7 @@
+/**
+ * @typedef {import('mdast').Root} Root
+ */
+
 import fs from 'node:fs'
 import path from 'node:path'
 import test from 'tape'
@@ -27,9 +31,10 @@ test('Level 7', (t) => {
 function check(t, test, message) {
   const input = fs.readFileSync(path.join('test', 'fixture', test + '.in'))
   const output = fs.readFileSync(path.join('test', 'fixture', test + '.out'))
+  const root = /** @type {Root} */ (remark().parse(input))
 
   t.deepEqual(
-    removePosition(normalizeHeadings(remark().parse(input)), true),
+    removePosition(normalizeHeadings(root), true),
     removePosition(remark().parse(output), true),
     message
   )