feat: unsigned sweep for wemix

Ticket: WIN-4884

TICKET: WIN-4884

Author: prithvishet2503

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

@@ -6,6 +6,7 @@ import { TransferBuilder } from './transferBuilder';
 
 export class TransactionBuilder extends AbstractTransactionBuilder {
   protected _transfer: TransferBuilder;
+  private _signatures: any;
 
   constructor(_coinConfig: Readonly<CoinConfig>) {
     super(_coinConfig);
@@ -24,6 +25,11 @@ export class TransactionBuilder extends AbstractTransactionBuilder {
     return this._transfer;
   }
 
+  addSignature(publicKey, signature) {
+    this._signatures = [];
+    this._signatures.push({ publicKey, signature });
+  }
+
   protected getContractData(addresses: string[]): string {
     throw new Error('Method not implemented.');
   }

modules/sdk-coin-wemix/src/wemix.ts

@@ -8,7 +8,13 @@
 
 import { BaseCoin, BitGoBase, common, MPCAlgorithm, MultisigType, multisigTypes } from '@bitgo/sdk-core';
 import { BaseCoin as StaticsBaseCoin, coins } from '@bitgo/statics';
-import { AbstractEthLikeNewCoins, recoveryBlockchainExplorerQuery } from '@bitgo/abstract-eth';
+import {
+  AbstractEthLikeNewCoins,
+  recoveryBlockchainExplorerQuery,
+  UnsignedSweepTxMPCv2,
+  RecoverOptions,
+  OfflineVaultTxInfo,
+} from '@bitgo/abstract-eth';
 import { TransactionBuilder } from './lib';
 
 export class Wemix extends AbstractEthLikeNewCoins {
@@ -39,6 +45,10 @@ export class Wemix extends AbstractEthLikeNewCoins {
     return 'ecdsa';
   }
 
+  protected async buildUnsignedSweepTxnTSS(params: RecoverOptions): Promise<OfflineVaultTxInfo | UnsignedSweepTxMPCv2> {
+    return this.buildUnsignedSweepTxnMPCv2(params);
+  }
+
   async recoveryBlockchainExplorerQuery(query: Record<string, string>): Promise<Record<string, unknown>> {
     const apiToken = common.Environments[this.bitgo.getEnv()].wemixExplorerApiToken;
     const explorerUrl = common.Environments[this.bitgo.getEnv()].wemixExplorerBaseUrl;

modules/sdk-coin-wemix/test/resources.ts

@@ -9,3 +9,48 @@ export const SEND_TX_BROADCAST_LEGACY =
 
 export const SEND_TX_AMOUNT_ZERO_BROADCAST =
   '0xf901cc02843b9aca0083b8a1a0948f977e912ef500548a0c3be6ddde9899f1199b8180b901643912521500000000000000000000000019645032c7f1533395d44a629462e751084d3e4c000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000c0000000000000000000000000000000000000000000000000000000005ec67da8000000000000000000000000000000000000000000000000000000000000000500000000000000000000000000000000000000000000000000000000000000e0000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000411badc5af505b10ebf6bc80322567ac71af7970a9ac1ebdf33f656bbfa2b40f652239e4ecd9b7b391db68fb84467e11dacb5d54f38770d79f4c95c9c1d4eba0a21c000000000000000000000000000000000000000000000000000000000000008208d3a01c830bee5c43a339f4f6bec2e705045986482d72cf0c504183c121a4c246a57ba068b7460d6e0b0e5b354053c78cd412286adc79e21ec3bb9169930b47e455b071';
+
+const getTxListRequestUnsignedSweep: Record<string, string> = {
+  module: 'account',
+  action: 'txlist',
+  address: '0x742838193c4169f6b2ba7b0e03f723c3ba0928e1',
+};
+
+const getTxListResponseUnsignedSweep: Record<string, unknown> = {
+  status: '1',
+  result: [
+    {
+      hash: '0xede855d43d70ea1bb75db63d4f75113dae0845f0d4bdb0b2d8bda55249c70812',
+      nonce: '23',
+      from: '0x742838193c4169f6b2ba7b0e03f723c3ba0928e1',
+    },
+  ],
+  message: 'OK',
+};
+
+const getBalanceRequestUnsignedSweep: Record<string, string> = {
+  module: 'account',
+  action: 'balance',
+  address: '0x742838193c4169f6b2ba7b0e03f723c3ba0928e1',
+};
+
+const getBalanceResponseUnsignedSweep: Record<string, unknown> = {
+  status: '1',
+  result: '100000000000000000',
+  message: 'OK',
+};
+
+export const mockDataUnsignedSweep = {
+  userKey:
+    '029d2ded2d39ee7cd8d8bbba8b25e4c60bb09297936fa6b223de1f495b5ee20dcaf762367f9691f7719cb5e13e59d725669a18aad1e2522dd141fa4c7fd3d25c17',
+  backupKey:
+    '029d2ded2d39ee7cd8d8bbba8b25e4c60bb09297936fa6b223de1f495b5ee20dcaf762367f9691f7719cb5e13e59d725669a18aad1e2522dd141fa4c7fd3d25c17',
+  derivationPath: 'm/0',
+  derivationSeed: '',
+  walletBaseAddress: '0x742838193c4169f6b2ba7b0e03f723c3ba0928e1',
+  recoveryDestination: '0xd76b586901850f2c656db0cbef795c0851bbec35',
+  getTxListRequest: getTxListRequestUnsignedSweep,
+  getTxListResponse: getTxListResponseUnsignedSweep,
+  getBalanceRequest: getBalanceRequestUnsignedSweep,
+  getBalanceResponse: getBalanceResponseUnsignedSweep,
+};

modules/sdk-coin-wemix/test/unit/wemix.ts

@@ -1,9 +1,13 @@
-import 'should';
+import * as should from 'should';
 
 import { TestBitGo, TestBitGoAPI } from '@bitgo/sdk-test';
 import { BitGoAPI } from '@bitgo/sdk-api';
 
 import { Wemix, Twemix } from '../../src/index';
+import { UnsignedSweepTxMPCv2 } from '@bitgo/abstract-eth';
+import { mockDataUnsignedSweep } from '../resources';
+import nock from 'nock';
+import { common } from '@bitgo/sdk-core';
 
 const bitgo: TestBitGoAPI = TestBitGo.decorate(BitGoAPI, { env: 'test' });
 
@@ -40,3 +44,55 @@ describe('wemix', function () {
     });
   });
 });
+
+describe('Build Unsigned Sweep for Self-Custody Cold Wallets (MPCv2)', function () {
+  const bitgo = TestBitGo.decorate(BitGoAPI, { env: 'test' });
+  const explorerUrl = common.Environments[bitgo.getEnv()].wemixExplorerBaseUrl as string;
+
+  it('should generate an unsigned sweep without derivation path', async () => {
+    nock(explorerUrl)
+      .get('/api')
+      .twice()
+      .query(mockDataUnsignedSweep.getTxListRequest)
+      .reply(200, mockDataUnsignedSweep.getTxListResponse);
+    nock(explorerUrl)
+      .get('/api')
+      .query(mockDataUnsignedSweep.getBalanceRequest)
+      .reply(200, mockDataUnsignedSweep.getBalanceResponse);
+
+    const baseCoin: any = bitgo.coin('twemix');
+    const transaction = (await baseCoin.recover({
+      userKey: mockDataUnsignedSweep.userKey,
+      backupKey: mockDataUnsignedSweep.backupKey,
+      walletContractAddress: mockDataUnsignedSweep.walletBaseAddress,
+      recoveryDestination: mockDataUnsignedSweep.recoveryDestination,
+      isTss: true,
+      eip1559: { maxFeePerGas: 20000000000, maxPriorityFeePerGas: 10000000000 },
+      gasLimit: 500000,
+      replayProtectionOptions: {
+        chain: 1112,
+        hardfork: 'london',
+      },
+    })) as UnsignedSweepTxMPCv2;
+    should.exist(transaction);
+    transaction.should.have.property('txRequests');
+    transaction.txRequests.length.should.equal(1);
+    const txRequest = transaction.txRequests[0];
+    txRequest.should.have.property('walletCoin');
+    txRequest.walletCoin.should.equal('twemix');
+    txRequest.should.have.property('transactions');
+    txRequest.transactions.length.should.equal(1);
+    const tx = txRequest.transactions[0];
+    tx.should.have.property('nonce');
+    tx.should.have.property('unsignedTx');
+    tx.unsignedTx.should.have.property('serializedTxHex');
+    tx.unsignedTx.should.have.property('signableHex');
+    tx.unsignedTx.should.have.property('derivationPath');
+    tx.unsignedTx.should.have.property('feeInfo');
+    tx.unsignedTx.feeInfo?.should.have.property('fee');
+    tx.unsignedTx.feeInfo?.should.have.property('feeString');
+    tx.unsignedTx.should.have.property('parsedTx');
+    tx.unsignedTx.parsedTx?.should.have.property('spendAmount');
+    tx.unsignedTx.parsedTx?.should.have.property('outputs');
+  });
+});