feat: support tron with native wallet

packages/hdwallet-core/src/index.ts

@@ -18,6 +18,7 @@ export * from "./terra";
 export * from "./thorchain";
 export * from "./solana";
 export * from "./transport";
+export * from "./tron";
 export * from "./utils";
 export * from "./utxoUtils";
 export * from "./wallet";

packages/hdwallet-core/src/tron.ts

@@ -0,0 +1,87 @@
+import { addressNListToBIP32, slip44ByCoin } from "./utils";
+import { BIP32Path, HDWallet, HDWalletInfo, PathDescription } from "./wallet";
+
+export interface TronGetAddress {
+  addressNList: BIP32Path;
+  showDisplay?: boolean;
+  pubKey?: string;
+}
+
+export interface TronSignTx {
+  addressNList: BIP32Path;
+  /** Raw transaction data in hex format */
+  rawDataHex: string;
+}
+
+export interface TronSignedTx {
+  serialized: string;
+  signature: string;
+}
+
+export interface TronTxSignature {
+  signature: string;
+}
+
+export interface TronGetAccountPaths {
+  accountIdx: number;
+}
+
+export interface TronAccountPath {
+  addressNList: BIP32Path;
+}
+
+export interface TronWalletInfo extends HDWalletInfo {
+  readonly _supportsTronInfo: boolean;
+
+  /**
+   * Returns a list of bip32 paths for a given account index in preferred order
+   * from most to least preferred.
+   */
+  tronGetAccountPaths(msg: TronGetAccountPaths): Array<TronAccountPath>;
+
+  /**
+   * Returns the "next" account path, if any.
+   */
+  tronNextAccountPath(msg: TronAccountPath): TronAccountPath | undefined;
+}
+
+export interface TronWallet extends TronWalletInfo, HDWallet {
+  readonly _supportsTron: boolean;
+
+  tronGetAddress(msg: TronGetAddress): Promise<string | null>;
+  tronGetAddresses?(msgs: TronGetAddress[]): Promise<string[]>;
+  tronSignTx(msg: TronSignTx): Promise<TronSignedTx | null>;
+  tronSendTx?(msg: TronSignTx): Promise<TronTxSignature | null>;
+}
+
+export function tronDescribePath(path: BIP32Path): PathDescription {
+  const pathStr = addressNListToBIP32(path);
+  const unknown: PathDescription = {
+    verbose: pathStr,
+    coin: "Tron",
+    isKnown: false,
+  };
+
+  if (path.length != 5) return unknown;
+  if (path[0] != 0x80000000 + 44) return unknown;
+  if (path[1] != 0x80000000 + slip44ByCoin("Tron")) return unknown;
+  if ((path[2] & 0x80000000) >>> 0 !== 0x80000000) return unknown;
+  if (path[3] !== 0) return unknown;
+  if (path[4] !== 0) return unknown;
+
+  const index = path[2] & 0x7fffffff;
+  return {
+    verbose: `Tron Account #${index}`,
+    accountIdx: index,
+    wholeAccount: true,
+    coin: "Tron",
+    isKnown: true,
+  };
+}
+
+// The standard BIP44 derivation path for Tron is: m/44'/195'/<account>'/0/0
+// https://github.com/tronprotocol/tips/issues/102
+export function tronGetAccountPaths(msg: TronGetAccountPaths): Array<TronAccountPath> {
+  const slip44 = slip44ByCoin("Tron");
+  return [{ addressNList: [0x80000000 + 44, 0x80000000 + slip44, 0x80000000 + msg.accountIdx, 0, 0] }];
+}

packages/hdwallet-core/src/utils.ts

@@ -156,6 +156,7 @@ export const slip44Table = Object.freeze({
   Terra: 330,
   Kava: 459,
   Solana: 501,
+  Tron: 195,
   // EVM chains all use the same SLIP44
   Ethereum: 60,
   Avalanche: 60,

packages/hdwallet-core/src/wallet.ts

@@ -16,6 +16,7 @@ import { SolanaWallet, SolanaWalletInfo } from "./solana";
 import { TerraWallet, TerraWalletInfo } from "./terra";
 import { ThorchainWallet, ThorchainWalletInfo } from "./thorchain";
 import { Transport } from "./transport";
+import { TronWallet, TronWalletInfo } from "./tron";
 
 export type BIP32Path = Array<number>;
 
@@ -257,6 +258,14 @@ export function infoSolana(info: HDWalletInfo): info is SolanaWalletInfo {
   return isObject(info) && (info as any)._supportsSolanaInfo;
 }
 
+export function supportsTron(wallet: HDWallet): wallet is TronWallet {
+  return isObject(wallet) && (wallet as any)._supportsTron;
+}
+
+export function infoTron(info: HDWalletInfo): info is TronWalletInfo {
+  return isObject(info) && (info as any)._supportsTronInfo;
+}
+
 export function supportsDebugLink(wallet: HDWallet): wallet is DebugLinkWallet {
   return isObject(wallet) && (wallet as any)._supportsDebugLink;
 }

packages/hdwallet-native/src/crypto/isolation/adapters/index.ts

@@ -6,3 +6,4 @@ export { default as Binance } from "./binance";
 export { default as Cosmos } from "./cosmos";
 export { default as CosmosDirect } from "./cosmosDirect";
 export { default as Solana } from "./solana";
+export { default as Tron } from "./tron";

packages/hdwallet-native/src/crypto/isolation/adapters/tron.ts

@@ -0,0 +1,69 @@
+import { keccak256 } from "@ethersproject/keccak256";
+import * as core from "@shapeshiftoss/hdwallet-core";
+import * as bs58check from "bs58check";
+
+import { Isolation } from "../..";
+import { SecP256K1 } from "../core";
+
+const TRON_ADDRESS_PREFIX = 0x41;
+
+export class TronAdapter {
+  protected readonly nodeAdapter: Isolation.Adapters.BIP32;
+
+  constructor(nodeAdapter: Isolation.Adapters.BIP32) {
+    this.nodeAdapter = nodeAdapter;
+  }
+
+  /**
+   * Get TRON address from BIP32 path
+   * Address generation:
+   * 1. Get uncompressed public key (65 bytes)
+   * 2. Remove first byte (0x04)
+   * 3. Hash with Keccak256
+   * 4. Take last 20 bytes
+   * 5. Prepend 0x41
+   * 6. Base58check encode
+   */
+  async getAddress(addressNList: core.BIP32Path): Promise<string> {
+    const nodeAdapter = await this.nodeAdapter.derivePath(core.addressNListToBIP32(addressNList));
+    const publicKey = SecP256K1.UncompressedPoint.from(nodeAdapter.getPublicKey());
+
+    // Remove the 0x04 prefix from uncompressed public key
+    const publicKeyBytes = publicKey.slice(1);
+
+    // Hash with Keccak256
+    const hash = keccak256("0x" + Buffer.from(publicKeyBytes).toString("hex"));
+
+    // Take last 20 bytes and prepend 0x41 (TRON prefix)
+    const addressBytes = Buffer.concat([Buffer.from([TRON_ADDRESS_PREFIX]), Buffer.from(hash.slice(-40), "hex")]);
+
+    // Base58check encode
+    return bs58check.encode(addressBytes);
+  }
+
+  /**
+   * Sign TRON transaction
+   * Transaction signing:
+   * 1. Get raw transaction hex
+   * 2. Hash with SHA256 (NOT Keccak256)
+   * 3. Sign with secp256k1
+   * 4. Return hex signature
+   */
+  async signTransaction(rawDataHex: string, addressNList: core.BIP32Path): Promise<string> {
+    const nodeAdapter = await this.nodeAdapter.derivePath(core.addressNListToBIP32(addressNList));
+    const txBuf = Buffer.from(rawDataHex, "hex");
+
+    // TRON uses SHA256 for transaction signing
+    const recoverableSig = await SecP256K1.RecoverableSignature.signCanonically(nodeAdapter.node, "sha256", txBuf);
+
+    const sig = SecP256K1.RecoverableSignature.sig(recoverableSig);
+    const recoveryParam = SecP256K1.RecoverableSignature.recoveryParam(recoverableSig);
+
+    // Concatenate signature (64 bytes) + recovery param (1 byte)
+    const fullSig = core.compatibleBufferConcat([sig, Buffer.from([recoveryParam])]);
+
+    return fullSig.toString("hex");
+  }
+}
+
+export default TronAdapter;

packages/hdwallet-native/src/native.ts

@@ -18,6 +18,7 @@ import { MixinNativeSecretWallet, MixinNativeSecretWalletInfo } from "./secret";
 import { MixinNativeSolanaWallet, MixinNativeSolanaWalletInfo } from "./solana";
 import { MixinNativeTerraWallet, MixinNativeTerraWalletInfo } from "./terra";
 import { MixinNativeThorchainWallet, MixinNativeThorchainWalletInfo } from "./thorchain";
+import { MixinNativeTronWallet, MixinNativeTronWalletInfo } from "./tron";
 
 export enum NativeEvents {
   MNEMONIC_REQUIRED = "MNEMONIC_REQUIRED",
@@ -126,12 +127,14 @@ class NativeHDWalletInfo
       MixinNativeCosmosWalletInfo(
         MixinNativeBinanceWalletInfo(
           MixinNativeSolanaWalletInfo(
-            MixinNativeThorchainWalletInfo(
-              MixinNativeMayachainWalletInfo(
-                MixinNativeSecretWalletInfo(
-                  MixinNativeTerraWalletInfo(
-                    MixinNativeKavaWalletInfo(
-                      MixinNativeArkeoWalletInfo(MixinNativeOsmosisWalletInfo(NativeHDWalletBase))
+            MixinNativeTronWalletInfo(
+              MixinNativeThorchainWalletInfo(
+                MixinNativeMayachainWalletInfo(
+                  MixinNativeSecretWalletInfo(
+                    MixinNativeTerraWalletInfo(
+                      MixinNativeKavaWalletInfo(
+                        MixinNativeArkeoWalletInfo(MixinNativeOsmosisWalletInfo(NativeHDWalletBase))
+                      )
                     )
                   )
                 )
@@ -149,6 +152,7 @@ class NativeHDWalletInfo
     core.CosmosWalletInfo,
     core.BinanceWalletInfo,
     core.SolanaWalletInfo,
+    core.TronWalletInfo,
     core.ThorchainWalletInfo,
     core.MayachainWalletInfo,
     core.SecretWalletInfo,
@@ -203,6 +207,9 @@ class NativeHDWalletInfo
         return core.osmosisDescribePath(msg.path);
       case "arkeo":
         return core.arkeoDescribePath(msg.path);
+      case "tron":
+      case "trx":
+        return core.tronDescribePath(msg.path);
       default:
         throw new Error("Unsupported path");
     }
@@ -215,11 +222,13 @@ export class NativeHDWallet
       MixinNativeCosmosWallet(
         MixinNativeBinanceWallet(
           MixinNativeSolanaWallet(
-            MixinNativeThorchainWallet(
-              MixinNativeMayachainWallet(
-                MixinNativeSecretWallet(
-                  MixinNativeTerraWallet(
-                    MixinNativeKavaWallet(MixinNativeOsmosisWallet(MixinNativeArkeoWallet(NativeHDWalletInfo)))
+            MixinNativeTronWallet(
+              MixinNativeThorchainWallet(
+                MixinNativeMayachainWallet(
+                  MixinNativeSecretWallet(
+                    MixinNativeTerraWallet(
+                      MixinNativeKavaWallet(MixinNativeOsmosisWallet(MixinNativeArkeoWallet(NativeHDWalletInfo)))
+                    )
                   )
                 )
               )
@@ -236,6 +245,7 @@ export class NativeHDWallet
     core.CosmosWallet,
     core.BinanceWallet,
     core.SolanaWallet,
+    core.TronWallet,
     core.ThorchainWallet,
     core.MayachainWallet,
     core.SecretWallet,
@@ -359,6 +369,7 @@ export class NativeHDWallet
           super.cosmosInitializeWallet(secp256k1MasterKey),
           super.osmosisInitializeWallet(secp256k1MasterKey),
           super.binanceInitializeWallet(secp256k1MasterKey),
+          super.tronInitializeWallet(secp256k1MasterKey),
           super.thorchainInitializeWallet(secp256k1MasterKey),
           super.mayachainInitializeWallet(secp256k1MasterKey),
           super.secretInitializeWallet(secp256k1MasterKey),
@@ -412,6 +423,7 @@ export class NativeHDWallet
     super.cosmosWipe();
     super.osmosisWipe();
     super.binanceWipe();
+    super.tronWipe();
     super.thorchainWipe();
     super.mayachainWipe();
     super.secretWipe();

packages/hdwallet-native/src/tron.ts

@@ -0,0 +1,67 @@
+import * as core from "@shapeshiftoss/hdwallet-core";
+
+import { Isolation } from "./crypto";
+import { TronAdapter } from "./crypto/isolation/adapters/tron";
+import { NativeHDWalletBase } from "./native";
+
+export function MixinNativeTronWalletInfo<TBase extends core.Constructor<core.HDWalletInfo>>(Base: TBase) {
+  // eslint-disable-next-line @typescript-eslint/no-shadow
+  return class MixinNativeTronWalletInfo extends Base implements core.TronWalletInfo {
+    readonly _supportsTronInfo = true;
+
+    tronGetAccountPaths(msg: core.TronGetAccountPaths): Array<core.TronAccountPath> {
+      return core.tronGetAccountPaths(msg);
+    }
+
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    tronNextAccountPath(msg: core.TronAccountPath): core.TronAccountPath | undefined {
+      throw new Error("Method not implemented");
+    }
+  };
+}
+
+export function MixinNativeTronWallet<TBase extends core.Constructor<NativeHDWalletBase>>(Base: TBase) {
+  // eslint-disable-next-line @typescript-eslint/no-shadow
+  return class MixinNativeTronWallet extends Base {
+    readonly _supportsTron = true;
+
+    tronAdapter: TronAdapter | undefined;
+
+    async tronInitializeWallet(masterKey: Isolation.Core.BIP32.Node): Promise<void> {
+      const nodeAdapter = await Isolation.Adapters.BIP32.create(masterKey);
+      this.tronAdapter = new TronAdapter(nodeAdapter);
+    }
+
+    tronWipe() {
+      this.tronAdapter = undefined;
+    }
+
+    async tronGetAddress(msg: core.TronGetAddress): Promise<string | null> {
+      return this.needsMnemonic(!!this.tronAdapter, () => {
+        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+        return this.tronAdapter!.getAddress(msg.addressNList);
+      });
+    }
+
+    async tronSignTx(msg: core.TronSignTx): Promise<core.TronSignedTx | null> {
+      return this.needsMnemonic(!!this.tronAdapter, async () => {
+        const address = await this.tronGetAddress({
+          addressNList: msg.addressNList,
+          showDisplay: false,
+        });
+
+        if (!address) throw new Error("Failed to get TRON address");
+
+        const signature = await this.tronAdapter!.signTransaction(msg.rawDataHex, msg.addressNList);
+
+        // Serialized transaction = rawDataHex + signature
+        const serialized = msg.rawDataHex + signature;
+
+        return {
+          serialized,
+          signature,
+        };
+      });
+    }
+  };
+}