Add mongodb.ts and initial refactor

Author: baileympearson

src/cmap/auth/mongodb_oidc.ts

@@ -6,10 +6,10 @@ import { type AuthContext, AuthProvider } from './auth_provider';
 import type { MongoCredentials } from './mongo_credentials';
 import { AutomatedCallbackWorkflow } from './mongodb_oidc/automated_callback_workflow';
 import { callback as azureCallback } from './mongodb_oidc/azure_machine_workflow';
-import { callback as gcpCallback } from './mongodb_oidc/gcp_machine_workflow';
-import { callback as k8sCallback } from './mongodb_oidc/k8s_machine_workflow';
+import { gcpCallback as gcpCallback } from './mongodb_oidc/gcp_machine_workflow';
+import { k8sCallback as k8sCallback } from './mongodb_oidc/k8s_machine_workflow';
 import { TokenCache } from './mongodb_oidc/token_cache';
-import { callback as testCallback } from './mongodb_oidc/token_machine_workflow';
+import { tokenMachineCallback as testCallback } from './mongodb_oidc/token_machine_workflow';
 
 /** Error when credentials are missing. */
 const MISSING_CREDENTIALS_ERROR = 'AuthContext must provide credentials.';

src/cmap/auth/mongodb_oidc/gcp_machine_workflow.ts

@@ -18,7 +18,7 @@ const TOKEN_RESOURCE_MISSING_ERROR =
  * @param params - The OIDC callback parameters.
  * @returns The OIDC response.
  */
-export const callback: OIDCCallbackFunction = async (
+export const gcpCallback: OIDCCallbackFunction = async (
   params: OIDCCallbackParams
 ): Promise<OIDCResponse> => {
   const tokenAudience = params.tokenAudience;

src/cmap/auth/mongodb_oidc/k8s_machine_workflow.ts

@@ -17,7 +17,7 @@ const AWS_FILENAME = 'AWS_WEB_IDENTITY_TOKEN_FILE';
  * @param params - The OIDC callback parameters.
  * @returns The OIDC response.
  */
-export const callback: OIDCCallbackFunction = async (): Promise<OIDCResponse> => {
+export const k8sCallback: OIDCCallbackFunction = async (): Promise<OIDCResponse> => {
   let filename: string;
   if (process.env[AZURE_FILENAME]) {
     filename = process.env[AZURE_FILENAME];

src/cmap/auth/mongodb_oidc/token_machine_workflow.ts

@@ -12,7 +12,7 @@ const TOKEN_MISSING_ERROR = 'OIDC_TOKEN_FILE must be set in the environment.';
  * @param params - The OIDC callback parameters.
  * @returns The OIDC response.
  */
-export const callback: OIDCCallbackFunction = async (): Promise<OIDCResponse> => {
+export const tokenMachineCallback: OIDCCallbackFunction = async (): Promise<OIDCResponse> => {
   const tokenFile = process.env.OIDC_TOKEN_FILE;
   if (!tokenFile) {
     throw new MongoAWSError(TOKEN_MISSING_ERROR);

test/mongodb.ts

@@ -0,0 +1,135 @@
+/**
+ * Auto-generated file that exports everything from src/
+ * Generated on: 2026-02-03T22:38:18.484Z
+ */
+
+export * from '../src/admin';
+export * from '../src/bson';
+export * from '../src/bulk/common';
+export * from '../src/bulk/ordered';
+export * from '../src/bulk/unordered';
+export * from '../src/change_stream';
+export * from '../src/client-side-encryption/auto_encrypter';
+export * from '../src/client-side-encryption/client_encryption';
+export * from '../src/client-side-encryption/errors';
+export * from '../src/client-side-encryption/mongocryptd_manager';
+export * from '../src/client-side-encryption/providers/aws';
+export * from '../src/client-side-encryption/providers/azure';
+export * from '../src/client-side-encryption/providers/gcp';
+export * from '../src/client-side-encryption/providers/index';
+export * from '../src/client-side-encryption/state_machine';
+export * from '../src/cmap/auth/auth_provider';
+export * from '../src/cmap/auth/aws4';
+export * from '../src/cmap/auth/aws_temporary_credentials';
+export * from '../src/cmap/auth/gssapi';
+export * from '../src/cmap/auth/mongo_credentials';
+export * from '../src/cmap/auth/mongodb_aws';
+export * from '../src/cmap/auth/mongodb_oidc';
+export * from '../src/cmap/auth/mongodb_oidc/automated_callback_workflow';
+export * from '../src/cmap/auth/mongodb_oidc/azure_machine_workflow';
+export * from '../src/cmap/auth/mongodb_oidc/callback_workflow';
+export * from '../src/cmap/auth/mongodb_oidc/command_builders';
+export * from '../src/cmap/auth/mongodb_oidc/gcp_machine_workflow';
+export * from '../src/cmap/auth/mongodb_oidc/human_callback_workflow';
+export * from '../src/cmap/auth/mongodb_oidc/k8s_machine_workflow';
+export * from '../src/cmap/auth/mongodb_oidc/token_cache';
+export * from '../src/cmap/auth/mongodb_oidc/token_machine_workflow';
+export * from '../src/cmap/auth/plain';
+export * from '../src/cmap/auth/providers';
+export * from '../src/cmap/auth/scram';
+export * from '../src/cmap/auth/x509';
+export * from '../src/cmap/command_monitoring_events';
+export * from '../src/cmap/commands';
+export * from '../src/cmap/connect';
+export * from '../src/cmap/connection';
+export * from '../src/cmap/connection_pool';
+export * from '../src/cmap/connection_pool_events';
+export * from '../src/cmap/errors';
+export * from '../src/cmap/handshake/client_metadata';
+export * from '../src/cmap/metrics';
+export * from '../src/cmap/stream_description';
+export * from '../src/cmap/wire_protocol/compression';
+export * from '../src/cmap/wire_protocol/constants';
+export * from '../src/cmap/wire_protocol/on_data';
+export * from '../src/cmap/wire_protocol/on_demand/document';
+export * from '../src/cmap/wire_protocol/responses';
+export * from '../src/cmap/wire_protocol/shared';
+export * from '../src/collection';
+export * from '../src/connection_string';
+export * from '../src/constants';
+export * from '../src/cursor/abstract_cursor';
+export * from '../src/cursor/aggregation_cursor';
+export * from '../src/cursor/change_stream_cursor';
+export * from '../src/cursor/client_bulk_write_cursor';
+export * from '../src/cursor/explainable_cursor';
+export * from '../src/cursor/find_cursor';
+export * from '../src/cursor/list_collections_cursor';
+export * from '../src/cursor/list_indexes_cursor';
+export * from '../src/cursor/list_search_indexes_cursor';
+export * from '../src/cursor/run_command_cursor';
+export * from '../src/db';
+export * from '../src/deps';
+export * from '../src/encrypter';
+export * from '../src/error';
+export * from '../src/explain';
+export * from '../src/gridfs/download';
+export * from '../src/gridfs/index';
+export * from '../src/gridfs/upload';
+export * from '../src/index';
+export * from '../src/mongo_client';
+export * from '../src/mongo_client_auth_providers';
+export * from '../src/mongo_logger';
+export * from '../src/mongo_types';
+export * from '../src/operations/aggregate';
+export * from '../src/operations/client_bulk_write/client_bulk_write';
+export * from '../src/operations/client_bulk_write/command_builder';
+export * from '../src/operations/client_bulk_write/common';
+export * from '../src/operations/client_bulk_write/executor';
+export * from '../src/operations/client_bulk_write/results_merger';
+export * from '../src/operations/command';
+export * from '../src/operations/count';
+export * from '../src/operations/create_collection';
+export * from '../src/operations/delete';
+export * from '../src/operations/distinct';
+export * from '../src/operations/drop';
+export * from '../src/operations/end_sessions';
+export * from '../src/operations/estimated_document_count';
+export * from '../src/operations/execute_operation';
+export * from '../src/operations/find';
+export * from '../src/operations/find_and_modify';
+export * from '../src/operations/get_more';
+export * from '../src/operations/indexes';
+export * from '../src/operations/insert';
+export * from '../src/operations/kill_cursors';
+export * from '../src/operations/list_collections';
+export * from '../src/operations/list_databases';
+export * from '../src/operations/operation';
+export * from '../src/operations/profiling_level';
+export * from '../src/operations/remove_user';
+export * from '../src/operations/rename';
+export * from '../src/operations/run_command';
+export * from '../src/operations/search_indexes/create';
+export * from '../src/operations/search_indexes/drop';
+export * from '../src/operations/search_indexes/update';
+export * from '../src/operations/set_profiling_level';
+export * from '../src/operations/stats';
+export * from '../src/operations/update';
+export * from '../src/operations/validate_collection';
+export * from '../src/read_concern';
+export * from '../src/read_preference';
+export * from '../src/sdam/common';
+export * from '../src/sdam/events';
+export * from '../src/sdam/monitor';
+export * from '../src/sdam/server';
+export * from '../src/sdam/server_description';
+export * from '../src/sdam/server_selection';
+export * from '../src/sdam/server_selection_events';
+export * from '../src/sdam/srv_polling';
+export * from '../src/sdam/topology';
+export * from '../src/sdam/topology_description';
+export * from '../src/sessions';
+export * from '../src/sort';
+export * from '../src/timeout';
+export * from '../src/transactions';
+export * from '../src/utils';
+export * from '../src/write_concern';

test/unit/cmap/auth/mongodb_oidc/automated_callback_workflow.test.ts

@@ -4,7 +4,7 @@ import * as sinon from 'sinon';
 import { MongoCredentials } from '../../../../../src/cmap/auth/mongo_credentials';
 import { AutomatedCallbackWorkflow } from '../../../../../src/cmap/auth/mongodb_oidc/automated_callback_workflow';
 import { CallbackWorkflow } from '../../../../../src/cmap/auth/mongodb_oidc/callback_workflow';
-import { callback } from '../../../../../src/cmap/auth/mongodb_oidc/gcp_machine_workflow';
+import { gcpCallback } from '../../../../../src/cmap/auth/mongodb_oidc/gcp_machine_workflow';
 import { TokenCache } from '../../../../../src/cmap/auth/mongodb_oidc/token_cache';
 import { Connection } from '../../../../../src/cmap/connection';
 
@@ -20,7 +20,7 @@ describe('AutomatedCallbackWorkflow', function () {
         const connection = sandbox.createStubInstance(Connection);
         const credentials = sandbox.createStubInstance(MongoCredentials);
         sandbox.stub(CallbackWorkflow.prototype, 'finishAuthentication').resolves();
-        const workflow = new AutomatedCallbackWorkflow(cache, callback);
+        const workflow = new AutomatedCallbackWorkflow(cache, gcpCallback);
 
         beforeEach(function () {
           cache.put({ accessToken: 'test', expiresInSeconds: 7200 });

test/unit/cmap/auth/mongodb_oidc/gcp_machine_workflow.test.ts

@@ -1,7 +1,7 @@
 import { expect } from 'chai';
 
 import { OIDC_VERSION, type OIDCCallbackParams } from '../../../../../src/cmap/auth/mongodb_oidc';
-import { callback } from '../../../../../src/cmap/auth/mongodb_oidc/gcp_machine_workflow';
+import { gcpCallback } from '../../../../../src/cmap/auth/mongodb_oidc/gcp_machine_workflow';
 
 describe('GCP machine workflow', function () {
   describe('#callback', function () {
@@ -13,7 +13,7 @@ describe('GCP machine workflow', function () {
       };
 
       it('throws an error', async function () {
-        const error = await callback(params).catch(error => error);
+        const error = await gcpCallback(params).catch(error => error);
         expect(error.message).to.include('TOKEN_RESOURCE');
       });
     });

test/unit/cmap/auth/mongodb_oidc/token_machine_workflow.test.ts

@@ -1,7 +1,7 @@
 import { expect } from 'chai';
 import * as process from 'process';
 
-import { callback } from '../../../../../src/cmap/auth/mongodb_oidc/token_machine_workflow';
+import { tokenMachineCallback } from '../../../../../src/cmap/auth/mongodb_oidc/token_machine_workflow';
 
 describe('Token machine workflow', function () {
   describe('#callback', function () {
@@ -20,7 +20,7 @@ describe('Token machine workflow', function () {
       });
 
       it('throws an error', async function () {
-        const error = await callback().catch(error => error);
+        const error = await tokenMachineCallback().catch(error => error);
         expect(error.message).to.include('OIDC_TOKEN_FILE');
       });
     });