Function omniCandidateTypeToBase

backend/plugin/parser/mongodb/completion.go:150–168

omniCandidateTypeToBase maps omni completion candidate types to Bytebase base types.

(t omnicompletion.CandidateType)

// omniCandidateTypeToBase maps omni completion candidate types to Bytebase base types.
func omniCandidateTypeToBase(t omnicompletion.CandidateType) base.CandidateType {
	switch t {
	case omnicompletion.CandidateCollection:
		return base.CandidateTypeTable
	case omnicompletion.CandidateKeyword, omnicompletion.CandidateShowTarget:
		return base.CandidateTypeKeyword
	case omnicompletion.CandidateMethod,
		omnicompletion.CandidateCursorMethod,
		omnicompletion.CandidateAggStage,
		omnicompletion.CandidateQueryOperator,
		omnicompletion.CandidateBSONHelper,
		omnicompletion.CandidateDbMethod,
		omnicompletion.CandidateRsMethod,
		omnicompletion.CandidateShMethod:
		return base.CandidateTypeFunction
	default:
		return base.CandidateTypeFunction
	}
}