@depup/mongodb 7.1.0-depup.0

package/lib/operations/client_bulk_write/executor.js

@@ -0,0 +1,120 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ClientBulkWriteExecutor = void 0;
+const abstract_cursor_1 = require("../../cursor/abstract_cursor");
+const client_bulk_write_cursor_1 = require("../../cursor/client_bulk_write_cursor");
+const error_1 = require("../../error");
+const timeout_1 = require("../../timeout");
+const utils_1 = require("../../utils");
+const write_concern_1 = require("../../write_concern");
+const execute_operation_1 = require("../execute_operation");
+const client_bulk_write_1 = require("./client_bulk_write");
+const command_builder_1 = require("./command_builder");
+const results_merger_1 = require("./results_merger");
+/**
+ * Responsible for executing a client bulk write.
+ * @internal
+ */
+class ClientBulkWriteExecutor {
+    /**
+     * Instantiate the executor.
+     * @param client - The mongo client.
+     * @param operations - The user supplied bulk write models.
+     * @param options - The bulk write options.
+     */
+    constructor(client, operations, options) {
+        if (operations.length === 0) {
+            throw new error_1.MongoClientBulkWriteExecutionError('No client bulk write models were provided.');
+        }
+        this.client = client;
+        this.operations = operations;
+        this.options = {
+            ordered: true,
+            bypassDocumentValidation: false,
+            verboseResults: false,
+            ...options
+        };
+        // If no write concern was provided, we inherit one from the client.
+        if (!this.options.writeConcern) {
+            this.options.writeConcern = write_concern_1.WriteConcern.fromOptions(this.client.s.options);
+        }
+        if (this.options.writeConcern?.w === 0) {
+            if (this.options.verboseResults) {
+                throw new error_1.MongoInvalidArgumentError('Cannot request unacknowledged write concern and verbose results');
+            }
+            if (this.options.ordered) {
+                throw new error_1.MongoInvalidArgumentError('Cannot request unacknowledged write concern and ordered writes');
+            }
+        }
+    }
+    /**
+     * Execute the client bulk write. Will split commands into batches and exhaust the cursors
+     * for each, then merge the results into one.
+     * @returns The result.
+     */
+    async execute() {
+        // The command builder will take the user provided models and potential split the batch
+        // into multiple commands due to size.
+        const pkFactory = this.client.s.options.pkFactory;
+        const commandBuilder = new command_builder_1.ClientBulkWriteCommandBuilder(this.operations, this.options, pkFactory);
+        // Unacknowledged writes need to execute all batches and return { ok: 1}
+        const resolvedOptions = (0, utils_1.resolveTimeoutOptions)(this.client, this.options);
+        const context = timeout_1.TimeoutContext.create(resolvedOptions);
+        if (this.options.writeConcern?.w === 0) {
+            while (commandBuilder.hasNextBatch()) {
+                const operation = new client_bulk_write_1.ClientBulkWriteOperation(commandBuilder, this.options);
+                await (0, execute_operation_1.executeOperation)(this.client, operation, context);
+            }
+            return results_merger_1.ClientBulkWriteResultsMerger.unacknowledged();
+        }
+        else {
+            const resultsMerger = new results_merger_1.ClientBulkWriteResultsMerger(this.options);
+            // For each command will will create and exhaust a cursor for the results.
+            while (commandBuilder.hasNextBatch()) {
+                const cursorContext = new abstract_cursor_1.CursorTimeoutContext(context, Symbol());
+                const options = {
+                    ...this.options,
+                    timeoutContext: cursorContext,
+                    ...(resolvedOptions.timeoutMS != null && { timeoutMode: abstract_cursor_1.CursorTimeoutMode.LIFETIME })
+                };
+                const cursor = new client_bulk_write_cursor_1.ClientBulkWriteCursor(this.client, commandBuilder, options);
+                try {
+                    await resultsMerger.merge(cursor);
+                }
+                catch (error) {
+                    // Write concern errors are recorded in the writeConcernErrors field on MongoClientBulkWriteError.
+                    // When a write concern error is encountered, it should not terminate execution of the bulk write
+                    // for either ordered or unordered bulk writes. However, drivers MUST throw an exception at the end
+                    // of execution if any write concern errors were observed.
+                    if (error instanceof error_1.MongoServerError && !(error instanceof error_1.MongoClientBulkWriteError)) {
+                        // Server side errors need to be wrapped inside a MongoClientBulkWriteError, where the root
+                        // cause is the error property and a partial result is to be included.
+                        const bulkWriteError = new error_1.MongoClientBulkWriteError({
+                            message: 'Mongo client bulk write encountered an error during execution'
+                        });
+                        bulkWriteError.cause = error;
+                        bulkWriteError.partialResult = resultsMerger.bulkWriteResult;
+                        throw bulkWriteError;
+                    }
+                    else {
+                        // Client side errors are just thrown.
+                        throw error;
+                    }
+                }
+            }
+            // If we have write concern errors or unordered write errors at the end we throw.
+            if (resultsMerger.writeConcernErrors.length > 0 || resultsMerger.writeErrors.size > 0) {
+                const error = new error_1.MongoClientBulkWriteError({
+                    message: 'Mongo client bulk write encountered errors during execution.'
+                });
+                error.writeConcernErrors = resultsMerger.writeConcernErrors;
+                error.writeErrors = resultsMerger.writeErrors;
+                error.partialResult = resultsMerger.bulkWriteResult;
+                throw error;
+            }
+            return resultsMerger.bulkWriteResult;
+        }
+    }
+}
+exports.ClientBulkWriteExecutor = ClientBulkWriteExecutor;
+//# sourceMappingURL=executor.js.map

package/lib/operations/client_bulk_write/executor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"executor.js","sourceRoot":"","sources":["../../../src/operations/client_bulk_write/executor.ts"],"names":[],"mappings":";;;AACA,kEAAuF;AACvF,oFAA8E;AAC9E,uCAKqB;AAErB,2CAA+C;AAC/C,uCAAoD;AACpD,uDAAmD;AACnD,4DAAwD;AACxD,2DAA+D;AAC/D,uDAAkE;AAMlE,qDAAgE;AAEhE;;;GAGG;AACH,MAAa,uBAAuB;IAKlC;;;;;OAKG;IACH,YACE,MAAmB,EACnB,UAA4D,EAC5D,OAAgC;QAEhC,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,0CAAkC,CAAC,4CAA4C,CAAC,CAAC;QAC7F,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG;YACb,OAAO,EAAE,IAAI;YACb,wBAAwB,EAAE,KAAK;YAC/B,cAAc,EAAE,KAAK;YACrB,GAAG,OAAO;SACX,CAAC;QAEF,oEAAoE;QACpE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;YAC/B,IAAI,CAAC,OAAO,CAAC,YAAY,GAAG,4BAAY,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAC9E,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;YACvC,IAAI,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;gBAChC,MAAM,IAAI,iCAAyB,CACjC,iEAAiE,CAClE,CAAC;YACJ,CAAC;YAED,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;gBACzB,MAAM,IAAI,iCAAyB,CACjC,gEAAgE,CACjE,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,OAAO;QACX,uFAAuF;QACvF,sCAAsC;QACtC,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC;QAClD,MAAM,cAAc,GAAG,IAAI,+CAA6B,CACtD,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,OAAO,EACZ,SAAS,CACV,CAAC;QACF,wEAAwE;QACxE,MAAM,eAAe,GAAG,IAAA,6BAAqB,EAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACzE,MAAM,OAAO,GAAG,wBAAc,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;QAEvD,IAAI,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;YACvC,OAAO,cAAc,CAAC,YAAY,EAAE,EAAE,CAAC;gBACrC,MAAM,SAAS,GAAG,IAAI,4CAAwB,CAAC,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC7E,MAAM,IAAA,oCAAgB,EAAC,IAAI,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;YAC1D,CAAC;YACD,OAAO,6CAA4B,CAAC,cAAc,EAAE,CAAC;QACvD,CAAC;aAAM,CAAC;YACN,MAAM,aAAa,GAAG,IAAI,6CAA4B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrE,0EAA0E;YAC1E,OAAO,cAAc,CAAC,YAAY,EAAE,EAAE,CAAC;gBACrC,MAAM,aAAa,GAAG,IAAI,sCAAoB,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;gBAClE,MAAM,OAAO,GAAG;oBACd,GAAG,IAAI,CAAC,OAAO;oBACf,cAAc,EAAE,aAAa;oBAC7B,GAAG,CAAC,eAAe,CAAC,SAAS,IAAI,IAAI,IAAI,EAAE,WAAW,EAAE,mCAAiB,CAAC,QAAQ,EAAE,CAAC;iBACtF,CAAC;gBACF,MAAM,MAAM,GAAG,IAAI,gDAAqB,CAAC,IAAI,CAAC,MAAM,EAAE,cAAc,EAAE,OAAO,CAAC,CAAC;gBAC/E,IAAI,CAAC;oBACH,MAAM,aAAa,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBACpC,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,kGAAkG;oBAClG,iGAAiG;oBACjG,mGAAmG;oBACnG,0DAA0D;oBAC1D,IAAI,KAAK,YAAY,wBAAgB,IAAI,CAAC,CAAC,KAAK,YAAY,iCAAyB,CAAC,EAAE,CAAC;wBACvF,2FAA2F;wBAC3F,sEAAsE;wBACtE,MAAM,cAAc,GAAG,IAAI,iCAAyB,CAAC;4BACnD,OAAO,EAAE,+DAA+D;yBACzE,CAAC,CAAC;wBACH,cAAc,CAAC,KAAK,GAAG,KAAK,CAAC;wBAC7B,cAAc,CAAC,aAAa,GAAG,aAAa,CAAC,eAAe,CAAC;wBAC7D,MAAM,cAAc,CAAC;oBACvB,CAAC;yBAAM,CAAC;wBACN,sCAAsC;wBACtC,MAAM,KAAK,CAAC;oBACd,CAAC;gBACH,CAAC;YACH,CAAC;YAED,iFAAiF;YACjF,IAAI,aAAa,CAAC,kBAAkB,CAAC,MAAM,GAAG,CAAC,IAAI,aAAa,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBACtF,MAAM,KAAK,GAAG,IAAI,iCAAyB,CAAC;oBAC1C,OAAO,EAAE,8DAA8D;iBACxE,CAAC,CAAC;gBACH,KAAK,CAAC,kBAAkB,GAAG,aAAa,CAAC,kBAAkB,CAAC;gBAC5D,KAAK,CAAC,WAAW,GAAG,aAAa,CAAC,WAAW,CAAC;gBAC9C,KAAK,CAAC,aAAa,GAAG,aAAa,CAAC,eAAe,CAAC;gBACpD,MAAM,KAAK,CAAC;YACd,CAAC;YAED,OAAO,aAAa,CAAC,eAAe,CAAC;QACvC,CAAC;IACH,CAAC;CACF;AAzHD,0DAyHC"}

package/lib/operations/client_bulk_write/results_merger.js

@@ -0,0 +1,204 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ClientBulkWriteResultsMerger = void 0;
+const __1 = require("../..");
+const error_1 = require("../../error");
+/**
+ * Unacknowledged bulk writes are always the same.
+ */
+const UNACKNOWLEDGED = {
+    acknowledged: false,
+    insertedCount: 0,
+    upsertedCount: 0,
+    matchedCount: 0,
+    modifiedCount: 0,
+    deletedCount: 0,
+    insertResults: undefined,
+    updateResults: undefined,
+    deleteResults: undefined
+};
+/**
+ * Merges client bulk write cursor responses together into a single result.
+ * @internal
+ */
+class ClientBulkWriteResultsMerger {
+    /**
+     * @returns The standard unacknowledged bulk write result.
+     */
+    static unacknowledged() {
+        return UNACKNOWLEDGED;
+    }
+    /**
+     * Instantiate the merger.
+     * @param options - The options.
+     */
+    constructor(options) {
+        this.options = options;
+        this.currentBatchOffset = 0;
+        this.writeConcernErrors = [];
+        this.writeErrors = new Map();
+        this.result = {
+            acknowledged: true,
+            insertedCount: 0,
+            upsertedCount: 0,
+            matchedCount: 0,
+            modifiedCount: 0,
+            deletedCount: 0,
+            insertResults: undefined,
+            updateResults: undefined,
+            deleteResults: undefined
+        };
+        if (options.verboseResults) {
+            this.result.insertResults = new Map();
+            this.result.updateResults = new Map();
+            this.result.deleteResults = new Map();
+        }
+    }
+    /**
+     * Get the bulk write result object.
+     */
+    get bulkWriteResult() {
+        return {
+            acknowledged: this.result.acknowledged,
+            insertedCount: this.result.insertedCount,
+            upsertedCount: this.result.upsertedCount,
+            matchedCount: this.result.matchedCount,
+            modifiedCount: this.result.modifiedCount,
+            deletedCount: this.result.deletedCount,
+            insertResults: this.result.insertResults,
+            updateResults: this.result.updateResults,
+            deleteResults: this.result.deleteResults
+        };
+    }
+    /**
+     * Merge the results in the cursor to the existing result.
+     * @param currentBatchOffset - The offset index to the original models.
+     * @param response - The cursor response.
+     * @param documents - The documents in the cursor.
+     * @returns The current result.
+     */
+    async merge(cursor) {
+        let writeConcernErrorResult;
+        try {
+            for await (const document of cursor) {
+                // Only add to maps if ok: 1
+                if (document.ok === 1) {
+                    if (this.options.verboseResults) {
+                        this.processDocument(cursor, document);
+                    }
+                }
+                else {
+                    // If an individual write error is encountered during an ordered bulk write, drivers MUST
+                    // record the error in writeErrors and immediately throw the exception. Otherwise, drivers
+                    // MUST continue to iterate the results cursor and execute any further bulkWrite batches.
+                    if (this.options.ordered) {
+                        const error = new error_1.MongoClientBulkWriteError({
+                            message: 'Mongo client ordered bulk write encountered a write error.'
+                        });
+                        error.writeErrors.set(document.idx + this.currentBatchOffset, {
+                            code: document.code,
+                            message: document.errmsg
+                        });
+                        error.partialResult = this.result;
+                        throw error;
+                    }
+                    else {
+                        this.writeErrors.set(document.idx + this.currentBatchOffset, {
+                            code: document.code,
+                            message: document.errmsg
+                        });
+                    }
+                }
+            }
+        }
+        catch (error) {
+            if (error instanceof __1.MongoWriteConcernError) {
+                const result = error.result;
+                writeConcernErrorResult = {
+                    insertedCount: result.nInserted,
+                    upsertedCount: result.nUpserted,
+                    matchedCount: result.nMatched,
+                    modifiedCount: result.nModified,
+                    deletedCount: result.nDeleted,
+                    writeConcernError: result.writeConcernError
+                };
+                if (this.options.verboseResults && result.cursor.firstBatch) {
+                    for (const document of result.cursor.firstBatch) {
+                        if (document.ok === 1) {
+                            this.processDocument(cursor, document);
+                        }
+                    }
+                }
+            }
+            else {
+                throw error;
+            }
+        }
+        finally {
+            // Update the counts from the cursor response.
+            if (cursor.response) {
+                const response = cursor.response;
+                this.incrementCounts(response);
+            }
+            // Increment the batch offset.
+            this.currentBatchOffset += cursor.operations.length;
+        }
+        // If we have write concern errors ensure they are added.
+        if (writeConcernErrorResult) {
+            const writeConcernError = writeConcernErrorResult.writeConcernError;
+            this.incrementCounts(writeConcernErrorResult);
+            this.writeConcernErrors.push({
+                code: writeConcernError.code,
+                message: writeConcernError.errmsg
+            });
+        }
+        return this.result;
+    }
+    /**
+     * Process an individual document in the results.
+     * @param cursor - The cursor.
+     * @param document - The document to process.
+     */
+    processDocument(cursor, document) {
+        // Get the corresponding operation from the command.
+        const operation = cursor.operations[document.idx];
+        // Handle insert results.
+        if ('insert' in operation) {
+            this.result.insertResults?.set(document.idx + this.currentBatchOffset, {
+                insertedId: operation.document._id
+            });
+        }
+        // Handle update results.
+        if ('update' in operation) {
+            const result = {
+                matchedCount: document.n,
+                modifiedCount: document.nModified ?? 0,
+                // Check if the bulk did actually upsert.
+                didUpsert: document.upserted != null
+            };
+            if (document.upserted) {
+                result.upsertedId = document.upserted._id;
+            }
+            this.result.updateResults?.set(document.idx + this.currentBatchOffset, result);
+        }
+        // Handle delete results.
+        if ('delete' in operation) {
+            this.result.deleteResults?.set(document.idx + this.currentBatchOffset, {
+                deletedCount: document.n
+            });
+        }
+    }
+    /**
+     * Increment the result counts.
+     * @param document - The document with the results.
+     */
+    incrementCounts(document) {
+        this.result.insertedCount += document.insertedCount;
+        this.result.upsertedCount += document.upsertedCount;
+        this.result.matchedCount += document.matchedCount;
+        this.result.modifiedCount += document.modifiedCount;
+        this.result.deletedCount += document.deletedCount;
+    }
+}
+exports.ClientBulkWriteResultsMerger = ClientBulkWriteResultsMerger;
+//# sourceMappingURL=results_merger.js.map

package/lib/operations/client_bulk_write/results_merger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"results_merger.js","sourceRoot":"","sources":["../../../src/operations/client_bulk_write/results_merger.ts"],"names":[],"mappings":";;;AAAA,6BAA+C;AAG/C,uCAAwD;AAUxD;;GAEG;AACH,MAAM,cAAc,GAAG;IACrB,YAAY,EAAE,KAAK;IACnB,aAAa,EAAE,CAAC;IAChB,aAAa,EAAE,CAAC;IAChB,YAAY,EAAE,CAAC;IACf,aAAa,EAAE,CAAC;IAChB,YAAY,EAAE,CAAC;IACf,aAAa,EAAE,SAAS;IACxB,aAAa,EAAE,SAAS;IACxB,aAAa,EAAE,SAAS;CACzB,CAAC;AAyCF;;;GAGG;AACH,MAAa,4BAA4B;IAOvC;;OAEG;IACH,MAAM,CAAC,cAAc;QACnB,OAAO,cAAc,CAAC;IACxB,CAAC;IAED;;;OAGG;IACH,YAAY,OAA+B;QACzC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC;QAC5B,IAAI,CAAC,kBAAkB,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC,WAAW,GAAG,IAAI,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG;YACZ,YAAY,EAAE,IAAI;YAClB,aAAa,EAAE,CAAC;YAChB,aAAa,EAAE,CAAC;YAChB,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,SAAS;YACxB,aAAa,EAAE,SAAS;YACxB,aAAa,EAAE,SAAS;SACzB,CAAC;QAEF,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,IAAI,GAAG,EAAiC,CAAC;YACrE,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,IAAI,GAAG,EAA8B,CAAC;YAClE,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,IAAI,GAAG,EAA8B,CAAC;QACpE,CAAC;IACH,CAAC;IAED;;OAEG;IACH,IAAI,eAAe;QACjB,OAAO;YACL,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;YACtC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;YACxC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;YACxC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;YACtC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;YACxC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;YACtC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;YACxC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;YACxC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;SACzC,CAAC;IACJ,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,KAAK,CAAC,MAA6B;QACvC,IAAI,uBAAuB,CAAC;QAC5B,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,MAAM,EAAE,CAAC;gBACpC,4BAA4B;gBAC5B,IAAI,QAAQ,CAAC,EAAE,KAAK,CAAC,EAAE,CAAC;oBACtB,IAAI,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;wBAChC,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;oBACzC,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,yFAAyF;oBACzF,0FAA0F;oBAC1F,yFAAyF;oBACzF,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;wBACzB,MAAM,KAAK,GAAG,IAAI,iCAAyB,CAAC;4BAC1C,OAAO,EAAE,4DAA4D;yBACtE,CAAC,CAAC;wBACH,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,GAAG,IAAI,CAAC,kBAAkB,EAAE;4BAC5D,IAAI,EAAE,QAAQ,CAAC,IAAI;4BACnB,OAAO,EAAE,QAAQ,CAAC,MAAM;yBACzB,CAAC,CAAC;wBACH,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC;wBAClC,MAAM,KAAK,CAAC;oBACd,CAAC;yBAAM,CAAC;wBACN,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,GAAG,IAAI,CAAC,kBAAkB,EAAE;4BAC3D,IAAI,EAAE,QAAQ,CAAC,IAAI;4BACnB,OAAO,EAAE,QAAQ,CAAC,MAAM;yBACzB,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,0BAAsB,EAAE,CAAC;gBAC5C,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;gBAC5B,uBAAuB,GAAG;oBACxB,aAAa,EAAE,MAAM,CAAC,SAAS;oBAC/B,aAAa,EAAE,MAAM,CAAC,SAAS;oBAC/B,YAAY,EAAE,MAAM,CAAC,QAAQ;oBAC7B,aAAa,EAAE,MAAM,CAAC,SAAS;oBAC/B,YAAY,EAAE,MAAM,CAAC,QAAQ;oBAC7B,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;iBAC5C,CAAC;gBACF,IAAI,IAAI,CAAC,OAAO,CAAC,cAAc,IAAI,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;oBAC5D,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;wBAChD,IAAI,QAAQ,CAAC,EAAE,KAAK,CAAC,EAAE,CAAC;4BACtB,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;wBACzC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,8CAA8C;YAC9C,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACpB,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;gBACjC,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;YACjC,CAAC;YAED,8BAA8B;YAC9B,IAAI,CAAC,kBAAkB,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;QACtD,CAAC;QAED,yDAAyD;QACzD,IAAI,uBAAuB,EAAE,CAAC;YAC5B,MAAM,iBAAiB,GAAG,uBAAuB,CAAC,iBAA6B,CAAC;YAChF,IAAI,CAAC,eAAe,CAAC,uBAAuB,CAAC,CAAC;YAC9C,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC;gBAC3B,IAAI,EAAE,iBAAiB,CAAC,IAAI;gBAC5B,OAAO,EAAE,iBAAiB,CAAC,MAAM;aAClC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;;;OAIG;IACK,eAAe,CAAC,MAA6B,EAAE,QAAkB;QACvE,oDAAoD;QACpD,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;QAClD,yBAAyB;QACzB,IAAI,QAAQ,IAAI,SAAS,EAAE,CAAC;YAC1B,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,QAAQ,CAAC,GAAG,GAAG,IAAI,CAAC,kBAAkB,EAAE;gBACrE,UAAU,EAAE,SAAS,CAAC,QAAQ,CAAC,GAAG;aACnC,CAAC,CAAC;QACL,CAAC;QACD,yBAAyB;QACzB,IAAI,QAAQ,IAAI,SAAS,EAAE,CAAC;YAC1B,MAAM,MAAM,GAAuB;gBACjC,YAAY,EAAE,QAAQ,CAAC,CAAC;gBACxB,aAAa,EAAE,QAAQ,CAAC,SAAS,IAAI,CAAC;gBACtC,yCAAyC;gBACzC,SAAS,EAAE,QAAQ,CAAC,QAAQ,IAAI,IAAI;aACrC,CAAC;YACF,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBACtB,MAAM,CAAC,UAAU,GAAG,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC;YAC5C,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,QAAQ,CAAC,GAAG,GAAG,IAAI,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC;QACjF,CAAC;QACD,yBAAyB;QACzB,IAAI,QAAQ,IAAI,SAAS,EAAE,CAAC;YAC1B,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,QAAQ,CAAC,GAAG,GAAG,IAAI,CAAC,kBAAkB,EAAE;gBACrE,YAAY,EAAE,QAAQ,CAAC,CAAC;aACzB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,eAAe,CAAC,QAAkB;QACxC,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,QAAQ,CAAC,aAAa,CAAC;QACpD,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,QAAQ,CAAC,aAAa,CAAC;QACpD,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,QAAQ,CAAC,YAAY,CAAC;QAClD,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,QAAQ,CAAC,aAAa,CAAC;QACpD,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,QAAQ,CAAC,YAAY,CAAC;IACpD,CAAC;CACF;AA5LD,oEA4LC"}

package/lib/operations/command.js

@@ -0,0 +1,83 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CommandOperation = void 0;
+const constants_1 = require("../cmap/wire_protocol/constants");
+const error_1 = require("../error");
+const explain_1 = require("../explain");
+const read_concern_1 = require("../read_concern");
+const utils_1 = require("../utils");
+const write_concern_1 = require("../write_concern");
+const operation_1 = require("./operation");
+/** @internal */
+class CommandOperation extends operation_1.AbstractOperation {
+    constructor(parent, options) {
+        super(options);
+        this.options = options ?? {};
+        // NOTE: this was explicitly added for the add/remove user operations, it's likely
+        //       something we'd want to reconsider. Perhaps those commands can use `Admin`
+        //       as a parent?
+        const dbNameOverride = options?.dbName || options?.authdb;
+        if (dbNameOverride) {
+            this.ns = new utils_1.MongoDBNamespace(dbNameOverride, '$cmd');
+        }
+        else {
+            this.ns = parent
+                ? parent.s.namespace.withCollection('$cmd')
+                : new utils_1.MongoDBNamespace('admin', '$cmd');
+        }
+        this.readConcern = read_concern_1.ReadConcern.fromOptions(options);
+        this.writeConcern = write_concern_1.WriteConcern.fromOptions(options);
+        if (this.hasAspect(operation_1.Aspect.EXPLAINABLE)) {
+            this.explain = explain_1.Explain.fromOptions(options);
+            if (this.explain)
+                (0, explain_1.validateExplainTimeoutOptions)(this.options, this.explain);
+        }
+        else if (options?.explain != null) {
+            throw new error_1.MongoInvalidArgumentError(`Option "explain" is not supported on this command`);
+        }
+    }
+    get canRetryWrite() {
+        if (this.hasAspect(operation_1.Aspect.EXPLAINABLE)) {
+            return this.explain == null;
+        }
+        return super.canRetryWrite;
+    }
+    buildOptions(timeoutContext) {
+        return {
+            ...this.options,
+            ...this.bsonOptions,
+            timeoutContext,
+            readPreference: this.readPreference,
+            session: this.session
+        };
+    }
+    buildCommand(connection, session) {
+        const command = this.buildCommandDocument(connection, session);
+        const inTransaction = this.session && this.session.inTransaction();
+        if (this.readConcern && (0, utils_1.commandSupportsReadConcern)(command) && !inTransaction) {
+            Object.assign(command, { readConcern: this.readConcern });
+        }
+        if (this.writeConcern && this.hasAspect(operation_1.Aspect.WRITE_OPERATION) && !inTransaction) {
+            write_concern_1.WriteConcern.apply(command, this.writeConcern);
+        }
+        if (this.options.collation &&
+            typeof this.options.collation === 'object' &&
+            !this.hasAspect(operation_1.Aspect.SKIP_COLLATION)) {
+            Object.assign(command, { collation: this.options.collation });
+        }
+        if (typeof this.options.maxTimeMS === 'number') {
+            command.maxTimeMS = this.options.maxTimeMS;
+        }
+        if (this.options.rawData != null &&
+            this.hasAspect(operation_1.Aspect.SUPPORTS_RAW_DATA) &&
+            (0, utils_1.maxWireVersion)(connection) >= constants_1.MIN_SUPPORTED_RAW_DATA_WIRE_VERSION) {
+            command.rawData = this.options.rawData;
+        }
+        if (this.hasAspect(operation_1.Aspect.EXPLAINABLE) && this.explain) {
+            return (0, explain_1.decorateWithExplain)(command, this.explain);
+        }
+        return command;
+    }
+}
+exports.CommandOperation = CommandOperation;
+//# sourceMappingURL=command.js.map

package/lib/operations/command.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"command.js","sourceRoot":"","sources":["../../src/operations/command.ts"],"names":[],"mappings":";;;AAEA,+DAAsF;AACtF,oCAAqD;AACrD,wCAKoB;AACpB,kDAA8C;AAK9C,oCAAwF;AACxF,oDAA0E;AAE1E,2CAA+E;AA4D/E,gBAAgB;AAChB,MAAsB,gBAAoB,SAAQ,6BAAoB;IAMpE,YAAY,MAAwB,EAAE,OAAiC;QACrE,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,EAAE,CAAC;QAE7B,kFAAkF;QAClF,kFAAkF;QAClF,qBAAqB;QACrB,MAAM,cAAc,GAAG,OAAO,EAAE,MAAM,IAAI,OAAO,EAAE,MAAM,CAAC;QAC1D,IAAI,cAAc,EAAE,CAAC;YACnB,IAAI,CAAC,EAAE,GAAG,IAAI,wBAAgB,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QACzD,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,EAAE,GAAG,MAAM;gBACd,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,MAAM,CAAC;gBAC3C,CAAC,CAAC,IAAI,wBAAgB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC5C,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,0BAAW,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QACpD,IAAI,CAAC,YAAY,GAAG,4BAAY,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAEtD,IAAI,IAAI,CAAC,SAAS,CAAC,kBAAM,CAAC,WAAW,CAAC,EAAE,CAAC;YACvC,IAAI,CAAC,OAAO,GAAG,iBAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAC5C,IAAI,IAAI,CAAC,OAAO;gBAAE,IAAA,uCAA6B,EAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9E,CAAC;aAAM,IAAI,OAAO,EAAE,OAAO,IAAI,IAAI,EAAE,CAAC;YACpC,MAAM,IAAI,iCAAyB,CAAC,mDAAmD,CAAC,CAAC;QAC3F,CAAC;IACH,CAAC;IAED,IAAa,aAAa;QACxB,IAAI,IAAI,CAAC,SAAS,CAAC,kBAAM,CAAC,WAAW,CAAC,EAAE,CAAC;YACvC,OAAO,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC;QAC9B,CAAC;QACD,OAAO,KAAK,CAAC,aAAa,CAAC;IAC7B,CAAC;IAIQ,YAAY,CAAC,cAA8B;QAClD,OAAO;YACL,GAAG,IAAI,CAAC,OAAO;YACf,GAAG,IAAI,CAAC,WAAW;YACnB,cAAc;YACd,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC;IAEQ,YAAY,CAAC,UAAsB,EAAE,OAAuB;QACnE,MAAM,OAAO,GAAG,IAAI,CAAC,oBAAoB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAE/D,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;QAEnE,IAAI,IAAI,CAAC,WAAW,IAAI,IAAA,kCAA0B,EAAC,OAAO,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YAC9E,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QAC5D,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,SAAS,CAAC,kBAAM,CAAC,eAAe,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YAClF,4BAAY,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACjD,CAAC;QAED,IACE,IAAI,CAAC,OAAO,CAAC,SAAS;YACtB,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,KAAK,QAAQ;YAC1C,CAAC,IAAI,CAAC,SAAS,CAAC,kBAAM,CAAC,cAAc,CAAC,EACtC,CAAC;YACD,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;QAChE,CAAC;QAED,IAAI,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;YAC/C,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC;QAC7C,CAAC;QAED,IACE,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,IAAI;YAC5B,IAAI,CAAC,SAAS,CAAC,kBAAM,CAAC,iBAAiB,CAAC;YACxC,IAAA,sBAAc,EAAC,UAAU,CAAC,IAAI,+CAAmC,EACjE,CAAC;YACD,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC;QACzC,CAAC;QAED,IAAI,IAAI,CAAC,SAAS,CAAC,kBAAM,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACvD,OAAO,IAAA,6BAAmB,EAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACpD,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AA3FD,4CA2FC"}

package/lib/operations/count.js

@@ -0,0 +1,45 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CountOperation = void 0;
+const responses_1 = require("../cmap/wire_protocol/responses");
+const command_1 = require("./command");
+const operation_1 = require("./operation");
+/** @internal */
+class CountOperation extends command_1.CommandOperation {
+    constructor(namespace, filter, options) {
+        super({ s: { namespace: namespace } }, options);
+        this.SERVER_COMMAND_RESPONSE_TYPE = responses_1.MongoDBResponse;
+        this.options = options;
+        this.collectionName = namespace.collection;
+        this.query = filter;
+    }
+    get commandName() {
+        return 'count';
+    }
+    buildCommandDocument(_connection, _session) {
+        const options = this.options;
+        const cmd = {
+            count: this.collectionName,
+            query: this.query
+        };
+        if (typeof options.limit === 'number') {
+            cmd.limit = options.limit;
+        }
+        if (typeof options.skip === 'number') {
+            cmd.skip = options.skip;
+        }
+        if (options.hint != null) {
+            cmd.hint = options.hint;
+        }
+        if (typeof options.maxTimeMS === 'number') {
+            cmd.maxTimeMS = options.maxTimeMS;
+        }
+        return cmd;
+    }
+    handleOk(response) {
+        return response.getNumber('n') ?? 0;
+    }
+}
+exports.CountOperation = CountOperation;
+(0, operation_1.defineAspects)(CountOperation, [operation_1.Aspect.READ_OPERATION, operation_1.Aspect.RETRYABLE, operation_1.Aspect.SUPPORTS_RAW_DATA]);
+//# sourceMappingURL=count.js.map

package/lib/operations/count.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"count.js","sourceRoot":"","sources":["../../src/operations/count.ts"],"names":[],"mappings":";;;AAEA,+DAAkE;AAIlE,uCAA2E;AAC3E,2CAAoD;AAgBpD,gBAAgB;AAChB,MAAa,cAAe,SAAQ,0BAAwB;IAM1D,YAAY,SAA2B,EAAE,MAAgB,EAAE,OAAqB;QAC9E,KAAK,CAAC,EAAE,CAAC,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,EAA2B,EAAE,OAAO,CAAC,CAAC;QANlE,iCAA4B,GAAG,2BAAe,CAAC;QAQtD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC,UAAU,CAAC;QAC3C,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC;IACtB,CAAC;IAED,IAAa,WAAW;QACtB,OAAO,OAAgB,CAAC;IAC1B,CAAC;IAEQ,oBAAoB,CAAC,WAAuB,EAAE,QAAwB;QAC7E,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,MAAM,GAAG,GAAa;YACpB,KAAK,EAAE,IAAI,CAAC,cAAc;YAC1B,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC;QAEF,IAAI,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;YACtC,GAAG,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC5B,CAAC;QAED,IAAI,OAAO,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACrC,GAAG,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QAC1B,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC;YACzB,GAAG,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QAC1B,CAAC;QAED,IAAI,OAAO,OAAO,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;YAC1C,GAAG,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACpC,CAAC;QAED,OAAO,GAAG,CAAC;IACb,CAAC;IAEQ,QAAQ,CAAC,QAAgE;QAChF,OAAO,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC;CACF;AA/CD,wCA+CC;AAED,IAAA,yBAAa,EAAC,cAAc,EAAE,CAAC,kBAAM,CAAC,cAAc,EAAE,kBAAM,CAAC,SAAS,EAAE,kBAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC"}

package/lib/operations/create_collection.js

@@ -0,0 +1,109 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CreateCollectionOperation = void 0;
+exports.createCollections = createCollections;
+const constants_1 = require("../cmap/wire_protocol/constants");
+const responses_1 = require("../cmap/wire_protocol/responses");
+const collection_1 = require("../collection");
+const error_1 = require("../error");
+const timeout_1 = require("../timeout");
+const utils_1 = require("../utils");
+const command_1 = require("./command");
+const execute_operation_1 = require("./execute_operation");
+const indexes_1 = require("./indexes");
+const operation_1 = require("./operation");
+const ILLEGAL_COMMAND_FIELDS = new Set([
+    'w',
+    'wtimeout',
+    'timeoutMS',
+    'j',
+    'fsync',
+    'pkFactory',
+    'raw',
+    'readPreference',
+    'session',
+    'readConcern',
+    'writeConcern',
+    'raw',
+    'fieldsAsRaw',
+    'useBigInt64',
+    'promoteLongs',
+    'promoteValues',
+    'promoteBuffers',
+    'bsonRegExp',
+    'serializeFunctions',
+    'ignoreUndefined',
+    'enableUtf8Validation'
+]);
+/* @internal */
+const INVALID_QE_VERSION = 'Driver support of Queryable Encryption is incompatible with server. Upgrade server to use Queryable Encryption.';
+/** @internal */
+class CreateCollectionOperation extends command_1.CommandOperation {
+    constructor(db, name, options = {}) {
+        super(db, options);
+        this.SERVER_COMMAND_RESPONSE_TYPE = responses_1.MongoDBResponse;
+        this.options = options;
+        this.db = db;
+        this.name = name;
+    }
+    get commandName() {
+        return 'create';
+    }
+    buildCommandDocument(_connection, _session) {
+        const isOptionValid = ([k, v]) => v != null && typeof v !== 'function' && !ILLEGAL_COMMAND_FIELDS.has(k);
+        return {
+            create: this.name,
+            ...Object.fromEntries(Object.entries(this.options).filter(isOptionValid))
+        };
+    }
+    handleOk(_response) {
+        return new collection_1.Collection(this.db, this.name, this.options);
+    }
+}
+exports.CreateCollectionOperation = CreateCollectionOperation;
+async function createCollections(db, name, options) {
+    const timeoutContext = timeout_1.TimeoutContext.create({
+        session: options.session,
+        serverSelectionTimeoutMS: db.client.s.options.serverSelectionTimeoutMS,
+        waitQueueTimeoutMS: db.client.s.options.waitQueueTimeoutMS,
+        timeoutMS: options.timeoutMS
+    });
+    const encryptedFields = options.encryptedFields ??
+        db.client.s.options.autoEncryption?.encryptedFieldsMap?.[`${db.databaseName}.${name}`];
+    if (encryptedFields) {
+        class CreateSupportingFLEv2CollectionOperation extends CreateCollectionOperation {
+            buildCommandDocument(connection, session) {
+                if (!connection.description.loadBalanced &&
+                    (0, utils_1.maxWireVersion)(connection) < constants_1.MIN_SUPPORTED_QE_WIRE_VERSION) {
+                    throw new error_1.MongoCompatibilityError(`${INVALID_QE_VERSION} The minimum server version required is ${constants_1.MIN_SUPPORTED_QE_SERVER_VERSION}`);
+                }
+                return super.buildCommandDocument(connection, session);
+            }
+        }
+        // Create auxilliary collections for queryable encryption support.
+        const escCollection = encryptedFields.escCollection ?? `enxcol_.${name}.esc`;
+        const ecocCollection = encryptedFields.ecocCollection ?? `enxcol_.${name}.ecoc`;
+        for (const collectionName of [escCollection, ecocCollection]) {
+            const createOp = new CreateSupportingFLEv2CollectionOperation(db, collectionName, {
+                clusteredIndex: {
+                    key: { _id: 1 },
+                    unique: true
+                },
+                session: options.session
+            });
+            await (0, execute_operation_1.executeOperation)(db.client, createOp, timeoutContext);
+        }
+        if (!options.encryptedFields) {
+            options = { ...options, encryptedFields };
+        }
+    }
+    const coll = await (0, execute_operation_1.executeOperation)(db.client, new CreateCollectionOperation(db, name, options), timeoutContext);
+    if (encryptedFields) {
+        // Create the required index for queryable encryption support.
+        const createIndexOp = indexes_1.CreateIndexesOperation.fromIndexSpecification(db, name, { __safeContent__: 1 }, { session: options.session });
+        await (0, execute_operation_1.executeOperation)(db.client, createIndexOp, timeoutContext);
+    }
+    return coll;
+}
+(0, operation_1.defineAspects)(CreateCollectionOperation, [operation_1.Aspect.WRITE_OPERATION]);
+//# sourceMappingURL=create_collection.js.map

package/lib/operations/create_collection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"create_collection.js","sourceRoot":"","sources":["../../src/operations/create_collection.ts"],"names":[],"mappings":";;;AAiJA,8CAsEC;AArND,+DAGyC;AACzC,+DAAkE;AAClE,8CAA2C;AAE3C,oCAAmD;AAGnD,wCAA4C;AAC5C,oCAA0C;AAC1C,uCAA2E;AAC3E,2DAAuD;AACvD,uCAAmD;AACnD,2CAAoD;AAEpD,MAAM,sBAAsB,GAAG,IAAI,GAAG,CAAC;IACrC,GAAG;IACH,UAAU;IACV,WAAW;IACX,GAAG;IACH,OAAO;IACP,WAAW;IACX,KAAK;IACL,gBAAgB;IAChB,SAAS;IACT,aAAa;IACb,cAAc;IACd,KAAK;IACL,aAAa;IACb,aAAa;IACb,cAAc;IACd,eAAe;IACf,gBAAgB;IAChB,YAAY;IACZ,oBAAoB;IACpB,iBAAiB;IACjB,sBAAsB;CACvB,CAAC,CAAC;AAiEH,eAAe;AACf,MAAM,kBAAkB,GACtB,iHAAiH,CAAC;AAEpH,gBAAgB;AAChB,MAAa,yBAA0B,SAAQ,0BAA4B;IAMzE,YAAY,EAAM,EAAE,IAAY,EAAE,UAAmC,EAAE;QACrE,KAAK,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QANZ,iCAA4B,GAAG,2BAAe,CAAC;QAQtD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,IAAa,WAAW;QACtB,OAAO,QAAiB,CAAC;IAC3B,CAAC;IAEQ,oBAAoB,CAAC,WAAuB,EAAE,QAAwB;QAC7E,MAAM,aAAa,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAA0B,EAAE,EAAE,CACxD,CAAC,IAAI,IAAI,IAAI,OAAO,CAAC,KAAK,UAAU,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACzE,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,IAAI;YACjB,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;SAC1E,CAAC;IACJ,CAAC;IAEQ,QAAQ,CACf,SAAiE;QAEjE,OAAO,IAAI,uBAAU,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1D,CAAC;CACF;AAhCD,8DAgCC;AAEM,KAAK,UAAU,iBAAiB,CACrC,EAAM,EACN,IAAY,EACZ,OAAgC;IAEhC,MAAM,cAAc,GAAG,wBAAc,CAAC,MAAM,CAAC;QAC3C,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,wBAAwB,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,wBAAwB;QACtE,kBAAkB,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,kBAAkB;QAC1D,SAAS,EAAE,OAAO,CAAC,SAAS;KAC7B,CAAC,CAAC;IAEH,MAAM,eAAe,GACnB,OAAO,CAAC,eAAe;QACvB,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,EAAE,kBAAkB,EAAE,CAAC,GAAG,EAAE,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC,CAAC;IAEzF,IAAI,eAAe,EAAE,CAAC;QACpB,MAAM,wCAAyC,SAAQ,yBAAyB;YACrE,oBAAoB,CAAC,UAAsB,EAAE,OAAuB;gBAC3E,IACE,CAAC,UAAU,CAAC,WAAW,CAAC,YAAY;oBACpC,IAAA,sBAAc,EAAC,UAAU,CAAC,GAAG,yCAA6B,EAC1D,CAAC;oBACD,MAAM,IAAI,+BAAuB,CAC/B,GAAG,kBAAkB,2CAA2C,2CAA+B,EAAE,CAClG,CAAC;gBACJ,CAAC;gBAED,OAAO,KAAK,CAAC,oBAAoB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YACzD,CAAC;SACF;QAED,kEAAkE;QAClE,MAAM,aAAa,GAAG,eAAe,CAAC,aAAa,IAAI,WAAW,IAAI,MAAM,CAAC;QAC7E,MAAM,cAAc,GAAG,eAAe,CAAC,cAAc,IAAI,WAAW,IAAI,OAAO,CAAC;QAEhF,KAAK,MAAM,cAAc,IAAI,CAAC,aAAa,EAAE,cAAc,CAAC,EAAE,CAAC;YAC7D,MAAM,QAAQ,GAAG,IAAI,wCAAwC,CAAC,EAAE,EAAE,cAAc,EAAE;gBAChF,cAAc,EAAE;oBACd,GAAG,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE;oBACf,MAAM,EAAE,IAAI;iBACb;gBACD,OAAO,EAAE,OAAO,CAAC,OAAO;aACzB,CAAC,CAAC;YACH,MAAM,IAAA,oCAAgB,EAAC,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,cAAc,CAAC,CAAC;QAC9D,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC;YAC7B,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,eAAe,EAAE,CAAC;QAC5C,CAAC;IACH,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,IAAA,oCAAgB,EACjC,EAAE,CAAC,MAAM,EACT,IAAI,yBAAyB,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,EAChD,cAAc,CACf,CAAC;IAEF,IAAI,eAAe,EAAE,CAAC;QACpB,8DAA8D;QAC9D,MAAM,aAAa,GAAG,gCAAsB,CAAC,sBAAsB,CACjE,EAAE,EACF,IAAI,EACJ,EAAE,eAAe,EAAE,CAAC,EAAE,EACtB,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAC7B,CAAC;QACF,MAAM,IAAA,oCAAgB,EAAC,EAAE,CAAC,MAAM,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC;IACnE,CAAC;IAED,OAAO,IAAsC,CAAC;AAChD,CAAC;AAED,IAAA,yBAAa,EAAC,yBAAyB,EAAE,CAAC,kBAAM,CAAC,eAAe,CAAC,CAAC,CAAC"}