Function Completion

backend/plugin/parser/mongodb/completion.go:24–70

Completion is the entry point for MongoDB code completion.

(ctx context.Context, cCtx base.CompletionContext, statement string, caretLine int, caretOffset int)

// Completion is the entry point for MongoDB code completion.
func Completion(ctx context.Context, cCtx base.CompletionContext, statement string, caretLine int, caretOffset int) ([]base.Candidate, error) {
	// Build omni catalog from Bytebase metadata.
	cat := buildCatalog(ctx, cCtx)

	// Convert caret line:column to byte offset.
	byteOffset := caretToByteOffset(statement, caretLine, caretOffset)

	// Get omni completion candidates.
	candidates := omnicompletion.Complete(statement, byteOffset, cat)

	// Detect whether cursor is inside bracket notation (db[|).
	inBracket := isAfterOpenBracket(statement, byteOffset)

	// Convert to base.Candidate, formatting collection names for context.
	var result []base.Candidate
	for _, c := range candidates {
		text := c.Text
		if c.Type == omnicompletion.CandidateCollection {
			text = formatCollectionCandidate(c.Text, inBracket)
		}
		result = append(result, base.Candidate{
			Type: omniCandidateTypeToBase(c.Type),
			Text: text,
		})
	}

	// Sort and deduplicate.
	slices.SortFunc(result, func(a, b base.Candidate) int {
		if a.Type != b.Type {
			if a.Type < b.Type {
				return -1
			}
			return 1
		}
		if a.Text < b.Text {
			return -1
		}
		if a.Text > b.Text {
			return 1
		}
		return 0
	})

	return slices.CompactFunc(result, func(a, b base.Candidate) bool {
		return a.Type == b.Type && a.Text == b.Text
	}), nil
}

// buildCatalog creates an omni catalog from Bytebase completion context.
func buildCatalog(ctx context.Context, cCtx base.CompletionContext) *catalog.Catalog {