feat: add utils to select asset from network (#33)

Author: LuizAsFight

.changeset/popular-lions-approve.md

@@ -0,0 +1,5 @@
+---
+'@fuels/assets': patch
+---
+
+Asset package: Add network utils

package.json

@@ -77,7 +77,8 @@
     },
     "overrides": {
       "semver@<7.5.2": ">=7.5.2",
-      "word-wrap": "npm:@aashutoshrathi/word-wrap"
+      "word-wrap": "npm:@aashutoshrathi/word-wrap",
+      "@babel/traverse@<7.23.2": ">=7.23.2"
     }
   }
 }

packages/assets/src/index.ts

@@ -1,6 +1,7 @@
 import { assets as assetList } from './assets';
 import { resolveIconPath } from './utils/resolveIconPath';
 export * from './constants';
+export * from './utils/network';
 export * from './utils/resolveIconPath';
 export * from './types';
 

packages/assets/src/types.ts

@@ -34,3 +34,6 @@ export type Asset = {
 };
 
 export type Assets = Array<Asset>;
+
+export type AssetEth = Omit<Asset, 'networks'> & Ethereum;
+export type AssetFuel = Omit<Asset, 'networks'> & Fuel;

packages/assets/src/utils/network.ts

@@ -0,0 +1,83 @@
+import { CHAIN_IDS } from '../constants';
+import type { Asset, AssetEth, AssetFuel, Ethereum, Fuel } from '../types';
+
+type Network = Ethereum | Fuel;
+export type NetworkTypes = Ethereum['type'] | Fuel['type'];
+type NetworkTypeToNetwork<T> = T extends 'ethereum'
+  ? Ethereum
+  : T extends 'fuel'
+  ? Fuel
+  : Network;
+
+export const getDefaultChainId = (
+  networkType: NetworkTypes,
+): number | undefined => {
+  if (networkType === 'ethereum') return CHAIN_IDS.eth.sepolia;
+  if (networkType === 'fuel') return CHAIN_IDS.fuel.beta4;
+
+  return undefined;
+};
+
+export type GetAssetNetworkParams<T extends NetworkTypes | undefined> = {
+  asset: Asset;
+  chainId?: number;
+  networkType: T;
+};
+
+export const getAssetNetwork = <T extends NetworkTypes | undefined>({
+  asset,
+  chainId,
+  networkType,
+}: GetAssetNetworkParams<T>): NetworkTypeToNetwork<T> => {
+  const network = asset.networks.find(
+    (network) => network.chainId === chainId && network.type === networkType,
+  ) as NetworkTypeToNetwork<T>;
+
+  return network;
+};
+
+export const getAssetWithNetwork = <T extends NetworkTypes>({
+  asset,
+  chainId,
+  networkType,
+}: GetAssetNetworkParams<T>): AssetEth | AssetFuel | undefined => {
+  const { networks: _, ...assetRest } = asset;
+
+  const chainIdToUse = chainId ?? getDefaultChainId(networkType);
+  if (!chainIdToUse) return undefined;
+
+  const assetNetwork = getAssetNetwork({
+    asset,
+    chainId: chainIdToUse,
+    networkType,
+  });
+
+  if (!assetNetwork) return undefined;
+
+  return {
+    ...assetRest,
+    ...assetNetwork,
+  };
+};
+
+export const getAssetEth = (
+  asset: Asset,
+  chainId?: number,
+): AssetEth | undefined => {
+  return getAssetWithNetwork({
+    asset,
+    networkType: 'ethereum',
+    chainId,
+  }) as AssetEth;
+};
+
+export const getAssetFuel = (
+  asset: Asset,
+  chainId?: number,
+): AssetFuel | undefined => {
+  return getAssetWithNetwork({
+    asset,
+    networkType: 'fuel',
+    chainId,
+  }) as AssetFuel;
+};