Feat: Extend Evm Vault swaps functionality

bouncer/package.json

@@ -6,7 +6,7 @@
     "prettier:write": "prettier --write ."
   },
   "dependencies": {
-    "@chainflip/cli": "1.6.3",
+    "@chainflip/cli": "1.8.0-cf-parameters-rename.5",
     "@chainflip/utils": "^0.4.0",
     "@coral-xyz/anchor": "^0.30.1",
     "@iarna/toml": "^2.2.5",

bouncer/shared/contract_swap.ts

@@ -5,44 +5,67 @@ import {
   approveVault,
   Asset as SCAsset,
   Chains,
-  InternalAsset,
   Chain,
 } from '@chainflip/cli';
-import { HDNodeWallet, Wallet, getDefaultProvider } from 'ethers';
+import { HDNodeWallet } from 'ethers';
+import { randomBytes } from 'crypto';
 import {
   observeBalanceIncrease,
   getContractAddress,
   observeCcmReceived,
   amountToFineAmount,
   defaultAssetAmounts,
   chainFromAsset,
-  getEvmEndpoint,
   assetDecimals,
   stateChainAssetFromAsset,
-  chainGasAsset,
+  createEvmWalletAndFund,
+  newAddress,
 } from './utils';
 import { getBalance } from './get_balance';
-import { CcmDepositMetadata } from '../shared/new_swap';
-import { send } from './send';
+import { CcmDepositMetadata, DcaParams, FillOrKillParamsX128 } from '../shared/new_swap';
 import { SwapContext, SwapStatus } from './swap_context';
 
 const erc20Assets: Asset[] = ['Flip', 'Usdc', 'Usdt', 'ArbUsdc'];
 
 export async function executeContractSwap(
-  srcAsset: Asset,
+  sourceAsset: Asset,
   destAsset: Asset,
   destAddress: string,
-  wallet: HDNodeWallet,
   messageMetadata?: CcmDepositMetadata,
+  amount?: string,
+  boostFeeBps?: number,
+  fillOrKillParams?: FillOrKillParamsX128,
+  dcaParams?: DcaParams,
+  wallet?: HDNodeWallet,
 ): ReturnType<typeof executeSwap> {
-  const srcChain = chainFromAsset(srcAsset);
+  const srcChain = chainFromAsset(sourceAsset);
   const destChain = chainFromAsset(destAsset);
+  const amountToSwap = amount ?? defaultAssetAmounts(sourceAsset);
+
+  const refundAddress = await newAddress(sourceAsset, randomBytes(32).toString('hex'));
+  const fokParams = fillOrKillParams ?? {
+    retryDurationBlocks: 0,
+    refundAddress,
+    minPriceX128: '0',
+  };
+
+  const evmWallet = wallet ?? (await createEvmWalletAndFund(sourceAsset));
+
+  if (erc20Assets.includes(sourceAsset)) {
+    // Doing effectively infinite approvals to make sure it doesn't fail.
+    // eslint-disable-next-line @typescript-eslint/no-use-before-define
+    await approveTokenVault(
+      sourceAsset,
+      (BigInt(amountToFineAmount(amountToSwap, assetDecimals(sourceAsset))) * 100n).toString(),
+      evmWallet,
+    );
+  }
 
   const networkOptions = {
-    signer: wallet,
+    signer: evmWallet,
     network: 'localnet',
     vaultContractAddress: getContractAddress(srcChain, 'VAULT'),
-    srcTokenContractAddress: getContractAddress(srcChain, srcAsset),
+    srcTokenContractAddress: getContractAddress(srcChain, sourceAsset),
   } as const;
   const txOptions = {
     // This is run with fresh addresses to prevent nonce issues. Will be 1 for ERC20s.
@@ -55,15 +78,21 @@ export async function executeContractSwap(
       destAsset: stateChainAssetFromAsset(destAsset),
       // It is important that this is large enough to result in
       // an amount larger than existential (e.g. on Polkadot):
-      amount: amountToFineAmount(defaultAssetAmounts(srcAsset), assetDecimals(srcAsset)),
+      amount: amountToFineAmount(amountToSwap, assetDecimals(sourceAsset)),
       destAddress,
-      srcAsset: stateChainAssetFromAsset(srcAsset),
+      srcAsset: stateChainAssetFromAsset(sourceAsset),
       srcChain,
       ccmParams: messageMetadata && {
         gasBudget: messageMetadata.gasBudget.toString(),
         message: messageMetadata.message,
-        cfParameters: messageMetadata.cfParameters,
+        ccmAdditionalData: messageMetadata.ccmAdditionalData,
       },
+      // The SDK will encode these parameters and the ccmAdditionalData
+      // into the `cfParameters` field for the vault swap.
+      boostFeeBps,
+      fillOrKillParams: fokParams,
+      dcaParams,
+      beneficiaries: undefined,
     } as ExecuteSwapParams,
     networkOptions,
     txOptions,
@@ -86,37 +115,18 @@ export async function performSwapViaContract(
   messageMetadata?: CcmDepositMetadata,
   swapContext?: SwapContext,
   log = true,
+  amount?: string,
+  boostFeeBps?: number,
+  fillOrKillParams?: FillOrKillParamsX128,
+  dcaParams?: DcaParams,
 ): Promise<ContractSwapParams> {
   const tag = swapTag ?? '';
-
-  const srcChain = chainFromAsset(sourceAsset);
-
-  // Generate a new wallet for each contract swap to prevent nonce issues when running in parallel
-  // with other swaps via deposit channels.
-  const mnemonic = Wallet.createRandom().mnemonic?.phrase ?? '';
-  if (mnemonic === '') {
-    throw new Error('Failed to create random mnemonic');
-  }
-  const wallet = Wallet.fromPhrase(mnemonic).connect(getDefaultProvider(getEvmEndpoint(srcChain)));
+  const amountToSwap = amount ?? defaultAssetAmounts(sourceAsset);
 
   try {
-    // Fund new key with native asset and asset to swap.
-    await send(chainGasAsset(srcChain) as InternalAsset, wallet.address);
-    await send(sourceAsset, wallet.address);
-
-    if (erc20Assets.includes(sourceAsset)) {
-      // Doing effectively infinite approvals to make sure it doesn't fail.
-      // eslint-disable-next-line @typescript-eslint/no-use-before-define
-      await approveTokenVault(
-        sourceAsset,
-        (
-          BigInt(amountToFineAmount(defaultAssetAmounts(sourceAsset), assetDecimals(sourceAsset))) *
-          100n
-        ).toString(),
-        wallet,
-      );
-    }
-    swapContext?.updateStatus(swapTag, SwapStatus.ContractApproved);
+    // Generate a new wallet for each contract swap to prevent nonce issues when running in parallel
+    // with other swaps via deposit channels.
+    const wallet = await createEvmWalletAndFund(sourceAsset);
 
     const oldBalance = await getBalance(destAsset, destAddress);
     if (log) {
@@ -133,8 +143,12 @@ export async function performSwapViaContract(
       sourceAsset,
       destAsset,
       destAddress,
-      wallet,
       messageMetadata,
+      amountToSwap,
+      boostFeeBps,
+      fillOrKillParams,
+      dcaParams,
+      wallet,
     );
     swapContext?.updateStatus(swapTag, SwapStatus.ContractExecuted);
 
@@ -171,24 +185,24 @@ export async function performSwapViaContract(
     throw new Error(`${tag} ${err}`);
   }
 }
-export async function approveTokenVault(srcAsset: Asset, amount: string, wallet: HDNodeWallet) {
-  if (!erc20Assets.includes(srcAsset)) {
-    throw new Error(`Unsupported asset, not an ERC20: ${srcAsset}`);
+export async function approveTokenVault(sourceAsset: Asset, amount: string, wallet: HDNodeWallet) {
+  if (!erc20Assets.includes(sourceAsset)) {
+    throw new Error(`Unsupported asset, not an ERC20: ${sourceAsset}`);
   }
 
-  const chain = chainFromAsset(srcAsset as Asset);
+  const chain = chainFromAsset(sourceAsset as Asset);
 
   await approveVault(
     {
       amount,
       srcChain: chain as Chain,
-      srcAsset: stateChainAssetFromAsset(srcAsset) as SCAsset,
+      srcAsset: stateChainAssetFromAsset(sourceAsset) as SCAsset,
     },
     {
       signer: wallet,
       network: 'localnet',
       vaultContractAddress: getContractAddress(chain, 'VAULT'),
-      srcTokenContractAddress: getContractAddress(chain, srcAsset),
+      srcTokenContractAddress: getContractAddress(chain, sourceAsset),
     },
     // This is run with fresh addresses to prevent nonce issues
     {

bouncer/shared/new_swap.ts

@@ -38,7 +38,7 @@ export async function newSwap(
           ccmParams: messageMetadata && {
             message: messageMetadata.message as `0x${string}`,
             gasBudget: messageMetadata.gasBudget.toString(),
-            cfParameters: messageMetadata.cfParameters as `0x${string}`,
+            ccmAdditionalData: messageMetadata.ccmAdditionalData as `0x${string}`,
           },
           commissionBps: brokerCommissionBps,
           maxBoostFeeBps: boostFeeBps,

bouncer/shared/perform_swap.ts

@@ -70,7 +70,7 @@ export async function requestNewSwap(
         ? event.data.channelMetadata !== null &&
           event.data.channelMetadata.message === messageMetadata.message &&
           event.data.channelMetadata.gasBudget.replace(/,/g, '') === messageMetadata.gasBudget &&
-          event.data.channelMetadata.cfParameters === messageMetadata.cfParameters
+          event.data.channelMetadata.ccmAdditionalData === messageMetadata.ccmAdditionalData
         : event.data.channelMetadata === null;
 
       return destAddressMatches && destAssetMatches && sourceAssetMatches && ccmMetadataMatches;

bouncer/shared/send.ts

@@ -20,9 +20,7 @@ import { sendSolUsdc } from './send_solusdc';
 const cfTesterAbi = await getCFTesterAbi();
 
 export async function send(asset: Asset, address: string, amount?: string, log = true) {
-  // TODO: Remove this any when we have Sol assets in the Asset type.
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  switch (asset as any) {
+  switch (asset) {
     case 'Btc':
       await sendBtc(address, amount ?? defaultAssetAmounts(asset));
       break;

bouncer/shared/swap_context.ts

@@ -3,8 +3,6 @@ import assert from 'assert';
 export enum SwapStatus {
   Initiated,
   Funded,
-  // Contract swap specific statuses
-  ContractApproved,
   ContractExecuted,
   SwapScheduled,
   Success,
@@ -37,16 +35,9 @@ export class SwapContext {
         );
         break;
       }
-      case SwapStatus.ContractApproved: {
-        assert(
-          currentStatus === SwapStatus.Initiated,
-          `Unexpected status transition for ${tag}. Transitioning from ${currentStatus} to ${status}`,
-        );
-        break;
-      }
       case SwapStatus.ContractExecuted: {
         assert(
-          currentStatus === SwapStatus.ContractApproved,
+          currentStatus === SwapStatus.Initiated,
           `Unexpected status transition for ${tag}. Transitioning from ${currentStatus} to ${status}`,
         );
         break;

bouncer/shared/swapping.ts

@@ -12,7 +12,7 @@ import {
   defaultAssetAmounts,
   ccmSupportedChains,
   assetDecimals,
-  solCfParamsCodec,
+  solCcmAdditionalDataCodec,
 } from '../shared/utils';
 import { BtcAddressType } from '../shared/new_btc_address';
 import { CcmDepositMetadata } from '../shared/new_swap';
@@ -69,7 +69,7 @@ function newAbiEncodedMessage(types?: SolidityType[]): string {
   return web3.eth.abi.encodeParameters(typesArray, variables);
 }
 
-export function newSolanaCfParameters(maxAccounts: number) {
+export function newSolanaCcmAdditionalData(maxAccounts: number) {
   const cfReceiverAddress = getContractAddress('Solana', 'CFTESTER');
 
   const fallbackAddress = Keypair.generate().publicKey.toBytes();
@@ -93,7 +93,7 @@ export function newSolanaCfParameters(maxAccounts: number) {
     fallback_address: fallbackAddress,
   };
 
-  return u8aToHex(solCfParamsCodec.enc(cfParameters));
+  return u8aToHex(solCcmAdditionalDataCodec.enc(cfParameters));
 }
 
 // Solana CCM-related parameters. These are values in the protocol.
@@ -107,7 +107,7 @@ function newCcmArbitraryBytes(maxLength: number): string {
   return randomAsHex(Math.floor(Math.random() * Math.max(0, maxLength - 10)) + 10);
 }
 
-function newCfParameters(destAsset: Asset, message?: string): string {
+function newCcmAdditionalData(destAsset: Asset, message?: string): string {
   const destChain = chainFromAsset(destAsset);
   switch (destChain) {
     case 'Ethereum':
@@ -123,7 +123,7 @@ function newCfParameters(destAsset: Asset, message?: string): string {
 
       // The maximum number of extra accounts that can be passed is limited by the tx size
       // and therefore also depends on the message length.
-      return newSolanaCfParameters(maxAccounts);
+      return newSolanaCcmAdditionalData(maxAccounts);
     }
     default:
       throw new Error(`Unsupported chain: ${destChain}`);
@@ -151,7 +151,7 @@ export function newCcmMetadata(
   cfParamsArray?: string,
 ): CcmDepositMetadata {
   const message = ccmMessage ?? newCcmMessage(destAsset);
-  const cfParameters = cfParamsArray ?? newCfParameters(destAsset, message);
+  const ccmAdditionalData = cfParamsArray ?? newCcmAdditionalData(destAsset, message);
   const gasDiv = gasBudgetFraction ?? 2;
 
   const gasBudget = Math.floor(
@@ -162,7 +162,7 @@ export function newCcmMetadata(
   return {
     message,
     gasBudget,
-    cfParameters,
+    ccmAdditionalData,
   };
 }
 
@@ -250,7 +250,7 @@ export async function testSwapViaContract(
     destAsset,
     addressType,
     messageMetadata,
-    (tagSuffix ?? '') + ' Contract',
+    (tagSuffix ?? '') + 'Contract',
     log,
     swapContext,
   );