Function GetStatementRanges

backend/plugin/parser/mongodb/statement_ranges.go:20–43

GetStatementRanges returns the ranges of statements in the MongoDB shell script. Uses best-effort parsing so that valid statements are returned even when other statements in the input have syntax errors (e.g. user is mid-typing).

(_ context.Context, _ base.StatementRangeContext, statement string)

// Uses best-effort parsing so that valid statements are returned even when other
// statements in the input have syntax errors (e.g. user is mid-typing).
func GetStatementRanges(_ context.Context, _ base.StatementRangeContext, statement string) ([]base.Range, error) {
	stmts, _ := ParseMongoShellBestEffort(statement)

	// Build byte offset to LSP position mapping.
	positions := buildBytePositionMap(statement)

	var ranges []base.Range
	for _, stmt := range stmts {
		if stmt.Range == nil {
			continue
		}
		startPos := getPositionByByteOffset(int(stmt.Range.Start), positions)
		endPos := getPositionByByteOffset(int(stmt.Range.End), positions)
		if startPos == nil || endPos == nil {
			continue
		}
		ranges = append(ranges, base.Range{
			Start: *startPos,
			End:   *endPos,
		})
	}

	return ranges, nil
}

// bytePositionEntry stores the LSP position for a given byte offset.
type bytePositionEntry struct {