From 906657875db3ffdbe0078073381887e937d41ac4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Juan=20Cort=C3=A9s=20Ross?= <juan@bohem.io>
Date: Fri, 14 Jan 2022 15:14:21 +0100
Subject: [PATCH] LL-8941 Automatically release USB connection after 5s (#761)

---
 .../src/TransportNodeHid.ts                   | 32 ++++++++++++++++++-
 1 file changed, 31 insertions(+), 1 deletion(-)

diff --git a/packages/hw-transport-node-hid-singleton/src/TransportNodeHid.ts b/packages/hw-transport-node-hid-singleton/src/TransportNodeHid.ts
index ecfa18ccd..30dbb4e54 100644
--- a/packages/hw-transport-node-hid-singleton/src/TransportNodeHid.ts
+++ b/packages/hw-transport-node-hid-singleton/src/TransportNodeHid.ts
@@ -12,6 +12,22 @@ import { identifyUSBProductId } from "@ledgerhq/devices";
 import { CantOpenDevice } from "@ledgerhq/errors";
 import { listenDevices } from "./listenDevices";
 let transportInstance;
+
+const DISCONNECT_TIMEOUT = 5000;
+let disconnectTimeout;
+const clearDisconnectTimeout = () => {
+  if (disconnectTimeout) {
+    clearTimeout(disconnectTimeout);
+  }
+};
+
+const setDisconnectTimeout = () => {
+  disconnectTimeout = setTimeout(
+    () => TransportNodeHidSingleton.disconnect(),
+    DISCONNECT_TIMEOUT
+  );
+};
+
 /**
  * node-hid Transport implementation
  * @example
@@ -97,12 +113,14 @@ export default class TransportNodeHidSingleton extends TransportNodeHidNoEvents
       transportInstance.emit("disconnect");
       transportInstance = null;
     }
+    clearDisconnectTimeout();
   }
 
   /**
    * if path="" is not provided, the library will take the first device
    */
   static open(): Promise<TransportNodeHidSingleton> {
+    clearDisconnectTimeout();
     return Promise.resolve().then(() => {
       if (transportInstance) {
         log("hid-verbose", "reusing opened transport instance");
@@ -138,7 +156,19 @@ export default class TransportNodeHidSingleton extends TransportNodeHidNoEvents
     });
   }
 
-  close() {
+  /**
+   * Exchange with the device using APDU protocol.
+   * @param apdu
+   * @returns a promise of apdu response
+   */
+  async exchange(apdu: Buffer): Promise<Buffer> {
+    clearDisconnectTimeout();
+    const result = await super.exchange(apdu);
+    setDisconnectTimeout();
+    return result;
+  }
+
+  close(): Promise<void> {
     // intentionally, a close will not effectively close the hid connection
     return Promise.resolve();
   }