add xcmPallet.limitedTeleportAssets support

galacticcouncil · Feb 6, 2024 · 80a1e96 · 80a1e96
1 parent 9c8c989
commit 80a1e96
Show file tree

Hide file tree

Showing 4 changed files with 49 additions and 21 deletions.
diff --git a/packages/xcm-cfg/src/builders/pallets/xcmPallet.ts b/packages/xcm-cfg/src/builders/pallets/xcmPallet.ts
@@ -2,13 +2,14 @@ import {
   XcmVersion,
   ExtrinsicConfigBuilder,
   ExtrinsicConfig,
+  Parents,
 } from '@moonbeam-network/xcm-builder';
 import { toAssets, toBeneficiary, toDest } from './xcmPallet.utils';
 import { getExtrinsicAccount } from '../ExtrinsicBuilder.utils';
 
 const pallet = 'xcmPallet';
 
-const limitedReserveTransferAssets = () => {
+const limitedReserveTransferAssets = (parent: Parents) => {
   const func = 'limitedReserveTransferAssets';
   return {
     here: (): ExtrinsicConfigBuilder => ({
@@ -22,7 +23,31 @@ const limitedReserveTransferAssets = () => {
             return [
               toDest(version, destination),
               toBeneficiary(version, account),
-              toAssets(version, amount),
+              toAssets(version, parent, 'Here', amount),
+              0,
+              'Unlimited',
+            ];
+          },
+        }),
+    }),
+  };
+};
+
+const limitedTeleportAssets = (parent: Parents) => {
+  const func = 'limitedTeleportAssets';
+  return {
+    here: (): ExtrinsicConfigBuilder => ({
+      build: ({ address, amount, destination }) =>
+        new ExtrinsicConfig({
+          module: pallet,
+          func,
+          getArgs: () => {
+            const version = XcmVersion.v3;
+            const account = getExtrinsicAccount(address);
+            return [
+              toDest(version, destination),
+              toBeneficiary(version, account),
+              toAssets(version, parent, 'Here', amount),
               0,
               'Unlimited',
             ];
@@ -35,5 +60,6 @@ const limitedReserveTransferAssets = () => {
 export const xcmPallet = () => {
   return {
     limitedReserveTransferAssets,
+    limitedTeleportAssets,
   };
 };
diff --git a/packages/xcm-cfg/src/builders/pallets/xcmPallet.utils.ts b/packages/xcm-cfg/src/builders/pallets/xcmPallet.utils.ts
@@ -1,4 +1,4 @@
-import { XcmVersion } from '@moonbeam-network/xcm-builder';
+import { Parents, XcmVersion } from '@moonbeam-network/xcm-builder';
 import { AnyChain } from '@moonbeam-network/xcm-types';
 
 export const toDest = (version: XcmVersion, destination: AnyChain) => {
@@ -23,14 +23,19 @@ export const toBeneficiary = (version: XcmVersion, account: any) => {
   };
 };
 
-export const toAssets = (version: XcmVersion, amount: any) => {
+export const toAssets = (
+  version: XcmVersion,
+  parents: Parents,
+  interior: any,
+  amount: any
+) => {
   return {
     [version]: [
       {
         id: {
           Concrete: {
-            parents: 0,
-            interior: 'Here',
+            parents: parents,
+            interior: interior,
           },
         },
         fun: {

diff --git a/packages/xcm-cfg/src/configs/assethub.ts b/packages/xcm-cfg/src/configs/assethub.ts
@@ -3,7 +3,11 @@ import {
   BalanceBuilder,
   ExtrinsicBuilder,
 } from '@moonbeam-network/xcm-builder';
-import { AssetConfig, ChainConfig, polkadot } from '@moonbeam-network/xcm-config';
+import {
+  AssetConfig,
+  ChainConfig,
+  polkadot,
+} from '@moonbeam-network/xcm-config';
 
 import { dot, usdc, usdt } from '../assets';
 import { assetHub, hydraDX } from '../chains';
@@ -79,7 +83,10 @@ const toPolkadot: AssetConfig[] = [
       asset: dot,
       balance: BalanceBuilder().substrate().system().account(),
     },
-    extrinsic: ExtrinsicBuilderV2().polkadotXcm().limitedTeleportAssets().here(),
+    extrinsic: ExtrinsicBuilderV2()
+      .polkadotXcm()
+      .limitedTeleportAssets(0)
+      .here(),
     fee: {
       asset: dot,
       balance: BalanceBuilder().substrate().system().account(),
@@ -89,9 +96,6 @@ const toPolkadot: AssetConfig[] = [
 ];
 
 export const assetHubConfig = new ChainConfig({
-  assets: [
-    ...toHydraDX,
-    ...toPolkadot,
-  ],
+  assets: [...toHydraDX, ...toPolkadot],
   chain: assetHub,
 });
diff --git a/packages/xcm-cfg/src/configs/polkadot.ts b/packages/xcm-cfg/src/configs/polkadot.ts
@@ -56,18 +56,11 @@ const toAssetHub: AssetConfig[] = [
       asset: dot,
       balance: BalanceBuilder().substrate().system().account(),
     },
-    extrinsic: ExtrinsicBuilderV2()
-      .polkadotXcm()
-      .limitedTeleportAssets()
-      .here(),
+    extrinsic: ExtrinsicBuilderV2().xcmPallet().limitedTeleportAssets(0).here(),
   }),
 ];
 
 export const polkadotConfig = new ChainConfig({
-  assets: [
-    ...toHydraDX,
-    ...toBifrost,
-    ...toAssetHub,
-  ],
+  assets: [...toHydraDX, ...toBifrost, ...toAssetHub],
   chain: polkadot,
 });