Revert compass-collection diffs

Author: kpamaran

packages/compass-collection/src/index.ts

@@ -14,7 +14,6 @@ import { workspacesServiceLocator } from '@mongodb-js/compass-workspaces/provide
 import { experimentationServiceLocator } from '@mongodb-js/compass-telemetry/provider';
 import { createLoggerLocator } from '@mongodb-js/compass-logging/provider';
 import { preferencesLocator } from 'compass-preferences-model/provider';
-import { atlasAiServiceLocator } from '@mongodb-js/compass-generative-ai/provider';
 import {
   CollectionWorkspaceTitle,
   CollectionPluginTitleComponent,
@@ -33,7 +32,6 @@ export const WorkspaceTab: WorkspacePlugin<typeof CollectionWorkspaceTitle> = {
     {
       dataService: dataServiceLocator as DataServiceLocator<keyof DataService>,
       collection: collectionModelLocator,
-      atlasAiService: atlasAiServiceLocator,
       workspaces: workspacesServiceLocator,
       experimentationServices: experimentationServiceLocator,
       connectionInfoRef: connectionInfoRefLocator,

packages/compass-collection/src/mock-data-generator-types.ts

@@ -4,18 +4,12 @@ import { analyzeDocuments } from 'mongodb-schema';
 import type { CollectionMetadata } from 'mongodb-collection-model';
 import type { ThunkAction } from 'redux-thunk';
 import type AppRegistry from '@mongodb-js/compass-app-registry';
-import type { ConnectionInfo } from '@mongodb-js/connection-info';
 import type { workspacesServiceLocator } from '@mongodb-js/compass-workspaces/provider';
-import type { DataService } from '@mongodb-js/compass-connections/provider';
 import type { CollectionSubtab } from '@mongodb-js/compass-workspaces';
-import type { AtlasAiService } from '@mongodb-js/compass-generative-ai/provider';
+import type { DataService } from '@mongodb-js/compass-connections/provider';
 import type { experimentationServiceLocator } from '@mongodb-js/compass-telemetry/provider';
 import { type Logger, mongoLogId } from '@mongodb-js/compass-logging/provider';
 import { type PreferencesAccess } from 'compass-preferences-model/provider';
-import {
-  MockDataSchemaRequestShape,
-  type MockDataSchemaResponse,
-} from '@mongodb-js/compass-generative-ai';
 import { isInternalFieldPath } from 'hadron-document';
 import toNS from 'mongodb-ns';
 import {
@@ -31,13 +25,6 @@ import { calculateSchemaDepth } from '../calculate-schema-depth';
 import { processSchema } from '../transform-schema-to-field-info';
 import type { Document, MongoError } from 'mongodb';
 import { MockDataGeneratorStep } from '../components/mock-data-generator-modal/types';
-import {
-  MOCK_DATA_GENERATOR_STATE_IDLE,
-  MOCK_DATA_GENERATOR_STATE_GENERATING,
-  MOCK_DATA_GENERATOR_STATE_COMPLETED,
-  MOCK_DATA_GENERATOR_STATE_ERROR,
-} from '../mock-data-generator-types';
-import type { MockDataGeneratorState } from '../mock-data-generator-types';
 
 const DEFAULT_SAMPLE_SIZE = 100;
 
@@ -74,7 +61,6 @@ type CollectionThunkAction<R, A extends AnyAction = AnyAction> = ThunkAction<
   {
     localAppRegistry: AppRegistry;
     dataService: DataService;
-    atlasAiService: AtlasAiService;
     workspaces: ReturnType<typeof workspacesServiceLocator>;
     experimentationServices: ReturnType<typeof experimentationServiceLocator>;
     logger: Logger;
@@ -93,10 +79,9 @@ export type CollectionState = {
     isModalOpen: boolean;
     currentStep: MockDataGeneratorStep;
   };
-  fakerSchemaGeneration: MockDataGeneratorState;
 };
 
-export enum CollectionActions {
+enum CollectionActions {
   CollectionMetadataFetched = 'compass-collection/CollectionMetadataFetched',
   SchemaAnalysisStarted = 'compass-collection/SchemaAnalysisStarted',
   SchemaAnalysisFinished = 'compass-collection/SchemaAnalysisFinished',
@@ -106,9 +91,6 @@ export enum CollectionActions {
   MockDataGeneratorModalClosed = 'compass-collection/MockDataGeneratorModalClosed',
   MockDataGeneratorNextButtonClicked = 'compass-collection/MockDataGeneratorNextButtonClicked',
   MockDataGeneratorPreviousButtonClicked = 'compass-collection/MockDataGeneratorPreviousButtonClicked',
-  FakerMappingGenerationStarted = 'compass-collection/FakerMappingGenerationStarted',
-  FakerMappingGenerationCompleted = 'compass-collection/FakerMappingGenerationCompleted',
-  FakerMappingGenerationFailed = 'compass-collection/FakerMappingGenerationFailed',
 }
 
 interface CollectionMetadataFetchedAction {
@@ -155,23 +137,6 @@ interface MockDataGeneratorPreviousButtonClickedAction {
   type: CollectionActions.MockDataGeneratorPreviousButtonClicked;
 }
 
-interface FakerMappingGenerationStartedAction {
-  type: CollectionActions.FakerMappingGenerationStarted;
-  requestId: string;
-}
-
-interface FakerMappingGenerationCompletedAction {
-  type: CollectionActions.FakerMappingGenerationCompleted;
-  fakerSchema: MockDataSchemaResponse;
-  requestId: string;
-}
-
-interface FakerMappingGenerationFailedAction {
-  type: CollectionActions.FakerMappingGenerationFailed;
-  error: string;
-  requestId: string;
-}
-
 const reducer: Reducer<CollectionState, Action> = (
   state = {
     // TODO(COMPASS-7782): use hook to get the workspace tab id instead
@@ -185,9 +150,6 @@ const reducer: Reducer<CollectionState, Action> = (
       isModalOpen: false,
       currentStep: MockDataGeneratorStep.AI_DISCLAIMER,
     },
-    fakerSchemaGeneration: {
-      status: MOCK_DATA_GENERATOR_STATE_IDLE,
-    },
   },
   action
 ) => {
@@ -370,53 +332,6 @@ const reducer: Reducer<CollectionState, Action> = (
     };
   }
 
-  if (
-    isAction<FakerMappingGenerationStartedAction>(
-      action,
-      CollectionActions.FakerMappingGenerationStarted
-    )
-  ) {
-    return {
-      ...state,
-      fakerSchemaGeneration: {
-        status: MOCK_DATA_GENERATOR_STATE_GENERATING,
-        requestId: action.requestId,
-      },
-    };
-  }
-
-  if (
-    isAction<FakerMappingGenerationCompletedAction>(
-      action,
-      CollectionActions.FakerMappingGenerationCompleted
-    )
-  ) {
-    return {
-      ...state,
-      fakerSchemaGeneration: {
-        status: MOCK_DATA_GENERATOR_STATE_COMPLETED,
-        fakerSchema: action.fakerSchema,
-        requestId: action.requestId,
-      },
-    };
-  }
-
-  if (
-    isAction<FakerMappingGenerationFailedAction>(
-      action,
-      CollectionActions.FakerMappingGenerationFailed
-    )
-  ) {
-    return {
-      ...state,
-      fakerSchemaGeneration: {
-        status: MOCK_DATA_GENERATOR_STATE_ERROR,
-        error: action.error,
-        requestId: action.requestId,
-      },
-    };
-  }
-
   return state;
 };
 
@@ -543,77 +458,6 @@ export const analyzeCollectionSchema = (): CollectionThunkAction<
   };
 };
 
-export const generateFakerMappings = (
-  connectionInfo: ConnectionInfo
-): CollectionThunkAction<Promise<void>> => {
-  return async (dispatch, getState, { logger, atlasAiService }) => {
-    const { schemaAnalysis, fakerSchemaGeneration, namespace } = getState();
-    if (schemaAnalysis.status !== SCHEMA_ANALYSIS_STATE_COMPLETE) {
-      logger.log.error(
-        mongoLogId(1_001_000_305),
-        'Collection',
-        'Cannot call `generateFakeMappings` unless schema analysis is complete'
-      );
-      return;
-    }
-
-    if (fakerSchemaGeneration.status === MOCK_DATA_GENERATOR_STATE_GENERATING) {
-      logger.debug(
-        'Faker mapping generation is already in progress, skipping new generation.'
-      );
-      return;
-    }
-
-    // todo: dedup/abort around requestId
-    const requestId = 'some-request-id';
-
-    try {
-      logger.debug('Generating faker mappings');
-
-      const { database, collection } = toNS(namespace);
-
-      dispatch({
-        type: CollectionActions.FakerMappingGenerationStarted,
-        requestId: requestId,
-      });
-
-      const mockDataSchemaRequest = MockDataSchemaRequestShape.parse({
-        databaseName: database,
-        collectionName: collection,
-        schema: schemaAnalysis.processedSchema,
-        validationRules: schemaAnalysis.schemaMetadata.validationRules,
-      });
-
-      const response = await atlasAiService.getMockDataSchema(
-        mockDataSchemaRequest,
-        connectionInfo
-      );
-      dispatch({
-        type: CollectionActions.FakerMappingGenerationCompleted,
-        fakerSchema: response,
-        requestId: requestId,
-      });
-    } catch (e) {
-      const errorMessage = e instanceof Error ? e.message : String(e);
-
-      logger.log.error(
-        mongoLogId(1_001_000_312),
-        'Collection',
-        'Failed to generate faker.js mappings',
-        {
-          error: errorMessage,
-          namespace,
-        }
-      );
-      dispatch({
-        type: CollectionActions.FakerMappingGenerationFailed,
-        error: 'Experienced an issue generating faker.js mappings',
-        requestId,
-      });
-    }
-  };
-};
-
 export type CollectionTabPluginMetadata = CollectionMetadata & {
   /**
    * Initial query for the query bar