diff --git a/services/web/frontend/js/features/ide-react/hooks/use-is-network-stalled.ts b/services/web/frontend/js/features/ide-react/hooks/use-is-network-stalled.ts index 0cea76cca4..8b533e9761 100644 --- a/services/web/frontend/js/features/ide-react/hooks/use-is-network-stalled.ts +++ b/services/web/frontend/js/features/ide-react/hooks/use-is-network-stalled.ts @@ -1,10 +1,14 @@ import { useFeatureFlag } from '@/shared/context/split-test-context' import { useUnsavedDocsContext } from '@/features/ide-react/context/unsaved-docs-context' import { useIdeReactContext } from '@/features/ide-react/context/ide-react-context' +import { useConnectionContext } from '../context/connection-context' export default function useIsNetworkStalled(): boolean { const enabled = useFeatureFlag('intermittent-connection-improvements') const { isSavingStalled } = useUnsavedDocsContext() const { outOfSync } = useIdeReactContext() - return enabled && (isSavingStalled || outOfSync) + const { isConnected, connectionState } = useConnectionContext() + const websocketDisconnected = + !isConnected || connectionState.reconnectAt !== null + return enabled && (isSavingStalled || outOfSync || websocketDisconnected) }