Function optimizeComponentTree

packages/extension/mcp/tools/code/text/render.ts:204–250

(node: DevComponent | string, classProp: string)

}

function optimizeComponentTree(node: DevComponent | string, classProp: string) {
  if (typeof node === 'string') return

  if (node.children) {
    for (const child of node.children) {
      optimizeComponentTree(child, classProp)
    }
  }

  const UNWRAP_WHITELIST = new Set([
    'li',
    'p',
    'h1',
    'h2',
    'h3',
    'h4',
    'h5',
    'h6',
    'strong',
    'em',
    'u',
    's',
    'code',
    'a',
    'span'
  ])

  if (UNWRAP_WHITELIST.has(node.name) && node.children && node.children.length === 1) {
    const child = node.children[0]
    if (typeof child !== 'string' && child.name === 'span') {
      const childProps = child.props || {}
      const extraChildProps = Object.keys(childProps).filter((key) => key !== classProp)
      if (extraChildProps.length === 0) {
        const parentClass = String(node.props?.[classProp] || '')
        const childClass = String(childProps?.[classProp] || '')
        const mergedClass = joinClassNames([parentClass, childClass])

        if (mergedClass) {
          if (!node.props) node.props = {}
          node.props[classProp] = mergedClass
        }

        node.children = child.children
      }
    }
  }
}

function buildInlineTree(
  runs: TextRun[],