Function buildVisibleTree

packages/extension/mcp/tools/code/tree.ts:24–168

(roots: SceneNode[])

  | { type: string; value: unknown }

export function buildVisibleTree(roots: SceneNode[]): VisibleTree {
  const depthLimit = suggestDepthLimit(roots)
  const stats: TreeStats = {
    totalNodes: 0,
    maxDepth: 0,
    depthLimit,
    capped: false,
    cappedNodeIds: []
  }

  const collectionCache = new Map<string, { name: string; modes: Map<string, string> } | null>()
  const collectionIdByName = new Map<string, string>()
  const warnedDuplicateCollections = new Set<string>()

  const nodes = new Map<string, NodeSnapshot>()
  const order: string[] = []
  const rootIds: string[] = []

  const getCollectionInfo = (id: string) => {
    if (collectionCache.has(id)) return collectionCache.get(id)
    try {
      const collection = figma.variables.getVariableCollectionById(id)
      if (!collection) {
        collectionCache.set(id, null)
        return null
      }

      const name = collection.name || id
      if (name) {
        const existing = collectionIdByName.get(name)
        if (!existing) {
          collectionIdByName.set(name, collection.id)
        } else if (existing !== collection.id && !warnedDuplicateCollections.has(name)) {
          warnedDuplicateCollections.add(name)
          logger.warn(`Duplicate variable collection name "${name}" detected.`)
        }
      }

      const modes = new Map<string, string>()
      if (Array.isArray(collection.modes)) {
        collection.modes.forEach((mode) => {
          modes.set(mode.modeId, mode.name || mode.modeId)
        })
      }

      const info = { name, modes }
      collectionCache.set(id, info)
      return info
    } catch {
      collectionCache.set(id, null)
      return null
    }
  }

  const resolveVariableModeHint = (node: SceneNode): string | undefined => {
    if (!('explicitVariableModes' in node)) return undefined
    const explicit = (node as { explicitVariableModes?: Record<string, string> })
      .explicitVariableModes