Function handleGetCode

packages/extension/mcp/tools/code/index.ts:128–293

(
  nodes: SceneNode[],
  preferredLang?: CodeLanguage,
  resolveTokens?: boolean,
  vectorMode: GetCodeParametersInput['vectorMode'] = 'smart',
  runtimeOptions: GetCodeRuntimeOptions = {}
)

type RenderStep = 'render' | 'stringify' | 'transform'

export async function handleGetCode(
  nodes: SceneNode[],
  preferredLang?: CodeLanguage,
  resolveTokens?: boolean,
  vectorMode: GetCodeParametersInput['vectorMode'] = 'smart',
  runtimeOptions: GetCodeRuntimeOptions = {}
): Promise<GetCodeResult> {
  const trace = createTrace()
  const { now, stamp } = trace
  const traceInfo: TraceInfo = { now, stamp }

  if (nodes.length !== 1) {
    throw new Error('Select exactly one node or provide a single root node id.')
  }

  const node = nodes[0]
  if (!node.visible) {
    throw new Error('The selected node is not visible.')
  }

  let t = now()
  const tree = buildVisibleTree(nodes)
  stamp('tree', t)
  const rootId = tree.rootIds[0]
  if (!rootId) {
    throw new Error('No renderable nodes found for the current selection.')
  }
  if (tree.stats.capped) {
    const depth = tree.stats.depthLimit ?? tree.stats.maxDepth
    logger.warn(`[get_code] Tree depth capped at ${depth}; output may be incomplete.`)
  }

  const config = currentCodegenConfig()
  const pluginCode = activePlugin.value?.code
  const codeBudget = runtimeOptions.unbounded ? resolveUnlimitedCodeBudget() : resolveCodeBudget()
  const budgetPreflight = preflightGetCodeBudget(tree, rootId, {
    maxResultBytes: codeBudget.maxResultBytes,
    pluginEnabled: !!pluginCode,
    unbounded: !!runtimeOptions.unbounded
  })
  const earlyShell = budgetPreflight.kind === 'shell'

  t = now()
  const variableCache = new Map<string, Variable | null>()
  const cache = createGetCodeCacheContext(variableCache, { metrics: true })
  const mappings = buildVariableMappings(nodes, variableCache, cache.readers, {
    traverseChildren: !earlyShell
  })
  stamp('vars', t)

  const { pluginComponents, pluginSkipped } =
    pluginCode && !earlyShell
      ? await collectPluginOutput(tree, config, pluginCode, preferredLang)
      : { pluginComponents: undefined, pluginSkipped: new Set<string>() }

  t = now()
  const plan = earlyShell