move the context to sidebar

Author: djechlin-mongodb

packages/compass-sidebar/src/components/multiple-connections/connections-navigation.tsx

@@ -1,10 +1,12 @@
 import toNS from 'mongodb-ns';
-import React, { useCallback, useContext, useEffect, useMemo } from 'react';
+import React, {
+  createContext,
+  useCallback,
+  useContext,
+  useEffect,
+  useMemo,
+} from 'react';
 import { connect } from 'react-redux';
-import {
-  connectionStorageLocator,
-  IsAtlasConnectionStorageContext,
-} from '@mongodb-js/connection-storage/provider';
 import {
   ChevronCollapse,
   type ItemAction,
@@ -95,6 +97,12 @@ const noDeploymentStyles = css({
   gap: spacing[200],
 });
 
+/**
+ * Indicates only Atlas cluster connections are supported, and the user cannot navigate
+ * to other types of connections from this UI.
+ */
+export const AtlasClusterConnectionsOnly = createContext<boolean>(false);
+
 function findCollection(ns: string, databases: Database[]) {
   const { database, collection } = toNS(ns);
 
@@ -481,7 +489,7 @@ const ConnectionsNavigation: React.FC<ConnectionsNavigationProps> = ({
     }
   }, [activeWorkspace, onDatabaseToggle, onConnectionToggle]);
 
-  const isAtlasConnectionStorage = useContext(IsAtlasConnectionStorageContext);
+  const isAtlasConnectionStorage = useContext(AtlasClusterConnectionsOnly);
 
   return (
     <div className={connectionsContainerStyles}>

packages/compass-sidebar/src/components/multiple-connections/sidebar.tsx

@@ -1,4 +1,4 @@
-import React, { useCallback, useContext, useState } from 'react';
+import React, { useCallback, useState } from 'react';
 import { type MapStateToProps, connect } from 'react-redux';
 import {
   ConnectionStatus,

packages/compass-sidebar/src/index.ts

@@ -1,5 +1,6 @@
 import type { ActivateHelpers } from 'hadron-app-registry';
 import { registerHadronPlugin, type AppRegistry } from 'hadron-app-registry';
+import { createContext } from 'react';
 import SidebarPlugin from './plugin';
 import { createSidebarStore } from './stores';
 import {
@@ -11,6 +12,7 @@ import type { ConnectionsService } from '@mongodb-js/compass-connections/provide
 import { connectionsLocator } from '@mongodb-js/compass-connections/provider';
 import type { Logger } from '@mongodb-js/compass-logging/provider';
 import { createLoggerLocator } from '@mongodb-js/compass-logging/provider';
+export { AtlasClusterConnectionsOnly } from './components/multiple-connections/connections-navigation';
 
 export const CompassSidebarPlugin = registerHadronPlugin(
   {

packages/compass-web/src/connection-storage.tsx

@@ -1,5 +1,4 @@
 import React, { useContext, useRef } from 'react';
-import { IsAtlasConnectionStorageContext } from '@mongodb-js/connection-storage/provider';
 import type {
   ConnectionStorage,
   ConnectionInfo,
@@ -375,9 +374,7 @@ export const AtlasCloudConnectionStorageProvider = createServiceProvider(
       <ConnectionStorageProvider
         value={sandboxConnectionStorage ?? storage.current}
       >
-        <IsAtlasConnectionStorageContext.Provider value={true}>
-          {children}
-        </IsAtlasConnectionStorageContext.Provider>
+        {children}
       </ConnectionStorageProvider>
     );
   }

packages/compass-web/src/entrypoint.tsx

@@ -19,7 +19,10 @@ import {
   WorkspaceTab as CollectionWorkspace,
   CollectionTabsProvider,
 } from '@mongodb-js/compass-collection';
-import { CompassSidebarPlugin } from '@mongodb-js/compass-sidebar';
+import {
+  CompassSidebarPlugin,
+  AtlasClusterConnectionsOnly,
+} from '@mongodb-js/compass-sidebar';
 import CompassQueryBarPlugin from '@mongodb-js/compass-query-bar';
 import { CompassDocumentsPlugin } from '@mongodb-js/compass-crud';
 import {
@@ -61,11 +64,13 @@ import { useCompassWebPreferences } from './preferences';
 const WithAtlasProviders: React.FC = ({ children }) => {
   return (
     <AtlasCloudAuthServiceProvider>
-      <AtlasServiceProvider>
-        <AtlasAiServiceProvider apiURLPreset="cloud">
-          {children}
-        </AtlasAiServiceProvider>
-      </AtlasServiceProvider>
+      <AtlasClusterConnectionsOnly.Provider value={true}>
+        <AtlasServiceProvider>
+          <AtlasAiServiceProvider apiURLPreset="cloud">
+            {children}
+          </AtlasAiServiceProvider>
+        </AtlasServiceProvider>
+      </AtlasClusterConnectionsOnly.Provider>
     </AtlasCloudAuthServiceProvider>
   );
 };

packages/connection-storage/src/provider.ts

@@ -15,8 +15,6 @@ export const ConnectionStorageContext = createContext<ConnectionStorage | null>(
   null
 );
 
-export const IsAtlasConnectionStorageContext = createContext<boolean>(false);
-
 export const ConnectionStorageProvider = ConnectionStorageContext.Provider;
 
 // TODO(COMPASS-7397): storage context should not be leaking out of the service