chore: fix wc throw location

Author: nbbeeken

src/client-side-encryption/client_encryption.ts

@@ -719,8 +719,8 @@ export class ClientEncryption {
     });
     const context = this._mongoCrypt.makeExplicitEncryptionContext(valueBuffer, contextOptions);
 
-    const result = deserialize(await stateMachine.execute(this, context));
-    return result.v;
+    const { v } = deserialize(await stateMachine.execute(this, context));
+    return v;
   }
 }
 

src/client-side-encryption/state_machine.ts

@@ -112,8 +112,25 @@ export type CSFLEKMSTlsOptions = {
   azure?: ClientEncryptionTlsOptions;
 };
 
-/** `{ v: [] }` */
-const EMPTY_V = Uint8Array.from([13, 0, 0, 0, 4, 118, 0, 5, 0, 0, 0, 0, 0]);
+/**
+ * This is kind of a hack.  For `rewrapManyDataKey`, we have tests that
+ * guarantee that when there are no matching keys, `rewrapManyDataKey` returns
+ * nothing.  We also have tests for auto encryption that guarantee for `encrypt`
+ * we return an error when there are no matching keys.  This error is generated in
+ * subsequent iterations of the state machine.
+ * Some apis (`encrypt`) throw if there are no filter matches and others (`rewrapManyDataKey`)
+ * do not.  We set the result manually here, and let the state machine continue.  `libmongocrypt`
+ * will inform us if we need to error by setting the state to `MONGOCRYPT_CTX_ERROR` but
+ * otherwise we'll return `{ v: [] }`.
+ */
+const EMPTY_V = Uint8Array.from([
+  ...[13, 0, 0, 0], // document size = 13 bytes
+  ...[
+    ...[4, 118, 0], // array type (4), "v\x00" basic latin "v"
+    ...[5, 0, 0, 0, 0] // empty document (5 byte size, null terminator)
+  ],
+  0 // null terminator
+]);
 
 /**
  * @internal
@@ -211,15 +228,7 @@ export class StateMachine {
           const keys = await this.fetchKeys(keyVaultClient, keyVaultNamespace, filter);
 
           if (keys.length === 0) {
-            // This is kind of a hack.  For `rewrapManyDataKey`, we have tests that
-            // guarantee that when there are no matching keys, `rewrapManyDataKey` returns
-            // nothing.  We also have tests for auto encryption that guarantee for `encrypt`
-            // we return an error when there are no matching keys.  This error is generated in
-            // subsequent iterations of the state machine.
-            // Some apis (`encrypt`) throw if there are no filter matches and others (`rewrapManyDataKey`)
-            // do not.  We set the result manually here, and let the state machine continue.  `libmongocrypt`
-            // will inform us if we need to error by setting the state to `MONGOCRYPT_CTX_ERROR` but
-            // otherwise we'll return `{ v: [] }`.
+            // See docs on EMPTY_V
             result = EMPTY_V;
           }
           for await (const key of keys) {

src/cmap/wire_protocol/responses.ts

@@ -66,6 +66,14 @@ export type MongoDBResponseConstructor = {
 
 /** @internal */
 export class MongoDBResponse extends OnDemandDocument {
+  /**
+   * Devtools need to know which keys were encrypted before the driver automatically decrypted them.
+   * If decorating is enabled (`Symbol.for('@@mdb.decorateDecryptionResult')`), this field will be set,
+   * storing the original encrypted response from the server, so that we can build an object that has
+   * the list of BSON keys that were encrypted stored at a well known symbol: `Symbol.for('@@mdb.decryptedKeys')`.
+   */
+  encryptedResponse?: MongoDBResponse;
+
   static is(value: unknown): value is MongoDBResponse {
     return value instanceof MongoDBResponse;
   }
@@ -161,13 +169,11 @@ export class MongoDBResponse extends OnDemandDocument {
     }
     return { utf8: { writeErrors: false } };
   }
-
-  // TODO: Supports decorating result
-  encryptedResponse?: MongoDBResponse;
 }
 
 // Here's a little blast from the past.
 // OLD style method definition so that I can override get without redefining ALL the fancy TS :/
+// TODO there must be a better way...
 Object.defineProperty(MongoDBResponse.prototype, 'get', {
   value: function get(name: any, as: any, required: any) {
     try {

src/cursor/abstract_cursor.ts

@@ -294,8 +294,7 @@ export abstract class AbstractCursor<
 
     return bufferedDocs;
   }
-
-  private async *asyncIterator() {
+  async *[Symbol.asyncIterator](): AsyncGenerator<TSchema, void, void> {
     if (this.closed) {
       return;
     }
@@ -343,10 +342,6 @@ export abstract class AbstractCursor<
     }
   }
 
-  async *[Symbol.asyncIterator](): AsyncGenerator<TSchema, void, void> {
-    yield* this.asyncIterator();
-  }
-
   stream(options?: CursorStreamOptions): Readable & AsyncIterable<TSchema> {
     if (options?.transform) {
       const transform = options.transform;

src/error.ts

@@ -1177,17 +1177,16 @@ export class MongoWriteConcernError extends MongoServerError {
    *
    * @public
    **/
-  constructor(
-    result: {
-      writeConcernError: {
-        code: number;
-        errmsg: string;
-        codeName?: string;
-        errInfo?: Document;
-      };
-    } & Document
-  ) {
-    super(result.writeConcernError);
+  constructor(result: {
+    writeConcernError: {
+      code: number;
+      errmsg: string;
+      codeName?: string;
+      errInfo?: Document;
+    };
+    errorLabels?: string[];
+  }) {
+    super({ ...result, ...result.writeConcernError });
     this.errInfo = result.writeConcernError.errInfo;
     this.result = result;
   }

src/operations/bulk_write.ts

@@ -7,7 +7,6 @@ import type {
 import type { Collection } from '../collection';
 import type { Server } from '../sdam/server';
 import type { ClientSession } from '../sessions';
-import { throwIfWriteConcernError } from '../utils';
 import { AbstractOperation, Aspect, defineAspects } from './operation';
 
 /** @internal */
@@ -51,9 +50,7 @@ export class BulkWriteOperation extends AbstractOperation<BulkWriteResult> {
     }
 
     // Execute the bulk
-    const result = await bulk.execute({ ...options, session });
-    throwIfWriteConcernError(result);
-    return result;
+    return await bulk.execute({ ...options, session });
   }
 }
 

src/operations/delete.ts

@@ -4,8 +4,8 @@ import { MongoCompatibilityError, MongoServerError } from '../error';
 import { type TODO_NODE_3286 } from '../mongo_types';
 import type { Server } from '../sdam/server';
 import type { ClientSession } from '../sessions';
-import { type MongoDBNamespace, throwIfWriteConcernError } from '../utils';
-import type { WriteConcernOptions } from '../write_concern';
+import { type MongoDBNamespace } from '../utils';
+import { type WriteConcernOptions } from '../write_concern';
 import { type CollationOptions, CommandOperation, type CommandOperationOptions } from './command';
 import { Aspect, defineAspects, type Hint } from './operation';
 
@@ -96,7 +96,6 @@ export class DeleteOperation extends CommandOperation<DeleteResult> {
     }
 
     const res: TODO_NODE_3286 = await super.executeCommand(server, session, command);
-    throwIfWriteConcernError(res);
     return res;
   }
 }

src/operations/find_and_modify.ts

@@ -5,13 +5,8 @@ import { ReadPreference } from '../read_preference';
 import type { Server } from '../sdam/server';
 import type { ClientSession } from '../sessions';
 import { formatSort, type Sort, type SortForCmd } from '../sort';
-import {
-  decorateWithCollation,
-  hasAtomicOperators,
-  maxWireVersion,
-  throwIfWriteConcernError
-} from '../utils';
-import type { WriteConcern, WriteConcernSettings } from '../write_concern';
+import { decorateWithCollation, hasAtomicOperators, maxWireVersion } from '../utils';
+import { type WriteConcern, type WriteConcernSettings } from '../write_concern';
 import { CommandOperation, type CommandOperationOptions } from './command';
 import { Aspect, defineAspects } from './operation';
 
@@ -219,7 +214,6 @@ export class FindAndModifyOperation extends CommandOperation<Document> {
 
     // Execute the command
     const result = await super.executeCommand(server, session, cmd);
-    throwIfWriteConcernError(result);
     return options.includeResultMetadata ? result : result.value ?? null;
   }
 }

src/operations/update.ts

@@ -4,7 +4,7 @@ import { MongoCompatibilityError, MongoInvalidArgumentError, MongoServerError }
 import type { InferIdType, TODO_NODE_3286 } from '../mongo_types';
 import type { Server } from '../sdam/server';
 import type { ClientSession } from '../sessions';
-import { hasAtomicOperators, type MongoDBNamespace, throwIfWriteConcernError } from '../utils';
+import { hasAtomicOperators, type MongoDBNamespace } from '../utils';
 import { type CollationOptions, CommandOperation, type CommandOperationOptions } from './command';
 import { Aspect, defineAspects, type Hint } from './operation';
 
@@ -123,7 +123,6 @@ export class UpdateOperation extends CommandOperation<Document> {
     }
 
     const res = await super.executeCommand(server, session, command);
-    throwIfWriteConcernError(res);
     return res;
   }
 }

src/sdam/server.ts

@@ -46,9 +46,9 @@ import {
   makeStateMachine,
   maxWireVersion,
   type MongoDBNamespace,
-  supportsRetryableWrites,
-  throwIfWriteConcernError
+  supportsRetryableWrites
 } from '../utils';
+import { throwIfWriteConcernError } from '../write_concern';
 import {
   type ClusterTime,
   STATE_CLOSED,
@@ -337,7 +337,9 @@ export class Server extends TypedEventEmitter<ServerEvents> {
       ) {
         await this.pool.reauthenticate(conn);
         try {
-          return await conn.command(ns, cmd, finalOptions, responseType);
+          const res = await conn.command(ns, cmd, finalOptions, responseType);
+          throwIfWriteConcernError(res);
+          return res;
         } catch (commandError) {
           throw this.decorateCommandError(conn, cmd, finalOptions, commandError);
         }