@metamask/transaction-pay-controller 16.5.0 → 17.1.0

package/dist/utils/token.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"token.cjs","sourceRoot":"","sources":["../../src/utils/token.ts"],"names":[],"mappings":";;;AAAA,wDAAoD;AACpD,wDAAwD;AACxD,iEAAkE;AAClE,mEAAuD;AAEvD,+CAAyC;AACzC,mCAA8B;AAE9B,gDAIsB;AAGtB;;;;;;;;;;GAUG;AACH,SAAgB,WAAW,CACzB,MAAsC,EACtC,MAAsC;IAEtC,OAAO,CACL,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE;QAC7D,MAAM,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,CAClC,CAAC;AACJ,CAAC;AARD,kCAQC;AAED;;;;;;;;GAQG;AACH,SAAgB,eAAe,CAC7B,SAA4C,EAC5C,OAAY,EACZ,OAAY,EACZ,YAAiB;IAEjB,MAAM,2BAA2B,GAAG,SAAS,CAAC,IAAI,CAChD,kCAAkC,CACnC,CAAC;IAEF,MAAM,iBAAiB,GAAG,OAAO,CAAC,WAAW,EAAS,CAAC;IACvD,MAAM,sBAAsB,GAAG,IAAA,uCAAoB,EAAC,YAAY,CAAQ,CAAC;IACzE,MAAM,QAAQ,GAAG,sBAAsB,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC;IAEpE,MAAM,UAAU,GACd,2BAA2B,CAAC,aAAa,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,CACzE,sBAAsB,CACvB,CAAC;IAEJ,IAAI,CAAC,QAAQ,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC1C,OAAO,GAAG,CAAC;IACb,CAAC;IAED,IAAI,CAAC,QAAQ,IAAI,UAAU,EAAE,CAAC;QAC5B,OAAO,IAAI,wBAAS,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACpD,CAAC;IAED,MAAM,6BAA6B,GAAG,SAAS,CAAC,IAAI,CAClD,mCAAmC,CACpC,CAAC;IAEF,MAAM,aAAa,GACjB,6BAA6B,CAAC,iBAAiB,EAAE,CAAC,OAAO,CAAC,CAAC;IAE7D,MAAM,eAAe,GAAG,IAAA,uCAAoB,EAAC,iBAAiB,CAAQ,CAAC;IACvE,MAAM,gBAAgB,GAAG,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE,OAAc,CAAC;IAE1E,OAAO,IAAI,wBAAS,CAAC,gBAAgB,IAAI,KAAK,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;AACnE,CAAC;AAtCD,0CAsCC;AAED;;;;;;GAMG;AACH,SAAgB,mBAAmB,CACjC,SAA4C,EAC5C,OAAY;IAMZ,MAAM,2BAA2B,GAAG,SAAS,CAAC,IAAI,CAChD,kCAAkC,CACnC,CAAC;IAEF,MAAM,6BAA6B,GAAG,SAAS,CAAC,IAAI,CAClD,mCAAmC,CACpC,CAAC;IAEF,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAChC,6BAA6B,CAAC,iBAAiB,CACvC,CAAC;IAEX,MAAM,iBAAiB,GAAG,OAAO,CAAC,WAAW,EAAS,CAAC;IAEvD,MAAM,sBAAsB,GAC1B,2BAA2B,CAAC,aAAa,EAAE,CAAC,iBAAiB,CAAC,CAAC;IAEjE,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAU,CAAC;IACnE,MAAM,QAAQ,GAAG,IAAA,aAAI,EAAC,CAAC,GAAG,aAAa,EAAE,GAAG,cAAc,CAAC,CAAC,CAAC;IAE7D,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAClC,MAAM,cAAc,GAAG;YACrB,GAAI,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,IAAI,EAAE,CAAW;YAChE,cAAc,CAAC,OAAO,CAAC;SACxB,CAAC;QAEF,OAAO,cAAc,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;YAC3C,OAAO;YACP,YAAY;YACZ,OAAO,EAAE,eAAe,CAAC,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY,CAAC;SACpE,CAAC,CAAC,CAAC;IACN,CAAC,CAAC,CAAC;AACL,CAAC;AAxCD,kDAwCC;AAED;;;;;;;GAOG;AACH,SAAgB,YAAY,CAC1B,SAA4C,EAC5C,YAAiB,EACjB,OAAY;IAEZ,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;IACpE,MAAM,sBAAsB,GAAG,YAAY,CAAC,WAAW,EAAS,CAAC;IAEjE,MAAM,QAAQ,GACZ,sBAAsB,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;IAEnE,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,SAAS,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;SACpE,IAAI,EAAE;SACN,IAAI,CACH,CAAC,WAAW,EAAE,EAAE,CACd,WAAW,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,sBAAsB,CAC/D,CAAC;IAEJ,IAAI,CAAC,KAAK,IAAI,CAAC,QAAQ,EAAE,CAAC;QACxB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,KAAK,IAAI,CAAC,QAAQ,EAAE,CAAC;QACvB,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;IACpE,CAAC;IAED,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAE7C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;AAC1C,CAAC;AAjCD,oCAiCC;AAED;;;;;;;GAOG;AACH,SAAgB,gBAAgB,CAC9B,SAA4C,EAC5C,YAAiB,EACjB,OAAY;IAEZ,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAE7C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,mBAAmB,GAAG,SAAS,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;IAE5E,MAAM,2BAA2B,GAAG,SAAS,CAAC,IAAI,CAChD,iCAAiC,CAClC,CAAC;IAEF,MAAM,sBAAsB,GAAG,IAAA,uCAAoB,EAAC,YAAY,CAAQ,CAAC;IACzE,MAAM,QAAQ,GAAG,sBAAsB,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC;IAEpE,MAAM,iBAAiB,GACrB,mBAAmB,CAAC,UAAU,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,sBAAsB,CAAC,EAAE,KAAK,CAAC;IAE7E,IAAI,iBAAiB,KAAK,SAAS,IAAI,CAAC,QAAQ,EAAE,CAAC;QACjD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,EACJ,cAAc,EAAE,gBAAgB,EAChC,iBAAiB,EAAE,eAAe,GACnC,GAAG,2BAA2B,CAAC,aAAa,EAAE,CAAC,MAAM,CAAC,IAAI;QACzD,cAAc,EAAE,IAAI;QACpB,iBAAiB,EAAE,IAAI;KACxB,CAAC;IAEF,IAAI,gBAAgB,KAAK,IAAI,IAAI,eAAe,KAAK,IAAI,EAAE,CAAC;QAC1D,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,YAAY,GAAG,uBAAW,CAAC,OAAO,CAAC,EAAE,QAAQ,CACjD,YAAY,CAAC,WAAW,EAAS,CAClC,CAAC;IAEF,MAAM,OAAO,GAAG,YAAY;QAC1B,CAAC,CAAC,GAAG;QACL,CAAC,CAAC,IAAI,wBAAS,CAAC,iBAAiB,IAAI,CAAC,CAAC;aAClC,YAAY,CAAC,eAAe,CAAC;aAC7B,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEpB,MAAM,QAAQ,GAAG,IAAI,wBAAS,CAAC,iBAAiB,IAAI,CAAC,CAAC;SACnD,YAAY,CAAC,gBAAgB,CAAC;SAC9B,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEhB,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;AAC/B,CAAC;AArDD,4CAqDC;AAED;;;;;;;GAOG;AACH,SAAgB,mBAAmB,CACjC,QAAyB,EACzB,QAAgB,EAChB,SAAoB;IAOpB,MAAM,QAAQ,GAAG,IAAI,wBAAS,CAAC,QAAQ,CAAC,CAAC;IACzC,MAAM,UAAU,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,CAAC;IAEjD,OAAO;QACL,GAAG,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QACxB,KAAK,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC9B,GAAG,EAAE,UAAU,CAAC,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5D,IAAI,EAAE,UAAU,CAAC,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC;KAC/D,CAAC;AACJ,CAAC;AAnBD,kDAmBC;AAED;;;;;GAKG;AACH,SAAgB,cAAc,CAAC,OAAY;IACzC,QAAQ,OAAO,EAAE,CAAC;QAChB,KAAK,MAAM;YACT,OAAO,4CAA4C,CAAC;QACtD;YACE,OAAO,gCAAoB,CAAC;IAChC,CAAC;AACH,CAAC;AAPD,wCAOC;AAED;;;;;;;;;;;;GAYG;AACI,KAAK,UAAU,mBAAmB,CACvC,SAA4C,EAC5C,OAAY,EACZ,OAAY,EACZ,YAAiB;IAEjB,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CACpC,gDAAgD,EAChD,OAAO,CACR,CAAC;IAEF,MAAM,EAAE,QAAQ,EAAE,GAAG,SAAS,CAAC,IAAI,CACjC,wCAAwC,EACxC,eAAe,CAChB,CAAC;IAEF,MAAM,cAAc,GAAG,IAAI,wBAAY,CAAC,QAAQ,CAAC,CAAC;IAClD,MAAM,QAAQ,GACZ,YAAY,CAAC,WAAW,EAAE,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;IAEvE,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACzD,OAAO,OAAO,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAAC,YAAY,EAAE,4BAAQ,EAAE,cAAc,CAAC,CAAC;IACtE,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAClD,OAAO,OAAO,CAAC,QAAQ,EAAE,CAAC;AAC5B,CAAC;AA5BD,kDA4BC;AAED,SAAS,SAAS,CAChB,OAAY,EACZ,SAA4C;IAE5C,IAAI,CAAC;QACH,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CACpC,gDAAgD,EAChD,OAAO,CACR,CAAC;QAEF,MAAM,oBAAoB,GAAG,SAAS,CAAC,IAAI,CACzC,wCAAwC,EACxC,eAAe,CAChB,CAAC;QAEF,OAAO,oBAAoB,CAAC,aAAa,CAAC,MAAM,CAAC;IACnD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,IAAY,kBAGX;AAHD,WAAY,kBAAkB;IAC5B,qCAAe,CAAA;IACf,2CAAqB,CAAA;AACvB,CAAC,EAHW,kBAAkB,kCAAlB,kBAAkB,QAG7B;AAED;;;;;;;;;;;;GAYG;AACH,SAAgB,qBAAqB,CACnC,YAAiB,EACjB,OAAY,EACZ,MAA2B;IAE3B,IAAI,OAAO,KAAK,4BAAgB,EAAE,CAAC;QACjC,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,MAAM,kBAAkB,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC,WAAW,EAAS,CAAC;IACxE,MAAM,sBAAsB,GAAG,YAAY,CAAC,WAAW,EAAE,CAAC;IAE1D,IACE,MAAM,KAAK,kBAAkB,CAAC,KAAK;QACnC,sBAAsB,KAAK,kBAAkB,EAC7C,CAAC;QACD,OAAO,gCAAoB,CAAC;IAC9B,CAAC;IAED,IACE,MAAM,KAAK,kBAAkB,CAAC,QAAQ;QACtC,sBAAsB,KAAK,gCAAoB,CAAC,WAAW,EAAE,EAC7D,CAAC;QACD,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC;AA3BD,sDA2BC","sourcesContent":["import { Contract } from '@ethersproject/contracts';\nimport { Web3Provider } from '@ethersproject/providers';\nimport { toChecksumHexAddress } from '@metamask/controller-utils';\nimport { abiERC20 } from '@metamask/metamask-eth-abis';\nimport type { Hex } from '@metamask/utils';\nimport { BigNumber } from 'bignumber.js';\nimport { uniq } from 'lodash';\n\nimport {\n  CHAIN_ID_POLYGON,\n  NATIVE_TOKEN_ADDRESS,\n  STABLECOINS,\n} from '../constants';\nimport type { FiatRates, TransactionPayControllerMessenger } from '../types';\n\n/**\n * Check if two tokens are the same (same address and chain).\n *\n * @param token1 - First token identifier.\n * @param token1.address - Token address.\n * @param token1.chainId - Token chain ID.\n * @param token2 - Second token identifier.\n * @param token2.address - Token address.\n * @param token2.chainId - Token chain ID.\n * @returns True if tokens are the same, false otherwise.\n */\nexport function isSameToken(\n  token1: { address: Hex; chainId: Hex },\n  token2: { address: Hex; chainId: Hex },\n): boolean {\n  return (\n    token1.address.toLowerCase() === token2.address.toLowerCase() &&\n    token1.chainId === token2.chainId\n  );\n}\n\n/**\n * Get the token balance for a specific account and token.\n *\n * @param messenger - Controller messenger.\n * @param account - Address of the account.\n * @param chainId - Id of the chain.\n * @param tokenAddress - Address of the token contract.\n * @returns Raw token balance as a decimal string.\n */\nexport function getTokenBalance(\n  messenger: TransactionPayControllerMessenger,\n  account: Hex,\n  chainId: Hex,\n  tokenAddress: Hex,\n): string {\n  const tokenBalanceControllerState = messenger.call(\n    'TokenBalancesController:getState',\n  );\n\n  const normalizedAccount = account.toLowerCase() as Hex;\n  const normalizedTokenAddress = toChecksumHexAddress(tokenAddress) as Hex;\n  const isNative = normalizedTokenAddress === getNativeToken(chainId);\n\n  const balanceHex =\n    tokenBalanceControllerState.tokenBalances?.[normalizedAccount]?.[chainId]?.[\n      normalizedTokenAddress\n    ];\n\n  if (!isNative && balanceHex === undefined) {\n    return '0';\n  }\n\n  if (!isNative && balanceHex) {\n    return new BigNumber(balanceHex, 16).toString(10);\n  }\n\n  const accountTrackerControllerState = messenger.call(\n    'AccountTrackerController:getState',\n  );\n\n  const chainAccounts =\n    accountTrackerControllerState.accountsByChainId?.[chainId];\n\n  const checksumAccount = toChecksumHexAddress(normalizedAccount) as Hex;\n  const nativeBalanceHex = chainAccounts?.[checksumAccount]?.balance as Hex;\n\n  return new BigNumber(nativeBalanceHex ?? '0x0', 16).toString(10);\n}\n\n/**\n * Get the token balance for a specific account and token.\n *\n * @param messenger - Controller messenger.\n * @param account - Address of the account.\n * @returns The token balance as a BigNumber.\n */\nexport function getAllTokenBalances(\n  messenger: TransactionPayControllerMessenger,\n  account: Hex,\n): {\n  balance: string;\n  chainId: Hex;\n  tokenAddress: Hex;\n}[] {\n  const tokenBalanceControllerState = messenger.call(\n    'TokenBalancesController:getState',\n  );\n\n  const accountTrackerControllerState = messenger.call(\n    'AccountTrackerController:getState',\n  );\n\n  const nativeChainIds = Object.keys(\n    accountTrackerControllerState.accountsByChainId,\n  ) as Hex[];\n\n  const normalizedAccount = account.toLowerCase() as Hex;\n\n  const balancesByTokenByChain =\n    tokenBalanceControllerState.tokenBalances?.[normalizedAccount];\n\n  const tokenChainIds = Object.keys(balancesByTokenByChain) as Hex[];\n  const chainIds = uniq([...tokenChainIds, ...nativeChainIds]);\n\n  return chainIds.flatMap((chainId) => {\n    const tokenAddresses = [\n      ...(Object.keys(balancesByTokenByChain[chainId] ?? {}) as Hex[]),\n      getNativeToken(chainId),\n    ];\n\n    return tokenAddresses.map((tokenAddress) => ({\n      chainId,\n      tokenAddress,\n      balance: getTokenBalance(messenger, account, chainId, tokenAddress),\n    }));\n  });\n}\n\n/**\n * Get the token decimals for a specific token.\n *\n * @param messenger - Controller messenger.\n * @param tokenAddress - Address of the token contract.\n * @param chainId - Id of the chain.\n * @returns The token decimals or undefined if the token is not found.\n */\nexport function getTokenInfo(\n  messenger: TransactionPayControllerMessenger,\n  tokenAddress: Hex,\n  chainId: Hex,\n): { decimals: number; symbol: string } | undefined {\n  const controllerState = messenger.call('TokensController:getState');\n  const normalizedTokenAddress = tokenAddress.toLowerCase() as Hex;\n\n  const isNative =\n    normalizedTokenAddress === getNativeToken(chainId).toLowerCase();\n\n  const token = Object.values(controllerState.allTokens?.[chainId] ?? {})\n    .flat()\n    .find(\n      (singleToken) =>\n        singleToken.address.toLowerCase() === normalizedTokenAddress,\n    );\n\n  if (!token && !isNative) {\n    return undefined;\n  }\n\n  if (token && !isNative) {\n    return { decimals: Number(token.decimals), symbol: token.symbol };\n  }\n\n  const ticker = getTicker(chainId, messenger);\n\n  if (!ticker) {\n    return undefined;\n  }\n\n  return { decimals: 18, symbol: ticker };\n}\n\n/**\n * Calculate fiat rates for a specific token.\n *\n * @param messenger - Controller messenger.\n * @param tokenAddress - Address of the token contract.\n * @param chainId - Id of the chain.\n * @returns An object containing the USD and fiat rates, or undefined if rates are not available.\n */\nexport function getTokenFiatRate(\n  messenger: TransactionPayControllerMessenger,\n  tokenAddress: Hex,\n  chainId: Hex,\n): FiatRates | undefined {\n  const ticker = getTicker(chainId, messenger);\n\n  if (!ticker) {\n    return undefined;\n  }\n\n  const rateControllerState = messenger.call('TokenRatesController:getState');\n\n  const currencyRateControllerState = messenger.call(\n    'CurrencyRateController:getState',\n  );\n\n  const normalizedTokenAddress = toChecksumHexAddress(tokenAddress) as Hex;\n  const isNative = normalizedTokenAddress === getNativeToken(chainId);\n\n  const tokenToNativeRate =\n    rateControllerState.marketData?.[chainId]?.[normalizedTokenAddress]?.price;\n\n  if (tokenToNativeRate === undefined && !isNative) {\n    return undefined;\n  }\n\n  const {\n    conversionRate: nativeToFiatRate,\n    usdConversionRate: nativeToUsdRate,\n  } = currencyRateControllerState.currencyRates?.[ticker] ?? {\n    conversionRate: null,\n    usdConversionRate: null,\n  };\n\n  if (nativeToFiatRate === null || nativeToUsdRate === null) {\n    return undefined;\n  }\n  const isStablecoin = STABLECOINS[chainId]?.includes(\n    tokenAddress.toLowerCase() as Hex,\n  );\n\n  const usdRate = isStablecoin\n    ? '1'\n    : new BigNumber(tokenToNativeRate ?? 1)\n        .multipliedBy(nativeToUsdRate)\n        .toString(10);\n\n  const fiatRate = new BigNumber(tokenToNativeRate ?? 1)\n    .multipliedBy(nativeToFiatRate)\n    .toString(10);\n\n  return { usdRate, fiatRate };\n}\n\n/**\n * Calculate the human-readable, raw, USD, and fiat representations of a token amount.\n *\n * @param rawInput - Raw token amount (decimal string, hex, or BigNumber).\n * @param decimals - Number of decimals for the token.\n * @param fiatRates - Fiat rates for the token.\n * @returns Object containing the amount in raw, human-readable, USD, and fiat formats.\n */\nexport function computeTokenAmounts(\n  rawInput: BigNumber.Value,\n  decimals: number,\n  fiatRates: FiatRates,\n): {\n  raw: string;\n  human: string;\n  usd: string;\n  fiat: string;\n} {\n  const rawValue = new BigNumber(rawInput);\n  const humanValue = rawValue.shiftedBy(-decimals);\n\n  return {\n    raw: rawValue.toFixed(0),\n    human: humanValue.toString(10),\n    usd: humanValue.multipliedBy(fiatRates.usdRate).toString(10),\n    fiat: humanValue.multipliedBy(fiatRates.fiatRate).toString(10),\n  };\n}\n\n/**\n * Get the native token address for a given chain ID.\n *\n * @param chainId - Chain ID.\n * @returns - Native token address for the given chain ID.\n */\nexport function getNativeToken(chainId: Hex): Hex {\n  switch (chainId) {\n    case '0x89':\n      return '0x0000000000000000000000000000000000001010';\n    default:\n      return NATIVE_TOKEN_ADDRESS;\n  }\n}\n\n/**\n * Get the live on-chain token balance via an RPC `eth_call` to the ERC-20\n * `balanceOf` function, or `eth_getBalance` for native tokens.\n *\n * Unlike {@link getTokenBalance}, this bypasses the cached state in\n * `TokenBalancesController` and reads directly from the chain.\n *\n * @param messenger - Controller messenger.\n * @param account - Address of the account.\n * @param chainId - Chain ID.\n * @param tokenAddress - Address of the token contract.\n * @returns Raw token balance as a decimal string.\n */\nexport async function getLiveTokenBalance(\n  messenger: TransactionPayControllerMessenger,\n  account: Hex,\n  chainId: Hex,\n  tokenAddress: Hex,\n): Promise<string> {\n  const networkClientId = messenger.call(\n    'NetworkController:findNetworkClientIdByChainId',\n    chainId,\n  );\n\n  const { provider } = messenger.call(\n    'NetworkController:getNetworkClientById',\n    networkClientId,\n  );\n\n  const ethersProvider = new Web3Provider(provider);\n  const isNative =\n    tokenAddress.toLowerCase() === getNativeToken(chainId).toLowerCase();\n\n  if (isNative) {\n    const balance = await ethersProvider.getBalance(account);\n    return balance.toString();\n  }\n\n  const contract = new Contract(tokenAddress, abiERC20, ethersProvider);\n  const balance = await contract.balanceOf(account);\n  return balance.toString();\n}\n\nfunction getTicker(\n  chainId: Hex,\n  messenger: TransactionPayControllerMessenger,\n): string | undefined {\n  try {\n    const networkClientId = messenger.call(\n      'NetworkController:findNetworkClientIdByChainId',\n      chainId,\n    );\n\n    const networkConfiguration = messenger.call(\n      'NetworkController:getNetworkClientById',\n      networkClientId,\n    );\n\n    return networkConfiguration.configuration.ticker;\n  } catch {\n    return undefined;\n  }\n}\n\nexport enum TokenAddressTarget {\n  Relay = 'relay',\n  MetaMask = 'metamask',\n}\n\n/**\n * Normalize token address formats between MetaMask and Relay for Polygon native\n * token handling.\n *\n * MetaMask uses Polygon's native token contract-like address (`0x...1010`),\n * while Relay expects the zero address for native tokens.\n *\n * @param tokenAddress - Token address to normalize.\n * @param chainId - Chain ID for the token.\n * @param target - Optional target system format.\n * @returns Normalized token address for the target system, or the original\n * address if no target is provided.\n */\nexport function normalizeTokenAddress(\n  tokenAddress: Hex,\n  chainId: Hex,\n  target?: TokenAddressTarget,\n): Hex {\n  if (chainId !== CHAIN_ID_POLYGON) {\n    return tokenAddress;\n  }\n\n  const nativeTokenAddress = getNativeToken(chainId).toLowerCase() as Hex;\n  const normalizedTokenAddress = tokenAddress.toLowerCase();\n\n  if (\n    target === TokenAddressTarget.Relay &&\n    normalizedTokenAddress === nativeTokenAddress\n  ) {\n    return NATIVE_TOKEN_ADDRESS;\n  }\n\n  if (\n    target === TokenAddressTarget.MetaMask &&\n    normalizedTokenAddress === NATIVE_TOKEN_ADDRESS.toLowerCase()\n  ) {\n    return nativeTokenAddress;\n  }\n\n  return tokenAddress;\n}\n"]}
+{"version":3,"file":"token.cjs","sourceRoot":"","sources":["../../src/utils/token.ts"],"names":[],"mappings":";;;AAAA,wDAAoD;AACpD,wDAAwD;AAExD,iEAAkE;AAClE,mEAAuD;AAEvD,+CAAyC;AAEzC,uDAA6D;AAC7D,gDAIsB;AAGtB;;;;;;;;;;GAUG;AACH,SAAgB,WAAW,CACzB,MAAsC,EACtC,MAAsC;IAEtC,OAAO,CACL,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE;QAC7D,MAAM,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,CAClC,CAAC;AACJ,CAAC;AARD,kCAQC;AAED;;;;;;;;GAQG;AACH,SAAgB,eAAe,CAC7B,SAA4C,EAC5C,OAAY,EACZ,OAAY,EACZ,YAAiB;IAEjB,MAAM,8BAA8B,GAAG,IAAA,0CAA0B,EAAC,SAAS,CAAC,CAAC;IAE7E,IAAI,aAAa,CAAC;IAClB,IAAI,iBAAiB,CAAC;IACtB,IAAI,8BAA8B,EAAE,CAAC;QACnC,MAAM,qBAAqB,GAAG,SAAS,CAAC,IAAI,CAC1C,4CAA4C,CAC7C,CAAC;QAEF,aAAa,GAAG,qBAAqB,EAAE,aAAa,CAAC;QACrD,iBAAiB,GAAG,qBAAqB,EAAE,iBAAiB,CAAC;IAC/D,CAAC;SAAM,CAAC;QACN,aAAa,GAAG,SAAS,CAAC,IAAI,CAC5B,kCAAkC,CACnC,EAAE,aAAa,CAAC;QACjB,iBAAiB,GAAG,SAAS,CAAC,IAAI,CAChC,mCAAmC,CACpC,EAAE,iBAAiB,CAAC;IACvB,CAAC;IAED,MAAM,iBAAiB,GAAG,OAAO,CAAC,WAAW,EAAS,CAAC;IACvD,MAAM,sBAAsB,GAAG,IAAA,uCAAoB,EAAC,YAAY,CAAQ,CAAC;IACzE,MAAM,QAAQ,GAAG,sBAAsB,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC;IAEpE,MAAM,UAAU,GACd,aAAa,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,sBAAsB,CAAC,CAAC;IAE1E,IAAI,CAAC,QAAQ,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC1C,OAAO,GAAG,CAAC;IACb,CAAC;IAED,IAAI,CAAC,QAAQ,IAAI,UAAU,EAAE,CAAC;QAC5B,OAAO,IAAI,wBAAS,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACpD,CAAC;IAED,MAAM,aAAa,GAAG,iBAAiB,EAAE,CAAC,OAAO,CAAC,CAAC;IAEnD,MAAM,eAAe,GAAG,IAAA,uCAAoB,EAAC,iBAAiB,CAAQ,CAAC;IACvE,MAAM,gBAAgB,GAAG,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE,OAAc,CAAC;IAE1E,OAAO,IAAI,wBAAS,CAAC,gBAAgB,IAAI,KAAK,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;AACnE,CAAC;AA/CD,0CA+CC;AAED;;;;;;;GAOG;AACH,SAAgB,YAAY,CAC1B,SAA4C,EAC5C,YAAiB,EACjB,OAAY;IAEZ,MAAM,8BAA8B,GAAG,IAAA,0CAA0B,EAAC,SAAS,CAAC,CAAC;IAE7E,IAAI,SAA6C,CAAC;IAClD,IAAI,8BAA8B,EAAE,CAAC;QACnC,SAAS,GAAG,SAAS,CAAC,IAAI,CACxB,4CAA4C,CAC7C,EAAE,SAAS,CAAC;IACf,CAAC;SAAM,CAAC;QACN,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,2BAA2B,CAAC,EAAE,SAAS,CAAC;IACrE,CAAC;IAED,MAAM,sBAAsB,GAAG,YAAY,CAAC,WAAW,EAAS,CAAC;IAEjE,MAAM,QAAQ,GACZ,sBAAsB,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;IAEnE,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;SACpD,IAAI,EAAE;SACN,IAAI,CACH,CAAC,WAAW,EAAE,EAAE,CACd,WAAW,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,sBAAsB,CAC/D,CAAC;IAEJ,IAAI,CAAC,KAAK,IAAI,CAAC,QAAQ,EAAE,CAAC;QACxB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,KAAK,IAAI,CAAC,QAAQ,EAAE,CAAC;QACvB,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;IACpE,CAAC;IAED,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAE7C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;AAC1C,CAAC;AA3CD,oCA2CC;AAED;;;;;;;GAOG;AACH,SAAgB,gBAAgB,CAC9B,SAA4C,EAC5C,YAAiB,EACjB,OAAY;IAEZ,MAAM,8BAA8B,GAAG,IAAA,0CAA0B,EAAC,SAAS,CAAC,CAAC;IAE7E,IAAI,UAAU,CAAC;IACf,IAAI,aAAa,CAAC;IAClB,IAAI,8BAA8B,EAAE,CAAC;QACnC,MAAM,qBAAqB,GAAG,SAAS,CAAC,IAAI,CAC1C,4CAA4C,CAC7C,CAAC;QAEF,UAAU,GAAG,qBAAqB,EAAE,UAAU,CAAC;QAC/C,aAAa,GAAG,qBAAqB,EAAE,aAAa,CAAC;IACvD,CAAC;SAAM,CAAC;QACN,UAAU,GAAG,SAAS,CAAC,IAAI,CAAC,+BAA+B,CAAC,EAAE,UAAU,CAAC;QACzE,aAAa,GAAG,SAAS,CAAC,IAAI,CAC5B,iCAAiC,CAClC,EAAE,aAAa,CAAC;IACnB,CAAC;IAED,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAE7C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,sBAAsB,GAAG,IAAA,uCAAoB,EAAC,YAAY,CAAQ,CAAC;IACzE,MAAM,QAAQ,GAAG,sBAAsB,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC;IAEpE,MAAM,iBAAiB,GACrB,UAAU,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,sBAAsB,CAAC,EAAE,KAAK,CAAC;IAEzD,IAAI,iBAAiB,KAAK,SAAS,IAAI,CAAC,QAAQ,EAAE,CAAC;QACjD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,EACJ,cAAc,EAAE,gBAAgB,EAChC,iBAAiB,EAAE,eAAe,GACnC,GAAG,aAAa,EAAE,CAAC,MAAM,CAAC,IAAI;QAC7B,cAAc,EAAE,IAAI;QACpB,iBAAiB,EAAE,IAAI;KACxB,CAAC;IAEF,IAAI,gBAAgB,KAAK,IAAI,IAAI,eAAe,KAAK,IAAI,EAAE,CAAC;QAC1D,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,YAAY,GAAG,uBAAW,CAAC,OAAO,CAAC,EAAE,QAAQ,CACjD,YAAY,CAAC,WAAW,EAAS,CAClC,CAAC;IAEF,MAAM,OAAO,GAAG,YAAY;QAC1B,CAAC,CAAC,GAAG;QACL,CAAC,CAAC,IAAI,wBAAS,CAAC,iBAAiB,IAAI,CAAC,CAAC;aAClC,YAAY,CAAC,eAAe,CAAC;aAC7B,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEpB,MAAM,QAAQ,GAAG,IAAI,wBAAS,CAAC,iBAAiB,IAAI,CAAC,CAAC;SACnD,YAAY,CAAC,gBAAgB,CAAC;SAC9B,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEhB,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;AAC/B,CAAC;AAjED,4CAiEC;AAED;;;;;;;GAOG;AACH,SAAgB,mBAAmB,CACjC,QAAyB,EACzB,QAAgB,EAChB,SAAoB;IAOpB,MAAM,QAAQ,GAAG,IAAI,wBAAS,CAAC,QAAQ,CAAC,CAAC;IACzC,MAAM,UAAU,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,CAAC;IAEjD,OAAO;QACL,GAAG,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QACxB,KAAK,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC9B,GAAG,EAAE,UAAU,CAAC,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5D,IAAI,EAAE,UAAU,CAAC,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC;KAC/D,CAAC;AACJ,CAAC;AAnBD,kDAmBC;AAED;;;;;GAKG;AACH,SAAgB,cAAc,CAAC,OAAY;IACzC,QAAQ,OAAO,EAAE,CAAC;QAChB,KAAK,MAAM;YACT,OAAO,4CAA4C,CAAC;QACtD;YACE,OAAO,gCAAoB,CAAC;IAChC,CAAC;AACH,CAAC;AAPD,wCAOC;AAED;;;;;;;;;;;;GAYG;AACI,KAAK,UAAU,mBAAmB,CACvC,SAA4C,EAC5C,OAAY,EACZ,OAAY,EACZ,YAAiB;IAEjB,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CACpC,gDAAgD,EAChD,OAAO,CACR,CAAC;IAEF,MAAM,EAAE,QAAQ,EAAE,GAAG,SAAS,CAAC,IAAI,CACjC,wCAAwC,EACxC,eAAe,CAChB,CAAC;IAEF,MAAM,cAAc,GAAG,IAAI,wBAAY,CAAC,QAAQ,CAAC,CAAC;IAClD,MAAM,QAAQ,GACZ,YAAY,CAAC,WAAW,EAAE,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;IAEvE,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACzD,OAAO,OAAO,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAAC,YAAY,EAAE,4BAAQ,EAAE,cAAc,CAAC,CAAC;IACtE,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAClD,OAAO,OAAO,CAAC,QAAQ,EAAE,CAAC;AAC5B,CAAC;AA5BD,kDA4BC;AAED,SAAS,SAAS,CAChB,OAAY,EACZ,SAA4C;IAE5C,IAAI,CAAC;QACH,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CACpC,gDAAgD,EAChD,OAAO,CACR,CAAC;QAEF,MAAM,oBAAoB,GAAG,SAAS,CAAC,IAAI,CACzC,wCAAwC,EACxC,eAAe,CAChB,CAAC;QAEF,OAAO,oBAAoB,CAAC,aAAa,CAAC,MAAM,CAAC;IACnD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,IAAY,kBAGX;AAHD,WAAY,kBAAkB;IAC5B,qCAAe,CAAA;IACf,2CAAqB,CAAA;AACvB,CAAC,EAHW,kBAAkB,kCAAlB,kBAAkB,QAG7B;AAED;;;;;;;;;;;;GAYG;AACH,SAAgB,qBAAqB,CACnC,YAAiB,EACjB,OAAY,EACZ,MAA2B;IAE3B,IAAI,OAAO,KAAK,4BAAgB,EAAE,CAAC;QACjC,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,MAAM,kBAAkB,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC,WAAW,EAAS,CAAC;IACxE,MAAM,sBAAsB,GAAG,YAAY,CAAC,WAAW,EAAE,CAAC;IAE1D,IACE,MAAM,KAAK,kBAAkB,CAAC,KAAK;QACnC,sBAAsB,KAAK,kBAAkB,EAC7C,CAAC;QACD,OAAO,gCAAoB,CAAC;IAC9B,CAAC;IAED,IACE,MAAM,KAAK,kBAAkB,CAAC,QAAQ;QACtC,sBAAsB,KAAK,gCAAoB,CAAC,WAAW,EAAE,EAC7D,CAAC;QACD,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC;AA3BD,sDA2BC","sourcesContent":["import { Contract } from '@ethersproject/contracts';\nimport { Web3Provider } from '@ethersproject/providers';\nimport { TokensControllerState } from '@metamask/assets-controllers';\nimport { toChecksumHexAddress } from '@metamask/controller-utils';\nimport { abiERC20 } from '@metamask/metamask-eth-abis';\nimport type { Hex } from '@metamask/utils';\nimport { BigNumber } from 'bignumber.js';\n\nimport { getAssetsUnifyStateFeature } from './feature-flags';\nimport {\n  CHAIN_ID_POLYGON,\n  NATIVE_TOKEN_ADDRESS,\n  STABLECOINS,\n} from '../constants';\nimport type { FiatRates, TransactionPayControllerMessenger } from '../types';\n\n/**\n * Check if two tokens are the same (same address and chain).\n *\n * @param token1 - First token identifier.\n * @param token1.address - Token address.\n * @param token1.chainId - Token chain ID.\n * @param token2 - Second token identifier.\n * @param token2.address - Token address.\n * @param token2.chainId - Token chain ID.\n * @returns True if tokens are the same, false otherwise.\n */\nexport function isSameToken(\n  token1: { address: Hex; chainId: Hex },\n  token2: { address: Hex; chainId: Hex },\n): boolean {\n  return (\n    token1.address.toLowerCase() === token2.address.toLowerCase() &&\n    token1.chainId === token2.chainId\n  );\n}\n\n/**\n * Get the token balance for a specific account and token.\n *\n * @param messenger - Controller messenger.\n * @param account - Address of the account.\n * @param chainId - Id of the chain.\n * @param tokenAddress - Address of the token contract.\n * @returns Raw token balance as a decimal string.\n */\nexport function getTokenBalance(\n  messenger: TransactionPayControllerMessenger,\n  account: Hex,\n  chainId: Hex,\n  tokenAddress: Hex,\n): string {\n  const assetsUnifyStateFeatureEnabled = getAssetsUnifyStateFeature(messenger);\n\n  let tokenBalances;\n  let accountsByChainId;\n  if (assetsUnifyStateFeatureEnabled) {\n    const assetsControllerState = messenger.call(\n      'AssetsController:getStateForTransactionPay',\n    );\n\n    tokenBalances = assetsControllerState?.tokenBalances;\n    accountsByChainId = assetsControllerState?.accountsByChainId;\n  } else {\n    tokenBalances = messenger.call(\n      'TokenBalancesController:getState',\n    )?.tokenBalances;\n    accountsByChainId = messenger.call(\n      'AccountTrackerController:getState',\n    )?.accountsByChainId;\n  }\n\n  const normalizedAccount = account.toLowerCase() as Hex;\n  const normalizedTokenAddress = toChecksumHexAddress(tokenAddress) as Hex;\n  const isNative = normalizedTokenAddress === getNativeToken(chainId);\n\n  const balanceHex =\n    tokenBalances?.[normalizedAccount]?.[chainId]?.[normalizedTokenAddress];\n\n  if (!isNative && balanceHex === undefined) {\n    return '0';\n  }\n\n  if (!isNative && balanceHex) {\n    return new BigNumber(balanceHex, 16).toString(10);\n  }\n\n  const chainAccounts = accountsByChainId?.[chainId];\n\n  const checksumAccount = toChecksumHexAddress(normalizedAccount) as Hex;\n  const nativeBalanceHex = chainAccounts?.[checksumAccount]?.balance as Hex;\n\n  return new BigNumber(nativeBalanceHex ?? '0x0', 16).toString(10);\n}\n\n/**\n * Get the token decimals for a specific token.\n *\n * @param messenger - Controller messenger.\n * @param tokenAddress - Address of the token contract.\n * @param chainId - Id of the chain.\n * @returns The token decimals or undefined if the token is not found.\n */\nexport function getTokenInfo(\n  messenger: TransactionPayControllerMessenger,\n  tokenAddress: Hex,\n  chainId: Hex,\n): { decimals: number; symbol: string } | undefined {\n  const assetsUnifyStateFeatureEnabled = getAssetsUnifyStateFeature(messenger);\n\n  let allTokens: TokensControllerState['allTokens'];\n  if (assetsUnifyStateFeatureEnabled) {\n    allTokens = messenger.call(\n      'AssetsController:getStateForTransactionPay',\n    )?.allTokens;\n  } else {\n    allTokens = messenger.call('TokensController:getState')?.allTokens;\n  }\n\n  const normalizedTokenAddress = tokenAddress.toLowerCase() as Hex;\n\n  const isNative =\n    normalizedTokenAddress === getNativeToken(chainId).toLowerCase();\n\n  const token = Object.values(allTokens?.[chainId] ?? {})\n    .flat()\n    .find(\n      (singleToken) =>\n        singleToken.address.toLowerCase() === normalizedTokenAddress,\n    );\n\n  if (!token && !isNative) {\n    return undefined;\n  }\n\n  if (token && !isNative) {\n    return { decimals: Number(token.decimals), symbol: token.symbol };\n  }\n\n  const ticker = getTicker(chainId, messenger);\n\n  if (!ticker) {\n    return undefined;\n  }\n\n  return { decimals: 18, symbol: ticker };\n}\n\n/**\n * Calculate fiat rates for a specific token.\n *\n * @param messenger - Controller messenger.\n * @param tokenAddress - Address of the token contract.\n * @param chainId - Id of the chain.\n * @returns An object containing the USD and fiat rates, or undefined if rates are not available.\n */\nexport function getTokenFiatRate(\n  messenger: TransactionPayControllerMessenger,\n  tokenAddress: Hex,\n  chainId: Hex,\n): FiatRates | undefined {\n  const assetsUnifyStateFeatureEnabled = getAssetsUnifyStateFeature(messenger);\n\n  let marketData;\n  let currencyRates;\n  if (assetsUnifyStateFeatureEnabled) {\n    const assetsControllerState = messenger.call(\n      'AssetsController:getStateForTransactionPay',\n    );\n\n    marketData = assetsControllerState?.marketData;\n    currencyRates = assetsControllerState?.currencyRates;\n  } else {\n    marketData = messenger.call('TokenRatesController:getState')?.marketData;\n    currencyRates = messenger.call(\n      'CurrencyRateController:getState',\n    )?.currencyRates;\n  }\n\n  const ticker = getTicker(chainId, messenger);\n\n  if (!ticker) {\n    return undefined;\n  }\n\n  const normalizedTokenAddress = toChecksumHexAddress(tokenAddress) as Hex;\n  const isNative = normalizedTokenAddress === getNativeToken(chainId);\n\n  const tokenToNativeRate =\n    marketData?.[chainId]?.[normalizedTokenAddress]?.price;\n\n  if (tokenToNativeRate === undefined && !isNative) {\n    return undefined;\n  }\n\n  const {\n    conversionRate: nativeToFiatRate,\n    usdConversionRate: nativeToUsdRate,\n  } = currencyRates?.[ticker] ?? {\n    conversionRate: null,\n    usdConversionRate: null,\n  };\n\n  if (nativeToFiatRate === null || nativeToUsdRate === null) {\n    return undefined;\n  }\n  const isStablecoin = STABLECOINS[chainId]?.includes(\n    tokenAddress.toLowerCase() as Hex,\n  );\n\n  const usdRate = isStablecoin\n    ? '1'\n    : new BigNumber(tokenToNativeRate ?? 1)\n        .multipliedBy(nativeToUsdRate)\n        .toString(10);\n\n  const fiatRate = new BigNumber(tokenToNativeRate ?? 1)\n    .multipliedBy(nativeToFiatRate)\n    .toString(10);\n\n  return { usdRate, fiatRate };\n}\n\n/**\n * Calculate the human-readable, raw, USD, and fiat representations of a token amount.\n *\n * @param rawInput - Raw token amount (decimal string, hex, or BigNumber).\n * @param decimals - Number of decimals for the token.\n * @param fiatRates - Fiat rates for the token.\n * @returns Object containing the amount in raw, human-readable, USD, and fiat formats.\n */\nexport function computeTokenAmounts(\n  rawInput: BigNumber.Value,\n  decimals: number,\n  fiatRates: FiatRates,\n): {\n  raw: string;\n  human: string;\n  usd: string;\n  fiat: string;\n} {\n  const rawValue = new BigNumber(rawInput);\n  const humanValue = rawValue.shiftedBy(-decimals);\n\n  return {\n    raw: rawValue.toFixed(0),\n    human: humanValue.toString(10),\n    usd: humanValue.multipliedBy(fiatRates.usdRate).toString(10),\n    fiat: humanValue.multipliedBy(fiatRates.fiatRate).toString(10),\n  };\n}\n\n/**\n * Get the native token address for a given chain ID.\n *\n * @param chainId - Chain ID.\n * @returns - Native token address for the given chain ID.\n */\nexport function getNativeToken(chainId: Hex): Hex {\n  switch (chainId) {\n    case '0x89':\n      return '0x0000000000000000000000000000000000001010';\n    default:\n      return NATIVE_TOKEN_ADDRESS;\n  }\n}\n\n/**\n * Get the live on-chain token balance via an RPC `eth_call` to the ERC-20\n * `balanceOf` function, or `eth_getBalance` for native tokens.\n *\n * Unlike {@link getTokenBalance}, this bypasses the cached state in\n * `TokenBalancesController` and reads directly from the chain.\n *\n * @param messenger - Controller messenger.\n * @param account - Address of the account.\n * @param chainId - Chain ID.\n * @param tokenAddress - Address of the token contract.\n * @returns Raw token balance as a decimal string.\n */\nexport async function getLiveTokenBalance(\n  messenger: TransactionPayControllerMessenger,\n  account: Hex,\n  chainId: Hex,\n  tokenAddress: Hex,\n): Promise<string> {\n  const networkClientId = messenger.call(\n    'NetworkController:findNetworkClientIdByChainId',\n    chainId,\n  );\n\n  const { provider } = messenger.call(\n    'NetworkController:getNetworkClientById',\n    networkClientId,\n  );\n\n  const ethersProvider = new Web3Provider(provider);\n  const isNative =\n    tokenAddress.toLowerCase() === getNativeToken(chainId).toLowerCase();\n\n  if (isNative) {\n    const balance = await ethersProvider.getBalance(account);\n    return balance.toString();\n  }\n\n  const contract = new Contract(tokenAddress, abiERC20, ethersProvider);\n  const balance = await contract.balanceOf(account);\n  return balance.toString();\n}\n\nfunction getTicker(\n  chainId: Hex,\n  messenger: TransactionPayControllerMessenger,\n): string | undefined {\n  try {\n    const networkClientId = messenger.call(\n      'NetworkController:findNetworkClientIdByChainId',\n      chainId,\n    );\n\n    const networkConfiguration = messenger.call(\n      'NetworkController:getNetworkClientById',\n      networkClientId,\n    );\n\n    return networkConfiguration.configuration.ticker;\n  } catch {\n    return undefined;\n  }\n}\n\nexport enum TokenAddressTarget {\n  Relay = 'relay',\n  MetaMask = 'metamask',\n}\n\n/**\n * Normalize token address formats between MetaMask and Relay for Polygon native\n * token handling.\n *\n * MetaMask uses Polygon's native token contract-like address (`0x...1010`),\n * while Relay expects the zero address for native tokens.\n *\n * @param tokenAddress - Token address to normalize.\n * @param chainId - Chain ID for the token.\n * @param target - Optional target system format.\n * @returns Normalized token address for the target system, or the original\n * address if no target is provided.\n */\nexport function normalizeTokenAddress(\n  tokenAddress: Hex,\n  chainId: Hex,\n  target?: TokenAddressTarget,\n): Hex {\n  if (chainId !== CHAIN_ID_POLYGON) {\n    return tokenAddress;\n  }\n\n  const nativeTokenAddress = getNativeToken(chainId).toLowerCase() as Hex;\n  const normalizedTokenAddress = tokenAddress.toLowerCase();\n\n  if (\n    target === TokenAddressTarget.Relay &&\n    normalizedTokenAddress === nativeTokenAddress\n  ) {\n    return NATIVE_TOKEN_ADDRESS;\n  }\n\n  if (\n    target === TokenAddressTarget.MetaMask &&\n    normalizedTokenAddress === NATIVE_TOKEN_ADDRESS.toLowerCase()\n  ) {\n    return nativeTokenAddress;\n  }\n\n  return tokenAddress;\n}\n"]}

package/dist/utils/token.d.cts

@@ -29,18 +29,6 @@ export declare function isSameToken(token1: {
  * @returns Raw token balance as a decimal string.
  */
 export declare function getTokenBalance(messenger: TransactionPayControllerMessenger, account: Hex, chainId: Hex, tokenAddress: Hex): string;
-/**
- * Get the token balance for a specific account and token.
- *
- * @param messenger - Controller messenger.
- * @param account - Address of the account.
- * @returns The token balance as a BigNumber.
- */
-export declare function getAllTokenBalances(messenger: TransactionPayControllerMessenger, account: Hex): {
-    balance: string;
-    chainId: Hex;
-    tokenAddress: Hex;
-}[];
 /**
  * Get the token decimals for a specific token.
  *

package/dist/utils/token.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"token.d.cts","sourceRoot":"","sources":["../../src/utils/token.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAC3C,OAAO,EAAE,SAAS,EAAE,qBAAqB;AAQzC,OAAO,KAAK,EAAE,SAAS,EAAE,iCAAiC,EAAE,qBAAiB;AAE7E;;;;;;;;;;GAUG;AACH,wBAAgB,WAAW,CACzB,MAAM,EAAE;IAAE,OAAO,EAAE,GAAG,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,EACtC,MAAM,EAAE;IAAE,OAAO,EAAE,GAAG,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,GACrC,OAAO,CAKT;AAED;;;;;;;;GAQG;AACH,wBAAgB,eAAe,CAC7B,SAAS,EAAE,iCAAiC,EAC5C,OAAO,EAAE,GAAG,EACZ,OAAO,EAAE,GAAG,EACZ,YAAY,EAAE,GAAG,GAChB,MAAM,CAiCR;AAED;;;;;;GAMG;AACH,wBAAgB,mBAAmB,CACjC,SAAS,EAAE,iCAAiC,EAC5C,OAAO,EAAE,GAAG,GACX;IACD,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,GAAG,CAAC;IACb,YAAY,EAAE,GAAG,CAAC;CACnB,EAAE,CAiCF;AAED;;;;;;;GAOG;AACH,wBAAgB,YAAY,CAC1B,SAAS,EAAE,iCAAiC,EAC5C,YAAY,EAAE,GAAG,EACjB,OAAO,EAAE,GAAG,GACX;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,GAAG,SAAS,CA6BlD;AAED;;;;;;;GAOG;AACH,wBAAgB,gBAAgB,CAC9B,SAAS,EAAE,iCAAiC,EAC5C,YAAY,EAAE,GAAG,EACjB,OAAO,EAAE,GAAG,GACX,SAAS,GAAG,SAAS,CAiDvB;AAED;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,CACjC,QAAQ,EAAE,SAAS,CAAC,KAAK,EACzB,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,SAAS,GACnB;IACD,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;CACd,CAUA;AAED;;;;;GAKG;AACH,wBAAgB,cAAc,CAAC,OAAO,EAAE,GAAG,GAAG,GAAG,CAOhD;AAED;;;;;;;;;;;;GAYG;AACH,wBAAsB,mBAAmB,CACvC,SAAS,EAAE,iCAAiC,EAC5C,OAAO,EAAE,GAAG,EACZ,OAAO,EAAE,GAAG,EACZ,YAAY,EAAE,GAAG,GAChB,OAAO,CAAC,MAAM,CAAC,CAuBjB;AAuBD,oBAAY,kBAAkB;IAC5B,KAAK,UAAU;IACf,QAAQ,aAAa;CACtB;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,qBAAqB,CACnC,YAAY,EAAE,GAAG,EACjB,OAAO,EAAE,GAAG,EACZ,MAAM,CAAC,EAAE,kBAAkB,GAC1B,GAAG,CAuBL"}
+{"version":3,"file":"token.d.cts","sourceRoot":"","sources":["../../src/utils/token.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAC3C,OAAO,EAAE,SAAS,EAAE,qBAAqB;AAQzC,OAAO,KAAK,EAAE,SAAS,EAAE,iCAAiC,EAAE,qBAAiB;AAE7E;;;;;;;;;;GAUG;AACH,wBAAgB,WAAW,CACzB,MAAM,EAAE;IAAE,OAAO,EAAE,GAAG,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,EACtC,MAAM,EAAE;IAAE,OAAO,EAAE,GAAG,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,GACrC,OAAO,CAKT;AAED;;;;;;;;GAQG;AACH,wBAAgB,eAAe,CAC7B,SAAS,EAAE,iCAAiC,EAC5C,OAAO,EAAE,GAAG,EACZ,OAAO,EAAE,GAAG,EACZ,YAAY,EAAE,GAAG,GAChB,MAAM,CA0CR;AAED;;;;;;;GAOG;AACH,wBAAgB,YAAY,CAC1B,SAAS,EAAE,iCAAiC,EAC5C,YAAY,EAAE,GAAG,EACjB,OAAO,EAAE,GAAG,GACX;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,GAAG,SAAS,CAuClD;AAED;;;;;;;GAOG;AACH,wBAAgB,gBAAgB,CAC9B,SAAS,EAAE,iCAAiC,EAC5C,YAAY,EAAE,GAAG,EACjB,OAAO,EAAE,GAAG,GACX,SAAS,GAAG,SAAS,CA6DvB;AAED;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,CACjC,QAAQ,EAAE,SAAS,CAAC,KAAK,EACzB,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,SAAS,GACnB;IACD,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;CACd,CAUA;AAED;;;;;GAKG;AACH,wBAAgB,cAAc,CAAC,OAAO,EAAE,GAAG,GAAG,GAAG,CAOhD;AAED;;;;;;;;;;;;GAYG;AACH,wBAAsB,mBAAmB,CACvC,SAAS,EAAE,iCAAiC,EAC5C,OAAO,EAAE,GAAG,EACZ,OAAO,EAAE,GAAG,EACZ,YAAY,EAAE,GAAG,GAChB,OAAO,CAAC,MAAM,CAAC,CAuBjB;AAuBD,oBAAY,kBAAkB;IAC5B,KAAK,UAAU;IACf,QAAQ,aAAa;CACtB;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,qBAAqB,CACnC,YAAY,EAAE,GAAG,EACjB,OAAO,EAAE,GAAG,EACZ,MAAM,CAAC,EAAE,kBAAkB,GAC1B,GAAG,CAuBL"}

package/dist/utils/token.d.mts

@@ -29,18 +29,6 @@ export declare function isSameToken(token1: {
  * @returns Raw token balance as a decimal string.
  */
 export declare function getTokenBalance(messenger: TransactionPayControllerMessenger, account: Hex, chainId: Hex, tokenAddress: Hex): string;
-/**
- * Get the token balance for a specific account and token.
- *
- * @param messenger - Controller messenger.
- * @param account - Address of the account.
- * @returns The token balance as a BigNumber.
- */
-export declare function getAllTokenBalances(messenger: TransactionPayControllerMessenger, account: Hex): {
-    balance: string;
-    chainId: Hex;
-    tokenAddress: Hex;
-}[];
 /**
  * Get the token decimals for a specific token.
  *

package/dist/utils/token.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"token.d.mts","sourceRoot":"","sources":["../../src/utils/token.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAC3C,OAAO,EAAE,SAAS,EAAE,qBAAqB;AAQzC,OAAO,KAAK,EAAE,SAAS,EAAE,iCAAiC,EAAE,qBAAiB;AAE7E;;;;;;;;;;GAUG;AACH,wBAAgB,WAAW,CACzB,MAAM,EAAE;IAAE,OAAO,EAAE,GAAG,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,EACtC,MAAM,EAAE;IAAE,OAAO,EAAE,GAAG,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,GACrC,OAAO,CAKT;AAED;;;;;;;;GAQG;AACH,wBAAgB,eAAe,CAC7B,SAAS,EAAE,iCAAiC,EAC5C,OAAO,EAAE,GAAG,EACZ,OAAO,EAAE,GAAG,EACZ,YAAY,EAAE,GAAG,GAChB,MAAM,CAiCR;AAED;;;;;;GAMG;AACH,wBAAgB,mBAAmB,CACjC,SAAS,EAAE,iCAAiC,EAC5C,OAAO,EAAE,GAAG,GACX;IACD,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,GAAG,CAAC;IACb,YAAY,EAAE,GAAG,CAAC;CACnB,EAAE,CAiCF;AAED;;;;;;;GAOG;AACH,wBAAgB,YAAY,CAC1B,SAAS,EAAE,iCAAiC,EAC5C,YAAY,EAAE,GAAG,EACjB,OAAO,EAAE,GAAG,GACX;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,GAAG,SAAS,CA6BlD;AAED;;;;;;;GAOG;AACH,wBAAgB,gBAAgB,CAC9B,SAAS,EAAE,iCAAiC,EAC5C,YAAY,EAAE,GAAG,EACjB,OAAO,EAAE,GAAG,GACX,SAAS,GAAG,SAAS,CAiDvB;AAED;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,CACjC,QAAQ,EAAE,SAAS,CAAC,KAAK,EACzB,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,SAAS,GACnB;IACD,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;CACd,CAUA;AAED;;;;;GAKG;AACH,wBAAgB,cAAc,CAAC,OAAO,EAAE,GAAG,GAAG,GAAG,CAOhD;AAED;;;;;;;;;;;;GAYG;AACH,wBAAsB,mBAAmB,CACvC,SAAS,EAAE,iCAAiC,EAC5C,OAAO,EAAE,GAAG,EACZ,OAAO,EAAE,GAAG,EACZ,YAAY,EAAE,GAAG,GAChB,OAAO,CAAC,MAAM,CAAC,CAuBjB;AAuBD,oBAAY,kBAAkB;IAC5B,KAAK,UAAU;IACf,QAAQ,aAAa;CACtB;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,qBAAqB,CACnC,YAAY,EAAE,GAAG,EACjB,OAAO,EAAE,GAAG,EACZ,MAAM,CAAC,EAAE,kBAAkB,GAC1B,GAAG,CAuBL"}
+{"version":3,"file":"token.d.mts","sourceRoot":"","sources":["../../src/utils/token.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAC3C,OAAO,EAAE,SAAS,EAAE,qBAAqB;AAQzC,OAAO,KAAK,EAAE,SAAS,EAAE,iCAAiC,EAAE,qBAAiB;AAE7E;;;;;;;;;;GAUG;AACH,wBAAgB,WAAW,CACzB,MAAM,EAAE;IAAE,OAAO,EAAE,GAAG,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,EACtC,MAAM,EAAE;IAAE,OAAO,EAAE,GAAG,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,GACrC,OAAO,CAKT;AAED;;;;;;;;GAQG;AACH,wBAAgB,eAAe,CAC7B,SAAS,EAAE,iCAAiC,EAC5C,OAAO,EAAE,GAAG,EACZ,OAAO,EAAE,GAAG,EACZ,YAAY,EAAE,GAAG,GAChB,MAAM,CA0CR;AAED;;;;;;;GAOG;AACH,wBAAgB,YAAY,CAC1B,SAAS,EAAE,iCAAiC,EAC5C,YAAY,EAAE,GAAG,EACjB,OAAO,EAAE,GAAG,GACX;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,GAAG,SAAS,CAuClD;AAED;;;;;;;GAOG;AACH,wBAAgB,gBAAgB,CAC9B,SAAS,EAAE,iCAAiC,EAC5C,YAAY,EAAE,GAAG,EACjB,OAAO,EAAE,GAAG,GACX,SAAS,GAAG,SAAS,CA6DvB;AAED;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,CACjC,QAAQ,EAAE,SAAS,CAAC,KAAK,EACzB,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,SAAS,GACnB;IACD,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;CACd,CAUA;AAED;;;;;GAKG;AACH,wBAAgB,cAAc,CAAC,OAAO,EAAE,GAAG,GAAG,GAAG,CAOhD;AAED;;;;;;;;;;;;GAYG;AACH,wBAAsB,mBAAmB,CACvC,SAAS,EAAE,iCAAiC,EAC5C,OAAO,EAAE,GAAG,EACZ,OAAO,EAAE,GAAG,EACZ,YAAY,EAAE,GAAG,GAChB,OAAO,CAAC,MAAM,CAAC,CAuBjB;AAuBD,oBAAY,kBAAkB;IAC5B,KAAK,UAAU;IACf,QAAQ,aAAa;CACtB;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,qBAAqB,CACnC,YAAY,EAAE,GAAG,EACjB,OAAO,EAAE,GAAG,EACZ,MAAM,CAAC,EAAE,kBAAkB,GAC1B,GAAG,CAuBL"}

package/dist/utils/token.mjs

@@ -3,8 +3,7 @@ import { Web3Provider } from "@ethersproject/providers";
 import { toChecksumHexAddress } from "@metamask/controller-utils";
 import { abiERC20 } from "@metamask/metamask-eth-abis";
 import { BigNumber } from "bignumber.js";
-import $lodash from "lodash";
-const { uniq } = $lodash;
+import { getAssetsUnifyStateFeature } from "./feature-flags.mjs";
 import { CHAIN_ID_POLYGON, NATIVE_TOKEN_ADDRESS, STABLECOINS } from "../constants.mjs";
 /**
  * Check if two tokens are the same (same address and chain).
@@ -31,50 +30,33 @@ export function isSameToken(token1, token2) {
  * @returns Raw token balance as a decimal string.
  */
 export function getTokenBalance(messenger, account, chainId, tokenAddress) {
-    const tokenBalanceControllerState = messenger.call('TokenBalancesController:getState');
+    const assetsUnifyStateFeatureEnabled = getAssetsUnifyStateFeature(messenger);
+    let tokenBalances;
+    let accountsByChainId;
+    if (assetsUnifyStateFeatureEnabled) {
+        const assetsControllerState = messenger.call('AssetsController:getStateForTransactionPay');
+        tokenBalances = assetsControllerState?.tokenBalances;
+        accountsByChainId = assetsControllerState?.accountsByChainId;
+    }
+    else {
+        tokenBalances = messenger.call('TokenBalancesController:getState')?.tokenBalances;
+        accountsByChainId = messenger.call('AccountTrackerController:getState')?.accountsByChainId;
+    }
     const normalizedAccount = account.toLowerCase();
     const normalizedTokenAddress = toChecksumHexAddress(tokenAddress);
     const isNative = normalizedTokenAddress === getNativeToken(chainId);
-    const balanceHex = tokenBalanceControllerState.tokenBalances?.[normalizedAccount]?.[chainId]?.[normalizedTokenAddress];
+    const balanceHex = tokenBalances?.[normalizedAccount]?.[chainId]?.[normalizedTokenAddress];
     if (!isNative && balanceHex === undefined) {
         return '0';
     }
     if (!isNative && balanceHex) {
         return new BigNumber(balanceHex, 16).toString(10);
     }
-    const accountTrackerControllerState = messenger.call('AccountTrackerController:getState');
-    const chainAccounts = accountTrackerControllerState.accountsByChainId?.[chainId];
+    const chainAccounts = accountsByChainId?.[chainId];
     const checksumAccount = toChecksumHexAddress(normalizedAccount);
     const nativeBalanceHex = chainAccounts?.[checksumAccount]?.balance;
     return new BigNumber(nativeBalanceHex ?? '0x0', 16).toString(10);
 }
-/**
- * Get the token balance for a specific account and token.
- *
- * @param messenger - Controller messenger.
- * @param account - Address of the account.
- * @returns The token balance as a BigNumber.
- */
-export function getAllTokenBalances(messenger, account) {
-    const tokenBalanceControllerState = messenger.call('TokenBalancesController:getState');
-    const accountTrackerControllerState = messenger.call('AccountTrackerController:getState');
-    const nativeChainIds = Object.keys(accountTrackerControllerState.accountsByChainId);
-    const normalizedAccount = account.toLowerCase();
-    const balancesByTokenByChain = tokenBalanceControllerState.tokenBalances?.[normalizedAccount];
-    const tokenChainIds = Object.keys(balancesByTokenByChain);
-    const chainIds = uniq([...tokenChainIds, ...nativeChainIds]);
-    return chainIds.flatMap((chainId) => {
-        const tokenAddresses = [
-            ...Object.keys(balancesByTokenByChain[chainId] ?? {}),
-            getNativeToken(chainId),
-        ];
-        return tokenAddresses.map((tokenAddress) => ({
-            chainId,
-            tokenAddress,
-            balance: getTokenBalance(messenger, account, chainId, tokenAddress),
-        }));
-    });
-}
 /**
  * Get the token decimals for a specific token.
  *
@@ -84,10 +66,17 @@ export function getAllTokenBalances(messenger, account) {
  * @returns The token decimals or undefined if the token is not found.
  */
 export function getTokenInfo(messenger, tokenAddress, chainId) {
-    const controllerState = messenger.call('TokensController:getState');
+    const assetsUnifyStateFeatureEnabled = getAssetsUnifyStateFeature(messenger);
+    let allTokens;
+    if (assetsUnifyStateFeatureEnabled) {
+        allTokens = messenger.call('AssetsController:getStateForTransactionPay')?.allTokens;
+    }
+    else {
+        allTokens = messenger.call('TokensController:getState')?.allTokens;
+    }
     const normalizedTokenAddress = tokenAddress.toLowerCase();
     const isNative = normalizedTokenAddress === getNativeToken(chainId).toLowerCase();
-    const token = Object.values(controllerState.allTokens?.[chainId] ?? {})
+    const token = Object.values(allTokens?.[chainId] ?? {})
         .flat()
         .find((singleToken) => singleToken.address.toLowerCase() === normalizedTokenAddress);
     if (!token && !isNative) {
@@ -111,19 +100,29 @@ export function getTokenInfo(messenger, tokenAddress, chainId) {
  * @returns An object containing the USD and fiat rates, or undefined if rates are not available.
  */
 export function getTokenFiatRate(messenger, tokenAddress, chainId) {
+    const assetsUnifyStateFeatureEnabled = getAssetsUnifyStateFeature(messenger);
+    let marketData;
+    let currencyRates;
+    if (assetsUnifyStateFeatureEnabled) {
+        const assetsControllerState = messenger.call('AssetsController:getStateForTransactionPay');
+        marketData = assetsControllerState?.marketData;
+        currencyRates = assetsControllerState?.currencyRates;
+    }
+    else {
+        marketData = messenger.call('TokenRatesController:getState')?.marketData;
+        currencyRates = messenger.call('CurrencyRateController:getState')?.currencyRates;
+    }
     const ticker = getTicker(chainId, messenger);
     if (!ticker) {
         return undefined;
     }
-    const rateControllerState = messenger.call('TokenRatesController:getState');
-    const currencyRateControllerState = messenger.call('CurrencyRateController:getState');
     const normalizedTokenAddress = toChecksumHexAddress(tokenAddress);
     const isNative = normalizedTokenAddress === getNativeToken(chainId);
-    const tokenToNativeRate = rateControllerState.marketData?.[chainId]?.[normalizedTokenAddress]?.price;
+    const tokenToNativeRate = marketData?.[chainId]?.[normalizedTokenAddress]?.price;
     if (tokenToNativeRate === undefined && !isNative) {
         return undefined;
     }
-    const { conversionRate: nativeToFiatRate, usdConversionRate: nativeToUsdRate, } = currencyRateControllerState.currencyRates?.[ticker] ?? {
+    const { conversionRate: nativeToFiatRate, usdConversionRate: nativeToUsdRate, } = currencyRates?.[ticker] ?? {
         conversionRate: null,
         usdConversionRate: null,
     };

package/dist/utils/token.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"token.mjs","sourceRoot":"","sources":["../../src/utils/token.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,iCAAiC;AACpD,OAAO,EAAE,YAAY,EAAE,iCAAiC;AACxD,OAAO,EAAE,oBAAoB,EAAE,mCAAmC;AAClE,OAAO,EAAE,QAAQ,EAAE,oCAAoC;AAEvD,OAAO,EAAE,SAAS,EAAE,qBAAqB;;;AAGzC,OAAO,EACL,gBAAgB,EAChB,oBAAoB,EACpB,WAAW,EACZ,yBAAqB;AAGtB;;;;;;;;;;GAUG;AACH,MAAM,UAAU,WAAW,CACzB,MAAsC,EACtC,MAAsC;IAEtC,OAAO,CACL,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE;QAC7D,MAAM,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,CAClC,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,eAAe,CAC7B,SAA4C,EAC5C,OAAY,EACZ,OAAY,EACZ,YAAiB;IAEjB,MAAM,2BAA2B,GAAG,SAAS,CAAC,IAAI,CAChD,kCAAkC,CACnC,CAAC;IAEF,MAAM,iBAAiB,GAAG,OAAO,CAAC,WAAW,EAAS,CAAC;IACvD,MAAM,sBAAsB,GAAG,oBAAoB,CAAC,YAAY,CAAQ,CAAC;IACzE,MAAM,QAAQ,GAAG,sBAAsB,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC;IAEpE,MAAM,UAAU,GACd,2BAA2B,CAAC,aAAa,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,CACzE,sBAAsB,CACvB,CAAC;IAEJ,IAAI,CAAC,QAAQ,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC1C,OAAO,GAAG,CAAC;IACb,CAAC;IAED,IAAI,CAAC,QAAQ,IAAI,UAAU,EAAE,CAAC;QAC5B,OAAO,IAAI,SAAS,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACpD,CAAC;IAED,MAAM,6BAA6B,GAAG,SAAS,CAAC,IAAI,CAClD,mCAAmC,CACpC,CAAC;IAEF,MAAM,aAAa,GACjB,6BAA6B,CAAC,iBAAiB,EAAE,CAAC,OAAO,CAAC,CAAC;IAE7D,MAAM,eAAe,GAAG,oBAAoB,CAAC,iBAAiB,CAAQ,CAAC;IACvE,MAAM,gBAAgB,GAAG,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE,OAAc,CAAC;IAE1E,OAAO,IAAI,SAAS,CAAC,gBAAgB,IAAI,KAAK,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;AACnE,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,mBAAmB,CACjC,SAA4C,EAC5C,OAAY;IAMZ,MAAM,2BAA2B,GAAG,SAAS,CAAC,IAAI,CAChD,kCAAkC,CACnC,CAAC;IAEF,MAAM,6BAA6B,GAAG,SAAS,CAAC,IAAI,CAClD,mCAAmC,CACpC,CAAC;IAEF,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAChC,6BAA6B,CAAC,iBAAiB,CACvC,CAAC;IAEX,MAAM,iBAAiB,GAAG,OAAO,CAAC,WAAW,EAAS,CAAC;IAEvD,MAAM,sBAAsB,GAC1B,2BAA2B,CAAC,aAAa,EAAE,CAAC,iBAAiB,CAAC,CAAC;IAEjE,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAU,CAAC;IACnE,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,GAAG,aAAa,EAAE,GAAG,cAAc,CAAC,CAAC,CAAC;IAE7D,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAClC,MAAM,cAAc,GAAG;YACrB,GAAI,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,IAAI,EAAE,CAAW;YAChE,cAAc,CAAC,OAAO,CAAC;SACxB,CAAC;QAEF,OAAO,cAAc,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;YAC3C,OAAO;YACP,YAAY;YACZ,OAAO,EAAE,eAAe,CAAC,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY,CAAC;SACpE,CAAC,CAAC,CAAC;IACN,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,YAAY,CAC1B,SAA4C,EAC5C,YAAiB,EACjB,OAAY;IAEZ,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;IACpE,MAAM,sBAAsB,GAAG,YAAY,CAAC,WAAW,EAAS,CAAC;IAEjE,MAAM,QAAQ,GACZ,sBAAsB,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;IAEnE,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,SAAS,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;SACpE,IAAI,EAAE;SACN,IAAI,CACH,CAAC,WAAW,EAAE,EAAE,CACd,WAAW,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,sBAAsB,CAC/D,CAAC;IAEJ,IAAI,CAAC,KAAK,IAAI,CAAC,QAAQ,EAAE,CAAC;QACxB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,KAAK,IAAI,CAAC,QAAQ,EAAE,CAAC;QACvB,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;IACpE,CAAC;IAED,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAE7C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;AAC1C,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,gBAAgB,CAC9B,SAA4C,EAC5C,YAAiB,EACjB,OAAY;IAEZ,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAE7C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,mBAAmB,GAAG,SAAS,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;IAE5E,MAAM,2BAA2B,GAAG,SAAS,CAAC,IAAI,CAChD,iCAAiC,CAClC,CAAC;IAEF,MAAM,sBAAsB,GAAG,oBAAoB,CAAC,YAAY,CAAQ,CAAC;IACzE,MAAM,QAAQ,GAAG,sBAAsB,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC;IAEpE,MAAM,iBAAiB,GACrB,mBAAmB,CAAC,UAAU,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,sBAAsB,CAAC,EAAE,KAAK,CAAC;IAE7E,IAAI,iBAAiB,KAAK,SAAS,IAAI,CAAC,QAAQ,EAAE,CAAC;QACjD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,EACJ,cAAc,EAAE,gBAAgB,EAChC,iBAAiB,EAAE,eAAe,GACnC,GAAG,2BAA2B,CAAC,aAAa,EAAE,CAAC,MAAM,CAAC,IAAI;QACzD,cAAc,EAAE,IAAI;QACpB,iBAAiB,EAAE,IAAI;KACxB,CAAC;IAEF,IAAI,gBAAgB,KAAK,IAAI,IAAI,eAAe,KAAK,IAAI,EAAE,CAAC;QAC1D,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,YAAY,GAAG,WAAW,CAAC,OAAO,CAAC,EAAE,QAAQ,CACjD,YAAY,CAAC,WAAW,EAAS,CAClC,CAAC;IAEF,MAAM,OAAO,GAAG,YAAY;QAC1B,CAAC,CAAC,GAAG;QACL,CAAC,CAAC,IAAI,SAAS,CAAC,iBAAiB,IAAI,CAAC,CAAC;aAClC,YAAY,CAAC,eAAe,CAAC;aAC7B,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEpB,MAAM,QAAQ,GAAG,IAAI,SAAS,CAAC,iBAAiB,IAAI,CAAC,CAAC;SACnD,YAAY,CAAC,gBAAgB,CAAC;SAC9B,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEhB,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;AAC/B,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,mBAAmB,CACjC,QAAyB,EACzB,QAAgB,EAChB,SAAoB;IAOpB,MAAM,QAAQ,GAAG,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;IACzC,MAAM,UAAU,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,CAAC;IAEjD,OAAO;QACL,GAAG,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QACxB,KAAK,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC9B,GAAG,EAAE,UAAU,CAAC,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5D,IAAI,EAAE,UAAU,CAAC,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC;KAC/D,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,cAAc,CAAC,OAAY;IACzC,QAAQ,OAAO,EAAE,CAAC;QAChB,KAAK,MAAM;YACT,OAAO,4CAA4C,CAAC;QACtD;YACE,OAAO,oBAAoB,CAAC;IAChC,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,SAA4C,EAC5C,OAAY,EACZ,OAAY,EACZ,YAAiB;IAEjB,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CACpC,gDAAgD,EAChD,OAAO,CACR,CAAC;IAEF,MAAM,EAAE,QAAQ,EAAE,GAAG,SAAS,CAAC,IAAI,CACjC,wCAAwC,EACxC,eAAe,CAChB,CAAC;IAEF,MAAM,cAAc,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;IAClD,MAAM,QAAQ,GACZ,YAAY,CAAC,WAAW,EAAE,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;IAEvE,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACzD,OAAO,OAAO,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,YAAY,EAAE,QAAQ,EAAE,cAAc,CAAC,CAAC;IACtE,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAClD,OAAO,OAAO,CAAC,QAAQ,EAAE,CAAC;AAC5B,CAAC;AAED,SAAS,SAAS,CAChB,OAAY,EACZ,SAA4C;IAE5C,IAAI,CAAC;QACH,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CACpC,gDAAgD,EAChD,OAAO,CACR,CAAC;QAEF,MAAM,oBAAoB,GAAG,SAAS,CAAC,IAAI,CACzC,wCAAwC,EACxC,eAAe,CAChB,CAAC;QAEF,OAAO,oBAAoB,CAAC,aAAa,CAAC,MAAM,CAAC;IACnD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,MAAM,CAAN,IAAY,kBAGX;AAHD,WAAY,kBAAkB;IAC5B,qCAAe,CAAA;IACf,2CAAqB,CAAA;AACvB,CAAC,EAHW,kBAAkB,KAAlB,kBAAkB,QAG7B;AAED;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,qBAAqB,CACnC,YAAiB,EACjB,OAAY,EACZ,MAA2B;IAE3B,IAAI,OAAO,KAAK,gBAAgB,EAAE,CAAC;QACjC,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,MAAM,kBAAkB,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC,WAAW,EAAS,CAAC;IACxE,MAAM,sBAAsB,GAAG,YAAY,CAAC,WAAW,EAAE,CAAC;IAE1D,IACE,MAAM,KAAK,kBAAkB,CAAC,KAAK;QACnC,sBAAsB,KAAK,kBAAkB,EAC7C,CAAC;QACD,OAAO,oBAAoB,CAAC;IAC9B,CAAC;IAED,IACE,MAAM,KAAK,kBAAkB,CAAC,QAAQ;QACtC,sBAAsB,KAAK,oBAAoB,CAAC,WAAW,EAAE,EAC7D,CAAC;QACD,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC","sourcesContent":["import { Contract } from '@ethersproject/contracts';\nimport { Web3Provider } from '@ethersproject/providers';\nimport { toChecksumHexAddress } from '@metamask/controller-utils';\nimport { abiERC20 } from '@metamask/metamask-eth-abis';\nimport type { Hex } from '@metamask/utils';\nimport { BigNumber } from 'bignumber.js';\nimport { uniq } from 'lodash';\n\nimport {\n  CHAIN_ID_POLYGON,\n  NATIVE_TOKEN_ADDRESS,\n  STABLECOINS,\n} from '../constants';\nimport type { FiatRates, TransactionPayControllerMessenger } from '../types';\n\n/**\n * Check if two tokens are the same (same address and chain).\n *\n * @param token1 - First token identifier.\n * @param token1.address - Token address.\n * @param token1.chainId - Token chain ID.\n * @param token2 - Second token identifier.\n * @param token2.address - Token address.\n * @param token2.chainId - Token chain ID.\n * @returns True if tokens are the same, false otherwise.\n */\nexport function isSameToken(\n  token1: { address: Hex; chainId: Hex },\n  token2: { address: Hex; chainId: Hex },\n): boolean {\n  return (\n    token1.address.toLowerCase() === token2.address.toLowerCase() &&\n    token1.chainId === token2.chainId\n  );\n}\n\n/**\n * Get the token balance for a specific account and token.\n *\n * @param messenger - Controller messenger.\n * @param account - Address of the account.\n * @param chainId - Id of the chain.\n * @param tokenAddress - Address of the token contract.\n * @returns Raw token balance as a decimal string.\n */\nexport function getTokenBalance(\n  messenger: TransactionPayControllerMessenger,\n  account: Hex,\n  chainId: Hex,\n  tokenAddress: Hex,\n): string {\n  const tokenBalanceControllerState = messenger.call(\n    'TokenBalancesController:getState',\n  );\n\n  const normalizedAccount = account.toLowerCase() as Hex;\n  const normalizedTokenAddress = toChecksumHexAddress(tokenAddress) as Hex;\n  const isNative = normalizedTokenAddress === getNativeToken(chainId);\n\n  const balanceHex =\n    tokenBalanceControllerState.tokenBalances?.[normalizedAccount]?.[chainId]?.[\n      normalizedTokenAddress\n    ];\n\n  if (!isNative && balanceHex === undefined) {\n    return '0';\n  }\n\n  if (!isNative && balanceHex) {\n    return new BigNumber(balanceHex, 16).toString(10);\n  }\n\n  const accountTrackerControllerState = messenger.call(\n    'AccountTrackerController:getState',\n  );\n\n  const chainAccounts =\n    accountTrackerControllerState.accountsByChainId?.[chainId];\n\n  const checksumAccount = toChecksumHexAddress(normalizedAccount) as Hex;\n  const nativeBalanceHex = chainAccounts?.[checksumAccount]?.balance as Hex;\n\n  return new BigNumber(nativeBalanceHex ?? '0x0', 16).toString(10);\n}\n\n/**\n * Get the token balance for a specific account and token.\n *\n * @param messenger - Controller messenger.\n * @param account - Address of the account.\n * @returns The token balance as a BigNumber.\n */\nexport function getAllTokenBalances(\n  messenger: TransactionPayControllerMessenger,\n  account: Hex,\n): {\n  balance: string;\n  chainId: Hex;\n  tokenAddress: Hex;\n}[] {\n  const tokenBalanceControllerState = messenger.call(\n    'TokenBalancesController:getState',\n  );\n\n  const accountTrackerControllerState = messenger.call(\n    'AccountTrackerController:getState',\n  );\n\n  const nativeChainIds = Object.keys(\n    accountTrackerControllerState.accountsByChainId,\n  ) as Hex[];\n\n  const normalizedAccount = account.toLowerCase() as Hex;\n\n  const balancesByTokenByChain =\n    tokenBalanceControllerState.tokenBalances?.[normalizedAccount];\n\n  const tokenChainIds = Object.keys(balancesByTokenByChain) as Hex[];\n  const chainIds = uniq([...tokenChainIds, ...nativeChainIds]);\n\n  return chainIds.flatMap((chainId) => {\n    const tokenAddresses = [\n      ...(Object.keys(balancesByTokenByChain[chainId] ?? {}) as Hex[]),\n      getNativeToken(chainId),\n    ];\n\n    return tokenAddresses.map((tokenAddress) => ({\n      chainId,\n      tokenAddress,\n      balance: getTokenBalance(messenger, account, chainId, tokenAddress),\n    }));\n  });\n}\n\n/**\n * Get the token decimals for a specific token.\n *\n * @param messenger - Controller messenger.\n * @param tokenAddress - Address of the token contract.\n * @param chainId - Id of the chain.\n * @returns The token decimals or undefined if the token is not found.\n */\nexport function getTokenInfo(\n  messenger: TransactionPayControllerMessenger,\n  tokenAddress: Hex,\n  chainId: Hex,\n): { decimals: number; symbol: string } | undefined {\n  const controllerState = messenger.call('TokensController:getState');\n  const normalizedTokenAddress = tokenAddress.toLowerCase() as Hex;\n\n  const isNative =\n    normalizedTokenAddress === getNativeToken(chainId).toLowerCase();\n\n  const token = Object.values(controllerState.allTokens?.[chainId] ?? {})\n    .flat()\n    .find(\n      (singleToken) =>\n        singleToken.address.toLowerCase() === normalizedTokenAddress,\n    );\n\n  if (!token && !isNative) {\n    return undefined;\n  }\n\n  if (token && !isNative) {\n    return { decimals: Number(token.decimals), symbol: token.symbol };\n  }\n\n  const ticker = getTicker(chainId, messenger);\n\n  if (!ticker) {\n    return undefined;\n  }\n\n  return { decimals: 18, symbol: ticker };\n}\n\n/**\n * Calculate fiat rates for a specific token.\n *\n * @param messenger - Controller messenger.\n * @param tokenAddress - Address of the token contract.\n * @param chainId - Id of the chain.\n * @returns An object containing the USD and fiat rates, or undefined if rates are not available.\n */\nexport function getTokenFiatRate(\n  messenger: TransactionPayControllerMessenger,\n  tokenAddress: Hex,\n  chainId: Hex,\n): FiatRates | undefined {\n  const ticker = getTicker(chainId, messenger);\n\n  if (!ticker) {\n    return undefined;\n  }\n\n  const rateControllerState = messenger.call('TokenRatesController:getState');\n\n  const currencyRateControllerState = messenger.call(\n    'CurrencyRateController:getState',\n  );\n\n  const normalizedTokenAddress = toChecksumHexAddress(tokenAddress) as Hex;\n  const isNative = normalizedTokenAddress === getNativeToken(chainId);\n\n  const tokenToNativeRate =\n    rateControllerState.marketData?.[chainId]?.[normalizedTokenAddress]?.price;\n\n  if (tokenToNativeRate === undefined && !isNative) {\n    return undefined;\n  }\n\n  const {\n    conversionRate: nativeToFiatRate,\n    usdConversionRate: nativeToUsdRate,\n  } = currencyRateControllerState.currencyRates?.[ticker] ?? {\n    conversionRate: null,\n    usdConversionRate: null,\n  };\n\n  if (nativeToFiatRate === null || nativeToUsdRate === null) {\n    return undefined;\n  }\n  const isStablecoin = STABLECOINS[chainId]?.includes(\n    tokenAddress.toLowerCase() as Hex,\n  );\n\n  const usdRate = isStablecoin\n    ? '1'\n    : new BigNumber(tokenToNativeRate ?? 1)\n        .multipliedBy(nativeToUsdRate)\n        .toString(10);\n\n  const fiatRate = new BigNumber(tokenToNativeRate ?? 1)\n    .multipliedBy(nativeToFiatRate)\n    .toString(10);\n\n  return { usdRate, fiatRate };\n}\n\n/**\n * Calculate the human-readable, raw, USD, and fiat representations of a token amount.\n *\n * @param rawInput - Raw token amount (decimal string, hex, or BigNumber).\n * @param decimals - Number of decimals for the token.\n * @param fiatRates - Fiat rates for the token.\n * @returns Object containing the amount in raw, human-readable, USD, and fiat formats.\n */\nexport function computeTokenAmounts(\n  rawInput: BigNumber.Value,\n  decimals: number,\n  fiatRates: FiatRates,\n): {\n  raw: string;\n  human: string;\n  usd: string;\n  fiat: string;\n} {\n  const rawValue = new BigNumber(rawInput);\n  const humanValue = rawValue.shiftedBy(-decimals);\n\n  return {\n    raw: rawValue.toFixed(0),\n    human: humanValue.toString(10),\n    usd: humanValue.multipliedBy(fiatRates.usdRate).toString(10),\n    fiat: humanValue.multipliedBy(fiatRates.fiatRate).toString(10),\n  };\n}\n\n/**\n * Get the native token address for a given chain ID.\n *\n * @param chainId - Chain ID.\n * @returns - Native token address for the given chain ID.\n */\nexport function getNativeToken(chainId: Hex): Hex {\n  switch (chainId) {\n    case '0x89':\n      return '0x0000000000000000000000000000000000001010';\n    default:\n      return NATIVE_TOKEN_ADDRESS;\n  }\n}\n\n/**\n * Get the live on-chain token balance via an RPC `eth_call` to the ERC-20\n * `balanceOf` function, or `eth_getBalance` for native tokens.\n *\n * Unlike {@link getTokenBalance}, this bypasses the cached state in\n * `TokenBalancesController` and reads directly from the chain.\n *\n * @param messenger - Controller messenger.\n * @param account - Address of the account.\n * @param chainId - Chain ID.\n * @param tokenAddress - Address of the token contract.\n * @returns Raw token balance as a decimal string.\n */\nexport async function getLiveTokenBalance(\n  messenger: TransactionPayControllerMessenger,\n  account: Hex,\n  chainId: Hex,\n  tokenAddress: Hex,\n): Promise<string> {\n  const networkClientId = messenger.call(\n    'NetworkController:findNetworkClientIdByChainId',\n    chainId,\n  );\n\n  const { provider } = messenger.call(\n    'NetworkController:getNetworkClientById',\n    networkClientId,\n  );\n\n  const ethersProvider = new Web3Provider(provider);\n  const isNative =\n    tokenAddress.toLowerCase() === getNativeToken(chainId).toLowerCase();\n\n  if (isNative) {\n    const balance = await ethersProvider.getBalance(account);\n    return balance.toString();\n  }\n\n  const contract = new Contract(tokenAddress, abiERC20, ethersProvider);\n  const balance = await contract.balanceOf(account);\n  return balance.toString();\n}\n\nfunction getTicker(\n  chainId: Hex,\n  messenger: TransactionPayControllerMessenger,\n): string | undefined {\n  try {\n    const networkClientId = messenger.call(\n      'NetworkController:findNetworkClientIdByChainId',\n      chainId,\n    );\n\n    const networkConfiguration = messenger.call(\n      'NetworkController:getNetworkClientById',\n      networkClientId,\n    );\n\n    return networkConfiguration.configuration.ticker;\n  } catch {\n    return undefined;\n  }\n}\n\nexport enum TokenAddressTarget {\n  Relay = 'relay',\n  MetaMask = 'metamask',\n}\n\n/**\n * Normalize token address formats between MetaMask and Relay for Polygon native\n * token handling.\n *\n * MetaMask uses Polygon's native token contract-like address (`0x...1010`),\n * while Relay expects the zero address for native tokens.\n *\n * @param tokenAddress - Token address to normalize.\n * @param chainId - Chain ID for the token.\n * @param target - Optional target system format.\n * @returns Normalized token address for the target system, or the original\n * address if no target is provided.\n */\nexport function normalizeTokenAddress(\n  tokenAddress: Hex,\n  chainId: Hex,\n  target?: TokenAddressTarget,\n): Hex {\n  if (chainId !== CHAIN_ID_POLYGON) {\n    return tokenAddress;\n  }\n\n  const nativeTokenAddress = getNativeToken(chainId).toLowerCase() as Hex;\n  const normalizedTokenAddress = tokenAddress.toLowerCase();\n\n  if (\n    target === TokenAddressTarget.Relay &&\n    normalizedTokenAddress === nativeTokenAddress\n  ) {\n    return NATIVE_TOKEN_ADDRESS;\n  }\n\n  if (\n    target === TokenAddressTarget.MetaMask &&\n    normalizedTokenAddress === NATIVE_TOKEN_ADDRESS.toLowerCase()\n  ) {\n    return nativeTokenAddress;\n  }\n\n  return tokenAddress;\n}\n"]}
+{"version":3,"file":"token.mjs","sourceRoot":"","sources":["../../src/utils/token.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,iCAAiC;AACpD,OAAO,EAAE,YAAY,EAAE,iCAAiC;AAExD,OAAO,EAAE,oBAAoB,EAAE,mCAAmC;AAClE,OAAO,EAAE,QAAQ,EAAE,oCAAoC;AAEvD,OAAO,EAAE,SAAS,EAAE,qBAAqB;AAEzC,OAAO,EAAE,0BAA0B,EAAE,4BAAwB;AAC7D,OAAO,EACL,gBAAgB,EAChB,oBAAoB,EACpB,WAAW,EACZ,yBAAqB;AAGtB;;;;;;;;;;GAUG;AACH,MAAM,UAAU,WAAW,CACzB,MAAsC,EACtC,MAAsC;IAEtC,OAAO,CACL,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE;QAC7D,MAAM,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,CAClC,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,eAAe,CAC7B,SAA4C,EAC5C,OAAY,EACZ,OAAY,EACZ,YAAiB;IAEjB,MAAM,8BAA8B,GAAG,0BAA0B,CAAC,SAAS,CAAC,CAAC;IAE7E,IAAI,aAAa,CAAC;IAClB,IAAI,iBAAiB,CAAC;IACtB,IAAI,8BAA8B,EAAE,CAAC;QACnC,MAAM,qBAAqB,GAAG,SAAS,CAAC,IAAI,CAC1C,4CAA4C,CAC7C,CAAC;QAEF,aAAa,GAAG,qBAAqB,EAAE,aAAa,CAAC;QACrD,iBAAiB,GAAG,qBAAqB,EAAE,iBAAiB,CAAC;IAC/D,CAAC;SAAM,CAAC;QACN,aAAa,GAAG,SAAS,CAAC,IAAI,CAC5B,kCAAkC,CACnC,EAAE,aAAa,CAAC;QACjB,iBAAiB,GAAG,SAAS,CAAC,IAAI,CAChC,mCAAmC,CACpC,EAAE,iBAAiB,CAAC;IACvB,CAAC;IAED,MAAM,iBAAiB,GAAG,OAAO,CAAC,WAAW,EAAS,CAAC;IACvD,MAAM,sBAAsB,GAAG,oBAAoB,CAAC,YAAY,CAAQ,CAAC;IACzE,MAAM,QAAQ,GAAG,sBAAsB,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC;IAEpE,MAAM,UAAU,GACd,aAAa,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,sBAAsB,CAAC,CAAC;IAE1E,IAAI,CAAC,QAAQ,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC1C,OAAO,GAAG,CAAC;IACb,CAAC;IAED,IAAI,CAAC,QAAQ,IAAI,UAAU,EAAE,CAAC;QAC5B,OAAO,IAAI,SAAS,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACpD,CAAC;IAED,MAAM,aAAa,GAAG,iBAAiB,EAAE,CAAC,OAAO,CAAC,CAAC;IAEnD,MAAM,eAAe,GAAG,oBAAoB,CAAC,iBAAiB,CAAQ,CAAC;IACvE,MAAM,gBAAgB,GAAG,aAAa,EAAE,CAAC,eAAe,CAAC,EAAE,OAAc,CAAC;IAE1E,OAAO,IAAI,SAAS,CAAC,gBAAgB,IAAI,KAAK,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;AACnE,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,YAAY,CAC1B,SAA4C,EAC5C,YAAiB,EACjB,OAAY;IAEZ,MAAM,8BAA8B,GAAG,0BAA0B,CAAC,SAAS,CAAC,CAAC;IAE7E,IAAI,SAA6C,CAAC;IAClD,IAAI,8BAA8B,EAAE,CAAC;QACnC,SAAS,GAAG,SAAS,CAAC,IAAI,CACxB,4CAA4C,CAC7C,EAAE,SAAS,CAAC;IACf,CAAC;SAAM,CAAC;QACN,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,2BAA2B,CAAC,EAAE,SAAS,CAAC;IACrE,CAAC;IAED,MAAM,sBAAsB,GAAG,YAAY,CAAC,WAAW,EAAS,CAAC;IAEjE,MAAM,QAAQ,GACZ,sBAAsB,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;IAEnE,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;SACpD,IAAI,EAAE;SACN,IAAI,CACH,CAAC,WAAW,EAAE,EAAE,CACd,WAAW,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,sBAAsB,CAC/D,CAAC;IAEJ,IAAI,CAAC,KAAK,IAAI,CAAC,QAAQ,EAAE,CAAC;QACxB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,KAAK,IAAI,CAAC,QAAQ,EAAE,CAAC;QACvB,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;IACpE,CAAC;IAED,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAE7C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;AAC1C,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,gBAAgB,CAC9B,SAA4C,EAC5C,YAAiB,EACjB,OAAY;IAEZ,MAAM,8BAA8B,GAAG,0BAA0B,CAAC,SAAS,CAAC,CAAC;IAE7E,IAAI,UAAU,CAAC;IACf,IAAI,aAAa,CAAC;IAClB,IAAI,8BAA8B,EAAE,CAAC;QACnC,MAAM,qBAAqB,GAAG,SAAS,CAAC,IAAI,CAC1C,4CAA4C,CAC7C,CAAC;QAEF,UAAU,GAAG,qBAAqB,EAAE,UAAU,CAAC;QAC/C,aAAa,GAAG,qBAAqB,EAAE,aAAa,CAAC;IACvD,CAAC;SAAM,CAAC;QACN,UAAU,GAAG,SAAS,CAAC,IAAI,CAAC,+BAA+B,CAAC,EAAE,UAAU,CAAC;QACzE,aAAa,GAAG,SAAS,CAAC,IAAI,CAC5B,iCAAiC,CAClC,EAAE,aAAa,CAAC;IACnB,CAAC;IAED,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAE7C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,sBAAsB,GAAG,oBAAoB,CAAC,YAAY,CAAQ,CAAC;IACzE,MAAM,QAAQ,GAAG,sBAAsB,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC;IAEpE,MAAM,iBAAiB,GACrB,UAAU,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,sBAAsB,CAAC,EAAE,KAAK,CAAC;IAEzD,IAAI,iBAAiB,KAAK,SAAS,IAAI,CAAC,QAAQ,EAAE,CAAC;QACjD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,EACJ,cAAc,EAAE,gBAAgB,EAChC,iBAAiB,EAAE,eAAe,GACnC,GAAG,aAAa,EAAE,CAAC,MAAM,CAAC,IAAI;QAC7B,cAAc,EAAE,IAAI;QACpB,iBAAiB,EAAE,IAAI;KACxB,CAAC;IAEF,IAAI,gBAAgB,KAAK,IAAI,IAAI,eAAe,KAAK,IAAI,EAAE,CAAC;QAC1D,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,YAAY,GAAG,WAAW,CAAC,OAAO,CAAC,EAAE,QAAQ,CACjD,YAAY,CAAC,WAAW,EAAS,CAClC,CAAC;IAEF,MAAM,OAAO,GAAG,YAAY;QAC1B,CAAC,CAAC,GAAG;QACL,CAAC,CAAC,IAAI,SAAS,CAAC,iBAAiB,IAAI,CAAC,CAAC;aAClC,YAAY,CAAC,eAAe,CAAC;aAC7B,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEpB,MAAM,QAAQ,GAAG,IAAI,SAAS,CAAC,iBAAiB,IAAI,CAAC,CAAC;SACnD,YAAY,CAAC,gBAAgB,CAAC;SAC9B,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEhB,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;AAC/B,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,mBAAmB,CACjC,QAAyB,EACzB,QAAgB,EAChB,SAAoB;IAOpB,MAAM,QAAQ,GAAG,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;IACzC,MAAM,UAAU,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,CAAC;IAEjD,OAAO;QACL,GAAG,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QACxB,KAAK,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC9B,GAAG,EAAE,UAAU,CAAC,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5D,IAAI,EAAE,UAAU,CAAC,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC;KAC/D,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,cAAc,CAAC,OAAY;IACzC,QAAQ,OAAO,EAAE,CAAC;QAChB,KAAK,MAAM;YACT,OAAO,4CAA4C,CAAC;QACtD;YACE,OAAO,oBAAoB,CAAC;IAChC,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,SAA4C,EAC5C,OAAY,EACZ,OAAY,EACZ,YAAiB;IAEjB,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CACpC,gDAAgD,EAChD,OAAO,CACR,CAAC;IAEF,MAAM,EAAE,QAAQ,EAAE,GAAG,SAAS,CAAC,IAAI,CACjC,wCAAwC,EACxC,eAAe,CAChB,CAAC;IAEF,MAAM,cAAc,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;IAClD,MAAM,QAAQ,GACZ,YAAY,CAAC,WAAW,EAAE,KAAK,cAAc,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;IAEvE,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACzD,OAAO,OAAO,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,YAAY,EAAE,QAAQ,EAAE,cAAc,CAAC,CAAC;IACtE,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAClD,OAAO,OAAO,CAAC,QAAQ,EAAE,CAAC;AAC5B,CAAC;AAED,SAAS,SAAS,CAChB,OAAY,EACZ,SAA4C;IAE5C,IAAI,CAAC;QACH,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CACpC,gDAAgD,EAChD,OAAO,CACR,CAAC;QAEF,MAAM,oBAAoB,GAAG,SAAS,CAAC,IAAI,CACzC,wCAAwC,EACxC,eAAe,CAChB,CAAC;QAEF,OAAO,oBAAoB,CAAC,aAAa,CAAC,MAAM,CAAC;IACnD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,MAAM,CAAN,IAAY,kBAGX;AAHD,WAAY,kBAAkB;IAC5B,qCAAe,CAAA;IACf,2CAAqB,CAAA;AACvB,CAAC,EAHW,kBAAkB,KAAlB,kBAAkB,QAG7B;AAED;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,qBAAqB,CACnC,YAAiB,EACjB,OAAY,EACZ,MAA2B;IAE3B,IAAI,OAAO,KAAK,gBAAgB,EAAE,CAAC;QACjC,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,MAAM,kBAAkB,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC,WAAW,EAAS,CAAC;IACxE,MAAM,sBAAsB,GAAG,YAAY,CAAC,WAAW,EAAE,CAAC;IAE1D,IACE,MAAM,KAAK,kBAAkB,CAAC,KAAK;QACnC,sBAAsB,KAAK,kBAAkB,EAC7C,CAAC;QACD,OAAO,oBAAoB,CAAC;IAC9B,CAAC;IAED,IACE,MAAM,KAAK,kBAAkB,CAAC,QAAQ;QACtC,sBAAsB,KAAK,oBAAoB,CAAC,WAAW,EAAE,EAC7D,CAAC;QACD,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC","sourcesContent":["import { Contract } from '@ethersproject/contracts';\nimport { Web3Provider } from '@ethersproject/providers';\nimport { TokensControllerState } from '@metamask/assets-controllers';\nimport { toChecksumHexAddress } from '@metamask/controller-utils';\nimport { abiERC20 } from '@metamask/metamask-eth-abis';\nimport type { Hex } from '@metamask/utils';\nimport { BigNumber } from 'bignumber.js';\n\nimport { getAssetsUnifyStateFeature } from './feature-flags';\nimport {\n  CHAIN_ID_POLYGON,\n  NATIVE_TOKEN_ADDRESS,\n  STABLECOINS,\n} from '../constants';\nimport type { FiatRates, TransactionPayControllerMessenger } from '../types';\n\n/**\n * Check if two tokens are the same (same address and chain).\n *\n * @param token1 - First token identifier.\n * @param token1.address - Token address.\n * @param token1.chainId - Token chain ID.\n * @param token2 - Second token identifier.\n * @param token2.address - Token address.\n * @param token2.chainId - Token chain ID.\n * @returns True if tokens are the same, false otherwise.\n */\nexport function isSameToken(\n  token1: { address: Hex; chainId: Hex },\n  token2: { address: Hex; chainId: Hex },\n): boolean {\n  return (\n    token1.address.toLowerCase() === token2.address.toLowerCase() &&\n    token1.chainId === token2.chainId\n  );\n}\n\n/**\n * Get the token balance for a specific account and token.\n *\n * @param messenger - Controller messenger.\n * @param account - Address of the account.\n * @param chainId - Id of the chain.\n * @param tokenAddress - Address of the token contract.\n * @returns Raw token balance as a decimal string.\n */\nexport function getTokenBalance(\n  messenger: TransactionPayControllerMessenger,\n  account: Hex,\n  chainId: Hex,\n  tokenAddress: Hex,\n): string {\n  const assetsUnifyStateFeatureEnabled = getAssetsUnifyStateFeature(messenger);\n\n  let tokenBalances;\n  let accountsByChainId;\n  if (assetsUnifyStateFeatureEnabled) {\n    const assetsControllerState = messenger.call(\n      'AssetsController:getStateForTransactionPay',\n    );\n\n    tokenBalances = assetsControllerState?.tokenBalances;\n    accountsByChainId = assetsControllerState?.accountsByChainId;\n  } else {\n    tokenBalances = messenger.call(\n      'TokenBalancesController:getState',\n    )?.tokenBalances;\n    accountsByChainId = messenger.call(\n      'AccountTrackerController:getState',\n    )?.accountsByChainId;\n  }\n\n  const normalizedAccount = account.toLowerCase() as Hex;\n  const normalizedTokenAddress = toChecksumHexAddress(tokenAddress) as Hex;\n  const isNative = normalizedTokenAddress === getNativeToken(chainId);\n\n  const balanceHex =\n    tokenBalances?.[normalizedAccount]?.[chainId]?.[normalizedTokenAddress];\n\n  if (!isNative && balanceHex === undefined) {\n    return '0';\n  }\n\n  if (!isNative && balanceHex) {\n    return new BigNumber(balanceHex, 16).toString(10);\n  }\n\n  const chainAccounts = accountsByChainId?.[chainId];\n\n  const checksumAccount = toChecksumHexAddress(normalizedAccount) as Hex;\n  const nativeBalanceHex = chainAccounts?.[checksumAccount]?.balance as Hex;\n\n  return new BigNumber(nativeBalanceHex ?? '0x0', 16).toString(10);\n}\n\n/**\n * Get the token decimals for a specific token.\n *\n * @param messenger - Controller messenger.\n * @param tokenAddress - Address of the token contract.\n * @param chainId - Id of the chain.\n * @returns The token decimals or undefined if the token is not found.\n */\nexport function getTokenInfo(\n  messenger: TransactionPayControllerMessenger,\n  tokenAddress: Hex,\n  chainId: Hex,\n): { decimals: number; symbol: string } | undefined {\n  const assetsUnifyStateFeatureEnabled = getAssetsUnifyStateFeature(messenger);\n\n  let allTokens: TokensControllerState['allTokens'];\n  if (assetsUnifyStateFeatureEnabled) {\n    allTokens = messenger.call(\n      'AssetsController:getStateForTransactionPay',\n    )?.allTokens;\n  } else {\n    allTokens = messenger.call('TokensController:getState')?.allTokens;\n  }\n\n  const normalizedTokenAddress = tokenAddress.toLowerCase() as Hex;\n\n  const isNative =\n    normalizedTokenAddress === getNativeToken(chainId).toLowerCase();\n\n  const token = Object.values(allTokens?.[chainId] ?? {})\n    .flat()\n    .find(\n      (singleToken) =>\n        singleToken.address.toLowerCase() === normalizedTokenAddress,\n    );\n\n  if (!token && !isNative) {\n    return undefined;\n  }\n\n  if (token && !isNative) {\n    return { decimals: Number(token.decimals), symbol: token.symbol };\n  }\n\n  const ticker = getTicker(chainId, messenger);\n\n  if (!ticker) {\n    return undefined;\n  }\n\n  return { decimals: 18, symbol: ticker };\n}\n\n/**\n * Calculate fiat rates for a specific token.\n *\n * @param messenger - Controller messenger.\n * @param tokenAddress - Address of the token contract.\n * @param chainId - Id of the chain.\n * @returns An object containing the USD and fiat rates, or undefined if rates are not available.\n */\nexport function getTokenFiatRate(\n  messenger: TransactionPayControllerMessenger,\n  tokenAddress: Hex,\n  chainId: Hex,\n): FiatRates | undefined {\n  const assetsUnifyStateFeatureEnabled = getAssetsUnifyStateFeature(messenger);\n\n  let marketData;\n  let currencyRates;\n  if (assetsUnifyStateFeatureEnabled) {\n    const assetsControllerState = messenger.call(\n      'AssetsController:getStateForTransactionPay',\n    );\n\n    marketData = assetsControllerState?.marketData;\n    currencyRates = assetsControllerState?.currencyRates;\n  } else {\n    marketData = messenger.call('TokenRatesController:getState')?.marketData;\n    currencyRates = messenger.call(\n      'CurrencyRateController:getState',\n    )?.currencyRates;\n  }\n\n  const ticker = getTicker(chainId, messenger);\n\n  if (!ticker) {\n    return undefined;\n  }\n\n  const normalizedTokenAddress = toChecksumHexAddress(tokenAddress) as Hex;\n  const isNative = normalizedTokenAddress === getNativeToken(chainId);\n\n  const tokenToNativeRate =\n    marketData?.[chainId]?.[normalizedTokenAddress]?.price;\n\n  if (tokenToNativeRate === undefined && !isNative) {\n    return undefined;\n  }\n\n  const {\n    conversionRate: nativeToFiatRate,\n    usdConversionRate: nativeToUsdRate,\n  } = currencyRates?.[ticker] ?? {\n    conversionRate: null,\n    usdConversionRate: null,\n  };\n\n  if (nativeToFiatRate === null || nativeToUsdRate === null) {\n    return undefined;\n  }\n  const isStablecoin = STABLECOINS[chainId]?.includes(\n    tokenAddress.toLowerCase() as Hex,\n  );\n\n  const usdRate = isStablecoin\n    ? '1'\n    : new BigNumber(tokenToNativeRate ?? 1)\n        .multipliedBy(nativeToUsdRate)\n        .toString(10);\n\n  const fiatRate = new BigNumber(tokenToNativeRate ?? 1)\n    .multipliedBy(nativeToFiatRate)\n    .toString(10);\n\n  return { usdRate, fiatRate };\n}\n\n/**\n * Calculate the human-readable, raw, USD, and fiat representations of a token amount.\n *\n * @param rawInput - Raw token amount (decimal string, hex, or BigNumber).\n * @param decimals - Number of decimals for the token.\n * @param fiatRates - Fiat rates for the token.\n * @returns Object containing the amount in raw, human-readable, USD, and fiat formats.\n */\nexport function computeTokenAmounts(\n  rawInput: BigNumber.Value,\n  decimals: number,\n  fiatRates: FiatRates,\n): {\n  raw: string;\n  human: string;\n  usd: string;\n  fiat: string;\n} {\n  const rawValue = new BigNumber(rawInput);\n  const humanValue = rawValue.shiftedBy(-decimals);\n\n  return {\n    raw: rawValue.toFixed(0),\n    human: humanValue.toString(10),\n    usd: humanValue.multipliedBy(fiatRates.usdRate).toString(10),\n    fiat: humanValue.multipliedBy(fiatRates.fiatRate).toString(10),\n  };\n}\n\n/**\n * Get the native token address for a given chain ID.\n *\n * @param chainId - Chain ID.\n * @returns - Native token address for the given chain ID.\n */\nexport function getNativeToken(chainId: Hex): Hex {\n  switch (chainId) {\n    case '0x89':\n      return '0x0000000000000000000000000000000000001010';\n    default:\n      return NATIVE_TOKEN_ADDRESS;\n  }\n}\n\n/**\n * Get the live on-chain token balance via an RPC `eth_call` to the ERC-20\n * `balanceOf` function, or `eth_getBalance` for native tokens.\n *\n * Unlike {@link getTokenBalance}, this bypasses the cached state in\n * `TokenBalancesController` and reads directly from the chain.\n *\n * @param messenger - Controller messenger.\n * @param account - Address of the account.\n * @param chainId - Chain ID.\n * @param tokenAddress - Address of the token contract.\n * @returns Raw token balance as a decimal string.\n */\nexport async function getLiveTokenBalance(\n  messenger: TransactionPayControllerMessenger,\n  account: Hex,\n  chainId: Hex,\n  tokenAddress: Hex,\n): Promise<string> {\n  const networkClientId = messenger.call(\n    'NetworkController:findNetworkClientIdByChainId',\n    chainId,\n  );\n\n  const { provider } = messenger.call(\n    'NetworkController:getNetworkClientById',\n    networkClientId,\n  );\n\n  const ethersProvider = new Web3Provider(provider);\n  const isNative =\n    tokenAddress.toLowerCase() === getNativeToken(chainId).toLowerCase();\n\n  if (isNative) {\n    const balance = await ethersProvider.getBalance(account);\n    return balance.toString();\n  }\n\n  const contract = new Contract(tokenAddress, abiERC20, ethersProvider);\n  const balance = await contract.balanceOf(account);\n  return balance.toString();\n}\n\nfunction getTicker(\n  chainId: Hex,\n  messenger: TransactionPayControllerMessenger,\n): string | undefined {\n  try {\n    const networkClientId = messenger.call(\n      'NetworkController:findNetworkClientIdByChainId',\n      chainId,\n    );\n\n    const networkConfiguration = messenger.call(\n      'NetworkController:getNetworkClientById',\n      networkClientId,\n    );\n\n    return networkConfiguration.configuration.ticker;\n  } catch {\n    return undefined;\n  }\n}\n\nexport enum TokenAddressTarget {\n  Relay = 'relay',\n  MetaMask = 'metamask',\n}\n\n/**\n * Normalize token address formats between MetaMask and Relay for Polygon native\n * token handling.\n *\n * MetaMask uses Polygon's native token contract-like address (`0x...1010`),\n * while Relay expects the zero address for native tokens.\n *\n * @param tokenAddress - Token address to normalize.\n * @param chainId - Chain ID for the token.\n * @param target - Optional target system format.\n * @returns Normalized token address for the target system, or the original\n * address if no target is provided.\n */\nexport function normalizeTokenAddress(\n  tokenAddress: Hex,\n  chainId: Hex,\n  target?: TokenAddressTarget,\n): Hex {\n  if (chainId !== CHAIN_ID_POLYGON) {\n    return tokenAddress;\n  }\n\n  const nativeTokenAddress = getNativeToken(chainId).toLowerCase() as Hex;\n  const normalizedTokenAddress = tokenAddress.toLowerCase();\n\n  if (\n    target === TokenAddressTarget.Relay &&\n    normalizedTokenAddress === nativeTokenAddress\n  ) {\n    return NATIVE_TOKEN_ADDRESS;\n  }\n\n  if (\n    target === TokenAddressTarget.MetaMask &&\n    normalizedTokenAddress === NATIVE_TOKEN_ADDRESS.toLowerCase()\n  ) {\n    return nativeTokenAddress;\n  }\n\n  return tokenAddress;\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@metamask/transaction-pay-controller",
-  "version": "16.5.0",
+  "version": "17.1.0",
   "description": "Manages alternate payment strategies to provide required funds for transactions in MetaMask",
   "keywords": [
     "MetaMask",
@@ -40,6 +40,7 @@
     "build:docs": "typedoc",
     "changelog:update": "../../scripts/update-changelog.sh @metamask/transaction-pay-controller",
     "changelog:validate": "../../scripts/validate-changelog.sh @metamask/transaction-pay-controller",
+    "generate-method-action-types": "tsx ../../scripts/generate-method-action-types.ts",
     "prepare-manifest:preview": "../../scripts/prepare-preview-manifest.sh",
     "since-latest-release": "../../scripts/since-latest-release.sh",
     "test": "NODE_OPTIONS=--experimental-vm-modules jest --reporters=jest-silent-reporter",
@@ -51,17 +52,18 @@
     "@ethersproject/abi": "^5.7.0",
     "@ethersproject/contracts": "^5.7.0",
     "@ethersproject/providers": "^5.7.0",
-    "@metamask/assets-controllers": "^100.2.1",
+    "@metamask/assets-controller": "^2.4.0",
+    "@metamask/assets-controllers": "^101.0.0",
     "@metamask/base-controller": "^9.0.0",
-    "@metamask/bridge-controller": "^69.1.0",
-    "@metamask/bridge-status-controller": "^68.1.0",
+    "@metamask/bridge-controller": "^69.1.1",
+    "@metamask/bridge-status-controller": "^69.0.0",
     "@metamask/controller-utils": "^11.19.0",
-    "@metamask/gas-fee-controller": "^26.0.3",
+    "@metamask/gas-fee-controller": "^26.1.0",
     "@metamask/messenger": "^0.3.0",
     "@metamask/metamask-eth-abis": "^3.1.1",
     "@metamask/network-controller": "^30.0.0",
     "@metamask/remote-feature-flag-controller": "^4.1.0",
-    "@metamask/transaction-controller": "^62.21.0",
+    "@metamask/transaction-controller": "^63.0.0",
     "@metamask/utils": "^11.9.0",
     "bignumber.js": "^9.1.2",
     "bn.js": "^5.2.1",
@@ -75,6 +77,7 @@
     "deepmerge": "^4.2.2",
     "jest": "^29.7.0",
     "ts-jest": "^29.2.5",
+    "tsx": "^4.20.5",
     "typedoc": "^0.25.13",
     "typedoc-plugin-missing-exports": "^2.0.0",
     "typescript": "~5.3.3"