mongodb-js · syn-zhu · May 6, 2025 · Apr 30, 2025 · Apr 30, 2025 · Apr 30, 2025
@@ -5,12 +5,23 @@ import {
   css,
   ItemActionControls,
   cx,
+  Badge,
+  BadgeVariant,
+  Tooltip,
+  useDarkMode,
+  Body,
 } from '@mongodb-js/compass-components';
 import { type Actions, ROW_HEIGHT } from './constants';
 import { ExpandButton } from './tree-item';
 import { type NavigationItemActions } from './item-actions';
+import type {
+  ConnectedConnectionTreeItem,
+  NotConnectedConnectionTreeItem,
+  SidebarTreeItem,
+} from './tree-data';
 
 type NavigationBaseItemProps = {
+  item: SidebarTreeItem;
   name: string;
   isActive: boolean;
   isExpandVisible: boolean;
@@ -86,7 +97,66 @@ const actionControlsWrapperStyles = css({
   gap: spacing[100],
 });
 
+const ClusterStateBadge: React.FunctionComponent<{
+  state: string;
+}> = ({ state }) => {
+  const badgeVariant =
+    state === 'CREATING'
+      ? BadgeVariant.Blue
+      : state === 'DELETED'
+      ? BadgeVariant.Red
+      : BadgeVariant.LightGray;
+  const badgeText =
+    state === 'DELETING'
+      ? 'TERMINATING'
+      : state === 'DELETED'
+      ? 'TERMINATED'
+      : state;
+
+  return (
+    <Badge variant={badgeVariant} data-testid="navigation-item-state-badge">
+      {badgeText}
+    </Badge>
+  );
+};
+
+const ClusterStateBadgeWithTooltip: React.FunctionComponent<{
+  item: ConnectedConnectionTreeItem | NotConnectedConnectionTreeItem;
+}> = ({ item }) => {
+  const isDarkMode = useDarkMode();
+
+  const atlasClusterState = item.connectionInfo.atlasMetadata?.clusterState;
+  if (atlasClusterState === 'PAUSED') {
+    return (
+      <Tooltip
+        enabled={true}
+        darkMode={isDarkMode}
+        trigger={({
+          children: tooltipChildren,
+          ...tooltipTriggerProps
+        }: React.HTMLProps<HTMLDivElement>) => (
+          <div {...tooltipTriggerProps}>
+            <ClusterStateBadge state={atlasClusterState} />
+            {tooltipChildren}
+          </div>
+        )}
+      >
+        <Body>Unpause your cluster to connect to it</Body>
+      </Tooltip>
+    );
+  } else if (
+    atlasClusterState === 'DELETING' ||
+    atlasClusterState === 'CREATING' ||
+    atlasClusterState === 'DELETED'
+  ) {
+    return <ClusterStateBadge state={atlasClusterState} />;
+  }
+
+  return null;
+};
+
 export const NavigationBaseItem: React.FC<NavigationBaseItemProps> = ({
+  item,
   isActive,
   actionProps,
   name,
@@ -102,6 +172,7 @@ export const NavigationBaseItem: React.FC<NavigationBaseItemProps> = ({
   children,
 }) => {
   const [hoverProps, isHovered] = useHoverState();
+
   return (
     <div
       data-testid="base-navigation-item"
@@ -127,6 +198,9 @@ export const NavigationBaseItem: React.FC<NavigationBaseItemProps> = ({
           {icon}
           <span title={name}>{name}</span>
         </div>
+        {item.type === 'connection' && (
+          <ClusterStateBadgeWithTooltip item={item} />
+        )}
         <div className={actionControlsWrapperStyles}>
           <ItemActionControls
             menuClassName={menuStyles}

@@ -1,6 +1,6 @@
 import React, { useCallback, useMemo } from 'react';
 import AutoSizer from 'react-virtualized-auto-sizer';
-import { getVirtualTreeItems } from './tree-data';
+import { getConnectionId, getVirtualTreeItems } from './tree-data';
 import { ROW_HEIGHT } from './constants';
 import type { Actions } from './constants';
 import { VirtualTree } from './virtual-list/virtual-list';
@@ -18,6 +18,7 @@ import {
   spacing,
   useId,
 } from '@mongodb-js/compass-components';
+import { useConnectable } from '@mongodb-js/compass-connections/provider';
 import type { WorkspaceTab } from '@mongodb-js/compass-workspaces';
 import { usePreference } from 'compass-preferences-model/provider';
 import type { NavigationItemActions } from './item-actions';
@@ -57,6 +58,7 @@ const ConnectionsNavigationTree: React.FunctionComponent<
   );
 
   const id = useId();
+  const getConnectable = useConnectable();
 
   const treeData = useMemo(() => {
     return getVirtualTreeItems({
@@ -69,6 +71,11 @@ const ConnectionsNavigationTree: React.FunctionComponent<
 
   const onDefaultAction: OnDefaultAction<SidebarActionableItem> = useCallback(
     (item, evt) => {
+      const connectionId = getConnectionId(item);
+      if (!getConnectable(connectionId)) {
+        return;
+      }
+
       if (item.type === 'connection') {
         if (item.connectionStatus === 'connected') {
           onItemAction(item, 'select-connection');
@@ -83,7 +90,7 @@ const ConnectionsNavigationTree: React.FunctionComponent<
         }
       }
     },
-    [onItemAction]
+    [onItemAction, getConnectable]
   );
 
   const activeItemId = useMemo(() => {
@@ -144,6 +151,12 @@ const ConnectionsNavigationTree: React.FunctionComponent<
 
   const getItemActionsAndConfig = useCallback(
     (item: SidebarTreeItem) => {
+      const connectionId = getConnectionId(item);
+      if (!getConnectable(connectionId)) {
+        return {
+          actions: [],
+        };
+      }
       switch (item.type) {
         case 'placeholder':
           return {
@@ -192,7 +205,11 @@ const ConnectionsNavigationTree: React.FunctionComponent<
           };
       }
     },
-    [isRenameCollectionEnabled, getCollapseAfterForConnectedItem]
+    [
+      isRenameCollectionEnabled,
+      getCollapseAfterForConnectedItem,
+      getConnectable,
+    ]
   );
 
   const isTestEnv = process.env.NODE_ENV === 'test';

@@ -63,6 +63,22 @@ export const NavigationItemIcon = ({ item }: { item: SidebarTreeItem }) => {
     return <Icon glyph="TimeSeries" />;
   }
   if (item.type === 'connection') {
+    const atlasClusterState = item.connectionInfo.atlasMetadata?.clusterState;
+    if (atlasClusterState === 'DELETING' || atlasClusterState === 'CREATING') {
+      return (
+        <WithStatusMarker status={'disconnected'}>
+          <Icon glyph="Refresh" />
+        </WithStatusMarker>
+      );
+    }
+    if (atlasClusterState === 'PAUSED' || atlasClusterState === 'DELETED') {
+      return (
+        <WithStatusMarker status={'disconnected'}>
+          <ServerIcon />
+        </WithStatusMarker>
+      );
+    }
+
     const isFavorite = item.connectionInfo.savedConnectionType === 'favorite';
     if (isFavorite) {
       return (

@@ -212,6 +212,7 @@ export function NavigationItem({
         <PlaceholderItem level={item.level} />
       ) : (
         <NavigationBaseItem
+          item={item}
           isActive={isActive}
           isFocused={isFocused}
           isExpanded={!!item.isExpanded}

@@ -4,7 +4,8 @@ import {
   DefaultColorCode,
 } from '@mongodb-js/connection-form';
 import { palette, useDarkMode } from '@mongodb-js/compass-components';
-import type { SidebarTreeItem } from './tree-data';
+import { getConnectionId, type SidebarTreeItem } from './tree-data';
+import { useConnectable } from '@mongodb-js/compass-connections/provider';
 
 type AcceptedStyles = {
   '--item-bg-color'?: string;
@@ -29,9 +30,12 @@ export default function StyledNavigationItem({
     () => (isDarkMode ? palette.gray.light1 : palette.gray.dark1),
     [isDarkMode]
   );
+  const getConnectable = useConnectable();
 
   const style: React.CSSProperties & AcceptedStyles = useMemo(() => {
     const style: AcceptedStyles = {};
+    const connectionId = getConnectionId(item);
+    const isConnectable = getConnectable(connectionId);
     const isDisconnectedConnection =
       item.type === 'connection' && item.connectionStatus !== 'connected';
     const isNonExistentNamespace =
@@ -44,19 +48,20 @@ export default function StyledNavigationItem({
       style['--item-bg-color-active'] = connectionColorToHexActive(colorCode);
     }
 
-    if (isDisconnectedConnection || isNonExistentNamespace) {
+    if (isDisconnectedConnection || isNonExistentNamespace || !isConnectable) {
       style['--item-color'] = inactiveColor;
     }
 
-    // For a non-existent namespace, even if its active, we show it as inactive
-    if (isNonExistentNamespace) {
+    // We always show these as inactive
+    if (isNonExistentNamespace || !isConnectable) {
       style['--item-color-active'] = inactiveColor;
     }
     return style;
   }, [
     inactiveColor,
     item,
     colorCode,
+    getConnectable,
     connectionColorToHex,
     connectionColorToHexActive,
   ]);

@@ -124,6 +124,16 @@ export type SidebarActionableItem =
 
 export type SidebarTreeItem = PlaceholderTreeItem | SidebarActionableItem;
 
+export function getConnectionId(item: SidebarTreeItem): string {
+  if (item.type === 'placeholder') {
+    return '';
+  } else if (item.type === 'connection') {
+    return item.connectionInfo.id;
+  } else {
+    return item.connectionId;
+  }
+}
+
 const notConnectedConnectionToItems = ({
   connection: { name, connectionInfo, connectionStatus },
   connectionIndex,

@@ -0,0 +1,20 @@
+import { useStore } from '../stores/store-context';
+import { useCallback } from 'react';
+import { connectable } from '../utils/connection-supports';
+
+export function useConnectable(): (connectionId: string) => boolean {
+  const store = useStore();
+  const getConnectable = useCallback(
+    (connectionId: string) => {
+      const conn = store.getState().connections.byId[connectionId];
+      if (!conn) {
+        return false;
+      }
+
+      return connectable(conn.info);
+    },
+    [store]
+  );
+
+  return getConnectable;
+}
@@ -23,7 +23,7 @@ import {
   ConnectionActionsProvider,
 } from './stores/store-context';
 export type { ConnectionFeature } from './utils/connection-supports';
-export { connectionSupports } from './utils/connection-supports';
+export { connectionSupports, connectable } from './utils/connection-supports';
 
 const ConnectionsComponent: React.FunctionComponent<{
   /**

@@ -69,6 +69,8 @@ export type { ConnectionsService } from './stores/store-context';
 
 export { useConnectionSupports } from './hooks/use-connection-supports';
 
+export { useConnectable } from './hooks/use-connectable';
+
 const ConnectionStatus = {
   /**
    * @deprecated use a string literal directly

@@ -32,6 +32,7 @@ import EventEmitter from 'events';
 import { showNonGenuineMongoDBWarningModal as _showNonGenuineMongoDBWarningModal } from '../components/non-genuine-connection-modal';
 import ConnectionString from 'mongodb-connection-string-url';
 import type { ExtraConnectionData as ExtraConnectionDataForTelemetry } from '@mongodb-js/compass-telemetry';
+import { connectable } from '../utils/connection-supports';
 
 export type ConnectionsEventMap = {
   connected: (
@@ -530,9 +531,12 @@ function mergeConnections(
     ? newConnections
     : [newConnections];
 
+  const removedConnectionIds = new Set(connectionsState.ids);
+
   let newConnectionsById = connectionsState.byId;
 
   for (const connectionInfo of newConnections) {
+    removedConnectionIds.delete(connectionInfo.id);
     const existingConnection = newConnectionsById[connectionInfo.id];
 
     // If we got a new connection, just create a default state for this
@@ -557,6 +561,30 @@ function mergeConnections(
           info: connectionInfo,
         },
       };
+
+      // TODO: if an Atlas connection is going from PAUSED state to unpaused, we should
+      // reconnect the data service, since it would previously have been disconnected
+      // due to non-retryable error code.
+    }
+  }
+
+  // If an Atlas connection was removed, it means that the cluster was deleted
+  for (const connectionId of removedConnectionIds) {
+    const removedConnection = newConnectionsById[connectionId];
+    if (removedConnection.info.atlasMetadata) {
+      newConnectionsById = {
+        ...newConnectionsById,
+        [connectionId]: {
+          ...removedConnection,
+          info: {
+            ...removedConnection.info,
+            atlasMetadata: {
+              ...removedConnection.info.atlasMetadata,
+              clusterState: 'DELETED',
+            },
+          },
+        },
+      };
     }
   }
 
@@ -1399,7 +1427,9 @@ function getDescriptionForNonRetryableError(error: Error): string {
   // to the generic error description.
   const reason = error.message.match(/code: \d+, reason: (.*)$/)?.[1];
   return reason && reason.length > 0
-    ? reason
+    ? reason.endsWith('.')
+      ? reason.slice(0, -1)
+      : reason // Remove trailing period
     : NonRetryableErrorDescriptionFallbacks[
         Number(
           error.message.match(/code: (\d+),/)?.[1]
@@ -1489,6 +1519,10 @@ const connectWithOptions = (
           return;
         }
 
+        if (!connectable(connectionInfo)) {
+          return;
+        }
+
         const isAutoconnectAttempt = isAutoconnectInfo(
           getState(),
           connectionInfo.id