Function buildBytePositionMap

backend/plugin/parser/mongodb/statement_ranges.go:52–80

buildBytePositionMap creates a mapping from byte offset to LSP position.

(statement string)

// buildBytePositionMap creates a mapping from byte offset to LSP position.
func buildBytePositionMap(statement string) map[int]bytePositionEntry {
	positions := make(map[int]bytePositionEntry, len(statement)+1)

	var line uint32
	var character uint32
	i := 0

	for i < len(statement) {
		positions[i] = bytePositionEntry{line: line, character: character}

		r, size := utf8.DecodeRuneInString(statement[i:])
		if r == '\n' {
			line++
			character = 0
		} else {
			// Count UTF-16 code units.
			if r > 0xFFFF {
				character += 2 // surrogate pair
			} else {
				character++
			}
		}
		i += size
	}
	// Position after the last character.
	positions[i] = bytePositionEntry{line: line, character: character}

	return positions
}

// getPositionByByteOffset converts a byte offset to an LSP Position.
func getPositionByByteOffset(byteOffset int, positions map[int]bytePositionEntry) *lsp.Position {