fix(suite): show correct step for disconnect event

trezor · Apr 30, 2024 · 199a5d0 · 199a5d0
1 parent 28afd8a
commit 199a5d0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/packages/suite/src/components/firmware/ReconnectDevicePrompt.tsx b/packages/suite/src/components/firmware/ReconnectDevicePrompt.tsx
@@ -4,7 +4,7 @@ import * as semver from 'semver';
 
 import { pickByDeviceModel, getFirmwareVersion } from '@trezor/device-utils';
 import { H2, Button, ConfirmOnDevice, variables, DeviceAnimation } from '@trezor/components';
-import { DEVICE, Device, DeviceModelInternal } from '@trezor/connect';
+import { DEVICE, Device, DeviceModelInternal, UI } from '@trezor/connect';
 import { Modal, Translation, WebUsbButton } from 'src/components/suite';
 import { DeviceConfirmImage } from 'src/components/suite/DeviceConfirmImage';
 import { useDevice, useFirmware } from 'src/hooks/suite';
@@ -183,8 +183,10 @@ export const ReconnectDevicePrompt = ({ onClose, onSuccess }: ReconnectDevicePro
         ) {
             return 'done';
         }
+        const rebootToBootloaderNotSupported = uiEvent?.type === UI.FIRMWARE_DISCONNECT;
+        const rebootToBootloaderCancelled = device?.connected && device?.mode !== 'bootloader';
 
-        return device?.connected && device?.mode !== 'bootloader'
+        return rebootToBootloaderNotSupported || rebootToBootloaderCancelled
             ? 'waiting-for-reboot'
             : 'disconnected';
     };