[Issue-4079] feat: handle n-length path

Author: bluezdot

packages/extension-base/src/koni/background/handlers/Extension.ts

@@ -1326,7 +1326,7 @@ export default class KoniExtension {
     const warnings: TransactionWarning[] = [];
     const chainInfo = this.#koniState.getChainInfo(chain);
 
-    const test = generateSwapPairs(this.#koniState.getSubstrateApi(chain), this.#koniState.chainService, transferTokenInfo).map((asset) => asset.slug);
+    const test = generateSwapPairs(this.#koniState.getSubstrateApi(chain), this.#koniState.chainService, transferTokenInfo, 3).map((asset) => asset.slug);
 
     console.log('test', test);
 

packages/extension-base/src/services/swap-service/utils.ts

@@ -121,23 +121,41 @@ export function getChainflipBroker (isTestnet: boolean) { // noted: currently no
 }
 
 export function generateSwapPairs (substrateApi: _SubstrateApi, chainService: ChainService, fromAsset: _ChainAsset, maxPathLength = 1) {
+  if (maxPathLength < 1) {
+    return [];
+  }
+
   let currentTargets: _ChainAsset[] = [];
   let newTargets: _ChainAsset[] = [];
-
-  if (maxPathLength < 1) {
-    return currentTargets;
+  let currentStep = 1;
+
+  // step 1:
+  newTargets = findDestinations(chainService, fromAsset);
+  currentTargets = mergeWithoutDuplicate<_ChainAsset>(currentTargets, newTargets);
+  currentStep += 1;
+
+  // step 2 - n
+  while (currentStep <= maxPathLength) { // todo: improve by stop when nothing new
+    newTargets = Array.from(newTargets.reduce((farChildTargets, currentTarget) => {
+      const farChildTargetsLocal = findDestinations(chainService, currentTarget);
+
+      return new Set([...farChildTargets, ...farChildTargetsLocal]);
+    }, new Set<_ChainAsset>()));
+    currentTargets = mergeWithoutDuplicate<_ChainAsset>(currentTargets, newTargets);
+    currentStep += 1;
   }
 
-  const xcmTargets = findXcmDestinations(chainService, fromAsset);
-  const swapTargets = findSwapDestinations(chainService, fromAsset);
+  return currentTargets;
+}
 
-  newTargets = Array.from(new Set([...xcmTargets, ...swapTargets]));
-  currentTargets = Array.from(new Set([...currentTargets, ...newTargets]));
+function findDestinations (chainService: ChainService, chainAsset: _ChainAsset) {
+  const xcmTargets = findXcmDestinations(chainService, chainAsset);
+  const swapTargets = findSwapDestinations(chainService, chainAsset);
 
-  return currentTargets;
+  return mergeWithoutDuplicate<_ChainAsset>(xcmTargets, swapTargets);
 }
 
-export function findXcmDestinations (chainService: ChainService, chainAsset: _ChainAsset) {
+function findXcmDestinations (chainService: ChainService, chainAsset: _ChainAsset) {
   const xcmTargets: _ChainAsset[] = [];
   const multichainAssetSlug = _getMultiChainAsset(chainAsset);
 
@@ -156,9 +174,9 @@ export function findXcmDestinations (chainService: ChainService, chainAsset: _Ch
   return xcmTargets;
 }
 
-export function findSwapDestinations (chainService: ChainService, chainAsset: _ChainAsset) {
-  const swapTargets: _ChainAsset[] = [];
+function findSwapDestinations (chainService: ChainService, chainAsset: _ChainAsset) {
   const chain = chainAsset.originChain;
+  const swapTargets: _ChainAsset[] = [];
 
   const availableChains = Object.values(_PROVIDER_TO_SUPPORTED_PAIR_MAP).reduce((remainChains, currentChains) => {
     if (currentChains.includes(chain)) {
@@ -179,6 +197,34 @@ export function findSwapDestinations (chainService: ChainService, chainAsset: _C
   return swapTargets;
 }
 
-export function isHasChannel (subtrateApi: _SubstrateApi, fromChain: _ChainAsset, toChain: _ChainAsset) {
+export function findSwapDestinationsV2 (chainService: ChainService, chainAsset: _ChainAsset) {
+  const chain = chainAsset.originChain;
+  const swapTargets: _ChainAsset[] = [];
+
+  // Convert to Set once at the start
+  const availableChains = new Set<string>(
+    Object.values(_PROVIDER_TO_SUPPORTED_PAIR_MAP)
+      .filter((chains) => chains.includes(chain))
+      .flat()
+  );
+
+  // Use Set for O(1) lookup instead of includes()
+  for (const candidate of availableChains) {
+    const assets = chainService.getAssetByChainAndType(
+      candidate,
+      _getFungibleAssetType()
+    );
+
+    swapTargets.push(...Object.values(assets));
+  }
+
+  return swapTargets;
+}
+
+function isHasChannel (subtrateApi: _SubstrateApi, fromChain: _ChainAsset, toChain: _ChainAsset) {
   return true;
 }
+
+function mergeWithoutDuplicate<T> (arr1: T[], arr2: T[]): T[] {
+  return Array.from(new Set([...arr1, ...arr2]));
+}