fix(language-service): use JSX node name map

Author: guyutongxue

packages/language-service/lib/virtual-code.js

@@ -23,27 +23,7 @@ import {isInjectableComponent, isInjectableEstree} from './jsx-utils.js'
  *   The string to use for the JSX import source tag.
  */
 const jsPrefix = (jsxImportSource) => `/* @jsxRuntime automatic
-@jsxImportSource ${jsxImportSource} */
-
-/**
- * @internal
- * @template T
- * @typedef {void extends MDX.IntrinsicElements ? any : T extends keyof MDX.IntrinsicElements ? MDX.IntrinsicElements[T] : any} _MDXProps
- */
-
-/**
- * @internal
- * @typedef {void extends MDX.Element ? JSX.Element : MDX.Element} _MDXElement
- */
-
-/**
- * @internal
- * @type {{
- *   [Name in keyof _MDXNodeNames]: (props: _MDXProps<Name>) => _MDXElement
- * }}
- */
-const _MDXNodes = /** @type {any} */(0);
-`
+@jsxImportSource ${jsxImportSource} */`
 
 /**
  * @param {string} propsName
@@ -244,6 +224,48 @@ function padOffsets(mapping, padding) {
   }
 }
 
+/** @type {Partial<Record<Nodes["type"], string>>} */
+const mdastElementNodeNameMap = {
+  blockquote: 'blockquote',
+  break: 'br',
+  delete: 'del',
+  emphasis: 'em',
+  html: 'html',
+  image: 'img',
+  imageReference: 'img',
+  inlineCode: 'code',
+  link: 'a',
+  linkReference: 'a',
+  listItem: 'li',
+  paragraph: 'p',
+  strong: 'strong',
+  table: 'table',
+  tableCell: 'td',
+  tableRow: 'tr',
+  thematicBreak: 'hr'
+}
+/**
+ * Get the JSX node names for a given AST node.
+ * @param {Nodes} node
+ * @returns {string[]}
+ */
+function getJsxNodeNameForMdast(node) {
+  if (node.type === 'code') {
+    return ['pre', 'code']
+  }
+  if (node.type === 'heading') {
+    return ['h' + node.depth]
+  }
+  if (node.type === 'list') {
+    return node.ordered ? ['ol'] : ['ul']
+  }
+  if (mdastElementNodeNameMap[node.type]) {
+    return [/** @type string */ (mdastElementNodeNameMap[node.type])]
+  }
+  // JSX Fragment
+  return ['']
+}
+
 /**
  * @param {string} mdx
  * @param {Root} ast
@@ -266,8 +288,6 @@ function getEmbeddedCodes(
   let jsxVariables = ''
   let markdown = ''
   let nextMarkdownSourceStart = 0
-  /** @type {Set<string>} */
-  const foundMdxNodeNames = new Set()
 
   const plugins = virtualCodePlugins.map((plugin) => plugin())
   /** @type {CodeMapping[]} */
@@ -760,8 +780,7 @@ function getEmbeddedCodes(
         }
 
         default: {
-          foundMdxNodeNames.add(node.type)
-          jsx += jsxIndent + `<_MDXNodes.${node.type}>`
+          jsx += jsxIndent + getJsxNodeNameForMdast(node).map((name) => `<${name}>`).join('')
           break
         }
       }
@@ -811,7 +830,7 @@ function getEmbeddedCodes(
         }
 
         default: {
-          jsx += jsxIndent + `</_MDXNodes.${node.type}>`
+          jsx += jsxIndent + getJsxNodeNameForMdast(node).map((name) => `</${name}>`).join('')
           break
         }
       }
@@ -842,12 +861,6 @@ function getEmbeddedCodes(
     esmMapping.lengths.unshift(0)
   }
 
-  esm += `/**
- * @typedef {object} _MDXNodeNames
-${[...foundMdxNodeNames].map((name) => ` * @property {unknown} ${name}`).join('\n')}
- */
-`
-
   updateMarkdownFromOffsets(mdx.length, mdx.length)
   esm += componentStart(hasAwait, programScope)
 
@@ -868,6 +881,7 @@ ${[...foundMdxNodeNames].map((name) => ` * @property {unknown} ${name}`).join('\
   if (jsxVariablesMapping.sourceOffsets.length > 0) {
     jsMappings.push(jsxVariablesMapping)
   }
+  console.log(esm)
 
   virtualCodes.unshift(
     {