From 013f14a52830a4a94e0be8fffc5413713869d641 Mon Sep 17 00:00:00 2001
From: Marek Polak <marek.polak@satoshilabs.com>
Date: Mon, 3 Feb 2025 16:18:52 +0100
Subject: [PATCH] feat(connect): upgrade to bridge when it starts

---
 packages/connect/src/device/DeviceList.ts | 28 +++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/packages/connect/src/device/DeviceList.ts b/packages/connect/src/device/DeviceList.ts
index 0f740d7d78a..e3b8b0f0a9c 100644
--- a/packages/connect/src/device/DeviceList.ts
+++ b/packages/connect/src/device/DeviceList.ts
@@ -203,6 +203,8 @@ export class DeviceList extends TypedEmitter<DeviceListEvents> implements IDevic
         return lock.promise as Promise<T>;
     }
 
+    private readonly scheduledUpgradeChecks: ApiTypeMap<ReturnType<typeof setTimeout>> = {};
+
     constructor({
         messages,
         priority,
@@ -378,6 +380,10 @@ export class DeviceList extends TypedEmitter<DeviceListEvents> implements IDevic
                     this.emit(TRANSPORT.START, getTransportInfo(transport));
                 }
             }
+            if (transport) {
+                // new transport started successfully or present transport kept, (re)plan check
+                this.scheduleUpgradeCheck(apiType, initParams);
+            }
         } catch (error) {
             this.emit(TRANSPORT.ERROR, { apiType, error: error?.message });
             if (initParams.transportReconnect && !abortSignal.aborted) {
@@ -396,6 +402,26 @@ export class DeviceList extends TypedEmitter<DeviceListEvents> implements IDevic
         return promise.finally(() => clearTimeout(timeout));
     }
 
+    private scheduleUpgradeCheck(apiType: TransportApiType, initParams: InitParams) {
+        clearTimeout(this.scheduledUpgradeChecks[apiType]);
+        this.scheduledUpgradeChecks[apiType] = setTimeout(async () => {
+            const transport = this.transport[apiType];
+            if (!transport) return;
+            const transports = this.transports.filter(t => t.apiType === apiType);
+            for (const t of transports) {
+                if (t === transport) break;
+                if (await t.ping()) {
+                    this.transportLock(apiType, 'Upgrading', signal =>
+                        this.createInitPromise(apiType, initParams, signal),
+                    ).catch(() => {});
+
+                    return;
+                }
+            }
+            this.scheduleUpgradeCheck(apiType, initParams);
+        }, 1000);
+    }
+
     private async selectTransport(
         [transport, ...rest]: Transport[],
         signal: AbortSignal,
@@ -542,6 +568,8 @@ export class DeviceList extends TypedEmitter<DeviceListEvents> implements IDevic
     }
 
     private async stopTransport(transport: Transport) {
+        clearTimeout(this.scheduledUpgradeChecks[transport.apiType]);
+
         const devices = this.devices.clear(transport);
 
         // disconnect devices