mongodb-data-service 21.18.0 → 21.21.0

package/lib/data-service.js

@@ -1,8 +1,28 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.DataServiceImpl = void 0;
 const util_1 = require("util");
 const async_1 = __importDefault(require("async"));
 const compass_logging_1 = __importDefault(require("@mongodb-js/compass-logging"));
@@ -14,6 +34,10 @@ const instance_detail_helper_1 = require("./instance-detail-helper");
 const redact_1 = require("./redact");
 const connect_mongo_client_1 = __importDefault(require("./connect-mongo-client"));
 const run_command_1 = require("./run-command");
+const csfle_collection_tracker_1 = require("./csfle-collection-tracker");
+const mongodb = __importStar(require("mongodb"));
+const mongodb_client_encryption_1 = require("mongodb-client-encryption");
+const ClientEncryption = (0, mongodb_client_encryption_1.extension)(mongodb).ClientEncryption;
 const { fetch: getIndexes } = require('mongodb-index-model');
 const { log, mongoLogId, debug } = (0, compass_logging_1.default)('COMPASS-DATA-SERVICE');
 function uniqueBy(values, key) {
@@ -23,10 +47,12 @@ function isEmptyObject(obj) {
     return Object.keys(obj).length === 0;
 }
 let id = 0;
-class DataService extends events_1.EventEmitter {
+const kSessionClientType = Symbol('kSessionClientType');
+class DataServiceImpl extends events_1.EventEmitter {
     constructor(connectionOptions) {
         super();
         this._isConnecting = false;
+        this._useCRUDClient = true;
         this._lastSeenTopology = null;
         this._isWritable = false;
         this._topologyType = 'Unknown';
@@ -46,7 +72,26 @@ class DataService extends events_1.EventEmitter {
         return new mongodb_connection_string_url_1.default(this._connectionOptions.connectionString);
     }
     getReadPreference() {
-        return this._initializedClient.readPreference;
+        return this._initializedClient('CRUD').readPreference;
+    }
+    setCSFLEEnabled(enabled) {
+        log.info(mongoLogId(1001000117), this._logCtx(), 'Setting CSFLE mode', {
+            enabled,
+        });
+        this._useCRUDClient = enabled;
+    }
+    getCSFLEMode() {
+        if (this._crudClient && (0, instance_detail_helper_1.checkIsCSFLEConnection)(this._crudClient)) {
+            if (this._useCRUDClient) {
+                return 'enabled';
+            }
+            else {
+                return 'disabled';
+            }
+        }
+        else {
+            return 'unavailable';
+        }
     }
     collection(ns, options, callback) {
         async_1.default.parallel({
@@ -76,7 +121,7 @@ class DataService extends events_1.EventEmitter {
     }
     collectionStats(databaseName, collectionName, callback) {
         const logop = this._startLogOp(mongoLogId(1001000031), 'Fetching collection info', { ns: `${databaseName}.${collectionName}` });
-        const db = this._initializedClient.db(databaseName);
+        const db = this._initializedClient('META').db(databaseName);
         db.command({ collStats: collectionName, verbose: true }, (error, data) => {
             logop(error);
             if (error && !error.message.includes('is a view, not a collection')) {
@@ -95,7 +140,7 @@ class DataService extends events_1.EventEmitter {
         }
     }
     command(databaseName, comm, callback) {
-        const db = this._initializedClient.db(databaseName);
+        const db = this._initializedClient('META').db(databaseName);
         db.command(comm, (error, result) => {
             if (error) {
                 return callback(this._translateMessage(error));
@@ -115,7 +160,7 @@ class DataService extends events_1.EventEmitter {
     async connectionStatus() {
         const logop = this._startLogOp(mongoLogId(1001000100), 'Running connectionStatus');
         try {
-            const adminDb = this._initializedClient.db('admin');
+            const adminDb = this._initializedClient('META').db('admin');
             const result = await (0, run_command_1.runCommand)(adminDb, {
                 connectionStatus: 1,
                 showPrivileges: true,
@@ -135,12 +180,25 @@ class DataService extends events_1.EventEmitter {
         const { authInfo: { authenticatedUserPrivileges }, } = await this.connectionStatus();
         return authenticatedUserPrivileges;
     }
+    async _getRolesOrFallback(roles = null) {
+        if (roles) {
+            return roles;
+        }
+        const { authInfo: { authenticatedUserRoles }, } = await this.connectionStatus();
+        return authenticatedUserRoles;
+    }
     async listCollections(databaseName, filter = {}, { nameOnly, privileges = null, } = {}) {
         const logop = this._startLogOp(mongoLogId(1001000032), 'Running listCollections', { db: databaseName, nameOnly: nameOnly !== null && nameOnly !== void 0 ? nameOnly : false });
-        const db = this._initializedClient.db(databaseName);
+        const db = this._initializedClient('CRUD').db(databaseName);
         const listCollections = async () => {
             try {
-                return await db.listCollections(filter, { nameOnly }).toArray();
+                const cursor = db.listCollections(filter, { nameOnly });
+                const results = [];
+                for await (const result of cursor) {
+                    this.emit('collectionInfoFetched', { databaseName, nameOnly }, result);
+                    results.push(result);
+                }
+                return results;
             }
             catch (err) {
                 log.warn(mongoLogId(1001000099), this._logCtx(), 'Failed to run listCollections', { message: err.message });
@@ -167,9 +225,9 @@ class DataService extends events_1.EventEmitter {
             throw err;
         }
     }
-    async listDatabases({ nameOnly, privileges = null, } = {}) {
+    async listDatabases({ nameOnly, privileges = null, roles = null, } = {}) {
         const logop = this._startLogOp(mongoLogId(1001000033), 'Running listDatabases', { nameOnly: nameOnly !== null && nameOnly !== void 0 ? nameOnly : false });
-        const adminDb = this._initializedClient.db('admin');
+        const adminDb = this._initializedClient('CRUD').db('admin');
         const listDatabases = async () => {
             try {
                 const { databases } = await (0, run_command_1.runCommand)(adminDb, {
@@ -189,12 +247,17 @@ class DataService extends events_1.EventEmitter {
                 .filter(Boolean)
                 .map((name) => ({ name }));
         };
+        const getDatabasesFromRoles = async () => {
+            const databases = (0, instance_detail_helper_1.getDatabasesByRoles)(await this._getRolesOrFallback(roles), ['read', 'readWrite', 'dbAdmin', 'dbOwner']);
+            return databases.map((name) => ({ name }));
+        };
         try {
-            const [listedDatabases, databasesFromPrivileges] = await Promise.all([
+            const [listedDatabases, databasesFromPrivileges, databasesFromRoles] = await Promise.all([
                 listDatabases(),
                 getDatabasesFromPrivileges(),
+                getDatabasesFromRoles(),
             ]);
-            const databases = uniqueBy([...databasesFromPrivileges, ...listedDatabases], 'name').map((db) => {
+            const databases = uniqueBy([...databasesFromRoles, ...databasesFromPrivileges, ...listedDatabases], 'name').map((db) => {
                 return { _id: db.name, name: db.name, ...(0, instance_detail_helper_1.adaptDatabaseInfo)(db) };
             });
             logop(null, { databasesCount: databases.length });
@@ -206,7 +269,7 @@ class DataService extends events_1.EventEmitter {
         }
     }
     async connect() {
-        if (this._client) {
+        if (this._metadataClient) {
             debug('already connected');
             return;
         }
@@ -218,18 +281,21 @@ class DataService extends events_1.EventEmitter {
         this._isConnecting = true;
         log.info(mongoLogId(1001000014), this._logCtx(), 'Connecting', {
             url: (0, redact_1.redactConnectionString)(this._connectionOptions.connectionString),
+            csfle: this._csfleLogInformation(this._connectionOptions.fleOptions),
         });
         try {
-            const [client, tunnel, connectionOptions] = await (0, connect_mongo_client_1.default)(this._connectionOptions, this.setupListeners.bind(this));
+            const [metadataClient, crudClient, tunnel, connectionOptions] = await (0, connect_mongo_client_1.default)(this._connectionOptions, this.setupListeners.bind(this));
             const attr = {
                 isWritable: this.isWritable(),
                 isMongos: this.isMongos(),
             };
             log.info(mongoLogId(1001000015), this._logCtx(), 'Connected', attr);
             debug('connected!', attr);
-            this._client = client;
+            this._metadataClient = metadataClient;
+            this._crudClient = crudClient;
             this._tunnel = tunnel;
             this._mongoClientConnectionOptions = connectionOptions;
+            this._csfleCollectionTracker = new csfle_collection_tracker_1.CSFLECollectionTrackerImpl(this, this._crudClient);
         }
         finally {
             this._isConnecting = false;
@@ -237,21 +303,21 @@ class DataService extends events_1.EventEmitter {
     }
     estimatedCount(ns, options, callback) {
         const logop = this._startLogOp(mongoLogId(1001000034), 'Running estimatedCount', { ns });
-        this._collection(ns).estimatedDocumentCount(options, (err, result) => {
+        this._collection(ns, 'CRUD').estimatedDocumentCount(options, (err, result) => {
             logop(err, result);
             callback(err, result);
         });
     }
     count(ns, filter, options, callback) {
         const logop = this._startLogOp(mongoLogId(1001000035), 'Running countDocuments', { ns });
-        this._collection(ns).countDocuments(filter, options, (err, result) => {
+        this._collection(ns, 'CRUD').countDocuments(filter, options, (err, result) => {
             logop(err, result);
             callback(err, result);
         });
     }
     createCollection(ns, options, callback) {
         const collectionName = this._collectionName(ns);
-        const db = this._initializedClient.db(this._databaseName(ns));
+        const db = this._initializedClient('CRUD').db(this._databaseName(ns));
         const logop = this._startLogOp(mongoLogId(1001000036), 'Running createCollection', { ns, options });
         db.createCollection(collectionName, options, (error, result) => {
             logop(error);
@@ -263,7 +329,7 @@ class DataService extends events_1.EventEmitter {
     }
     createIndex(ns, spec, options, callback) {
         const logop = this._startLogOp(mongoLogId(1001000037), 'Running createIndex', { ns, spec, options });
-        this._collection(ns).createIndex(spec, options, (error, result) => {
+        this._collection(ns, 'CRUD').createIndex(spec, options, (error, result) => {
             logop(error);
             if (error) {
                 return callback(this._translateMessage(error));
@@ -281,7 +347,7 @@ class DataService extends events_1.EventEmitter {
     }
     deleteOne(ns, filter, options, callback) {
         const logop = this._startLogOp(mongoLogId(1001000038), 'Running deleteOne', { ns });
-        this._collection(ns).deleteOne(filter, options, (error, result) => {
+        this._collection(ns, 'CRUD').deleteOne(filter, options, (error, result) => {
             logop(error, result);
             if (error) {
                 return callback(this._translateMessage(error));
@@ -291,7 +357,7 @@ class DataService extends events_1.EventEmitter {
     }
     deleteMany(ns, filter, options, callback) {
         const logop = this._startLogOp(mongoLogId(1001000039), 'Running deleteMany', { ns });
-        this._collection(ns).deleteMany(filter, options, (error, result) => {
+        this._collection(ns, 'CRUD').deleteMany(filter, options, (error, result) => {
             logop(error, result);
             if (error) {
                 return callback(this._translateMessage(error));
@@ -300,11 +366,15 @@ class DataService extends events_1.EventEmitter {
         });
     }
     async disconnect() {
-        var _a, _b;
+        var _a, _b, _c;
         log.info(mongoLogId(1001000016), this._logCtx(), 'Disconnecting');
         try {
-            await ((_a = this._client) === null || _a === void 0 ? void 0 : _a.close(true).catch((err) => debug('failed to close MongoClient', err)));
-            await ((_b = this._tunnel) === null || _b === void 0 ? void 0 : _b.close().catch((err) => debug('failed to close tunnel', err)));
+            await Promise.all([
+                (_a = this._metadataClient) === null || _a === void 0 ? void 0 : _a.close(true).catch((err) => debug('failed to close MongoClient', err)),
+                this._crudClient !== this._metadataClient &&
+                    ((_b = this._crudClient) === null || _b === void 0 ? void 0 : _b.close(true).catch((err) => debug('failed to close MongoClient', err))),
+                (_c = this._tunnel) === null || _c === void 0 ? void 0 : _c.close().catch((err) => debug('failed to close tunnel', err)),
+            ]);
         }
         finally {
             this._cleanup();
@@ -313,7 +383,7 @@ class DataService extends events_1.EventEmitter {
     }
     dropCollection(ns, callback) {
         const logop = this._startLogOp(mongoLogId(1001000059), 'Running dropCollection', { ns });
-        this._collection(ns).drop((error, result) => {
+        this._collection(ns, 'CRUD').drop((error, result) => {
             logop(error, result);
             if (error) {
                 return callback(this._translateMessage(error));
@@ -323,7 +393,7 @@ class DataService extends events_1.EventEmitter {
     }
     dropDatabase(name, callback) {
         const logop = this._startLogOp(mongoLogId(1001000040), 'Running dropDatabase', { db: name });
-        this._initializedClient
+        this._initializedClient('CRUD')
             .db(this._databaseName(name))
             .dropDatabase((error, result) => {
             logop(error, result);
@@ -335,7 +405,7 @@ class DataService extends events_1.EventEmitter {
     }
     dropIndex(ns, name, callback) {
         const logop = this._startLogOp(mongoLogId(1001000060), 'Running dropIndex', { ns, name });
-        this._collection(ns).dropIndex(name, (error, result) => {
+        this._collection(ns, 'CRUD').dropIndex(name, (error, result) => {
             logop(error, result);
             if (error) {
                 return callback(this._translateMessage(error));
@@ -352,7 +422,7 @@ class DataService extends events_1.EventEmitter {
             callback = options;
             options = undefined;
         }
-        const cursor = this._collection(ns).aggregate(pipeline, options);
+        const cursor = this._collection(ns, 'CRUD').aggregate(pipeline, options);
         if ((0, lodash_1.isFunction)(callback)) {
             process.nextTick(callback, null, cursor);
             return;
@@ -363,7 +433,7 @@ class DataService extends events_1.EventEmitter {
         const logop = this._startLogOp(mongoLogId(1001000042), 'Running find', {
             ns,
         });
-        const cursor = this._collection(ns).find(filter, options);
+        const cursor = this._collection(ns, 'CRUD').find(filter, options);
         cursor.toArray((error, documents) => {
             logop(error);
             if (error) {
@@ -375,11 +445,11 @@ class DataService extends events_1.EventEmitter {
     fetch(ns, filter, options) {
         const logop = this._startLogOp(mongoLogId(1001000043), 'Running raw find', { ns });
         logop(null);
-        return this._collection(ns).find(filter, options);
+        return this._collection(ns, 'CRUD').find(filter, options);
     }
     findOneAndReplace(ns, filter, replacement, options, callback) {
         const logop = this._startLogOp(mongoLogId(1001000044), 'Running findOneAndReplace', { ns });
-        this._collection(ns).findOneAndReplace(filter, replacement, options, (error, result) => {
+        this._collection(ns, 'CRUD').findOneAndReplace(filter, replacement, options, (error, result) => {
             logop(error);
             if (error) {
                 return callback(this._translateMessage(error));
@@ -389,7 +459,7 @@ class DataService extends events_1.EventEmitter {
     }
     findOneAndUpdate(ns, filter, update, options, callback) {
         const logop = this._startLogOp(mongoLogId(1001000045), 'Running findOneAndUpdate', { ns });
-        this._collection(ns).findOneAndUpdate(filter, update, options, (error, result) => {
+        this._collection(ns, 'CRUD').findOneAndUpdate(filter, update, options, (error, result) => {
             logop(error);
             if (error) {
                 return callback(this._translateMessage(error));
@@ -399,7 +469,7 @@ class DataService extends events_1.EventEmitter {
     }
     explain(ns, filter, options, callback) {
         const logop = this._startLogOp(mongoLogId(1001000046), 'Running find explain', { ns });
-        this._collection(ns)
+        this._collection(ns, 'CRUD')
             .find(filter, options)
             .explain((error, explanation) => {
             logop(error);
@@ -411,7 +481,7 @@ class DataService extends events_1.EventEmitter {
     }
     indexes(ns, options, callback) {
         const logop = this._startLogOp(mongoLogId(1001000047), 'Listing indexes', { ns });
-        getIndexes(this._initializedClient, ns, (error, data) => {
+        getIndexes(this._initializedClient('META'), ns, (error, data) => {
             logop(error);
             if (error) {
                 return callback(this._translateMessage(error));
@@ -421,7 +491,10 @@ class DataService extends events_1.EventEmitter {
     }
     async instance() {
         try {
-            const instanceData = await (0, instance_detail_helper_1.getInstance)(this._initializedClient);
+            const instanceData = {
+                ...(await (0, instance_detail_helper_1.getInstance)(this._initializedClient('META'))),
+                csfleMode: this.getCSFLEMode(),
+            };
             log.info(mongoLogId(1001000024), this._logCtx(), 'Fetched instance information', {
                 serverVersion: instanceData.build.version,
                 genuineMongoDB: instanceData.genuineMongoDB,
@@ -436,7 +509,7 @@ class DataService extends events_1.EventEmitter {
     }
     insertOne(ns, doc, options, callback) {
         const logop = this._startLogOp(mongoLogId(1001000048), 'Running insertOne', { ns });
-        this._collection(ns).insertOne(doc, options, (error, result) => {
+        this._collection(ns, 'CRUD').insertOne(doc, options, (error, result) => {
             logop(error, { acknowledged: result === null || result === void 0 ? void 0 : result.acknowledged });
             if (error) {
                 return callback(this._translateMessage(error));
@@ -446,7 +519,7 @@ class DataService extends events_1.EventEmitter {
     }
     insertMany(ns, docs, options, callback) {
         const logop = this._startLogOp(mongoLogId(1001000049), 'Running insertOne', { ns });
-        this._collection(ns).insertMany(docs, options, (error, result) => {
+        this._collection(ns, 'CRUD').insertMany(docs, options, (error, result) => {
             logop(error, {
                 acknowledged: result === null || result === void 0 ? void 0 : result.acknowledged,
                 insertedCount: result === null || result === void 0 ? void 0 : result.insertedCount,
@@ -458,12 +531,12 @@ class DataService extends events_1.EventEmitter {
         });
     }
     putMany(ns, docs, options) {
-        return this._collection(ns).insertMany(docs, options);
+        return this._collection(ns, 'CRUD').insertMany(docs, options);
     }
     updateCollection(ns, flags, callback) {
         const logop = this._startLogOp(mongoLogId(1001000050), 'Running updateCollection', { ns });
         const collectionName = this._collectionName(ns);
-        const db = this._initializedClient.db(this._databaseName(ns));
+        const db = this._initializedClient('CRUD').db(this._databaseName(ns));
         const command = {
             collMod: collectionName,
             ...flags,
@@ -478,7 +551,7 @@ class DataService extends events_1.EventEmitter {
     }
     updateOne(ns, filter, update, options, callback) {
         const logop = this._startLogOp(mongoLogId(1001000051), 'Running updateOne', { ns });
-        this._collection(ns).updateOne(filter, update, options, (error, result) => {
+        this._collection(ns, 'CRUD').updateOne(filter, update, options, (error, result) => {
             logop(error, result);
             if (error) {
                 return callback(this._translateMessage(error));
@@ -488,7 +561,7 @@ class DataService extends events_1.EventEmitter {
     }
     updateMany(ns, filter, update, options, callback) {
         const logop = this._startLogOp(mongoLogId(1001000052), 'Running updateMany', { ns });
-        this._collection(ns).updateMany(filter, update, options, (error, result) => {
+        this._collection(ns, 'CRUD').updateMany(filter, update, options, (error, result) => {
             logop(error, result);
             if (error) {
                 return callback(this._translateMessage(error));
@@ -498,13 +571,13 @@ class DataService extends events_1.EventEmitter {
     }
     currentOp(includeAll, callback) {
         const logop = this._startLogOp(mongoLogId(1001000053), 'Running currentOp');
-        this._initializedClient
+        this._initializedClient('META')
             .db('admin')
             .command({ currentOp: 1, $all: includeAll }, (error, result) => {
             logop(error);
             if (error) {
                 const logop = this._startLogOp(mongoLogId(1001000054), 'Searching $cmd.sys.inprog manually');
-                this._initializedClient
+                this._initializedClient('META')
                     .db('admin')
                     .collection('$cmd.sys.inprog')
                     .findOne({ $all: includeAll }, (error2, result2) => {
@@ -524,7 +597,10 @@ class DataService extends events_1.EventEmitter {
     }
     serverstats(callback) {
         const logop = this._startLogOp(mongoLogId(1001000061), 'Running serverStats');
-        this._defaultDb.admin().serverStatus((error, result) => {
+        this._initializedClient('META')
+            .db()
+            .admin()
+            .serverStatus((error, result) => {
             logop(error);
             if (error) {
                 return callback(this._translateMessage(error));
@@ -534,7 +610,10 @@ class DataService extends events_1.EventEmitter {
     }
     top(callback) {
         const logop = this._startLogOp(mongoLogId(1001000062), 'Running top');
-        this._defaultDb.admin().command({ top: 1 }, (error, result) => {
+        this._initializedClient('META')
+            .db()
+            .admin()
+            .command({ top: 1 }, (error, result) => {
             logop(error);
             if (error) {
                 return callback(this._translateMessage(error));
@@ -551,7 +630,7 @@ class DataService extends events_1.EventEmitter {
             stages: pipeline.map((stage) => Object.keys(stage)[0]),
             options,
         });
-        this._initializedClient
+        this._initializedClient('CRUD')
             .db(this._databaseName(sourceNs))
             .createCollection(name, options, (error, result) => {
             logop(error, result);
@@ -568,7 +647,7 @@ class DataService extends events_1.EventEmitter {
             collMod: name,
             ...options,
         };
-        const db = this._initializedClient.db(this._databaseName(sourceNs));
+        const db = this._initializedClient('META').db(this._databaseName(sourceNs));
         const logop = this._startLogOp(mongoLogId(1001000056), 'Running updateView', {
             name,
             sourceNs,
@@ -608,21 +687,30 @@ class DataService extends events_1.EventEmitter {
             ...options,
         });
     }
-    startSession() {
-        return this._initializedClient.startSession();
+    startSession(clientType) {
+        const session = this._initializedClient(clientType).startSession();
+        session[kSessionClientType] = clientType;
+        return session;
     }
     killSessions(sessions) {
-        return this._initializedClient.db('admin').command({
-            killSessions: Array.isArray(sessions)
-                ? sessions.map((s) => s.id)
-                : [sessions.id],
+        const sessionsArray = Array.isArray(sessions) ? sessions : [sessions];
+        const clientTypes = new Set(sessionsArray.map((s) => s[kSessionClientType]));
+        if (clientTypes.size !== 1) {
+            throw new Error(`Cannot kill sessions without a specific client type: [${[
+                ...clientTypes,
+            ].join(', ')}]`);
+        }
+        const [clientType] = clientTypes;
+        return this._initializedClient(clientType)
+            .db('admin')
+            .command({
+            killSessions: sessionsArray.map((s) => s.id),
         });
     }
     isConnected() {
-        return !!this._client;
+        return !!this._metadataClient;
     }
     setupListeners(client) {
-        this._client = client;
         if (client) {
             client.on('serverDescriptionChanged', (evt) => {
                 var _a, _b;
@@ -743,19 +831,28 @@ class DataService extends events_1.EventEmitter {
             });
         }
     }
-    get _initializedClient() {
-        if (!this._client) {
-            throw new Error('client not yet initialized');
+    _initializedClient(type) {
+        if (type !== 'CRUD' && type !== 'META') {
+            throw new Error(`Invalid client type: ${type}`);
         }
-        return this._client;
+        const client = type === 'CRUD' && this._useCRUDClient
+            ? this._crudClient
+            : this._metadataClient;
+        if (!client) {
+            throw new Error('Client not yet initialized');
+        }
+        return client;
     }
-    get _defaultDb() {
-        return this._initializedClient.db();
+    getCSFLECollectionTracker() {
+        if (!this._csfleCollectionTracker) {
+            throw new Error('Client not yet initialized');
+        }
+        return this._csfleCollectionTracker;
     }
     async databaseStats(name) {
         const logop = this._startLogOp(mongoLogId(1001000057), 'Running databaseStats', { db: name });
         try {
-            const db = this._initializedClient.db(name);
+            const db = this._initializedClient('META').db(name);
             const stats = await (0, run_command_1.runCommand)(db, { dbStats: 1 });
             const normalized = (0, instance_detail_helper_1.adaptDatabaseInfo)(stats);
             return { name, ...normalized };
@@ -809,8 +906,8 @@ class DataService extends events_1.EventEmitter {
             collections: db.collections,
         };
     }
-    _collection(ns) {
-        return this._initializedClient
+    _collection(ns, type) {
+        return this._initializedClient(type)
             .db(this._databaseName(ns))
             .collection(this._collectionName(ns));
     }
@@ -841,10 +938,11 @@ class DataService extends events_1.EventEmitter {
         return error;
     }
     _cleanup() {
-        if (this._client) {
-            this._client.removeAllListeners();
-        }
-        this._client = undefined;
+        var _a, _b, _c, _d;
+        (_b = (_a = this._metadataClient) === null || _a === void 0 ? void 0 : _a.removeAllListeners) === null || _b === void 0 ? void 0 : _b.call(_a);
+        (_d = (_c = this._crudClient) === null || _c === void 0 ? void 0 : _c.removeAllListeners) === null || _d === void 0 ? void 0 : _d.call(_c);
+        this._metadataClient = undefined;
+        this._crudClient = undefined;
         this._tunnel = undefined;
         this._mongoClientConnectionOptions = undefined;
         this._isWritable = false;
@@ -874,6 +972,57 @@ class DataService extends events_1.EventEmitter {
             }
         };
     }
+    configuredKMSProviders() {
+        var _a;
+        return (0, instance_detail_helper_1.configuredKMSProviders)((_a = this._connectionOptions.fleOptions) === null || _a === void 0 ? void 0 : _a.autoEncryption);
+    }
+    _csfleLogInformation(fleOptions) {
+        var _a, _b, _c;
+        const kmsProviders = (0, instance_detail_helper_1.configuredKMSProviders)(fleOptions === null || fleOptions === void 0 ? void 0 : fleOptions.autoEncryption);
+        if (kmsProviders.length === 0)
+            return null;
+        return {
+            storeCredentials: fleOptions === null || fleOptions === void 0 ? void 0 : fleOptions.storeCredentials,
+            encryptedFieldsMapNamespaces: Object.keys({
+                ...(_a = fleOptions === null || fleOptions === void 0 ? void 0 : fleOptions.autoEncryption) === null || _a === void 0 ? void 0 : _a.encryptedFieldsMap,
+                ...(_b = fleOptions === null || fleOptions === void 0 ? void 0 : fleOptions.autoEncryption) === null || _b === void 0 ? void 0 : _b.schemaMap,
+            }),
+            keyVaultNamespace: (_c = fleOptions === null || fleOptions === void 0 ? void 0 : fleOptions.autoEncryption) === null || _c === void 0 ? void 0 : _c.keyVaultNamespace,
+            kmsProviders,
+        };
+    }
+    async createDataKey(provider, options) {
+        const logop = this._startLogOp(mongoLogId(1001000123), 'Running createDataKey', { provider });
+        const clientEncryption = this._getClientEncryption();
+        let result;
+        try {
+            result = await clientEncryption.createDataKey(provider, options !== null && options !== void 0 ? options : {});
+        }
+        catch (err) {
+            logop(err);
+            throw err;
+        }
+        logop(null);
+        return result;
+    }
+    _getClientEncryption() {
+        const crudClient = this._initializedClient('CRUD');
+        const autoEncryptionOptions = crudClient.options.autoEncryption;
+        const { proxyHost, proxyPort, proxyUsername, proxyPassword } = crudClient.options;
+        return new ClientEncryption(crudClient, {
+            keyVaultNamespace: autoEncryptionOptions.keyVaultNamespace,
+            kmsProviders: autoEncryptionOptions.kmsProviders,
+            tlsOptions: autoEncryptionOptions.tlsOptions,
+            proxyOptions: proxyHost
+                ? {
+                    proxyHost,
+                    proxyPort,
+                    proxyUsername,
+                    proxyPassword,
+                }
+                : undefined,
+        });
+    }
 }
-exports.default = DataService;
+exports.DataServiceImpl = DataServiceImpl;
 //# sourceMappingURL=data-service.js.map