Merge pull request #2 from haqq-network/issue/HW-731/refactor-ui-trx-addresses-should-be-displayed-correctly

feat(HW-731): TRON private key and address generation

Author: devkudasov

.yarnrc.yml

@@ -0,0 +1 @@
+nodeLinker: node-modules

package.json

@@ -8,7 +8,6 @@
     "dist/**/*"
   ],
   "license": "MIT",
-  "private": false,
   "repository": {
     "type": "git",
     "url": "https://github.com/haqq-network/haqq-rn-wallet-providers.git"
@@ -33,6 +32,7 @@
     "@ledgerhq/hw-transport": "6.30.3",
     "@ledgerhq/react-native-hw-transport-ble": "6.30.0",
     "bech32-converting": "^1.0.9",
+    "bip32": "1.0.4",
     "bip39": "^3.1.0",
     "ethers": "^5.7.2",
     "events": "^3.3.0",
@@ -48,11 +48,11 @@
     "react-native-encrypted-storage": "*"
   },
   "devDependencies": {
-    "@react-native-community/eslint-config": "3.2.0",
     "@haqq/encryption-react-native": "^0.0.3",
     "@haqq/provider-web3-utils": "^0.0.13",
     "@haqq/shared-react-native": "^0.0.8",
     "@ledgerhq/types-devices": "^6.23.0",
+    "@react-native-community/eslint-config": "3.2.0",
     "@types/jest": "^29.5.1",
     "@types/node": "^20.1.2",
     "@types/react-native": "^0.71.12",
@@ -90,15 +90,15 @@
     "vm": "vm-browserify"
   },
   "browser": {
+    "_stream_duplex": "readable-stream/duplex",
+    "_stream_passthrough": "readable-stream/passthrough",
+    "_stream_readable": "readable-stream/readable",
+    "_stream_transform": "readable-stream/transform",
+    "_stream_writable": "readable-stream/writable",
     "crypto": "react-native-crypto",
     "http": "@tradle/react-native-http",
     "https": "https-browserify",
     "os": "react-native-os",
-    "_stream_transform": "readable-stream/transform",
-    "_stream_readable": "readable-stream/readable",
-    "_stream_writable": "readable-stream/writable",
-    "_stream_duplex": "readable-stream/duplex",
-    "_stream_passthrough": "readable-stream/passthrough",
     "stream": "stream-browserify",
     "vm": "vm-browserify"
   }

src/providers/mnemonic/provider.ts

@@ -7,10 +7,11 @@ import EncryptedStorage from 'react-native-encrypted-storage';
 import {ProviderMnemonicBaseOptions} from './types';
 
 import {ITEM_KEYS, WalletType} from '../../constants';
+import {Multichain} from '../../services/multichain';
 import {compressPublicKey} from '../../utils';
 import {getMnemonic} from '../../utils/mnemonic/get-mnemonic';
 import {ProviderBase} from '../base-provider';
-import {ProviderBaseOptions, ProviderInterface} from '../types';
+import {NETWORK_TYPE, ProviderBaseOptions, ProviderInterface} from '../types';
 
 export class ProviderMnemonicBase
   extends ProviderBase<ProviderMnemonicBaseOptions>
@@ -27,6 +28,7 @@ export class ProviderMnemonicBase
       mnemonic === null
         ? (await generateEntropy(16)).toString('hex')
         : mnemonicToEntropy(mnemonic);
+
     const seed = await mnemonicToSeed(entropyToMnemonic(entropy));
 
     const privateData = await encryptShare(
@@ -117,7 +119,7 @@ export class ProviderMnemonicBase
   }
 
   async getAccountInfo(hdPath: string) {
-    let resp = {publicKey: '', address: ''};
+    let resp = {publicKey: '', address: '', tronAddress: ''};
     try {
       const share = await getMnemonic(
         this._options.account,
@@ -130,17 +132,22 @@ export class ProviderMnemonicBase
 
       const seed = await ProviderMnemonicBase.shareToSeed(share);
 
-      const privateKey = await derive(seed, hdPath);
+      const ethPrivateKey = await derive(seed, hdPath);
 
-      if (!privateKey) {
+      if (!ethPrivateKey) {
         throw new Error('private_key_not_found');
       }
 
-      const account = await accountInfo(privateKey);
+      const account = await accountInfo(ethPrivateKey);
 
       resp = {
         publicKey: compressPublicKey(account.publicKey),
         address: account.address,
+        tronAddress: await Multichain.generateAddress(
+          NETWORK_TYPE.TRON,
+          hdPath,
+          await this.getMnemonicPhrase(),
+        ),
       };
       this.emit('getPublicKeyForHDPath', true);
     } catch (e) {

src/providers/sss/provider.ts

@@ -14,14 +14,15 @@ import EncryptedStorage from 'react-native-encrypted-storage';
 import {ProviderSSSBaseOptions, StorageInterface} from './types';
 
 import {ITEM_KEYS, WalletType} from '../../constants';
+import {Multichain} from '../../services/multichain';
 import {
   Polynomial,
   compressPublicKey,
   getSeed,
   lagrangeInterpolation,
 } from '../../utils';
 import {ProviderBase} from '../base-provider';
-import {ProviderBaseOptions, ProviderInterface} from '../types';
+import {NETWORK_TYPE, ProviderBaseOptions, ProviderInterface} from '../types';
 
 export class ProviderSSSBase
   extends ProviderBase<ProviderSSSBaseOptions>
@@ -244,7 +245,7 @@ export class ProviderSSSBase
   }
 
   async getAccountInfo(hdPath: string) {
-    let resp = {publicKey: '', address: ''};
+    let resp = {publicKey: '', address: '', tronAddress: ''};
     try {
       const {seed} = await getSeed(
         this._options.account,
@@ -267,6 +268,11 @@ export class ProviderSSSBase
       resp = {
         publicKey: compressPublicKey(account.publicKey),
         address: account.address,
+        tronAddress: await Multichain.generateAddress(
+          NETWORK_TYPE.TRON,
+          hdPath,
+          await this.getMnemonicPhrase(),
+        ),
       };
       this.emit('getPublicKeyForHDPath', true);
     } catch (e) {

src/providers/types.ts

@@ -111,3 +111,9 @@ export type ProviderBaseError = {
   error: Error;
   source: string;
 };
+
+export enum NETWORK_TYPE {
+  HAQQ,
+  ETH,
+  TRON,
+}

src/services/multichain.ts

@@ -0,0 +1,79 @@
+import * as bip32 from 'bip32';
+import {mnemonicToSeed} from 'bip39';
+import * as CryptoJS from 'crypto-js';
+import {ec} from 'elliptic';
+
+import {NETWORK_TYPE} from '../providers/types';
+import {encodeBase58} from '../utils';
+
+export class Multichain {
+  static async generatePrivateKey(
+    type: NETWORK_TYPE,
+    hdPath: string,
+    mnemonic: string,
+  ): Promise<string> {
+    switch (type) {
+      case NETWORK_TYPE.TRON:
+        return Multichain.generateTronPrivateKey(hdPath, mnemonic);
+      default:
+        return '';
+    }
+  }
+
+  private static async generateTronPrivateKey(
+    hdPath: string,
+    mnemonic: string,
+  ): Promise<string> {
+    const tronSeed = await mnemonicToSeed(mnemonic);
+    const root = bip32.fromSeed(tronSeed);
+    const child = root.derivePath(hdPath);
+    return child.privateKey.toString('hex');
+  }
+
+  static async generateAddress(
+    type: NETWORK_TYPE,
+    hdPath: string,
+    mnemonic: string,
+  ): Promise<string> {
+    switch (type) {
+      case NETWORK_TYPE.TRON:
+        return Multichain.generateTronAddress(
+          await this.generateTronPrivateKey(hdPath, mnemonic),
+        );
+      default:
+        return '';
+    }
+  }
+
+  private static async generateTronAddress(
+    privateKey: string,
+  ): Promise<string> {
+    const EC = new ec('secp256k1');
+
+    // Create key pair from private key
+    const keyPair = EC.keyFromPrivate(privateKey);
+
+    // Get public key and remove prefix 0x
+    const publicKey = keyPair.getPublic('hex').slice(2);
+
+    // Hashing public key with Keccak256
+    const publicKeyHash = CryptoJS.SHA3(publicKey, {
+      outputLength: 256,
+    }).toString();
+
+    // Get last 20 bites (40 hex symbols) from public key's hash
+    // Also add TRON prefix 41
+    const tronAddressHex = '41' + publicKeyHash.slice(-40);
+
+    // Convert hex-address to base58 which is TRON address format
+    const addressBuffer = Buffer.from(tronAddressHex, 'hex');
+    const addressHash = CryptoJS.SHA256(
+      CryptoJS.SHA256(addressBuffer).toString(CryptoJS.enc.Hex),
+    ).toString(CryptoJS.enc.Hex);
+    const checksum = addressHash.slice(0, 8);
+
+    return encodeBase58(
+      Buffer.concat([addressBuffer, Buffer.from(checksum, 'hex')]),
+    );
+  }
+}

src/utils/utils.ts

@@ -30,3 +30,28 @@ export function hexBuffer(str: string | Buffer): Buffer {
 export function normalize0x(strWith0x: string): string {
   return strWith0x.replace(/^(0x)*/, '0x');
 }
+
+export function encodeBase58(buffer: Buffer): string {
+  const alphabet = '123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz';
+  const base = BigInt(58);
+
+  let result = '';
+  let num = BigInt('0x' + buffer.toString('hex'));
+
+  while (num > 0) {
+    const remainder = num % base;
+    num = num / base;
+    result = alphabet[Number(remainder)] + result;
+  }
+
+  // Добавляем ведущие нули
+  for (const byte of buffer) {
+    if (byte === 0x00) {
+      result = alphabet[0] + result;
+    } else {
+      break;
+    }
+  }
+
+  return result;
+}

tsconfig.json

@@ -1,6 +1,6 @@
 {
   "compilerOptions": {
-    "target": "es5",
+    "target": "es2015",
     "module": "commonjs",
     "esModuleInterop": true,
     "skipLibCheck": true,