Add method for OneKey device analysis

CHANGELOG.md

@@ -8,6 +8,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Changed
 - **BREAKING:** Removed support for Node v12 in favor of v14 ([#137](https://github.com/MetaMask/eth-json-rpc-middleware/pull/137))
 
+## [0.10.1]
+- Support for differentiating hardware wallet vendors
+
 ## [0.10.0]
 ### Added
 - Support for EIP-721 signTypedData_v4 ([#117](https://github.com/MetaMask/eth-trezor-keyring/pull/117))

index.js

@@ -22,6 +22,9 @@ const TREZOR_CONNECT_MANIFEST = {
   appUrl: 'https://metamask.io',
 };
 
+const oneKeySpecialVersion = 99;
+const oneKeyVendor = 'onekey.so';
+
 function wait(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
@@ -58,6 +61,7 @@ class TrezorKeyring extends EventEmitter {
     this.perPage = 5;
     this.unlockedAccount = 0;
     this.paths = {};
+    this.vendor = undefined;
     this.deserialize(opts);
 
     TrezorConnect.on('DEVICE_EVENT', (event) => {
@@ -78,6 +82,64 @@ class TrezorKeyring extends EventEmitter {
     return this.model;
   }
 
+  /**
+   * Gets the vendor, if known.
+   *
+   * @returns {"trezor" | "onekey"}
+   */
+  async getVendor() {
+    return this.vendor || (this.vendor = await this.fetchVendor());
+  }
+
+  /**
+   * fetch vendor by call getFeatures
+   * @param {object} features
+   * @returns {'onekey' | 'trezor'}
+   */
+  fetchVendor() {
+    return new Promise((resolve) => {
+      TrezorConnect.getFeatures()
+        .then((response) => {
+          if (!response.success) {
+            resolve(null);
+            return;
+          }
+          const vendor = this.__getVendor(response.payload);
+          resolve(vendor);
+        })
+        .catch(() => {
+          resolve(null);
+        });
+    });
+  }
+
+  /**
+   * get the vendor name of the hardware wallet
+   * @param {object} features
+   * @returns {'onekey' | 'trezor'}
+   */
+  __getVendor(features) {
+    // If the value of features is null, set vendor to the default value
+    if (!features) {
+      return undefined;
+    }
+
+    // No special field, default is trezor device
+    if (!features.minor_version || !features.patch_version) {
+      return 'trezor';
+    }
+
+    if (
+      features.vendor === oneKeyVendor ||
+      (features.minor_version === oneKeySpecialVersion &&
+        features.patch_version === oneKeySpecialVersion)
+    ) {
+      return 'onekey';
+    }
+
+    return 'trezor';
+  }
+
   dispose() {
     // This removes the Trezor Connect iframe from the DOM
     // This method is not well documented, but the code it calls can be seen