feat(compass-collection): Connect schema redux state to modal and filter collections based on nesting depth - CLOUDP-342665

packages/compass-collection/src/components/collection-header-actions/collection-header-actions.spec.tsx

@@ -57,6 +57,8 @@ describe('CollectionHeaderActions [Component]', function () {
               namespace="test.test"
               isReadonly={false}
               onOpenMockDataModal={sinon.stub()}
+              hasData={true}
+              maxNestingDepth={2}
               {...props}
             />
           </PreferencesProvider>
@@ -374,5 +376,80 @@ describe('CollectionHeaderActions [Component]', function () {
 
       expect(onOpenMockDataModal).to.have.been.calledOnce;
     });
+
+    it('should disable Mock Data Generator button when hasData is false', async function () {
+      mockUseAssignment.returns({
+        assignment: {
+          assignmentData: {
+            variant: 'mockDataGeneratorVariant',
+          },
+        },
+      });
+
+      await renderCollectionHeaderActions(
+        {
+          namespace: 'test.collection',
+          isReadonly: false,
+          hasData: false,
+        },
+        {},
+        atlasConnectionInfo
+      );
+
+      const button = screen.getByTestId(
+        'collection-header-generate-mock-data-button'
+      );
+      expect(button).to.have.attribute('aria-disabled', 'true');
+    });
+
+    it('should enable Mock Data Generator button when hasData is true', async function () {
+      mockUseAssignment.returns({
+        assignment: {
+          assignmentData: {
+            variant: 'mockDataGeneratorVariant',
+          },
+        },
+      });
+
+      await renderCollectionHeaderActions(
+        {
+          namespace: 'test.collection',
+          isReadonly: false,
+          hasData: true,
+        },
+        {},
+        atlasConnectionInfo
+      );
+
+      const button = screen.getByTestId(
+        'collection-header-generate-mock-data-button'
+      );
+      expect(button).to.not.have.attribute('aria-disabled', 'true');
+    });
+
+    it('should not show Mock Data Generator button for collections with excessive nesting depth', async function () {
+      mockUseAssignment.returns({
+        assignment: {
+          assignmentData: {
+            variant: 'mockDataGeneratorVariant',
+          },
+        },
+      });
+
+      await renderCollectionHeaderActions(
+        {
+          namespace: 'test.collection',
+          isReadonly: false,
+          hasData: true,
+          maxNestingDepth: 4, // Exceeds the limit of 3
+        },
+        {},
+        atlasConnectionInfo
+      );
+
+      expect(
+        screen.queryByTestId('collection-header-generate-mock-data-button')
+      ).to.not.exist;
+    });
   });
 });

packages/compass-collection/src/components/collection-header-actions/collection-header-actions.tsx

@@ -47,6 +47,8 @@ type CollectionHeaderActionsProps = {
   sourceName?: string;
   sourcePipeline?: unknown[];
   onOpenMockDataModal: () => void;
+  hasData: boolean;
+  maxNestingDepth: number;
 };
 
 const CollectionHeaderActions: React.FunctionComponent<
@@ -58,6 +60,8 @@ const CollectionHeaderActions: React.FunctionComponent<
   sourceName,
   sourcePipeline,
   onOpenMockDataModal,
+  hasData,
+  maxNestingDepth,
 }: CollectionHeaderActionsProps) => {
   const connectionInfo = useConnectionInfo();
   const { id: connectionId, atlasMetadata } = connectionInfo;
@@ -84,10 +88,8 @@ const CollectionHeaderActions: React.FunctionComponent<
     isInMockDataTreatmentVariant &&
     atlasMetadata && // Only show in Atlas
     !isReadonly && // Don't show for readonly collections (views)
-    !sourceName; // sourceName indicates it's a view
-  // TODO: CLOUDP-337090: also filter out overly nested collections
-
-  const hasData = true; // TODO: CLOUDP-337090
+    !sourceName && // sourceName indicates it's a view
+    maxNestingDepth < 4; // Filter out overly nested collections (4+ levels)
 
   return (
     <div

packages/compass-collection/src/components/collection-header/collection-header.tsx

@@ -21,6 +21,8 @@ import { getConnectionTitle } from '@mongodb-js/connection-info';
 import MockDataGeneratorModal from '../mock-data-generator-modal/mock-data-generator-modal';
 import { connect } from 'react-redux';
 import { openMockDataGeneratorModal } from '../../modules/collection-tab';
+import type { CollectionState } from '../../modules/collection-tab';
+import { SCHEMA_ANALYSIS_STATE_COMPLETE } from '../../schema-analysis-types';
 
 const collectionHeaderStyles = css({
   padding: spacing[400],
@@ -62,6 +64,8 @@ type CollectionHeaderProps = {
   editViewName?: string;
   sourcePipeline?: unknown[];
   onOpenMockDataModal: () => void;
+  hasData: boolean;
+  maxNestingDepth: number;
 };
 
 const getInsightsForPipeline = (pipeline: any[], isAtlas: boolean) => {
@@ -97,6 +101,8 @@ const CollectionHeader: React.FunctionComponent<CollectionHeaderProps> = ({
   editViewName,
   sourcePipeline,
   onOpenMockDataModal,
+  hasData,
+  maxNestingDepth,
 }) => {
   const darkMode = useDarkMode();
   const showInsights = usePreference('showInsights');
@@ -174,14 +180,32 @@ const CollectionHeader: React.FunctionComponent<CollectionHeaderProps> = ({
           sourceName={sourceName}
           sourcePipeline={sourcePipeline}
           onOpenMockDataModal={onOpenMockDataModal}
+          hasData={hasData}
+          maxNestingDepth={maxNestingDepth}
         />
       </div>
       <MockDataGeneratorModal />
     </div>
   );
 };
 
-const ConnectedCollectionHeader = connect(undefined, {
+const mapStateToProps = (state: CollectionState) => {
+  const { schemaAnalysis } = state;
+
+  return {
+    hasData:
+      schemaAnalysis.status === SCHEMA_ANALYSIS_STATE_COMPLETE &&
+      schemaAnalysis.processedSchema &&
+      Object.keys(schemaAnalysis.processedSchema).length > 0,
+    maxNestingDepth:
+      schemaAnalysis.status === SCHEMA_ANALYSIS_STATE_COMPLETE &&
+      schemaAnalysis.schemaMetadata
+        ? schemaAnalysis.schemaMetadata.maxNestingDepth
+        : 0,
+  };
+};
+
+const ConnectedCollectionHeader = connect(mapStateToProps, {
   onOpenMockDataModal: openMockDataGeneratorModal,
 })(CollectionHeader);