chore: add missing classes from top-level

Author: nbbeeken

src/index.js

@@ -12,13 +12,19 @@ const {
   makeLegacyListCollectionsCursor,
   makeLegacyListIndexesCursor
 } = require('./legacy_wrappers/cursors');
-const { makeLegacyGridFSBucket } = require('./legacy_wrappers/gridfs');
+const {
+  makeLegacyGridFSBucket,
+  makeLegacyGridFSBucketWriteStream
+} = require('./legacy_wrappers/gridfs');
 const { makeLegacyChangeStream } = require('./legacy_wrappers/change_stream');
 const { makeLegacyClientSession } = require('./legacy_wrappers/session');
+const {
+  makeLegacyUnorderedBulkOperation,
+  makeLegacyOrderedBulkOperation
+} = require('./legacy_wrappers/bulk');
 
 /** @type {import('..')} */
 module.exports = Object.create(null);
-
 Object.defineProperty(module.exports, '__esModule', { value: true });
 
 const classesWithAsyncAPIs = new Map([
@@ -32,9 +38,26 @@ const classesWithAsyncAPIs = new Map([
   ['Db', makeLegacyDb],
   ['GridFSBucket', makeLegacyGridFSBucket],
   ['ClientSession', makeLegacyClientSession],
-  ['MongoClient', makeLegacyMongoClient]
+  ['MongoClient', makeLegacyMongoClient],
+
+  // Need to be exported top-level still
+  ['ClientSession', makeLegacyClientSession],
+  ['GridFSBucketWriteStream', makeLegacyGridFSBucketWriteStream],
+  ['OrderedBulkOperation', makeLegacyOrderedBulkOperation],
+  ['UnorderedBulkOperation', makeLegacyUnorderedBulkOperation]
 ]);
 
+const TODO_SPECIAL_IMPORTS = new Map([
+  ['ClientSession', '/lib/sessions'],
+  ['GridFSBucketWriteStream', '/lib/gridfs/upload'],
+  ['OrderedBulkOperation', '/lib/bulk/ordered'],
+  ['UnorderedBulkOperation', '/lib/bulk/unordered']
+]);
+
+for (const [missingTopLevelClassName, location] of TODO_SPECIAL_IMPORTS) {
+  mongodb[missingTopLevelClassName] = require(`mongodb${location}`)[missingTopLevelClassName];
+}
+
 for (const [mongodbExportName, mongodbExportValue] of Object.entries(mongodb)) {
   let makeLegacyClass = classesWithAsyncAPIs.get(mongodbExportName);
   if (makeLegacyClass != null) {

src/legacy_wrappers/admin.js

@@ -3,6 +3,7 @@
 const { toLegacy, maybeCallback } = require('../utils');
 
 module.exports = Object.create(null);
+Object.defineProperty(module.exports, '__esModule', { value: true });
 
 module.exports.makeLegacyAdmin = function (baseClass) {
   class LegacyAdmin extends baseClass {

src/legacy_wrappers/bulk.js

@@ -0,0 +1,20 @@
+'use strict';
+
+module.exports = Object.create(null);
+Object.defineProperty(module.exports, '__esModule', { value: true });
+
+module.exports.makeLegacyOrderedBulkOperation = function (baseClass) {
+  return class LegacyOrderedBulkOperation extends baseClass {
+    execute(callback) {
+
+    }
+  };
+};
+
+module.exports.makeLegacyUnorderedBulkOperation = function (baseClass) {
+  return class LegacyUnorderedBulkOperation extends baseClass {
+    execute(callback) {
+
+    }
+  };
+};

src/legacy_wrappers/change_stream.js

@@ -3,6 +3,7 @@
 const { toLegacy, maybeCallback } = require('../utils');
 
 module.exports = Object.create(null);
+Object.defineProperty(module.exports, '__esModule', { value: true });
 
 module.exports.makeLegacyChangeStream = function (baseClass) {
   class LegacyChangeStream extends baseClass {

src/legacy_wrappers/collection.js

@@ -3,6 +3,7 @@
 const { toLegacy, maybeCallback } = require('../utils');
 
 module.exports = Object.create(null);
+Object.defineProperty(module.exports, '__esModule', { value: true });
 
 module.exports.makeLegacyCollection = function (baseClass) {
   class LegacyCollection extends baseClass {

src/legacy_wrappers/cursors.js

@@ -4,6 +4,7 @@ const { maybeCallback } = require('../utils');
 const { toLegacy } = require('../utils');
 
 module.exports = Object.create(null);
+Object.defineProperty(module.exports, '__esModule', { value: true });
 
 const commonCursorFunctions = new Map([
   [

src/legacy_wrappers/db.js

@@ -3,6 +3,7 @@
 const { toLegacy, maybeCallback } = require('../utils');
 
 module.exports = Object.create(null);
+Object.defineProperty(module.exports, '__esModule', { value: true });
 
 module.exports.makeLegacyDb = function (baseClass) {
   class LegacyDb extends baseClass {

src/legacy_wrappers/gridfs.js

@@ -3,6 +3,7 @@
 const { toLegacy, maybeCallback } = require('../utils');
 
 module.exports = Object.create(null);
+Object.defineProperty(module.exports, '__esModule', { value: true });
 
 module.exports.makeLegacyGridFSBucket = function (baseClass) {
   class LegacyGridFSBucket extends baseClass {
@@ -41,3 +42,20 @@ module.exports.makeLegacyGridFSBucket = function (baseClass) {
 
   return LegacyGridFSBucket;
 };
+
+module.exports.makeLegacyGridFSBucketWriteStream = function (baseClass) {
+  class LegacyGridFSBucketWriteStream extends baseClass {
+    abort(callback) {
+      maybeCallback(super.abort(), callback);
+    }
+  }
+
+  Object.defineProperty(baseClass.prototype, toLegacy, {
+    enumerable: false,
+    value: function () {
+      return Object.setPrototypeOf(this, LegacyGridFSBucketWriteStream.prototype);
+    }
+  });
+
+  return LegacyGridFSBucketWriteStream;
+};

src/legacy_wrappers/mongo_client.js

@@ -3,6 +3,7 @@
 const { toLegacy, maybeCallback } = require('../utils');
 
 module.exports = Object.create(null);
+Object.defineProperty(module.exports, '__esModule', { value: true });
 
 module.exports.makeLegacyMongoClient = function (baseClass) {
   class LegacyMongoClient extends baseClass {

src/legacy_wrappers/session.js

@@ -3,6 +3,7 @@
 const { toLegacy } = require('../utils');
 
 module.exports = Object.create(null);
+Object.defineProperty(module.exports, '__esModule', { value: true });
 
 module.exports.makeLegacyClientSession = function (baseClass) {
   class LegacyClientSession extends baseClass {