return computed origin account info in moonChainData for parachain to…

… evm transactions
moonbeam-foundation · Dec 17, 2024 · 5906089 · 5906089
1 parent b094a24
commit 5906089
Show file tree

Hide file tree

Showing 10 changed files with 255 additions and 12 deletions.
diff --git a/packages/config/src/mrl-configs/fantomTestnet.ts b/packages/config/src/mrl-configs/fantomTestnet.ts
@@ -35,6 +35,7 @@ export const fantomTestnetRoutes = new MrlChainRoutes({
         transfer: MrlBuilder().wormhole().wormhole().tokenTransfer(),
         moonChain: {
           asset: ftmwh,
+          balance: BalanceBuilder().evm().erc20(),
           fee: {
             asset: dev,
             amount: 0.1,
@@ -66,6 +67,7 @@ export const fantomTestnetRoutes = new MrlChainRoutes({
         transfer: MrlBuilder().wormhole().wormhole().tokenTransfer(),
         moonChain: {
           asset: ftmwh,
+          balance: BalanceBuilder().evm().erc20(),
           fee: {
             asset: dev,
             amount: 0.1,
@@ -97,6 +99,8 @@ export const fantomTestnetRoutes = new MrlChainRoutes({
         transfer: MrlBuilder().wormhole().wormhole().tokenTransfer(),
         moonChain: {
           asset: agng,
+          balance: BalanceBuilder().substrate().assets().account(),
+
           fee: {
             asset: dev,
             amount: 0.1,
@@ -128,6 +132,7 @@ export const fantomTestnetRoutes = new MrlChainRoutes({
         transfer: MrlBuilder().wormhole().wormhole().tokenTransfer(),
         moonChain: {
           asset: ftmwh,
+          balance: BalanceBuilder().evm().erc20(),
           fee: {
             asset: dev,
             amount: 0.1,
@@ -159,6 +164,7 @@ export const fantomTestnetRoutes = new MrlChainRoutes({
         transfer: MrlBuilder().wormhole().wormhole().tokenTransfer(),
         moonChain: {
           asset: ftmwh,
+          balance: BalanceBuilder().evm().erc20(),
           fee: {
             asset: dev,
             amount: 0.1,
@@ -190,6 +196,7 @@ export const fantomTestnetRoutes = new MrlChainRoutes({
         transfer: MrlBuilder().wormhole().wormhole().tokenTransfer(),
         moonChain: {
           asset: dev,
+          balance: BalanceBuilder().substrate().system().account(),
           fee: {
             asset: dev,
             amount: 0.1,

diff --git a/packages/config/src/mrl-configs/moonbaseAlpha.ts b/packages/config/src/mrl-configs/moonbaseAlpha.ts
@@ -33,9 +33,10 @@ export const moonbaseAlphaRoutes = new MrlChainRoutes({
         transfer: MrlBuilder().wormhole().wormhole().tokenTransfer(),
         moonChain: {
           asset: ftmwh,
+          balance: BalanceBuilder().evm().erc20(),
           fee: {
             asset: dev,
-            amount: 0.1, // TODO not really, it would be the source fee as source is moonChain
+            amount: 0.1,
             balance: BalanceBuilder().substrate().system().account(),
           },
         },
@@ -64,6 +65,7 @@ export const moonbaseAlphaRoutes = new MrlChainRoutes({
         transfer: MrlBuilder().wormhole().wormhole().tokenTransfer(),
         moonChain: {
           asset: dev,
+          balance: BalanceBuilder().evm().erc20(),
           fee: {
             asset: dev,
             amount: 0,

diff --git a/packages/config/src/mrl-configs/peaqAlphanet.ts b/packages/config/src/mrl-configs/peaqAlphanet.ts
@@ -37,6 +37,7 @@ export const peaqAlphanetRoutes = new MrlChainRoutes({
         transfer: MrlBuilder().wormhole().extrinsic().polkadotXcm().send(),
         moonChain: {
           asset: ftmwh,
+          balance: BalanceBuilder().evm().erc20(),
           fee: {
             asset: dev,
             amount: 0.1,
@@ -72,6 +73,7 @@ export const peaqAlphanetRoutes = new MrlChainRoutes({
         transfer: MrlBuilder().wormhole().extrinsic().polkadotXcm().send(),
         moonChain: {
           asset: agng,
+          balance: BalanceBuilder().substrate().assets().account(),
           fee: {
             asset: dev,
             amount: 0.1,

diff --git a/packages/config/src/mrl-configs/peaqEvmAlphanet.ts b/packages/config/src/mrl-configs/peaqEvmAlphanet.ts
@@ -41,6 +41,7 @@ export const peaqEvmAlphanetRoutes = new MrlChainRoutes({
           .transferAssetsAndMessage(),
         moonChain: {
           asset: ftmwh,
+          balance: BalanceBuilder().evm().erc20(),
           fee: {
             asset: dev,
             amount: 0.1,

diff --git a/packages/config/src/types/AssetRoute.ts b/packages/config/src/types/AssetRoute.ts
@@ -58,6 +58,7 @@ export interface DestinationFeeConfig
 
 export interface MoonChainConfig {
   asset: Asset;
+  balance: BalanceConfigBuilder;
   fee: MoonChainFeeConfig;
 }
 

diff --git a/packages/mrl/package.json b/packages/mrl/package.json
@@ -42,6 +42,7 @@
     "@polkadot/api-augment": "14.3.1",
     "@polkadot/types": "14.3.1",
     "@polkadot/util": "13.2.3",
+    "@polkadot/util-crypto": "13.2.3",
     "@wormhole-foundation/sdk-connect": "^0.10.7",
     "@wormhole-foundation/sdk-evm": "^0.10.7",
     "viem": "^2.21.7"

diff --git a/packages/mrl/src/getTransferData/getMoonChainData.ts b/packages/mrl/src/getTransferData/getMoonChainData.ts
@@ -1,17 +1,20 @@
 import { type MrlAssetRoute, getMoonChain } from '@moonbeam-network/xcm-config';
 import { getBalance, getDestinationFee } from '@moonbeam-network/xcm-sdk';
-import { Parachain } from '@moonbeam-network/xcm-types';
+import { EvmChain, EvmParachain, Parachain } from '@moonbeam-network/xcm-types';
 import { getMultilocationDerivedAddresses } from '@moonbeam-network/xcm-utils';
+import { evmToAddress } from '@polkadot/util-crypto';
 import type { MoonChainTransferData } from '../mrl.interfaces';
 
 interface GetMoonChainDataParams {
   route: MrlAssetRoute;
   sourceAddress: string;
+  destinationAddress: string;
 }
 
 export async function getMoonChainData({
   route,
   sourceAddress,
+  destinationAddress,
 }: GetMoonChainDataParams): Promise<MoonChainTransferData> {
   if (!route.mrl) {
     throw new Error(
@@ -20,30 +23,44 @@ export async function getMoonChainData({
   }
 
   const moonChain = getMoonChain(route.source.chain);
+  const isDestinationMoonChain = moonChain.isEqual(route.destination.chain);
+  const isSourceEvmChain = EvmChain.is(route.source.chain);
+  let address =
+    isDestinationMoonChain || isSourceEvmChain
+      ? destinationAddress
+      : sourceAddress;
 
   const fee = await getDestinationFee({
-    address: sourceAddress, // TODO not correct
+    address,
     asset: route.source.asset,
     destination: moonChain,
     fee: route.mrl.moonChain.fee.amount,
     feeAsset: route.mrl.moonChain.fee.asset,
   });
 
-  let address = sourceAddress;
-
   if (
     Parachain.is(route.source.chain) &&
     !route.source.chain.isEqual(moonChain)
   ) {
+    const addressToUse = EvmParachain.is(route.source.chain)
+      ? evmToAddress(sourceAddress)
+      : address;
     const { address20 } = getMultilocationDerivedAddresses({
-      address: sourceAddress,
+      address: addressToUse,
       paraId: route.source.chain.parachainId,
       isParents: true,
     });
 
     address = address20;
   }
 
+  const balance = await getBalance({
+    address,
+    asset: moonChain.getChainAsset(route.mrl.moonChain.asset),
+    builder: route.mrl.moonChain.balance,
+    chain: moonChain,
+  });
+
   const feeBalance = await getBalance({
     address,
     asset: moonChain.getChainAsset(route.mrl.moonChain.fee.asset),
@@ -52,6 +69,8 @@ export async function getMoonChainData({
   });
 
   return {
+    address,
+    balance,
     feeBalance,
     chain: moonChain,
     fee,

diff --git a/packages/mrl/src/getTransferData/getTransferData.ts b/packages/mrl/src/getTransferData/getTransferData.ts
@@ -70,6 +70,7 @@ export async function getTransferData({
   const moonChainData = await getMoonChainData({
     route,
     sourceAddress,
+    destinationAddress,
   });
 
   return {

diff --git a/packages/mrl/src/mrl.interfaces.ts b/packages/mrl/src/mrl.interfaces.ts
@@ -40,10 +40,8 @@ export interface SourceTransferData extends SourceChainTransferData {
 
 export interface DestinationTransferData extends ChainTransferData {}
 
-export type MoonChainTransferData = Omit<
-  ChainTransferData,
-  'min' | 'balance'
-> & {
+export type MoonChainTransferData = Omit<ChainTransferData, 'min'> & {
+  address: string;
   feeBalance: AssetAmount;
 };