@mongosh/shell-api 3.13.0 → 3.16.0

package/lib/api-raw.d.ts

@@ -1,6 +1,7 @@
 /// <reference types="node" />
 
 import type { Agent } from 'https';
+import type { AutocompletionContext } from '@mongodb-js/mongodb-ts-autocomplete';
 import { Binary } from 'bson';
 import { BSONRegExp } from 'bson';
 import { BSONSymbol } from 'bson';
@@ -1126,8 +1127,8 @@ declare class Bulk extends ShellApiWithMongoClass {
     [asPrintable](): any;
     private _emitBulkApiCall;
     execute(writeConcern?: WriteConcern): Promise<BulkWriteResult_2>;
-    find(query: Document_2): BulkFindOp;
-    insert(document: Document_2): Bulk;
+    find(query: MQLQuery): BulkFindOp;
+    insert(document: MQLDocument): Bulk;
     toJSON(): Record<'nInsertOps' | 'nUpdateOps' | 'nRemoveOps' | 'nBatches', number>;
     toString(): string;
     getOperations(): Pick<Batch, 'originalZeroIndex' | 'batchType' | 'operations'>[];
@@ -2848,36 +2849,37 @@ declare class Collection<M extends GenericServerSideSchema = GenericServerSideSc
     _mongo: Mongo<M>;
     _database: DatabaseWithSchema<M, D>;
     _name: N;
+    _cachedSampleDocs: Document_2[];
     constructor(mongo: Mongo<M>, database: DatabaseWithSchema<M, D> | Database<M, D>, name: N);
     [namespaceInfo](): Namespace;
     [asPrintable](): string;
     private _emitCollectionApiCall;
-    aggregate(pipeline: Document_2[], options: AggregateOptions & {
+    aggregate(pipeline: MQLPipeline, options: AggregateOptions & {
         explain: ExplainVerbosityLike;
     }): Promise<Document_2>;
-    aggregate(pipeline: Document_2[], options?: AggregateOptions): Promise<AggregationCursor_2>;
-    aggregate(...stages: Document_2[]): Promise<AggregationCursor_2>;
+    aggregate(pipeline: MQLPipeline, options?: AggregateOptions): Promise<AggregationCursor_2>;
+    aggregate(...stages: MQLPipeline): Promise<AggregationCursor_2>;
     bulkWrite(operations: AnyBulkWriteOperation[], options?: BulkWriteOptions): Promise<BulkWriteResult_2>;
     count(query?: {}, options?: CountOptions): Promise<number>;
-    countDocuments(query?: Document_2, options?: CountDocumentsOptions): Promise<number>;
+    countDocuments(query?: MQLQuery, options?: CountDocumentsOptions): Promise<number>;
     deleteMany(filter: Document_2, options?: DeleteOptions): Promise<DeleteResult_2 | Document_2>;
     deleteOne(filter: Document_2, options?: DeleteOptions): Promise<DeleteResult_2 | Document_2>;
     distinct(field: string): Promise<Document_2>;
-    distinct(field: string, query: Document_2): Promise<Document_2>;
-    distinct(field: string, query: Document_2, options: DistinctOptions): Promise<Document_2>;
+    distinct(field: string, query: MQLQuery): Promise<Document_2>;
+    distinct(field: string, query: MQLQuery, options: DistinctOptions): Promise<Document_2>;
     estimatedDocumentCount(options?: EstimatedDocumentCountOptions): Promise<number>;
-    find(query?: Document_2, projection?: Document_2, options?: FindOptions): Promise<Cursor>;
+    find(query?: MQLQuery, projection?: Document_2, options?: FindOptions): Promise<Cursor>;
     findAndModify(options: FindAndModifyMethodShellOptions): Promise<Document_2 | null>;
-    findOne(query?: Document_2, projection?: Document_2, options?: FindOptions): Promise<C['schema'] | null>;
+    findOne(query?: MQLQuery, projection?: Document_2, options?: FindOptions): Promise<MQLDocument | null>;
     renameCollection(newName: string, dropTarget?: boolean): Promise<Document_2>;
     findOneAndDelete(filter: Document_2, options?: FindOneAndDeleteOptions): Promise<Document_2 | null>;
     findOneAndReplace(filter: Document_2, replacement: Document_2, options?: FindAndModifyShellOptions<FindOneAndReplaceOptions>): Promise<Document_2>;
     findOneAndUpdate(filter: Document_2, update: Document_2 | Document_2[], options?: FindAndModifyShellOptions<FindOneAndUpdateOptions>): Promise<Document_2>;
-    insert(docs: Document_2 | Document_2[], options?: BulkWriteOptions): Promise<InsertManyResult_2>;
-    insertMany(docs: Document_2[], options?: BulkWriteOptions): Promise<InsertManyResult_2>;
-    insertOne(doc: Document_2, options?: InsertOneOptions): Promise<InsertOneResult_2>;
+    insert(docs: MQLDocument | MQLDocument[], options?: BulkWriteOptions): Promise<InsertManyResult_2>;
+    insertMany(docs: MQLDocument[], options?: BulkWriteOptions): Promise<InsertManyResult_2>;
+    insertOne(doc: MQLDocument, options?: InsertOneOptions): Promise<InsertOneResult_2>;
     isCapped(): Promise<boolean>;
-    remove(query: Document_2, options?: boolean | RemoveShellOptions): Promise<DeleteResult_2 | Document_2>;
+    remove(query: MQLQuery, options?: boolean | RemoveShellOptions): Promise<DeleteResult_2 | Document_2>;
     replaceOne(filter: Document_2, replacement: Document_2, options?: ReplaceOptions): Promise<UpdateResult_2>;
     update(filter: Document_2, update: Document_2, options?: UpdateOptions & {
         multi?: boolean;
@@ -2923,7 +2925,11 @@ declare class Collection<M extends GenericServerSideSchema = GenericServerSideSc
     getShardVersion(): Promise<Document_2>;
     _getShardedCollectionInfo(config: DatabaseWithSchema<M, D>, collStats: Document_2[]): Promise<Document_2>;
     getShardDistribution(): Promise<CommandResult<GetShardDistributionResult>>;
-    watch(pipeline?: Document_2[] | ChangeStreamOptions, options?: ChangeStreamOptions): Promise<ChangeStreamCursor>;
+    getShardLocation(): Promise<{
+        shards: string[];
+        sharded: boolean;
+    }>;
+    watch(pipeline?: MQLPipeline | ChangeStreamOptions, options?: ChangeStreamOptions): Promise<ChangeStreamCursor>;
     hideIndex(index: string | Document_2): Promise<Document_2>;
     unhideIndex(index: string | Document_2): Promise<Document_2>;
     analyzeShardKey(key: Document_2, options?: Document_2): Promise<Document_2>;
@@ -2937,6 +2943,8 @@ declare class Collection<M extends GenericServerSideSchema = GenericServerSideSc
     createSearchIndexes(specs: SearchIndexDescription[]): Promise<string[]>;
     dropSearchIndex(indexName: string): Promise<void>;
     updateSearchIndex(indexName: string, definition: Document_2): Promise<void>;
+    _getSampleDocs(): Promise<Document_2[]>;
+    _getSampleDocsForCompletion(): Promise<Document_2[]>;
 }
 
 /**
@@ -4238,30 +4246,30 @@ declare class Cursor extends AggregateOrFindCursor<ServiceProviderFindCursor> {
     constructor(mongo: Mongo, cursor: ServiceProviderFindCursor);
     toJSON(): void;
     private _addFlag;
-    addOption(optionFlagNumber: number): Cursor;
-    allowDiskUse(allow?: boolean): Cursor;
-    allowPartialResults(): Cursor;
-    collation(spec: CollationOptions): Cursor;
-    comment(cmt: string): Cursor;
+    addOption(optionFlagNumber: number): this;
+    allowDiskUse(allow?: boolean): this;
+    allowPartialResults(): this;
+    collation(spec: CollationOptions): this;
+    comment(cmt: string): this;
     count(): Promise<number>;
     hasNext(): Promise<boolean>;
-    hint(index: string): Cursor;
-    limit(value: number): Cursor;
-    max(indexBounds: Document_2): Cursor;
-    maxAwaitTimeMS(value: number): Cursor;
-    min(indexBounds: Document_2): Cursor;
+    hint(index: string): this;
+    limit(value: number): this;
+    max(indexBounds: Document_2): this;
+    maxAwaitTimeMS(value: number): this;
+    min(indexBounds: Document_2): this;
     next(): Promise<Document_2 | null>;
-    noCursorTimeout(): Cursor;
-    oplogReplay(): Cursor;
-    readPref(mode: ReadPreferenceLike, tagSet?: TagSet[], hedgeOptions?: HedgeOptions): Cursor;
-    returnKey(enabled: boolean): Cursor;
+    noCursorTimeout(): this;
+    oplogReplay(): this;
+    readPref(mode: ReadPreferenceLike, tagSet?: TagSet[], hedgeOptions?: HedgeOptions): this;
+    returnKey(enabled: boolean): this;
     size(): Promise<number>;
     tailable(opts?: {
         awaitData: boolean;
-    }): Cursor;
+    }): this;
     maxScan(): void;
-    showRecordId(): Cursor;
-    readConcern(level: ReadConcernLevel): Cursor;
+    showRecordId(): this;
+    readConcern(level: ReadConcernLevel): this;
 }
 
 /** @public */
@@ -4351,11 +4359,11 @@ declare class Database<M extends GenericServerSideSchema = GenericServerSideSche
     getCollectionInfos(filter?: Document_2, options?: ListCollectionsOptions): Promise<Document_2[]>;
     runCommand(cmd: string | Document_2, options?: RunCommandOptions): Promise<Document_2>;
     adminCommand(cmd: string | Document_2): Promise<Document_2>;
-    aggregate(pipeline: Document_2[], options: AggregateOptions & {
+    aggregate(pipeline: MQLPipeline, options: AggregateOptions & {
         explain: ExplainVerbosityLike;
     }): Promise<Document_2>;
-    aggregate(pipeline: Document_2[], options?: AggregateOptions): Promise<AggregationCursor_2>;
-    aggregate(...stages: Document_2[]): Promise<AggregationCursor_2>;
+    aggregate(pipeline: MQLPipeline, options?: AggregateOptions): Promise<AggregationCursor_2>;
+    aggregate(...stages: MQLPipeline): Promise<AggregationCursor_2>;
     getSiblingDB<K extends StringKey<M>>(db: K): DatabaseWithSchema<M, M[K]>;
     getCollection<K extends StringKey<D>>(coll: K): CollectionWithSchema<M, D, D[K], K>;
     dropDatabase(writeConcern?: WriteConcern): Promise<Document_2>;
@@ -4379,7 +4387,7 @@ declare class Database<M extends GenericServerSideSchema = GenericServerSideSche
         collection: Collection;
         encryptedFields: Document_2;
     }>;
-    createView(name: string, source: string, pipeline: Document_2[], options?: CreateCollectionOptions): Promise<{
+    createView(name: string, source: string, pipeline: MQLPipeline, options?: CreateCollectionOptions): Promise<{
         ok: number;
     }>;
     createRole(role: Document_2, writeConcern?: WriteConcern): Promise<Document_2>;
@@ -4426,7 +4434,7 @@ declare class Database<M extends GenericServerSideSchema = GenericServerSideSche
     printReplicationInfo(): Promise<CommandResult>;
     printSlaveReplicationInfo(): never;
     setSecondaryOk(): Promise<void>;
-    watch(pipeline?: Document_2[] | ChangeStreamOptions, options?: ChangeStreamOptions): Promise<ChangeStreamCursor>;
+    watch(pipeline?: MQLPipeline | ChangeStreamOptions, options?: ChangeStreamOptions): Promise<ChangeStreamCursor>;
     sql(sqlString: string, options?: AggregateOptions): Promise<AggregationCursor_2>;
     checkMetadataConsistency(options?: CheckMetadataConsistencyOptions): Promise<RunCommandCursor_2>;
 }
@@ -4981,18 +4989,18 @@ declare class Explainable extends ShellApiWithMongoClass {
     getCollection(): CollectionWithSchema;
     getVerbosity(): ExplainVerbosityLike;
     setVerbosity(verbosity: ExplainVerbosityLike): void;
-    find(query?: Document_2, projection?: Document_2, options?: FindOptions): Promise<ExplainableCursor_2>;
-    aggregate(pipeline: Document_2[], options: Document_2): Promise<Document_2>;
-    aggregate(...stages: Document_2[]): Promise<Document_2>;
+    find(query?: MQLQuery, projection?: Document_2, options?: FindOptions): Promise<ExplainableCursor_2>;
+    aggregate(pipeline: MQLPipeline, options: Document_2): Promise<Document_2>;
+    aggregate(...stages: MQLPipeline): Promise<Document_2>;
     count(query?: {}, options?: CountOptions): Promise<Document_2>;
     distinct(field: string): Promise<Document_2>;
-    distinct(field: string, query: Document_2): Promise<Document_2>;
-    distinct(field: string, query: Document_2, options: DistinctOptions): Promise<Document_2>;
+    distinct(field: string, query: MQLQuery): Promise<Document_2>;
+    distinct(field: string, query: MQLQuery, options: DistinctOptions): Promise<Document_2>;
     findAndModify(options: FindAndModifyMethodShellOptions): Promise<Document_2 | null>;
-    findOneAndDelete(filter: Document_2, options?: FindOneAndDeleteOptions): Promise<Document_2 | null>;
-    findOneAndReplace(filter: Document_2, replacement: Document_2, options?: FindAndModifyShellOptions<FindOneAndReplaceOptions>): Promise<Document_2>;
-    findOneAndUpdate(filter: Document_2, update: Document_2, options?: FindAndModifyShellOptions<FindOneAndUpdateOptions>): Promise<Document_2>;
-    remove(query: Document_2, options?: boolean | RemoveShellOptions): Promise<Document_2>;
+    findOneAndDelete(filter: MQLQuery, options?: FindOneAndDeleteOptions): Promise<Document_2 | null>;
+    findOneAndReplace(filter: MQLQuery, replacement: MQLDocument, options?: FindAndModifyShellOptions<FindOneAndReplaceOptions>): Promise<Document_2>;
+    findOneAndUpdate(filter: MQLQuery, update: MQLDocument, options?: FindAndModifyShellOptions<FindOneAndUpdateOptions>): Promise<Document_2>;
+    remove(query: MQLQuery, options?: boolean | RemoveShellOptions): Promise<Document_2>;
     update(filter: Document_2, update: Document_2, options?: UpdateOptions): Promise<Document_2>;
     mapReduce(map: Function | string, reduce: Function | string, optionsOrOutString: MapReduceShellOptions): Promise<Document_2>;
 }
@@ -5035,6 +5043,7 @@ declare class ExplainableCursor_2 extends Cursor {
     _explained: any;
     constructor(mongo: Mongo, cursor: Cursor, verbosity: ExplainVerbosityLike);
     [asPrintable](): Promise<any>;
+    finish(): Promise<any>;
 }
 
 /** @public */
@@ -5159,7 +5168,7 @@ declare interface FilterOperators<TValue> extends NonObjectIdLikeDocument {
 }
 
 declare type FindAndModifyMethodShellOptions = {
-    query: Document_2;
+    query: MQLQuery;
     sort?: (FindOneAndDeleteOptions | FindOneAndReplaceOptions | FindOneAndUpdateOptions)['sort'];
     update?: Document_2 | Document_2[];
     remove?: boolean;
@@ -6060,6 +6069,7 @@ declare interface ModifyResult<TSchema = Document_2> {
 export declare class Mongo<M extends GenericServerSideSchema = GenericServerSideSchema> extends ShellApiClass {
     private __serviceProvider;
     readonly _databases: Record<StringKey<M>, DatabaseWithSchema<M>>;
+    private _connectionId;
     _instanceState: ShellInstanceState;
     _connectionInfo: ConnectionInfo;
     private _explicitEncryptionOnly;
@@ -6081,6 +6091,7 @@ export declare class Mongo<M extends GenericServerSideSchema = GenericServerSide
     _getDb<K extends StringKey<M>>(name: K): DatabaseWithSchema<M, M[K]>;
     getDB<K extends StringKey<M>>(db: K): DatabaseWithSchema<M, M[K]>;
     getCollection<KD extends StringKey<M>, KC extends StringKey<M[KD]>>(name: `${KD}.${KC}`): CollectionWithSchema<M, M[KD], M[KD][KC]>;
+    _getConnectionId(): string;
     getURI(): string;
     use(db: StringKey<M>): string;
     _listDatabases(opts?: ListDatabasesOptions): Promise<{
@@ -6122,7 +6133,7 @@ export declare class Mongo<M extends GenericServerSideSchema = GenericServerSide
     isCausalConsistency(): void;
     setSlaveOk(): void;
     setSecondaryOk(): Promise<void>;
-    watch(pipeline?: Document_2[] | ChangeStreamOptions, options?: ChangeStreamOptions): Promise<ChangeStreamCursor>;
+    watch(pipeline?: MQLPipeline | ChangeStreamOptions, options?: ChangeStreamOptions): Promise<ChangeStreamCursor>;
     getClientEncryption(): ClientEncryption_2;
     getKeyVault(): Promise<KeyVault>;
     convertShardKeyToHashed(value: any): Promise<unknown>;
@@ -6609,6 +6620,7 @@ declare interface MongoDBOIDCLogEventsMap {
     }) => void;
     'mongodb-oidc-plugin:state-updated': (event: {
         updateId: number;
+        tokenSetId: string;
         timerDuration: number | undefined;
     }) => void;
     'mongodb-oidc-plugin:local-redirect-accessed': (event: {
@@ -6662,10 +6674,14 @@ declare interface MongoDBOIDCLogEventsMap {
     'mongodb-oidc-plugin:open-browser-complete': () => void;
     'mongodb-oidc-plugin:notify-device-flow': () => void;
     'mongodb-oidc-plugin:auth-attempt-started': (event: {
+        authStateId: string;
         flow: string;
     }) => void;
-    'mongodb-oidc-plugin:auth-attempt-succeeded': () => void;
+    'mongodb-oidc-plugin:auth-attempt-succeeded': (event: {
+        authStateId: string;
+    }) => void;
     'mongodb-oidc-plugin:auth-attempt-failed': (event: {
+        authStateId: string;
         error: string;
     }) => void;
     'mongodb-oidc-plugin:refresh-skipped': (event: {
@@ -6687,12 +6703,15 @@ declare interface MongoDBOIDCLogEventsMap {
         refreshToken: string | null;
     }) => void;
     'mongodb-oidc-plugin:skip-auth-attempt': (event: {
+        authStateId: string;
         reason: string;
     }) => void;
     'mongodb-oidc-plugin:auth-failed': (event: {
+        authStateId: string;
         error: string;
     }) => void;
     'mongodb-oidc-plugin:auth-succeeded': (event: {
+        authStateId: string;
         tokenType: string | null;
         refreshToken: string | null;
         expiresAt: string | null;
@@ -6702,6 +6721,30 @@ declare interface MongoDBOIDCLogEventsMap {
             idToken: string | undefined;
             refreshToken: string | undefined;
         };
+        forceRefreshOrReauth: boolean;
+        willRetryWithForceRefreshOrReauth: boolean;
+        tokenSetId: string;
+    }) => void;
+    'mongodb-oidc-plugin:request-token-started': (event: {
+        authStateId: string;
+        isCurrentAuthAttemptSet: boolean;
+        tokenSetId: string | undefined;
+        username: string | undefined;
+        issuer: string;
+        clientId: string;
+        requestScopes: string[] | undefined;
+    }) => void;
+    'mongodb-oidc-plugin:request-token-ended': (event: {
+        authStateId: string;
+        isCurrentAuthAttemptSet: boolean;
+        tokenSetId: string | undefined;
+        username: string | undefined;
+        issuer: string;
+        clientId: string;
+        requestScopes: string[] | undefined;
+    }) => void;
+    'mongodb-oidc-plugin:discarding-token-set': (event: {
+        tokenSetId: string;
     }) => void;
     'mongodb-oidc-plugin:destroyed': () => void;
     'mongodb-oidc-plugin:missing-id-token': () => void;
@@ -7062,7 +7105,7 @@ declare interface MongoshBusEventsMap extends ConnectEventMap {
     'mongosh:crypt-library-load-found': (ev: CryptLibraryFoundEvent) => void;
     'mongosh:closed': () => void;
     'mongosh:eval-complete': () => void;
-    'mongosh:autocompletion-complete': () => void;
+    'mongosh:autocompletion-complete': (replResults: string[], mongoshResults: string[]) => void;
     'mongosh:interrupt-complete': () => void;
     'mongosh-snippets:loaded': (ev: SnippetsLoadedEvent) => void;
     'mongosh-snippets:npm-lookup': (ev: SnippetsNpmLookupEvent) => void;
@@ -7087,6 +7130,12 @@ declare interface MongoshBusEventsMap extends ConnectEventMap {
     'mongosh:log-initialized': () => void;
 }
 
+declare type MQLDocument = Document_2;
+
+declare type MQLPipeline = Document_2[];
+
+declare type MQLQuery = Document_2;
+
 declare interface Namespace {
     db: string;
     collection: string;
@@ -7276,8 +7325,8 @@ declare class PlanCache extends ShellApiWithMongoClass {
     get _mongo(): Mongo;
     [asPrintable](): string;
     clear(): Promise<Document_2>;
-    clearPlansByQuery(query: Document_2, projection?: Document_2, sort?: Document_2): Promise<Document_2>;
-    list(pipeline?: Document_2[]): Promise<Document_2[]>;
+    clearPlansByQuery(query: MQLQuery, projection?: Document_2, sort?: Document_2): Promise<Document_2>;
+    list(pipeline?: MQLPipeline): Promise<Document_2[]>;
     listQueryShapes(): never;
     getPlansByQuery(): never;
 }
@@ -8205,9 +8254,9 @@ export declare class Shard<M extends GenericServerSideSchema = GenericServerSide
     removeShardTag(shard: string, tag: string): Promise<Document_2>;
     enableAutoSplit(): Promise<UpdateResult_2>;
     disableAutoSplit(): Promise<UpdateResult_2>;
-    splitAt(ns: string, query: Document_2): Promise<Document_2>;
-    splitFind(ns: string, query: Document_2): Promise<Document_2>;
-    moveChunk(ns: string, query: Document_2, destination: string): Promise<Document_2>;
+    splitAt(ns: string, query: MQLQuery): Promise<Document_2>;
+    splitFind(ns: string, query: MQLQuery): Promise<Document_2>;
+    moveChunk(ns: string, query: MQLQuery, destination: string): Promise<Document_2>;
     moveRange(ns: string, toShard: string, min?: Document_2, max?: Document_2): Promise<Document_2>;
     balancerCollectionStatus(ns: string): Promise<Document_2>;
     enableBalancing(ns: string): Promise<UpdateResult_2>;
@@ -8435,6 +8484,8 @@ declare class ShellInstanceState {
     emitApiCallWithArgs(event: ApiEventWithArguments): void;
     emitApiCall(event: Omit<ApiEvent, 'callDepth'>): void;
     setEvaluationListener(listener: EvaluationListener): void;
+    getMongoByConnectionId(connectionId: string): Mongo;
+    getAutocompletionContext(): AutocompletionContext;
     getAutocompleteParameters(): AutocompleteParameters;
     apiVersionInfo(): Required<ServerApi> | undefined;
     onInterruptExecution(): Promise<boolean>;
@@ -8580,7 +8631,7 @@ declare class StreamProcessor extends ShellApiWithMongoClass {
     _drop(options?: Document_2): Promise<Document_2>;
     stats(options?: Document_2): Promise<Document_2>;
     modify(options: Document_2): Promise<Document_2>;
-    modify(pipeline: Document_2[], options?: Document_2): Promise<Document_2>;
+    modify(pipeline: MQLPipeline, options?: Document_2): Promise<Document_2>;
     sample(options?: Document_2): Promise<Document_2 | undefined>;
     _sampleFrom(cursorId: number): Promise<Document_2 | undefined>;
 }
@@ -8592,8 +8643,8 @@ export declare class Streams<M extends GenericServerSideSchema = GenericServerSi
     get _mongo(): Mongo<M>;
     [asPrintable](): string;
     getProcessor(name: string): StreamProcessor;
-    process(pipeline: Document_2[], options?: Document_2): Promise<void | Document_2>;
-    createStreamProcessor(name: string, pipeline: Document_2[], options?: Document_2): Promise<Document_2 | StreamProcessor>;
+    process(pipeline: MQLPipeline, options?: Document_2): Promise<void | Document_2>;
+    createStreamProcessor(name: string, pipeline: MQLPipeline, options?: Document_2): Promise<Document_2 | StreamProcessor>;
     listStreamProcessors(filter: Document_2): Promise<any>;
     listConnections(filter: Document_2): Promise<any>;
     _runStreamCommand(cmd: Document_2, options?: Document_2): Promise<Document_2>;

package/lib/bulk.d.ts

@@ -4,6 +4,7 @@ import type { Batch, Document, WriteConcern, OrderedBulkOperation, UnorderedBulk
 import { asPrintable } from './enums';
 import { BulkWriteResult } from './result';
 import type { CollectionWithSchema } from './collection';
+import type { MQLDocument, MQLQuery } from './mql-types';
 export declare class BulkFindOp extends ShellApiWithMongoClass {
     _serviceProviderBulkFindOp: FindOperators;
     _parentBulk: Bulk;
@@ -33,8 +34,8 @@ export default class Bulk extends ShellApiWithMongoClass {
     [asPrintable](): any;
     private _emitBulkApiCall;
     execute(writeConcern?: WriteConcern): Promise<BulkWriteResult>;
-    find(query: Document): BulkFindOp;
-    insert(document: Document): Bulk;
+    find(query: MQLQuery): BulkFindOp;
+    insert(document: MQLDocument): Bulk;
     toJSON(): Record<'nInsertOps' | 'nUpdateOps' | 'nRemoveOps' | 'nBatches', number>;
     toString(): string;
     getOperations(): Pick<Batch, 'originalZeroIndex' | 'batchType' | 'operations'>[];

package/lib/bulk.js.map

@@ -1 +1 @@
-{"version":3,"file":"bulk.js","sourceRoot":"","sources":["../src/bulk.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,6CAOsB;AAEtB,4CAAyE;AAUzE,mCAAsC;AACtC,uCAAgE;AAChE,qCAA2C;IAI9B,UAAU;4BADtB,iCAAoB;;;;sBACW,mCAAsB;;;;;;;;;;;;;uCAA9B,SAAQ,WAAsB;QAGpD,YAAY,SAAwB,EAAE,UAAgB;YACpD,KAAK,EAAE,CAAC;YAHV,+BAA0B,GADf,mDAAU,CACqB;YAIxC,IAAI,CAAC,0BAA0B,GAAG,SAAS,CAAC;YAC5C,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAChC,CAAC;QAED,IAAI,MAAM;YACR,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;QACjC,CAAC;QAED,CAAC,mBAAW,CAAC;YACX,OAAO,YAAY,CAAC;QACtB,CAAC;QAID,SAAS,CAAC,IAAsB;YAC9B,IAAI,CAAC,0BAA0B,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAChD,OAAO,IAAI,CAAC;QACd,CAAC;QAID,YAAY,CAAC,OAAmB;YAC9B,IAAI,CAAC,0BAA0B,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACtD,OAAO,IAAI,CAAC;QACd,CAAC;QAID,IAAI,CAAC,OAAiB;YACpB,IAAA,+BAAqB,EAAC,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,iBAAiB,CAAC,CAAC;YAC5D,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC9C,OAAO,IAAI,CAAC;QACd,CAAC;QAID,MAAM;YACJ,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAC3C,IAAI,CAAC,0BAA0B,CAAC,MAAM,EAAE,CAAC;YACzC,OAAO,IAAI,CAAC,WAAW,CAAC;QAC1B,CAAC;QAID,SAAS;YACP,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAC3C,IAAI,CAAC,0BAA0B,CAAC,SAAS,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC,WAAW,CAAC;QAC1B,CAAC;QAKD,MAAM;YACJ,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC;QACvB,CAAC;QAKD,SAAS;YACP,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC;QAC1B,CAAC;QAID,UAAU,CAAC,WAAqB;YAC9B,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAC3C,IAAA,+BAAqB,EAAC,CAAC,WAAW,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,wBAAwB,CAAC,CAAC;YACvE,MAAM,EAAE,GAAG,IAAA,sBAAY,EAAC,WAAW,CAAC,CAAC;YACrC,IAAI,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAC/C,OAAO,IAAI,CAAC,WAAW,CAAC;QAC1B,CAAC;QAID,SAAS,CAAC,MAA6B;YACrC,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAC3C,IAAA,+BAAqB,EAAC,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,mBAAmB,CAAC,CAAC;YAC7D,MAAM,EAAE,GAAG,IAAA,sBAAY,EAAC,MAAM,CAAC,CAAC;YAChC,IAAI,CAAC,0BAA0B,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAC9C,OAAO,IAAI,CAAC,WAAW,CAAC;QAC1B,CAAC;QAGD,MAAM,CAAC,MAA6B;YAClC,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAC3C,IAAA,+BAAqB,EAAC,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,mBAAmB,CAAC,CAAC;YAC7D,MAAM,EAAE,GAAG,IAAA,sBAAY,EAAC,MAAM,CAAC,CAAC;YAChC,IAAI,CAAC,0BAA0B,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC3C,OAAO,IAAI,CAAC,WAAW,CAAC;QAC1B,CAAC;QAGD,MAAM;YACJ,IAAI,CAAC,0BAA0B,CAAC,MAAM,EAAE,CAAC;YACzC,OAAO,IAAI,CAAC;QACd,CAAC;;;;;;iCArFA,IAAA,uBAAU,EAAC,YAAY,CAAC,EACxB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;oCAMhB,IAAA,uBAAU,EAAC,YAAY,CAAC,EACxB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;4BAMhB,IAAA,uBAAU,EAAC,YAAY,CAAC,EACxB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;8BAOhB,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;iCAOhB,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;8BAOhB,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC,EAChB,uBAAU;iCAKV,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC,EAChB,uBAAU;kCAKV,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;iCAShB,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;8BAShB,IAAA,uBAAU,EAAC,MAAM,CAAC;8BASlB,IAAA,uBAAU,EAAC,MAAM,CAAC;QA/EnB,oLAAA,SAAS,6DAGR;QAID,6LAAA,YAAY,6DAGX;QAID,qKAAA,IAAI,6DAIH;QAID,2KAAA,MAAM,6DAIL;QAID,oLAAA,SAAS,6DAIR;QAKD,2KAAA,MAAM,6DAEL;QAKD,oLAAA,SAAS,6DAER;QAID,uLAAA,UAAU,6DAMT;QAID,oLAAA,SAAS,6DAMR;QAGD,2KAAA,MAAM,6DAML;QAGD,2KAAA,MAAM,6DAGL;QAtGH,6KAuGC;;;QAvGY,uDAAU;;;;AAAV,gCAAU;;4BAyGtB,iCAAoB;;;;sBACa,mCAAsB;;;;;iCAA9B,SAAQ,WAAsB;QAQtD,YACE,UAAgC,EAChC,SAAwD,EACxD,OAAO,GAAG,KAAK;YAEf,KAAK,EAAE,CAAC;YAZV,WAAM,GADa,mDAAI,CACT;YAaZ,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;YAC9B,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;YAChC,IAAI,CAAC,sBAAsB,GAAG,SAAS,CAAC;YACxC,IAAI,CAAC,YAAY,GAAG;gBAClB,UAAU,EAAE,CAAC;gBACb,UAAU,EAAE,CAAC;gBACb,UAAU,EAAE,CAAC;aACd,CAAC;YACF,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;YACvB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QAC1B,CAAC;QAKD,CAAC,mBAAW,CAAC;YACX,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC;QACvB,CAAC;QASO,gBAAgB,CACtB,UAAkB,EAClB,kBAA4B,EAAE;YAE9B,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,mBAAmB,CAAC;gBAC7C,MAAM,EAAE,UAAU;gBAClB,KAAK,EAAE,MAAM;gBACb,EAAE,EAAE,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK;gBACpC,IAAI,EAAE,IAAI,CAAC,WAAW,CAAC,KAAK;gBAC5B,SAAS,EAAE,eAAe;aAC3B,CAAC,CAAC;QACL,CAAC;QAID,KAAK,CAAC,OAAO,CAAC,YAA2B;YACvC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,OAAO,EAAE,CAAC;YAC3D,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC,CAAC;YACjE,OAAO,IAAI,wBAAe,CACxB,CAAC,CAAC,MAAM,CAAC,EAAE,EACX,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,WAAW,EAClB,MAAM,CAAC,YAAY,EACnB,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,YAAY,EACnB,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,WAAW,CACnB,CAAC;QACJ,CAAC;QAID,IAAI,CAAC,KAAe;YAClB,IAAA,+BAAqB,EAAC,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,WAAW,CAAC,CAAC;YACpD,OAAO,IAAI,UAAU,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;QACvE,CAAC;QAID,MAAM,CAAC,QAAkB;YACvB,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAC/B,IAAA,+BAAqB,EAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,aAAa,CAAC,CAAC;YACzD,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC7C,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM;YAIJ,MAAM,OAAO,GAAG,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,MAAM,CAAC;YAE3D,OAAO;gBACL,GAAG,IAAI,CAAC,YAAY;gBACpB,QAAQ,EAAE,OAAO;aAClB,CAAC;QACJ,CAAC;QAED,QAAQ;YACN,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACvC,CAAC;QAED,aAAa;YAIX,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACpB,MAAM,IAAI,iCAAwB,CAChC,2DAA2D,EAC3D,qBAAY,CAAC,gBAAgB,CAC9B,CAAC;YACJ,CAAC;YACD,OAAO,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACrD,iBAAiB,EAAE,CAAC,CAAC,iBAAiB;gBACtC,SAAS,EAAE,CAAC,CAAC,SAAS;gBACtB,UAAU,EAAE,CAAC,CAAC,UAAU;aACzB,CAAC,CAAC,CAAC;QACN,CAAC;;;;;;+BAjEA,2BAAc,EACd,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;4BAiBhB,IAAA,uBAAU,EAAC,YAAY,CAAC,EACxB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;8BAMhB,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;QAxBjB,8KAAM,OAAO,6DAcZ;QAID,qKAAA,IAAI,6DAGH;QAID,2KAAA,MAAM,6DAKL;QArFH,6KAuHC;;;QAvHoB,uDAAI"}
+{"version":3,"file":"bulk.js","sourceRoot":"","sources":["../src/bulk.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,6CAOsB;AAEtB,4CAAyE;AAUzE,mCAAsC;AACtC,uCAAgE;AAChE,qCAA2C;IAK9B,UAAU;4BADtB,iCAAoB;;;;sBACW,mCAAsB;;;;;;;;;;;;;uCAA9B,SAAQ,WAAsB;QAGpD,YAAY,SAAwB,EAAE,UAAgB;YACpD,KAAK,EAAE,CAAC;YAHV,+BAA0B,GADf,mDAAU,CACqB;YAIxC,IAAI,CAAC,0BAA0B,GAAG,SAAS,CAAC;YAC5C,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAChC,CAAC;QAED,IAAI,MAAM;YACR,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;QACjC,CAAC;QAED,CAAC,mBAAW,CAAC;YACX,OAAO,YAAY,CAAC;QACtB,CAAC;QAID,SAAS,CAAC,IAAsB;YAC9B,IAAI,CAAC,0BAA0B,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAChD,OAAO,IAAI,CAAC;QACd,CAAC;QAID,YAAY,CAAC,OAAmB;YAC9B,IAAI,CAAC,0BAA0B,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACtD,OAAO,IAAI,CAAC;QACd,CAAC;QAID,IAAI,CAAC,OAAiB;YACpB,IAAA,+BAAqB,EAAC,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,iBAAiB,CAAC,CAAC;YAC5D,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC9C,OAAO,IAAI,CAAC;QACd,CAAC;QAID,MAAM;YACJ,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAC3C,IAAI,CAAC,0BAA0B,CAAC,MAAM,EAAE,CAAC;YACzC,OAAO,IAAI,CAAC,WAAW,CAAC;QAC1B,CAAC;QAID,SAAS;YACP,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAC3C,IAAI,CAAC,0BAA0B,CAAC,SAAS,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC,WAAW,CAAC;QAC1B,CAAC;QAKD,MAAM;YACJ,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC;QACvB,CAAC;QAKD,SAAS;YACP,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC;QAC1B,CAAC;QAID,UAAU,CAAC,WAAqB;YAC9B,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAC3C,IAAA,+BAAqB,EAAC,CAAC,WAAW,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,wBAAwB,CAAC,CAAC;YACvE,MAAM,EAAE,GAAG,IAAA,sBAAY,EAAC,WAAW,CAAC,CAAC;YACrC,IAAI,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAC/C,OAAO,IAAI,CAAC,WAAW,CAAC;QAC1B,CAAC;QAID,SAAS,CAAC,MAA6B;YACrC,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAC3C,IAAA,+BAAqB,EAAC,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,mBAAmB,CAAC,CAAC;YAC7D,MAAM,EAAE,GAAG,IAAA,sBAAY,EAAC,MAAM,CAAC,CAAC;YAChC,IAAI,CAAC,0BAA0B,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAC9C,OAAO,IAAI,CAAC,WAAW,CAAC;QAC1B,CAAC;QAGD,MAAM,CAAC,MAA6B;YAClC,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAC3C,IAAA,+BAAqB,EAAC,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,mBAAmB,CAAC,CAAC;YAC7D,MAAM,EAAE,GAAG,IAAA,sBAAY,EAAC,MAAM,CAAC,CAAC;YAChC,IAAI,CAAC,0BAA0B,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC3C,OAAO,IAAI,CAAC,WAAW,CAAC;QAC1B,CAAC;QAGD,MAAM;YACJ,IAAI,CAAC,0BAA0B,CAAC,MAAM,EAAE,CAAC;YACzC,OAAO,IAAI,CAAC;QACd,CAAC;;;;;;iCArFA,IAAA,uBAAU,EAAC,YAAY,CAAC,EACxB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;oCAMhB,IAAA,uBAAU,EAAC,YAAY,CAAC,EACxB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;4BAMhB,IAAA,uBAAU,EAAC,YAAY,CAAC,EACxB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;8BAOhB,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;iCAOhB,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;8BAOhB,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC,EAChB,uBAAU;iCAKV,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC,EAChB,uBAAU;kCAKV,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;iCAShB,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;8BAShB,IAAA,uBAAU,EAAC,MAAM,CAAC;8BASlB,IAAA,uBAAU,EAAC,MAAM,CAAC;QA/EnB,oLAAA,SAAS,6DAGR;QAID,6LAAA,YAAY,6DAGX;QAID,qKAAA,IAAI,6DAIH;QAID,2KAAA,MAAM,6DAIL;QAID,oLAAA,SAAS,6DAIR;QAKD,2KAAA,MAAM,6DAEL;QAKD,oLAAA,SAAS,6DAER;QAID,uLAAA,UAAU,6DAMT;QAID,oLAAA,SAAS,6DAMR;QAGD,2KAAA,MAAM,6DAML;QAGD,2KAAA,MAAM,6DAGL;QAtGH,6KAuGC;;;QAvGY,uDAAU;;;;AAAV,gCAAU;;4BAyGtB,iCAAoB;;;;sBACa,mCAAsB;;;;;iCAA9B,SAAQ,WAAsB;QAQtD,YACE,UAAgC,EAChC,SAAwD,EACxD,OAAO,GAAG,KAAK;YAEf,KAAK,EAAE,CAAC;YAZV,WAAM,GADa,mDAAI,CACT;YAaZ,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;YAC9B,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;YAChC,IAAI,CAAC,sBAAsB,GAAG,SAAS,CAAC;YACxC,IAAI,CAAC,YAAY,GAAG;gBAClB,UAAU,EAAE,CAAC;gBACb,UAAU,EAAE,CAAC;gBACb,UAAU,EAAE,CAAC;aACd,CAAC;YACF,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;YACvB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QAC1B,CAAC;QAKD,CAAC,mBAAW,CAAC;YACX,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC;QACvB,CAAC;QASO,gBAAgB,CACtB,UAAkB,EAClB,kBAA4B,EAAE;YAE9B,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,mBAAmB,CAAC;gBAC7C,MAAM,EAAE,UAAU;gBAClB,KAAK,EAAE,MAAM;gBACb,EAAE,EAAE,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK;gBACpC,IAAI,EAAE,IAAI,CAAC,WAAW,CAAC,KAAK;gBAC5B,SAAS,EAAE,eAAe;aAC3B,CAAC,CAAC;QACL,CAAC;QAID,KAAK,CAAC,OAAO,CAAC,YAA2B;YACvC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,OAAO,EAAE,CAAC;YAC3D,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC,CAAC;YACjE,OAAO,IAAI,wBAAe,CACxB,CAAC,CAAC,MAAM,CAAC,EAAE,EACX,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,WAAW,EAClB,MAAM,CAAC,YAAY,EACnB,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,YAAY,EACnB,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,WAAW,CACnB,CAAC;QACJ,CAAC;QAID,IAAI,CAAC,KAAe;YAClB,IAAA,+BAAqB,EAAC,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,WAAW,CAAC,CAAC;YACpD,OAAO,IAAI,UAAU,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;QACvE,CAAC;QAID,MAAM,CAAC,QAAqB;YAC1B,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAC/B,IAAA,+BAAqB,EAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,aAAa,CAAC,CAAC;YACzD,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC7C,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM;YAIJ,MAAM,OAAO,GAAG,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,MAAM,CAAC;YAE3D,OAAO;gBACL,GAAG,IAAI,CAAC,YAAY;gBACpB,QAAQ,EAAE,OAAO;aAClB,CAAC;QACJ,CAAC;QAED,QAAQ;YACN,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACvC,CAAC;QAED,aAAa;YAIX,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACpB,MAAM,IAAI,iCAAwB,CAChC,2DAA2D,EAC3D,qBAAY,CAAC,gBAAgB,CAC9B,CAAC;YACJ,CAAC;YACD,OAAO,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACrD,iBAAiB,EAAE,CAAC,CAAC,iBAAiB;gBACtC,SAAS,EAAE,CAAC,CAAC,SAAS;gBACtB,UAAU,EAAE,CAAC,CAAC,UAAU;aACzB,CAAC,CAAC,CAAC;QACN,CAAC;;;;;;+BAjEA,2BAAc,EACd,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;4BAiBhB,IAAA,uBAAU,EAAC,YAAY,CAAC,EACxB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;8BAMhB,IAAA,uBAAU,EAAC,MAAM,CAAC,EAClB,IAAA,wBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;QAxBjB,8KAAM,OAAO,6DAcZ;QAID,qKAAA,IAAI,6DAGH;QAID,2KAAA,MAAM,6DAKL;QArFH,6KAuHC;;;QAvHoB,uDAAI"}

package/lib/collection.d.ts

@@ -9,6 +9,7 @@ import { AggregationCursor, BulkWriteResult, CommandResult, Cursor, DeleteResult
 import Bulk from './bulk';
 import PlanCache from './plan-cache';
 import ChangeStreamCursor from './change-stream-cursor';
+import type { MQLDocument, MQLQuery, MQLPipeline } from './mql-types';
 export type CollectionWithSchema<M extends GenericServerSideSchema = GenericServerSideSchema, D extends GenericDatabaseSchema = M[keyof M], C extends GenericCollectionSchema = D[keyof D], N extends StringKey<D> = StringKey<D>> = Collection<M, D, C, N> & {
     [k in StringKey<D> as k extends `${N}.${infer S}` ? S : never]: Collection<M, D, D[k], k>;
 };
@@ -16,36 +17,37 @@ export declare class Collection<M extends GenericServerSideSchema = GenericServe
     _mongo: Mongo<M>;
     _database: DatabaseWithSchema<M, D>;
     _name: N;
+    _cachedSampleDocs: Document[];
     constructor(mongo: Mongo<M>, database: DatabaseWithSchema<M, D> | Database<M, D>, name: N);
     [namespaceInfo](): Namespace;
     [asPrintable](): string;
     private _emitCollectionApiCall;
-    aggregate(pipeline: Document[], options: AggregateOptions & {
+    aggregate(pipeline: MQLPipeline, options: AggregateOptions & {
         explain: ExplainVerbosityLike;
     }): Promise<Document>;
-    aggregate(pipeline: Document[], options?: AggregateOptions): Promise<AggregationCursor>;
-    aggregate(...stages: Document[]): Promise<AggregationCursor>;
+    aggregate(pipeline: MQLPipeline, options?: AggregateOptions): Promise<AggregationCursor>;
+    aggregate(...stages: MQLPipeline): Promise<AggregationCursor>;
     bulkWrite(operations: AnyBulkWriteOperation[], options?: BulkWriteOptions): Promise<BulkWriteResult>;
     count(query?: {}, options?: CountOptions): Promise<number>;
-    countDocuments(query?: Document, options?: CountDocumentsOptions): Promise<number>;
+    countDocuments(query?: MQLQuery, options?: CountDocumentsOptions): Promise<number>;
     deleteMany(filter: Document, options?: DeleteOptions): Promise<DeleteResult | Document>;
     deleteOne(filter: Document, options?: DeleteOptions): Promise<DeleteResult | Document>;
     distinct(field: string): Promise<Document>;
-    distinct(field: string, query: Document): Promise<Document>;
-    distinct(field: string, query: Document, options: DistinctOptions): Promise<Document>;
+    distinct(field: string, query: MQLQuery): Promise<Document>;
+    distinct(field: string, query: MQLQuery, options: DistinctOptions): Promise<Document>;
     estimatedDocumentCount(options?: EstimatedDocumentCountOptions): Promise<number>;
-    find(query?: Document, projection?: Document, options?: FindOptions): Promise<Cursor>;
+    find(query?: MQLQuery, projection?: Document, options?: FindOptions): Promise<Cursor>;
     findAndModify(options: FindAndModifyMethodShellOptions): Promise<Document | null>;
-    findOne(query?: Document, projection?: Document, options?: FindOptions): Promise<C['schema'] | null>;
+    findOne(query?: MQLQuery, projection?: Document, options?: FindOptions): Promise<MQLDocument | null>;
     renameCollection(newName: string, dropTarget?: boolean): Promise<Document>;
     findOneAndDelete(filter: Document, options?: FindOneAndDeleteOptions): Promise<Document | null>;
     findOneAndReplace(filter: Document, replacement: Document, options?: FindAndModifyShellOptions<FindOneAndReplaceOptions>): Promise<Document>;
     findOneAndUpdate(filter: Document, update: Document | Document[], options?: FindAndModifyShellOptions<FindOneAndUpdateOptions>): Promise<Document>;
-    insert(docs: Document | Document[], options?: BulkWriteOptions): Promise<InsertManyResult>;
-    insertMany(docs: Document[], options?: BulkWriteOptions): Promise<InsertManyResult>;
-    insertOne(doc: Document, options?: InsertOneOptions): Promise<InsertOneResult>;
+    insert(docs: MQLDocument | MQLDocument[], options?: BulkWriteOptions): Promise<InsertManyResult>;
+    insertMany(docs: MQLDocument[], options?: BulkWriteOptions): Promise<InsertManyResult>;
+    insertOne(doc: MQLDocument, options?: InsertOneOptions): Promise<InsertOneResult>;
     isCapped(): Promise<boolean>;
-    remove(query: Document, options?: boolean | RemoveShellOptions): Promise<DeleteResult | Document>;
+    remove(query: MQLQuery, options?: boolean | RemoveShellOptions): Promise<DeleteResult | Document>;
     replaceOne(filter: Document, replacement: Document, options?: ReplaceOptions): Promise<UpdateResult>;
     update(filter: Document, update: Document, options?: UpdateOptions & {
         multi?: boolean;
@@ -91,7 +93,11 @@ export declare class Collection<M extends GenericServerSideSchema = GenericServe
     getShardVersion(): Promise<Document>;
     _getShardedCollectionInfo(config: DatabaseWithSchema<M, D>, collStats: Document[]): Promise<Document>;
     getShardDistribution(): Promise<CommandResult<GetShardDistributionResult>>;
-    watch(pipeline?: Document[] | ChangeStreamOptions, options?: ChangeStreamOptions): Promise<ChangeStreamCursor>;
+    getShardLocation(): Promise<{
+        shards: string[];
+        sharded: boolean;
+    }>;
+    watch(pipeline?: MQLPipeline | ChangeStreamOptions, options?: ChangeStreamOptions): Promise<ChangeStreamCursor>;
     hideIndex(index: string | Document): Promise<Document>;
     unhideIndex(index: string | Document): Promise<Document>;
     analyzeShardKey(key: Document, options?: Document): Promise<Document>;
@@ -105,6 +111,8 @@ export declare class Collection<M extends GenericServerSideSchema = GenericServe
     createSearchIndexes(specs: SearchIndexDescription[]): Promise<string[]>;
     dropSearchIndex(indexName: string): Promise<void>;
     updateSearchIndex(indexName: string, definition: Document): Promise<void>;
+    _getSampleDocs(): Promise<Document[]>;
+    _getSampleDocsForCompletion(): Promise<Document[]>;
 }
 export type GetShardDistributionResult = {
     Totals: {

package/lib/collection.js

@@ -114,6 +114,7 @@ let Collection = (() => {
     let _validate_decorators;
     let _getShardVersion_decorators;
     let _getShardDistribution_decorators;
+    let _getShardLocation_decorators;
     let _watch_decorators;
     let _hideIndex_decorators;
     let _unhideIndex_decorators;
@@ -129,6 +130,7 @@ let Collection = (() => {
         constructor(mongo, database, name) {
             super();
             this._mongo = __runInitializers(this, _instanceExtraInitializers);
+            this._cachedSampleDocs = [];
             this._mongo = mongo;
             this._database = database;
             this._name = name;
@@ -1058,6 +1060,35 @@ let Collection = (() => {
             result.Totals = totalValue;
             return new index_1.CommandResult('StatsResult', result);
         }
+        async getShardLocation() {
+            this._emitCollectionApiCall('getShardLocation', {});
+            const result = await (await this._database.aggregate([
+                {
+                    $listClusterCatalog: {
+                        shards: true,
+                    },
+                },
+                {
+                    $match: {
+                        ns: this.getFullName(),
+                    },
+                },
+                {
+                    $project: {
+                        _id: 0,
+                        shards: 1,
+                        sharded: 1,
+                    },
+                },
+            ])).toArray();
+            if (result.length > 0) {
+                return {
+                    shards: result[0].shards,
+                    sharded: result[0].sharded,
+                };
+            }
+            throw new errors_1.MongoshRuntimeError(`Error finding location information for ${this.getFullName()}`, errors_1.CommonErrors.CommandFailed);
+        }
         async watch(pipeline = [], options = {}) {
             if (!Array.isArray(pipeline)) {
                 options = pipeline;
@@ -1102,7 +1133,7 @@ let Collection = (() => {
         }
         async checkMetadataConsistency(options = {}) {
             this._emitCollectionApiCall('checkMetadataConsistency', { options });
-            return this._database._runCursorCommand({
+            return await this._database._runCursorCommand({
                 checkMetadataConsistency: this._name,
             });
         }
@@ -1164,6 +1195,25 @@ let Collection = (() => {
             this._emitCollectionApiCall('updateSearchIndex', { indexName, definition });
             return await this._mongo._serviceProvider.updateSearchIndex(this._database._name, this._name, indexName, definition);
         }
+        async _getSampleDocs() {
+            this._cachedSampleDocs = await (await this.aggregate([{ $sample: { size: 10 } }], {
+                allowDiskUse: true,
+                maxTimeMS: 1000,
+                readPreference: 'secondaryPreferred',
+            })).toArray();
+            return this._cachedSampleDocs;
+        }
+        async _getSampleDocsForCompletion() {
+            return await Promise.race([
+                (async () => {
+                    return await this._getSampleDocs();
+                })(),
+                (async () => {
+                    await new Promise((resolve) => { var _a, _b; return (_b = (_a = setTimeout(resolve, 200)) === null || _a === void 0 ? void 0 : _a.unref) === null || _b === void 0 ? void 0 : _b.call(_a); });
+                    return this._cachedSampleDocs;
+                })(),
+            ]);
+        }
     };
     __setFunctionName(_classThis, "Collection");
     (() => {
@@ -1224,6 +1274,7 @@ let Collection = (() => {
         _validate_decorators = [decorators_1.returnsPromise, (0, decorators_1.apiVersions)([])];
         _getShardVersion_decorators = [decorators_1.returnsPromise, (0, decorators_1.topologies)([enums_1.Topologies.Sharded]), (0, decorators_1.apiVersions)([])];
         _getShardDistribution_decorators = [decorators_1.returnsPromise, (0, decorators_1.topologies)([enums_1.Topologies.Sharded]), (0, decorators_1.apiVersions)([])];
+        _getShardLocation_decorators = [decorators_1.returnsPromise, (0, decorators_1.topologies)([enums_1.Topologies.Sharded]), (0, decorators_1.apiVersions)([]), (0, decorators_1.serverVersions)(['8.0.10', enums_1.ServerVersions.latest])];
         _watch_decorators = [(0, decorators_1.serverVersions)(['3.1.0', enums_1.ServerVersions.latest]), (0, decorators_1.topologies)([enums_1.Topologies.ReplSet, enums_1.Topologies.Sharded]), (0, decorators_1.apiVersions)([1]), decorators_1.returnsPromise];
         _hideIndex_decorators = [(0, decorators_1.serverVersions)(['4.4.0', enums_1.ServerVersions.latest]), decorators_1.returnsPromise, (0, decorators_1.apiVersions)([1])];
         _unhideIndex_decorators = [(0, decorators_1.serverVersions)(['4.4.0', enums_1.ServerVersions.latest]), decorators_1.returnsPromise, (0, decorators_1.apiVersions)([1])];
@@ -1290,6 +1341,7 @@ let Collection = (() => {
         __esDecorate(_classThis, null, _validate_decorators, { kind: "method", name: "validate", static: false, private: false, access: { has: obj => "validate" in obj, get: obj => obj.validate }, metadata: _metadata }, null, _instanceExtraInitializers);
         __esDecorate(_classThis, null, _getShardVersion_decorators, { kind: "method", name: "getShardVersion", static: false, private: false, access: { has: obj => "getShardVersion" in obj, get: obj => obj.getShardVersion }, metadata: _metadata }, null, _instanceExtraInitializers);
         __esDecorate(_classThis, null, _getShardDistribution_decorators, { kind: "method", name: "getShardDistribution", static: false, private: false, access: { has: obj => "getShardDistribution" in obj, get: obj => obj.getShardDistribution }, metadata: _metadata }, null, _instanceExtraInitializers);
+        __esDecorate(_classThis, null, _getShardLocation_decorators, { kind: "method", name: "getShardLocation", static: false, private: false, access: { has: obj => "getShardLocation" in obj, get: obj => obj.getShardLocation }, metadata: _metadata }, null, _instanceExtraInitializers);
         __esDecorate(_classThis, null, _watch_decorators, { kind: "method", name: "watch", static: false, private: false, access: { has: obj => "watch" in obj, get: obj => obj.watch }, metadata: _metadata }, null, _instanceExtraInitializers);
         __esDecorate(_classThis, null, _hideIndex_decorators, { kind: "method", name: "hideIndex", static: false, private: false, access: { has: obj => "hideIndex" in obj, get: obj => obj.hideIndex }, metadata: _metadata }, null, _instanceExtraInitializers);
         __esDecorate(_classThis, null, _unhideIndex_decorators, { kind: "method", name: "unhideIndex", static: false, private: false, access: { has: obj => "unhideIndex" in obj, get: obj => obj.unhideIndex }, metadata: _metadata }, null, _instanceExtraInitializers);