mongodb 6.5.0-dev.20240426.sha.6d8ad33 → 6.5.0-dev.20240502.sha.9d73f45

package/src/sdam/server.ts

@@ -7,6 +7,7 @@ import {
   type ConnectionPoolOptions
 } from '../cmap/connection_pool';
 import { PoolClearedError } from '../cmap/errors';
+import { type MongoDBResponseConstructor } from '../cmap/wire_protocol/responses';
 import {
   APM_EVENTS,
   CLOSED,
@@ -262,11 +263,25 @@ export class Server extends TypedEventEmitter<ServerEvents> {
     }
   }
 
-  /**
-   * Execute a command
-   * @internal
-   */
-  async command(ns: MongoDBNamespace, cmd: Document, options: CommandOptions): Promise<Document> {
+  public async command<T extends MongoDBResponseConstructor>(
+    ns: MongoDBNamespace,
+    command: Document,
+    options: CommandOptions | undefined,
+    responseType: T | undefined
+  ): Promise<typeof responseType extends undefined ? Document : InstanceType<T>>;
+
+  public async command(
+    ns: MongoDBNamespace,
+    command: Document,
+    options?: CommandOptions
+  ): Promise<Document>;
+
+  public async command(
+    ns: MongoDBNamespace,
+    cmd: Document,
+    options: CommandOptions,
+    responseType?: MongoDBResponseConstructor
+  ): Promise<Document> {
     if (ns.db == null || typeof ns === 'string') {
       throw new MongoInvalidArgumentError('Namespace must not be a string');
     }
@@ -308,7 +323,7 @@ export class Server extends TypedEventEmitter<ServerEvents> {
 
     try {
       try {
-        return await conn.command(ns, cmd, finalOptions);
+        return await conn.command(ns, cmd, finalOptions, responseType);
       } catch (commandError) {
         throw this.decorateCommandError(conn, cmd, finalOptions, commandError);
       }
@@ -319,7 +334,7 @@ export class Server extends TypedEventEmitter<ServerEvents> {
       ) {
         await this.pool.reauthenticate(conn);
         try {
-          return await conn.command(ns, cmd, finalOptions);
+          return await conn.command(ns, cmd, finalOptions, responseType);
         } catch (commandError) {
           throw this.decorateCommandError(conn, cmd, finalOptions, commandError);
         }

package/src/utils.ts

@@ -64,6 +64,19 @@ export const ByteUtils = {
   }
 };
 
+/**
+ * Returns true if value is a Uint8Array or a Buffer
+ * @param value - any value that may be a Uint8Array
+ */
+export function isUint8Array(value: unknown): value is Uint8Array {
+  return (
+    value != null &&
+    typeof value === 'object' &&
+    Symbol.toStringTag in value &&
+    value[Symbol.toStringTag] === 'Uint8Array'
+  );
+}
+
 /**
  * Determines if a connection's address matches a user provided list
  * of domain wildcards.