Function encodeTree

internal/api/encoder/encoder.go:416–620

(rootNode *ast.Node, sourceFile *ast.SourceFile)

}

func encodeTree(rootNode *ast.Node, sourceFile *ast.SourceFile) ([]byte, *NodeIndexTable, error) {
	var parentIndex, nodeCount, prevIndex uint32
	var extendedData []byte
	var structuredData []byte
	var strs *stringTable
	var positionMap *ast.PositionMap
	if rootNode.Kind == ast.KindSourceFile {
		strs = newStringTable(sourceFile.Text(), sourceFile.TextCount)
		positionMap = sourceFile.GetPositionMap()
	} else {
		strs = newStringTable("", 0)
		if sourceFile != nil {
			positionMap = sourceFile.GetPositionMap()
		}
	}
	if positionMap == nil {
		positionMap = ast.ComputePositionMap("")
	}
	utf16 := func(pos int) uint32 {
		return uint32(positionMap.UTF8ToUTF16(pos))
	}
	var initialNodeCount int
	if sourceFile != nil {
		initialNodeCount = sourceFile.NodeCount
	}
	nodes := make([]byte, 0, (initialNodeCount+1)*NodeSize)

	// Build node index table for O(1) handle resolution.
	// Index 0 is a nil sentinel; real nodes start at index 1.
	nodeTable := make([]*ast.Node, 1, initialNodeCount+1) // index 0 = nil sentinel

	// Build a small map of nodes we need to track indices for (imports + moduleAugmentations).
	// Values start at 0 and are filled in during the walk.
	var nodeIndexMap map[*ast.Node]uint32
	var sfExtendedDataOffset int // byte offset in extendedData where SourceFile fields start
	if rootNode.Kind == ast.KindSourceFile {
		sf := rootNode.AsSourceFile()
		total := len(sf.Imports()) + len(sf.ModuleAugmentations)
		if sf.ExternalModuleIndicator != nil && sf.ExternalModuleIndicator != rootNode {
			total++
		}
		if total > 0 {
			nodeIndexMap = make(map[*ast.Node]uint32, total)
			for _, imp := range sf.Imports() {
				nodeIndexMap[imp.AsNode()] = 0
			}
			for _, aug := range sf.ModuleAugmentations {
				nodeIndexMap[aug.AsNode()] = 0
			}
			if sf.ExternalModuleIndicator != nil && sf.ExternalModuleIndicator != rootNode {
				nodeIndexMap[sf.ExternalModuleIndicator] = 0
			}
		}
	}

	visitor := &ast.NodeVisitor{
		Hooks: ast.NodeVisitorHooks{
			VisitNodes: func(nodeList *ast.NodeList, visitor *ast.NodeVisitor) *ast.NodeList {