feat: add sdk changes for mpc tss support for TRON

TICKET: CHALO-33

Author: bhuvanr159

modules/sdk-coin-trx/package.json

@@ -47,6 +47,7 @@
   },
   "dependencies": {
     "@bitgo/sdk-core": "^36.33.2",
+    "@bitgo/sdk-lib-mpc": "^10.9.0",
     "@bitgo/secp256k1": "^1.10.0",
     "@bitgo/statics": "^58.29.0",
     "@stablelib/hex": "^1.0.0",

modules/sdk-coin-trx/src/lib/transaction.ts

@@ -318,6 +318,14 @@ export class Transaction extends BaseTransaction {
     return this._inputs;
   }
 
+  /** @inheritDoc */
+  get signablePayload(): Buffer {
+    if (!this._transaction) {
+      throw new ParseTransactionError('Empty transaction');
+    }
+    return Buffer.from(this._transaction.raw_data_hex, 'hex');
+  }
+
   /** @inheritdoc */
   canSign(key: BaseKey): boolean {
     // Tron transaction do not contain the owners account address so it is not possible to check the

modules/sdk-coin-trx/src/lib/transactionBuilder.ts

@@ -8,6 +8,7 @@ import {
   BuildTransactionError,
   InvalidTransactionError,
   ParseTransactionError,
+  PublicKey as BasePublicKey,
   SigningError,
   ExtendTransactionError,
   InvalidParameterValueError,
@@ -36,6 +37,8 @@ export class TransactionBuilder extends BaseTransactionBuilder {
   protected _refBlockHash: string;
   protected _expiration: number;
   protected _timestamp: number;
+  protected _signature?: Buffer;
+  protected _signaturePublicKey?: string;
 
   /**
    * Public constructor.
@@ -162,12 +165,29 @@ export class TransactionBuilder extends BaseTransactionBuilder {
     return new Transaction(this._coinConfig, signedTransaction);
   }
 
+  /** @inheritDoc */
+  addSignature(publicKey: BasePublicKey, signature: Buffer): void {
+    this._signature = signature;
+    this._signaturePublicKey = publicKey.pub;
+  }
+
   /** @inheritdoc */
   protected async buildImplementation(): Promise<Transaction> {
-    // This is a no-op since Tron transactions are built from
     if (!this.transaction.id) {
       throw new BuildTransactionError('A valid transaction must have an id');
     }
+
+    if (this._signature) {
+      const txJson = this.transaction.toJson();
+      const signatures = txJson.signature ? [...txJson.signature] : [];
+      signatures.push(this._signature.toString('hex'));
+      const updatedReceipt: TransactionReceipt = {
+        ...txJson,
+        signature: signatures,
+      };
+      return new Transaction(this._coinConfig, updatedReceipt);
+    }
+
     return Promise.resolve(this.transaction);
   }
 

modules/sdk-coin-trx/src/lib/wrappedBuilder.ts

@@ -1,6 +1,6 @@
 import BigNumber from 'bignumber.js';
 import { BaseCoin as CoinConfig } from '@bitgo/statics';
-import { BaseKey, BaseTransaction, InvalidTransactionError } from '@bitgo/sdk-core';
+import { BaseKey, BaseTransaction, InvalidTransactionError, PublicKey as BasePublicKey } from '@bitgo/sdk-core';
 import { Transaction } from './transaction';
 import { Address } from './address';
 import { TransactionBuilder } from './transactionBuilder';
@@ -137,6 +137,11 @@ export class WrappedBuilder extends TransactionBuilder {
     this._builder.sign(key);
   }
 
+  /** @inheritDoc */
+  addSignature(publicKey: BasePublicKey, signature: Buffer): void {
+    this._builder.addSignature(publicKey, signature);
+  }
+
   /** @inheritdoc */
   async build(): Promise<BaseTransaction> {
     return this._builder.build();