feat(compass-web): add the AtlasUserData class to save user data COMPASS-9565

packages/atlas-service/src/atlas-service.ts

@@ -78,6 +78,11 @@ export class AtlasService {
     // https://github.com/10gen/mms/blob/9f858bb987aac6aa80acfb86492dd74c89cbb862/client/packages/project/common/ajaxPrefilter.ts#L34-L49
     return this.cloudEndpoint(path);
   }
+  userDataEndpoint(path?: string): string {
+    return `https://cluster-connection.cloud-dev.mongodb.com/userData${normalizePath(
+      path
+    )}`;
+  }
   driverProxyEndpoint(path?: string): string {
     return `${this.config.ccsBaseUrl}${normalizePath(path)}`;
   }
@@ -91,13 +96,14 @@ export class AtlasService {
       { url }
     );
     try {
+      const headers = {
+        ...this.options?.defaultHeaders,
+        ...(shouldAddCSRFHeaders(init?.method) && getCSRFHeaders()),
+        ...init?.headers,
+      };
       const res = await fetch(url, {
         ...init,
-        headers: {
-          ...this.options?.defaultHeaders,
-          ...(shouldAddCSRFHeaders(init?.method) && getCSRFHeaders()),
-          ...init?.headers,
-        },
+        headers,
       });
       this.logger.log.info(
         this.logger.mongoLogId(1_001_000_309),

packages/atlas-service/src/provider.tsx

@@ -48,7 +48,7 @@ export const AtlasServiceProvider: React.FC<{
   );
 });
 
-function useAtlasServiceContext(): AtlasService {
+export function useAtlasServiceContext(): AtlasService {
   const service = useContext(AtlasServiceContext);
   if (!service) {
     throw new Error('No AtlasService available in this context');

packages/compass-aggregations/src/components/pipeline-toolbar/index.spec.tsx

@@ -23,7 +23,7 @@ describe('PipelineToolbar', function () {
         />,
         { pipeline: [{ $match: { _id: 1 } }] },
         undefined,
-        { pipelineStorage: new CompassPipelineStorage() }
+        { pipelineStorage: { getStorage: () => new CompassPipelineStorage() } }
       );
       toolbar = screen.getByTestId('pipeline-toolbar');
     });

packages/compass-aggregations/src/components/pipeline-toolbar/pipeline-header/index.spec.tsx

@@ -23,7 +23,7 @@ describe('PipelineHeader', function () {
       />,
       undefined,
       undefined,
-      { pipelineStorage: new CompassPipelineStorage() }
+      { pipelineStorage: { getStorage: () => new CompassPipelineStorage() } }
     );
     container = screen.getByTestId('pipeline-header');
   });

packages/compass-aggregations/src/components/pipeline-toolbar/pipeline-settings/index.tsx

@@ -1,6 +1,6 @@
 import React from 'react';
 import { connect } from 'react-redux';
-import { Button, css, Icon, spacing } from '@mongodb-js/compass-components';
+import { Button, Icon, css, spacing } from '@mongodb-js/compass-components';
 import { exportToLanguage } from '../../../modules/export-to-language';
 import { SaveMenu } from './pipeline-menus';
 import PipelineName from './pipeline-name';

packages/compass-aggregations/src/stores/store.ts

@@ -33,7 +33,7 @@ import type { CollectionTabPluginMetadata } from '@mongodb-js/compass-collection
 import type { PreferencesAccess } from 'compass-preferences-model';
 import type { Logger } from '@mongodb-js/compass-logging/provider';
 import type { AtlasAiService } from '@mongodb-js/compass-generative-ai/provider';
-import type { PipelineStorage } from '@mongodb-js/my-queries-storage/provider';
+import type { PipelineStorageAccess } from '@mongodb-js/my-queries-storage/provider';
 import { maxTimeMSChanged } from '../modules/max-time-ms';
 import type {
   ConnectionInfoRef,
@@ -80,7 +80,7 @@ export type AggregationsPluginServices = {
   logger: Logger;
   track: TrackFunction;
   atlasAiService: AtlasAiService;
-  pipelineStorage?: PipelineStorage;
+  pipelineStorage?: PipelineStorageAccess;
   connectionInfoRef: ConnectionInfoRef;
   connectionScopedAppRegistry: ConnectionScopedAppRegistry<'open-export'>;
   collection: Collection;
@@ -178,7 +178,7 @@ export function activateAggregationsPlugin(
         globalAppRegistry,
         localAppRegistry,
         pipelineBuilder,
-        pipelineStorage,
+        pipelineStorage: pipelineStorage?.getStorage(),
         workspaces,
         instance,
         preferences,

packages/compass-preferences-model/src/preferences-schema.tsx

@@ -852,18 +852,6 @@ export const storedUserPreferencesProps: Required<{
     type: 'boolean',
   },
 
-  enableMyQueries: {
-    ui: true,
-    cli: true,
-    global: true,
-    description: {
-      short:
-        'Enable My Queries feature to save and manage favorite queries and aggregations',
-    },
-    validator: z.boolean().default(true),
-    type: 'boolean',
-  },
-
   enableAggregationBuilderRunPipeline: {
     ui: true,
     cli: true,
@@ -1032,6 +1020,17 @@ export const storedUserPreferencesProps: Required<{
     validator: z.boolean().default(true),
     type: 'boolean',
   },
+  enableMyQueries: {
+    ui: true,
+    cli: true,
+    global: true,
+    description: {
+      short:
+        'Enable My Queries feature to save and manage favorite queries and aggregations',
+    },
+    validator: z.boolean().default(true),
+    type: 'boolean',
+  },
 
   inferNamespacesFromPrivileges: {
     ui: true,

packages/compass-preferences-model/src/react.ts

@@ -126,3 +126,17 @@ export function withPreferences<
     return createElement(component, { ...prefs, ...props });
   };
 }
+
+/**
+ * Hook to check if the My Queries Data Explorer feature is enabled.
+ * This controls access to:
+ * - Saved queries and aggregations
+ * - Recent queries autocomplete
+ * - Favorite queries/aggregations
+ */
+export function useMyQueriesFeature(): boolean {
+  const enableMyQueries = usePreference('enableMyQueries');
+
+  // Default to true to match the preference schema default
+  return enableMyQueries ?? true;
+}

packages/compass-query-bar/src/stores/query-bar-reducer.spec.ts

@@ -216,7 +216,7 @@ describe('queryBarReducer', function () {
         .deep.eq(appliedQuery);
 
       // updateAttributes is called in saveRecentAsFavorite and updateFavoriteQuery
-      expect(updateAttributesStub).to.have.been.calledTwice;
+      expect(updateAttributesStub).to.have.been.calledOnce;
       expect(saveQueriesStub).not.to.have.been.calledTwice;
     });
   });

packages/compass-query-bar/src/stores/query-bar-reducer.ts

@@ -268,9 +268,20 @@ export const fetchRecents = (): QueryBarThunkAction<
   return async (
     dispatch,
     _getState,
-    { recentQueryStorage, logger: { debug } }
+    { recentQueryStorage, logger: { debug }, preferences }
   ) => {
     try {
+      // Check if My Queries feature is enabled
+      const { enableMyQueries } = preferences.getPreferences();
+      if (!enableMyQueries) {
+        // If feature is disabled, dispatch empty array
+        dispatch({
+          type: QueryBarActions.RecentQueriesFetched,
+          recents: [],
+        });
+        return;
+      }
+
       const {
         queryBar: { namespace },
       } = _getState();
@@ -286,7 +297,14 @@ export const fetchRecents = (): QueryBarThunkAction<
 };
 
 export const fetchSavedQueries = (): QueryBarThunkAction<void> => {
-  return (dispatch) => {
+  return (dispatch, _getState, { preferences }) => {
+    // Check if My Queries feature is enabled
+    const { enableMyQueries } = preferences.getPreferences();
+    if (!enableMyQueries) {
+      // If feature is disabled, don't fetch anything
+      return;
+    }
+
     void dispatch(fetchRecents());
     void dispatch(fetchFavorites());
   };
@@ -303,9 +321,20 @@ export const fetchFavorites = (): QueryBarThunkAction<
   return async (
     dispatch,
     _getState,
-    { favoriteQueryStorage, logger: { debug } }
+    { favoriteQueryStorage, logger: { debug }, preferences }
   ) => {
     try {
+      // Check if My Queries feature is enabled
+      const { enableMyQueries } = preferences.getPreferences();
+      if (!enableMyQueries) {
+        // If feature is disabled, dispatch empty array
+        dispatch({
+          type: QueryBarActions.FavoriteQueriesFetched,
+          favorites: [],
+        });
+        return;
+      }
+
       const {
         queryBar: { namespace },
       } = _getState();
@@ -356,10 +385,7 @@ export const saveRecentAsFavorite = (
       };
 
       // add it in the favorite
-      await favoriteQueryStorage?.updateAttributes(
-        favoriteQuery._id,
-        favoriteQuery
-      );
+      await favoriteQueryStorage?.saveQuery(favoriteQuery, favoriteQuery._id);
 
       // update favorites
       void dispatch(fetchFavorites());

packages/compass-query-bar/src/stores/query-bar-store.ts

@@ -130,6 +130,7 @@ export function activatePlugin(
 
   const favoriteQueryStorage = favoriteQueryStorageAccess?.getStorage();
   const recentQueryStorage = recentQueryStorageAccess?.getStorage();
+
   const store = configureStore(
     {
       namespace: namespace ?? '',

packages/compass-saved-aggregations-queries/src/stores/index.ts

@@ -12,6 +12,7 @@ import type { Logger } from '@mongodb-js/compass-logging/provider';
 import type { workspacesServiceLocator } from '@mongodb-js/compass-workspaces/provider';
 import type {
   FavoriteQueryStorageAccess,
+  PipelineStorageAccess,
   PipelineStorage,
   FavoriteQueryStorage,
 } from '@mongodb-js/my-queries-storage/provider';
@@ -25,7 +26,7 @@ type MyQueriesServices = {
   globalAppRegistry: AppRegistry;
   logger: Logger;
   track: TrackFunction;
-  pipelineStorage?: PipelineStorage;
+  pipelineStorage?: PipelineStorageAccess;
   workspaces: ReturnType<typeof workspacesServiceLocator>;
   favoriteQueryStorageAccess?: FavoriteQueryStorageAccess;
 };
@@ -55,7 +56,7 @@ export function configureStore({
         preferencesAccess,
         logger,
         track,
-        pipelineStorage,
+        pipelineStorage: pipelineStorage?.getStorage(),
         queryStorage: favoriteQueryStorageAccess?.getStorage(),
         workspaces,
       })
@@ -69,9 +70,10 @@ export type RootState = ReturnType<
 
 type SavedQueryAggregationExtraArgs = Omit<
   MyQueriesServices,
-  'locateFavoriteQueryStorage'
+  'favoriteQueryStorageAccess' | 'pipelineStorage'
 > & {
   queryStorage?: FavoriteQueryStorage;
+  pipelineStorage?: PipelineStorage;
 };
 
 export type SavedQueryAggregationThunkAction<