@metamask/transaction-pay-controller 16.3.0 → 16.4.1

package/dist/strategy/relay/relay-max-gas-station.mjs

@@ -0,0 +1,238 @@
+import { createModuleLogger } from "@metamask/utils";
+import { BigNumber } from "bignumber.js";
+import { getGasStationEligibility, getGasStationCostInSourceTokenRaw } from "./gas-station.mjs";
+import { projectLogger } from "../../logger.mjs";
+import { getNativeToken, getTokenBalance, getTokenInfo } from "../../utils/token.mjs";
+const log = createModuleLogger(projectLogger, 'relay-max-gas-station');
+const PROBE_AMOUNT_PERCENTAGE = 0.25;
+var GasCostEstimateSource;
+(function (GasCostEstimateSource) {
+    GasCostEstimateSource["GasStation"] = "gas-station";
+    GasCostEstimateSource["Probe"] = "probe";
+    GasCostEstimateSource["Quote"] = "quote";
+})(GasCostEstimateSource || (GasCostEstimateSource = {}));
+/**
+ * Returns a Relay max-amount quote using a two-phase gas-station fallback.
+ *
+ * It first requests a standard max quote (phase 1), then when needed estimates
+ * gas in source-token units (directly or via a probe quote), requests an
+ * adjusted max quote (phase 2), and accepts phase 2 only if validation passes
+ * (source gas fee token selected, gas-limit checks, affordability).
+ *
+ * If any step fails or validation is unsafe, it safely falls back to phase 1.
+ * Successful phase-2 quotes are tagged with `metamask.isMaxGasStation = true`.
+ *
+ * @param request - Relay quote request for a max-amount flow.
+ * @param fullRequest - Full quote request context including messenger and transaction.
+ * @param getSingleQuote - Quote fetcher used for phase-1, phase-2, and probe quotes.
+ * @returns The validated adjusted phase-2 quote, or the original phase-1 quote on fallback.
+ */
+export async function getRelayMaxGasStationQuote(request, fullRequest, getSingleQuote) {
+    const { messenger } = fullRequest;
+    const { sourceChainId, sourceTokenAmount } = request;
+    const context = {
+        fullRequest,
+        getSingleQuote,
+        messenger,
+        request,
+    };
+    const phase1Quote = await getSingleQuote(request, fullRequest);
+    const nativeBalanceCheck = checkEnoughNativeBalanceIfSourceGasFeeTokenNotUsed(phase1Quote, messenger, request);
+    if (nativeBalanceCheck.hasEnoughNativeBalance) {
+        return fallbackToPhase1(phase1Quote, 'Native balance is sufficient for gas', {
+            nativeBalance: nativeBalanceCheck.nativeBalance,
+            nativeGasCost: nativeBalanceCheck.nativeGasCostRaw,
+        });
+    }
+    const gasStationEligibility = getGasStationEligibility(messenger, sourceChainId);
+    if (!gasStationEligibility.isEligible) {
+        return fallbackToPhase1(phase1Quote, 'Gas station is disabled or unsupported');
+    }
+    const sourceAmountBN = new BigNumber(sourceTokenAmount);
+    const probeCost = await getProbeGasCostInSourceTokenRaw(context);
+    if (!probeCost) {
+        return fallbackToPhase1(phase1Quote, 'Unable to estimate gas-station source token cost');
+    }
+    const initialGasEstimate = {
+        amount: probeCost,
+        source: GasCostEstimateSource.Probe,
+    };
+    const adjustedSourceAmount = getAdjustedSourceAmount(sourceAmountBN, initialGasEstimate.amount);
+    if (!adjustedSourceAmount.isGreaterThan(0)) {
+        return fallbackToPhase1(phase1Quote, 'Insufficient balance for gas station after max adjustment');
+    }
+    const phase2Quote = await getAdjustedPhase2Quote(adjustedSourceAmount, initialGasEstimate, context);
+    if (!phase2Quote) {
+        return fallbackToPhase1(phase1Quote, 'Adjusted phase-2 quote request failed');
+    }
+    if (!phase2Quote.fees.isSourceGasFeeToken) {
+        return fallbackToPhase1(phase1Quote, 'Adjusted quote did not return source gas fee token pricing');
+    }
+    const phase1GasLimits = phase1Quote.original.metamask.gasLimits ?? [];
+    const phase2GasLimits = phase2Quote.original.metamask.gasLimits ?? [];
+    if (!hasMatchingGasLimitShape(phase1GasLimits, phase2GasLimits)) {
+        return fallbackToPhase1(phase1Quote, 'Adjusted quote gas limit shape changed between phases', {
+            phase1GasLimits,
+            phase2GasLimits,
+        });
+    }
+    const phase1TotalGasLimit = getTotalGasLimit(phase1GasLimits);
+    const phase2TotalGasLimit = getTotalGasLimit(phase2GasLimits);
+    if (phase2TotalGasLimit > phase1TotalGasLimit) {
+        return fallbackToPhase1(phase1Quote, 'Adjusted quote total gas limit increased between phases', {
+            phase1TotalGasLimit,
+            phase2TotalGasLimit,
+        });
+    }
+    const validationGasEstimate = new BigNumber(phase2Quote.fees.sourceNetwork.max.raw);
+    if (!isAdjustedAmountAffordable(adjustedSourceAmount, validationGasEstimate, sourceAmountBN)) {
+        return fallbackToPhase1(phase1Quote, 'Adjusted quote fails affordability validation', {
+            adjustedSourceAmount: adjustedSourceAmount.toString(10),
+            validationGasCost: validationGasEstimate.toString(10),
+        });
+    }
+    markQuoteAsMaxGasStation(phase2Quote);
+    return phase2Quote;
+}
+function checkEnoughNativeBalanceIfSourceGasFeeTokenNotUsed(quote, messenger, request) {
+    if (quote.fees.isSourceGasFeeToken) {
+        return { hasEnoughNativeBalance: false };
+    }
+    const nativeGasCostRaw = quote.fees.sourceNetwork.max.raw;
+    const nativeBalance = getTokenBalance(messenger, request.from, request.sourceChainId, getNativeToken(request.sourceChainId));
+    return {
+        hasEnoughNativeBalance: new BigNumber(nativeBalance).isGreaterThanOrEqualTo(nativeGasCostRaw),
+        nativeBalance,
+        nativeGasCostRaw,
+    };
+}
+function getAdjustedSourceAmount(sourceAmount, estimatedGasCost) {
+    return sourceAmount
+        .minus(estimatedGasCost)
+        .integerValue(BigNumber.ROUND_DOWN);
+}
+function isAdjustedAmountAffordable(adjustedSourceAmount, gasCost, originalSourceAmount) {
+    return adjustedSourceAmount
+        .plus(gasCost)
+        .isLessThanOrEqualTo(originalSourceAmount);
+}
+async function getAdjustedPhase2Quote(adjustedSourceAmount, initialGasEstimate, context) {
+    const { fullRequest, getSingleQuote, request } = context;
+    log('Requesting adjusted max quote', {
+        adjustedAmount: adjustedSourceAmount.toString(10),
+        gasCostInSourceToken: initialGasEstimate.amount.toString(10),
+        gasEstimateSource: initialGasEstimate.source,
+        originalAmount: request.sourceTokenAmount,
+    });
+    try {
+        return await getSingleQuote({
+            ...request,
+            sourceTokenAmount: adjustedSourceAmount.toFixed(0, BigNumber.ROUND_DOWN),
+        }, fullRequest);
+    }
+    catch (error) {
+        log('Adjusted quote request failed, falling back to phase-1 quote', {
+            error,
+        });
+        return undefined;
+    }
+}
+async function getGasCostFromQuoteOrGasStation(quote, messenger, request) {
+    const gasCost = quote.fees.sourceNetwork.max;
+    if (quote.fees.isSourceGasFeeToken) {
+        log('Gas cost already in source token units', { raw: gasCost.raw });
+        return {
+            amount: new BigNumber(gasCost.raw),
+            source: GasCostEstimateSource.Quote,
+        };
+    }
+    const firstStepData = quote.original.steps[0]?.items[0]?.data;
+    if (!firstStepData) {
+        return undefined;
+    }
+    const totalItemCount = quote.original.steps.reduce((count, step) => count + step.items.length, 0);
+    const totalGasEstimate = (quote.original.metamask.gasLimits ?? []).reduce((acc, gasLimit) => acc + gasLimit, 0);
+    const gasStationCost = await getGasStationCostInSourceTokenRaw({
+        firstStepData,
+        messenger,
+        request: {
+            from: request.from,
+            sourceChainId: request.sourceChainId,
+            sourceTokenAddress: request.sourceTokenAddress,
+        },
+        totalGasEstimate,
+        totalItemCount,
+    });
+    if (!gasStationCost) {
+        return undefined;
+    }
+    return {
+        amount: new BigNumber(gasStationCost.raw),
+        source: GasCostEstimateSource.GasStation,
+    };
+}
+async function getProbeGasCostInSourceTokenRaw(context) {
+    const { fullRequest, getSingleQuote, messenger, request } = context;
+    const sourceTokenInfo = getTokenInfo(messenger, request.sourceTokenAddress, request.sourceChainId);
+    if (!sourceTokenInfo) {
+        return undefined;
+    }
+    const probeAmount = getProbeSourceAmountRaw(request.sourceTokenAmount, sourceTokenInfo.decimals);
+    if (!probeAmount || probeAmount === request.sourceTokenAmount) {
+        return undefined;
+    }
+    log('Requesting probe quote for gas station estimation', {
+        originalSourceAmount: request.sourceTokenAmount,
+        probeAmount,
+    });
+    let probeQuote;
+    try {
+        probeQuote = await getSingleQuote({
+            ...request,
+            sourceTokenAmount: probeAmount,
+        }, fullRequest);
+    }
+    catch (error) {
+        log('Probe quote request failed', { error });
+        return undefined;
+    }
+    if (!probeQuote) {
+        return undefined;
+    }
+    const probeEstimate = await getGasCostFromQuoteOrGasStation(probeQuote, messenger, request);
+    return probeEstimate?.amount;
+}
+function getProbeSourceAmountRaw(sourceAmountRaw, sourceDecimals) {
+    const sourceAmount = new BigNumber(sourceAmountRaw);
+    if (sourceAmount.isLessThanOrEqualTo(0)) {
+        return undefined;
+    }
+    const probeRawAmount = sourceAmount
+        .multipliedBy(PROBE_AMOUNT_PERCENTAGE)
+        .integerValue(BigNumber.ROUND_FLOOR);
+    // Minimum probe size: ~0.01 token for tokens with >=2 decimals,
+    // otherwise one raw unit for low-decimal tokens.
+    const minimumProbeRaw = new BigNumber(1).shiftedBy(Math.max(sourceDecimals - 2, 0));
+    const probeRaw = BigNumber.minimum(sourceAmount, BigNumber.maximum(probeRawAmount, minimumProbeRaw)).integerValue(BigNumber.ROUND_FLOOR);
+    if (probeRaw.isLessThanOrEqualTo(0)) {
+        return undefined;
+    }
+    return probeRaw.toFixed(0);
+}
+function fallbackToPhase1(phase1Quote, message, paramsToLog) {
+    log(message, paramsToLog);
+    return phase1Quote;
+}
+function hasMatchingGasLimitShape(phase1, phase2) {
+    return phase1.length === phase2.length;
+}
+function getTotalGasLimit(gasLimits) {
+    return gasLimits.reduce((total, gasLimit) => total + gasLimit, 0);
+}
+function markQuoteAsMaxGasStation(quote) {
+    quote.original.metamask = {
+        ...quote.original.metamask,
+        isMaxGasStation: true,
+    };
+}
+//# sourceMappingURL=relay-max-gas-station.mjs.map

package/dist/strategy/relay/relay-max-gas-station.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"relay-max-gas-station.mjs","sourceRoot":"","sources":["../../../src/strategy/relay/relay-max-gas-station.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,wBAAwB;AACrD,OAAO,EAAE,SAAS,EAAE,qBAAqB;AAEzC,OAAO,EACL,wBAAwB,EACxB,iCAAiC,EAClC,0BAAsB;AAEvB,OAAO,EAAE,aAAa,EAAE,yBAAqB;AAO7C,OAAO,EACL,cAAc,EACd,eAAe,EACf,YAAY,EACb,8BAA0B;AAE3B,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,EAAE,uBAAuB,CAAC,CAAC;AAEvE,MAAM,uBAAuB,GAAG,IAAI,CAAC;AAErC,IAAK,qBAIJ;AAJD,WAAK,qBAAqB;IACxB,mDAA0B,CAAA;IAC1B,wCAAe,CAAA;IACf,wCAAe,CAAA;AACjB,CAAC,EAJI,qBAAqB,KAArB,qBAAqB,QAIzB;AAyBD;;;;;;;;;;;;;;;GAeG;AACH,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,OAAqB,EACrB,WAAwC,EACxC,cAAgC;IAEhC,MAAM,EAAE,SAAS,EAAE,GAAG,WAAW,CAAC;IAClC,MAAM,EAAE,aAAa,EAAE,iBAAiB,EAAE,GAAG,OAAO,CAAC;IACrD,MAAM,OAAO,GAA0B;QACrC,WAAW;QACX,cAAc;QACd,SAAS;QACT,OAAO;KACR,CAAC;IAEF,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAE/D,MAAM,kBAAkB,GAAG,kDAAkD,CAC3E,WAAW,EACX,SAAS,EACT,OAAO,CACR,CAAC;IAEF,IAAI,kBAAkB,CAAC,sBAAsB,EAAE,CAAC;QAC9C,OAAO,gBAAgB,CACrB,WAAW,EACX,sCAAsC,EACtC;YACE,aAAa,EAAE,kBAAkB,CAAC,aAAa;YAC/C,aAAa,EAAE,kBAAkB,CAAC,gBAAgB;SACnD,CACF,CAAC;IACJ,CAAC;IAED,MAAM,qBAAqB,GAAG,wBAAwB,CACpD,SAAS,EACT,aAAa,CACd,CAAC;IAEF,IAAI,CAAC,qBAAqB,CAAC,UAAU,EAAE,CAAC;QACtC,OAAO,gBAAgB,CACrB,WAAW,EACX,wCAAwC,CACzC,CAAC;IACJ,CAAC;IAED,MAAM,cAAc,GAAG,IAAI,SAAS,CAAC,iBAAiB,CAAC,CAAC;IACxD,MAAM,SAAS,GAAG,MAAM,+BAA+B,CAAC,OAAO,CAAC,CAAC;IAEjE,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,gBAAgB,CACrB,WAAW,EACX,kDAAkD,CACnD,CAAC;IACJ,CAAC;IAED,MAAM,kBAAkB,GAAoB;QAC1C,MAAM,EAAE,SAAS;QACjB,MAAM,EAAE,qBAAqB,CAAC,KAAK;KACpC,CAAC;IAEF,MAAM,oBAAoB,GAAG,uBAAuB,CAClD,cAAc,EACd,kBAAkB,CAAC,MAAM,CAC1B,CAAC;IAEF,IAAI,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3C,OAAO,gBAAgB,CACrB,WAAW,EACX,2DAA2D,CAC5D,CAAC;IACJ,CAAC;IAED,MAAM,WAAW,GAAG,MAAM,sBAAsB,CAC9C,oBAAoB,EACpB,kBAAkB,EAClB,OAAO,CACR,CAAC;IAEF,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,gBAAgB,CACrB,WAAW,EACX,uCAAuC,CACxC,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC1C,OAAO,gBAAgB,CACrB,WAAW,EACX,4DAA4D,CAC7D,CAAC;IACJ,CAAC;IAED,MAAM,eAAe,GAAG,WAAW,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,EAAE,CAAC;IACtE,MAAM,eAAe,GAAG,WAAW,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,EAAE,CAAC;IAEtE,IAAI,CAAC,wBAAwB,CAAC,eAAe,EAAE,eAAe,CAAC,EAAE,CAAC;QAChE,OAAO,gBAAgB,CACrB,WAAW,EACX,uDAAuD,EACvD;YACE,eAAe;YACf,eAAe;SAChB,CACF,CAAC;IACJ,CAAC;IAED,MAAM,mBAAmB,GAAG,gBAAgB,CAAC,eAAe,CAAC,CAAC;IAC9D,MAAM,mBAAmB,GAAG,gBAAgB,CAAC,eAAe,CAAC,CAAC;IAE9D,IAAI,mBAAmB,GAAG,mBAAmB,EAAE,CAAC;QAC9C,OAAO,gBAAgB,CACrB,WAAW,EACX,yDAAyD,EACzD;YACE,mBAAmB;YACnB,mBAAmB;SACpB,CACF,CAAC;IACJ,CAAC;IAED,MAAM,qBAAqB,GAAG,IAAI,SAAS,CACzC,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CACvC,CAAC;IAEF,IACE,CAAC,0BAA0B,CACzB,oBAAoB,EACpB,qBAAqB,EACrB,cAAc,CACf,EACD,CAAC;QACD,OAAO,gBAAgB,CACrB,WAAW,EACX,+CAA+C,EAC/C;YACE,oBAAoB,EAAE,oBAAoB,CAAC,QAAQ,CAAC,EAAE,CAAC;YACvD,iBAAiB,EAAE,qBAAqB,CAAC,QAAQ,CAAC,EAAE,CAAC;SACtD,CACF,CAAC;IACJ,CAAC;IAED,wBAAwB,CAAC,WAAW,CAAC,CAAC;IAEtC,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,SAAS,kDAAkD,CACzD,KAAsC,EACtC,SAA4C,EAC5C,OAAqB;IAErB,IAAI,KAAK,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACnC,OAAO,EAAE,sBAAsB,EAAE,KAAK,EAAE,CAAC;IAC3C,CAAC;IAED,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC;IAC1D,MAAM,aAAa,GAAG,eAAe,CACnC,SAAS,EACT,OAAO,CAAC,IAAI,EACZ,OAAO,CAAC,aAAa,EACrB,cAAc,CAAC,OAAO,CAAC,aAAa,CAAC,CACtC,CAAC;IAEF,OAAO;QACL,sBAAsB,EAAE,IAAI,SAAS,CAAC,aAAa,CAAC,CAAC,sBAAsB,CACzE,gBAAgB,CACjB;QACD,aAAa;QACb,gBAAgB;KACjB,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAC9B,YAAuB,EACvB,gBAA2B;IAE3B,OAAO,YAAY;SAChB,KAAK,CAAC,gBAAgB,CAAC;SACvB,YAAY,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;AACxC,CAAC;AAED,SAAS,0BAA0B,CACjC,oBAA+B,EAC/B,OAAkB,EAClB,oBAA+B;IAE/B,OAAO,oBAAoB;SACxB,IAAI,CAAC,OAAO,CAAC;SACb,mBAAmB,CAAC,oBAAoB,CAAC,CAAC;AAC/C,CAAC;AAED,KAAK,UAAU,sBAAsB,CACnC,oBAA+B,EAC/B,kBAAmC,EACnC,OAA8B;IAE9B,MAAM,EAAE,WAAW,EAAE,cAAc,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IAEzD,GAAG,CAAC,+BAA+B,EAAE;QACnC,cAAc,EAAE,oBAAoB,CAAC,QAAQ,CAAC,EAAE,CAAC;QACjD,oBAAoB,EAAE,kBAAkB,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5D,iBAAiB,EAAE,kBAAkB,CAAC,MAAM;QAC5C,cAAc,EAAE,OAAO,CAAC,iBAAiB;KAC1C,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,OAAO,MAAM,cAAc,CACzB;YACE,GAAG,OAAO;YACV,iBAAiB,EAAE,oBAAoB,CAAC,OAAO,CAC7C,CAAC,EACD,SAAS,CAAC,UAAU,CACrB;SACF,EACD,WAAW,CACZ,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,8DAA8D,EAAE;YAClE,KAAK;SACN,CAAC,CAAC;QACH,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,KAAK,UAAU,+BAA+B,CAC5C,KAAsC,EACtC,SAA4C,EAC5C,OAAqB;IAErB,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC;IAE7C,IAAI,KAAK,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACnC,GAAG,CAAC,wCAAwC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;QACpE,OAAO;YACL,MAAM,EAAE,IAAI,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC;YAClC,MAAM,EAAE,qBAAqB,CAAC,KAAK;SACpC,CAAC;IACJ,CAAC;IAED,MAAM,aAAa,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;IAE9D,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,cAAc,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAChD,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAC1C,CAAC,CACF,CAAC;IAEF,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,MAAM,CACvE,CAAC,GAAG,EAAE,QAAQ,EAAE,EAAE,CAAC,GAAG,GAAG,QAAQ,EACjC,CAAC,CACF,CAAC;IAEF,MAAM,cAAc,GAAG,MAAM,iCAAiC,CAAC;QAC7D,aAAa;QACb,SAAS;QACT,OAAO,EAAE;YACP,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,aAAa,EAAE,OAAO,CAAC,aAAa;YACpC,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;SAC/C;QACD,gBAAgB;QAChB,cAAc;KACf,CAAC,CAAC;IAEH,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO;QACL,MAAM,EAAE,IAAI,SAAS,CAAC,cAAc,CAAC,GAAG,CAAC;QACzC,MAAM,EAAE,qBAAqB,CAAC,UAAU;KACzC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,+BAA+B,CAC5C,OAA8B;IAE9B,MAAM,EAAE,WAAW,EAAE,cAAc,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IAEpE,MAAM,eAAe,GAAG,YAAY,CAClC,SAAS,EACT,OAAO,CAAC,kBAAkB,EAC1B,OAAO,CAAC,aAAa,CACtB,CAAC;IAEF,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,WAAW,GAAG,uBAAuB,CACzC,OAAO,CAAC,iBAAiB,EACzB,eAAe,CAAC,QAAQ,CACzB,CAAC;IAEF,IAAI,CAAC,WAAW,IAAI,WAAW,KAAK,OAAO,CAAC,iBAAiB,EAAE,CAAC;QAC9D,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,GAAG,CAAC,mDAAmD,EAAE;QACvD,oBAAoB,EAAE,OAAO,CAAC,iBAAiB;QAC/C,WAAW;KACZ,CAAC,CAAC;IAEH,IAAI,UAAuD,CAAC;IAE5D,IAAI,CAAC;QACH,UAAU,GAAG,MAAM,cAAc,CAC/B;YACE,GAAG,OAAO;YACV,iBAAiB,EAAE,WAAW;SAC/B,EACD,WAAW,CACZ,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,4BAA4B,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7C,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,aAAa,GAAG,MAAM,+BAA+B,CACzD,UAAU,EACV,SAAS,EACT,OAAO,CACR,CAAC;IAEF,OAAO,aAAa,EAAE,MAAM,CAAC;AAC/B,CAAC;AAED,SAAS,uBAAuB,CAC9B,eAAuB,EACvB,cAAsB;IAEtB,MAAM,YAAY,GAAG,IAAI,SAAS,CAAC,eAAe,CAAC,CAAC;IAEpD,IAAI,YAAY,CAAC,mBAAmB,CAAC,CAAC,CAAC,EAAE,CAAC;QACxC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,cAAc,GAAG,YAAY;SAChC,YAAY,CAAC,uBAAuB,CAAC;SACrC,YAAY,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAEvC,gEAAgE;IAChE,iDAAiD;IACjD,MAAM,eAAe,GAAG,IAAI,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAChD,IAAI,CAAC,GAAG,CAAC,cAAc,GAAG,CAAC,EAAE,CAAC,CAAC,CAChC,CAAC;IAEF,MAAM,QAAQ,GAAG,SAAS,CAAC,OAAO,CAChC,YAAY,EACZ,SAAS,CAAC,OAAO,CAAC,cAAc,EAAE,eAAe,CAAC,CACnD,CAAC,YAAY,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAEtC,IAAI,QAAQ,CAAC,mBAAmB,CAAC,CAAC,CAAC,EAAE,CAAC;QACpC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AAC7B,CAAC;AAED,SAAS,gBAAgB,CACvB,WAA4C,EAC5C,OAAe,EACf,WAAqC;IAErC,GAAG,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAC1B,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,SAAS,wBAAwB,CAAC,MAAgB,EAAE,MAAgB;IAClE,OAAO,MAAM,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM,CAAC;AACzC,CAAC;AAED,SAAS,gBAAgB,CAAC,SAAmB;IAC3C,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,KAAK,GAAG,QAAQ,EAAE,CAAC,CAAC,CAAC;AACpE,CAAC;AAED,SAAS,wBAAwB,CAC/B,KAAsC;IAEtC,KAAK,CAAC,QAAQ,CAAC,QAAQ,GAAG;QACxB,GAAG,KAAK,CAAC,QAAQ,CAAC,QAAQ;QAC1B,eAAe,EAAE,IAAI;KACtB,CAAC;AACJ,CAAC","sourcesContent":["import { createModuleLogger } from '@metamask/utils';\nimport { BigNumber } from 'bignumber.js';\n\nimport {\n  getGasStationEligibility,\n  getGasStationCostInSourceTokenRaw,\n} from './gas-station';\nimport type { RelayQuote } from './types';\nimport { projectLogger } from '../../logger';\nimport type {\n  PayStrategyGetQuotesRequest,\n  QuoteRequest,\n  TransactionPayControllerMessenger,\n  TransactionPayQuote,\n} from '../../types';\nimport {\n  getNativeToken,\n  getTokenBalance,\n  getTokenInfo,\n} from '../../utils/token';\n\nconst log = createModuleLogger(projectLogger, 'relay-max-gas-station');\n\nconst PROBE_AMOUNT_PERCENTAGE = 0.25;\n\nenum GasCostEstimateSource {\n  GasStation = 'gas-station',\n  Probe = 'probe',\n  Quote = 'quote',\n}\n\ntype GasCostEstimate = {\n  amount: BigNumber;\n  source: GasCostEstimateSource;\n};\n\ntype NativeBalanceCheckResult = {\n  hasEnoughNativeBalance: boolean;\n  nativeBalance?: string;\n  nativeGasCostRaw?: string;\n};\n\ntype GetSingleQuoteFn = (\n  request: QuoteRequest,\n  fullRequest: PayStrategyGetQuotesRequest,\n) => Promise<TransactionPayQuote<RelayQuote>>;\n\ntype MaxAmountQuoteContext = {\n  fullRequest: PayStrategyGetQuotesRequest;\n  getSingleQuote: GetSingleQuoteFn;\n  messenger: TransactionPayControllerMessenger;\n  request: QuoteRequest;\n};\n\n/**\n * Returns a Relay max-amount quote using a two-phase gas-station fallback.\n *\n * It first requests a standard max quote (phase 1), then when needed estimates\n * gas in source-token units (directly or via a probe quote), requests an\n * adjusted max quote (phase 2), and accepts phase 2 only if validation passes\n * (source gas fee token selected, gas-limit checks, affordability).\n *\n * If any step fails or validation is unsafe, it safely falls back to phase 1.\n * Successful phase-2 quotes are tagged with `metamask.isMaxGasStation = true`.\n *\n * @param request - Relay quote request for a max-amount flow.\n * @param fullRequest - Full quote request context including messenger and transaction.\n * @param getSingleQuote - Quote fetcher used for phase-1, phase-2, and probe quotes.\n * @returns The validated adjusted phase-2 quote, or the original phase-1 quote on fallback.\n */\nexport async function getRelayMaxGasStationQuote(\n  request: QuoteRequest,\n  fullRequest: PayStrategyGetQuotesRequest,\n  getSingleQuote: GetSingleQuoteFn,\n): Promise<TransactionPayQuote<RelayQuote>> {\n  const { messenger } = fullRequest;\n  const { sourceChainId, sourceTokenAmount } = request;\n  const context: MaxAmountQuoteContext = {\n    fullRequest,\n    getSingleQuote,\n    messenger,\n    request,\n  };\n\n  const phase1Quote = await getSingleQuote(request, fullRequest);\n\n  const nativeBalanceCheck = checkEnoughNativeBalanceIfSourceGasFeeTokenNotUsed(\n    phase1Quote,\n    messenger,\n    request,\n  );\n\n  if (nativeBalanceCheck.hasEnoughNativeBalance) {\n    return fallbackToPhase1(\n      phase1Quote,\n      'Native balance is sufficient for gas',\n      {\n        nativeBalance: nativeBalanceCheck.nativeBalance,\n        nativeGasCost: nativeBalanceCheck.nativeGasCostRaw,\n      },\n    );\n  }\n\n  const gasStationEligibility = getGasStationEligibility(\n    messenger,\n    sourceChainId,\n  );\n\n  if (!gasStationEligibility.isEligible) {\n    return fallbackToPhase1(\n      phase1Quote,\n      'Gas station is disabled or unsupported',\n    );\n  }\n\n  const sourceAmountBN = new BigNumber(sourceTokenAmount);\n  const probeCost = await getProbeGasCostInSourceTokenRaw(context);\n\n  if (!probeCost) {\n    return fallbackToPhase1(\n      phase1Quote,\n      'Unable to estimate gas-station source token cost',\n    );\n  }\n\n  const initialGasEstimate: GasCostEstimate = {\n    amount: probeCost,\n    source: GasCostEstimateSource.Probe,\n  };\n\n  const adjustedSourceAmount = getAdjustedSourceAmount(\n    sourceAmountBN,\n    initialGasEstimate.amount,\n  );\n\n  if (!adjustedSourceAmount.isGreaterThan(0)) {\n    return fallbackToPhase1(\n      phase1Quote,\n      'Insufficient balance for gas station after max adjustment',\n    );\n  }\n\n  const phase2Quote = await getAdjustedPhase2Quote(\n    adjustedSourceAmount,\n    initialGasEstimate,\n    context,\n  );\n\n  if (!phase2Quote) {\n    return fallbackToPhase1(\n      phase1Quote,\n      'Adjusted phase-2 quote request failed',\n    );\n  }\n\n  if (!phase2Quote.fees.isSourceGasFeeToken) {\n    return fallbackToPhase1(\n      phase1Quote,\n      'Adjusted quote did not return source gas fee token pricing',\n    );\n  }\n\n  const phase1GasLimits = phase1Quote.original.metamask.gasLimits ?? [];\n  const phase2GasLimits = phase2Quote.original.metamask.gasLimits ?? [];\n\n  if (!hasMatchingGasLimitShape(phase1GasLimits, phase2GasLimits)) {\n    return fallbackToPhase1(\n      phase1Quote,\n      'Adjusted quote gas limit shape changed between phases',\n      {\n        phase1GasLimits,\n        phase2GasLimits,\n      },\n    );\n  }\n\n  const phase1TotalGasLimit = getTotalGasLimit(phase1GasLimits);\n  const phase2TotalGasLimit = getTotalGasLimit(phase2GasLimits);\n\n  if (phase2TotalGasLimit > phase1TotalGasLimit) {\n    return fallbackToPhase1(\n      phase1Quote,\n      'Adjusted quote total gas limit increased between phases',\n      {\n        phase1TotalGasLimit,\n        phase2TotalGasLimit,\n      },\n    );\n  }\n\n  const validationGasEstimate = new BigNumber(\n    phase2Quote.fees.sourceNetwork.max.raw,\n  );\n\n  if (\n    !isAdjustedAmountAffordable(\n      adjustedSourceAmount,\n      validationGasEstimate,\n      sourceAmountBN,\n    )\n  ) {\n    return fallbackToPhase1(\n      phase1Quote,\n      'Adjusted quote fails affordability validation',\n      {\n        adjustedSourceAmount: adjustedSourceAmount.toString(10),\n        validationGasCost: validationGasEstimate.toString(10),\n      },\n    );\n  }\n\n  markQuoteAsMaxGasStation(phase2Quote);\n\n  return phase2Quote;\n}\n\nfunction checkEnoughNativeBalanceIfSourceGasFeeTokenNotUsed(\n  quote: TransactionPayQuote<RelayQuote>,\n  messenger: TransactionPayControllerMessenger,\n  request: QuoteRequest,\n): NativeBalanceCheckResult {\n  if (quote.fees.isSourceGasFeeToken) {\n    return { hasEnoughNativeBalance: false };\n  }\n\n  const nativeGasCostRaw = quote.fees.sourceNetwork.max.raw;\n  const nativeBalance = getTokenBalance(\n    messenger,\n    request.from,\n    request.sourceChainId,\n    getNativeToken(request.sourceChainId),\n  );\n\n  return {\n    hasEnoughNativeBalance: new BigNumber(nativeBalance).isGreaterThanOrEqualTo(\n      nativeGasCostRaw,\n    ),\n    nativeBalance,\n    nativeGasCostRaw,\n  };\n}\n\nfunction getAdjustedSourceAmount(\n  sourceAmount: BigNumber,\n  estimatedGasCost: BigNumber,\n): BigNumber {\n  return sourceAmount\n    .minus(estimatedGasCost)\n    .integerValue(BigNumber.ROUND_DOWN);\n}\n\nfunction isAdjustedAmountAffordable(\n  adjustedSourceAmount: BigNumber,\n  gasCost: BigNumber,\n  originalSourceAmount: BigNumber,\n): boolean {\n  return adjustedSourceAmount\n    .plus(gasCost)\n    .isLessThanOrEqualTo(originalSourceAmount);\n}\n\nasync function getAdjustedPhase2Quote(\n  adjustedSourceAmount: BigNumber,\n  initialGasEstimate: GasCostEstimate,\n  context: MaxAmountQuoteContext,\n): Promise<TransactionPayQuote<RelayQuote> | undefined> {\n  const { fullRequest, getSingleQuote, request } = context;\n\n  log('Requesting adjusted max quote', {\n    adjustedAmount: adjustedSourceAmount.toString(10),\n    gasCostInSourceToken: initialGasEstimate.amount.toString(10),\n    gasEstimateSource: initialGasEstimate.source,\n    originalAmount: request.sourceTokenAmount,\n  });\n\n  try {\n    return await getSingleQuote(\n      {\n        ...request,\n        sourceTokenAmount: adjustedSourceAmount.toFixed(\n          0,\n          BigNumber.ROUND_DOWN,\n        ),\n      },\n      fullRequest,\n    );\n  } catch (error) {\n    log('Adjusted quote request failed, falling back to phase-1 quote', {\n      error,\n    });\n    return undefined;\n  }\n}\n\nasync function getGasCostFromQuoteOrGasStation(\n  quote: TransactionPayQuote<RelayQuote>,\n  messenger: TransactionPayControllerMessenger,\n  request: QuoteRequest,\n): Promise<GasCostEstimate | undefined> {\n  const gasCost = quote.fees.sourceNetwork.max;\n\n  if (quote.fees.isSourceGasFeeToken) {\n    log('Gas cost already in source token units', { raw: gasCost.raw });\n    return {\n      amount: new BigNumber(gasCost.raw),\n      source: GasCostEstimateSource.Quote,\n    };\n  }\n\n  const firstStepData = quote.original.steps[0]?.items[0]?.data;\n\n  if (!firstStepData) {\n    return undefined;\n  }\n\n  const totalItemCount = quote.original.steps.reduce(\n    (count, step) => count + step.items.length,\n    0,\n  );\n\n  const totalGasEstimate = (quote.original.metamask.gasLimits ?? []).reduce(\n    (acc, gasLimit) => acc + gasLimit,\n    0,\n  );\n\n  const gasStationCost = await getGasStationCostInSourceTokenRaw({\n    firstStepData,\n    messenger,\n    request: {\n      from: request.from,\n      sourceChainId: request.sourceChainId,\n      sourceTokenAddress: request.sourceTokenAddress,\n    },\n    totalGasEstimate,\n    totalItemCount,\n  });\n\n  if (!gasStationCost) {\n    return undefined;\n  }\n\n  return {\n    amount: new BigNumber(gasStationCost.raw),\n    source: GasCostEstimateSource.GasStation,\n  };\n}\n\nasync function getProbeGasCostInSourceTokenRaw(\n  context: MaxAmountQuoteContext,\n): Promise<BigNumber | undefined> {\n  const { fullRequest, getSingleQuote, messenger, request } = context;\n\n  const sourceTokenInfo = getTokenInfo(\n    messenger,\n    request.sourceTokenAddress,\n    request.sourceChainId,\n  );\n\n  if (!sourceTokenInfo) {\n    return undefined;\n  }\n\n  const probeAmount = getProbeSourceAmountRaw(\n    request.sourceTokenAmount,\n    sourceTokenInfo.decimals,\n  );\n\n  if (!probeAmount || probeAmount === request.sourceTokenAmount) {\n    return undefined;\n  }\n\n  log('Requesting probe quote for gas station estimation', {\n    originalSourceAmount: request.sourceTokenAmount,\n    probeAmount,\n  });\n\n  let probeQuote: TransactionPayQuote<RelayQuote> | undefined;\n\n  try {\n    probeQuote = await getSingleQuote(\n      {\n        ...request,\n        sourceTokenAmount: probeAmount,\n      },\n      fullRequest,\n    );\n  } catch (error) {\n    log('Probe quote request failed', { error });\n    return undefined;\n  }\n\n  if (!probeQuote) {\n    return undefined;\n  }\n\n  const probeEstimate = await getGasCostFromQuoteOrGasStation(\n    probeQuote,\n    messenger,\n    request,\n  );\n\n  return probeEstimate?.amount;\n}\n\nfunction getProbeSourceAmountRaw(\n  sourceAmountRaw: string,\n  sourceDecimals: number,\n): string | undefined {\n  const sourceAmount = new BigNumber(sourceAmountRaw);\n\n  if (sourceAmount.isLessThanOrEqualTo(0)) {\n    return undefined;\n  }\n\n  const probeRawAmount = sourceAmount\n    .multipliedBy(PROBE_AMOUNT_PERCENTAGE)\n    .integerValue(BigNumber.ROUND_FLOOR);\n\n  // Minimum probe size: ~0.01 token for tokens with >=2 decimals,\n  // otherwise one raw unit for low-decimal tokens.\n  const minimumProbeRaw = new BigNumber(1).shiftedBy(\n    Math.max(sourceDecimals - 2, 0),\n  );\n\n  const probeRaw = BigNumber.minimum(\n    sourceAmount,\n    BigNumber.maximum(probeRawAmount, minimumProbeRaw),\n  ).integerValue(BigNumber.ROUND_FLOOR);\n\n  if (probeRaw.isLessThanOrEqualTo(0)) {\n    return undefined;\n  }\n\n  return probeRaw.toFixed(0);\n}\n\nfunction fallbackToPhase1(\n  phase1Quote: TransactionPayQuote<RelayQuote>,\n  message: string,\n  paramsToLog?: Record<string, unknown>,\n): TransactionPayQuote<RelayQuote> {\n  log(message, paramsToLog);\n  return phase1Quote;\n}\n\nfunction hasMatchingGasLimitShape(phase1: number[], phase2: number[]): boolean {\n  return phase1.length === phase2.length;\n}\n\nfunction getTotalGasLimit(gasLimits: number[]): number {\n  return gasLimits.reduce((total, gasLimit) => total + gasLimit, 0);\n}\n\nfunction markQuoteAsMaxGasStation(\n  quote: TransactionPayQuote<RelayQuote>,\n): void {\n  quote.original.metamask = {\n    ...quote.original.metamask,\n    isMaxGasStation: true,\n  };\n}\n"]}

package/dist/strategy/relay/relay-quotes.cjs

@@ -7,9 +7,12 @@ const controller_utils_1 = require("@metamask/controller-utils");
 const utils_1 = require("@metamask/utils");
 const bignumber_js_1 = require("bignumber.js");
 const constants_1 = require("./constants.cjs");
+const gas_station_1 = require("./gas-station.cjs");
+const relay_max_gas_station_1 = require("./relay-max-gas-station.cjs");
 const __1 = require("../../index.cjs");
 const constants_2 = require("../../constants.cjs");
 const logger_1 = require("../../logger.cjs");
+const amounts_1 = require("../../utils/amounts.cjs");
 const feature_flags_1 = require("../../utils/feature-flags.cjs");
 const gas_1 = require("../../utils/gas.cjs");
 const token_1 = require("../../utils/token.cjs");
@@ -31,7 +34,7 @@ async function getRelayQuotes(request) {
             singleRequest.isPostQuote)
             .map((singleRequest) => normalizeRequest(singleRequest));
         log('Normalized requests', normalizedRequests);
-        return await Promise.all(normalizedRequests.map((singleRequest) => getSingleQuote(singleRequest, request)));
+        return await Promise.all(normalizedRequests.map((singleRequest) => getQuoteWithMaxAmountHandling(singleRequest, request)));
     }
     catch (error) {
         log('Error fetching quotes', { error });
@@ -39,6 +42,13 @@ async function getRelayQuotes(request) {
     }
 }
 exports.getRelayQuotes = getRelayQuotes;
+async function getQuoteWithMaxAmountHandling(request, fullRequest) {
+    const { isMaxAmount } = request;
+    if (!isMaxAmount) {
+        return getSingleQuote(request, fullRequest);
+    }
+    return (0, relay_max_gas_station_1.getRelayMaxGasStationQuote)(request, fullRequest, getSingleQuote);
+}
 /**
  * Fetches a single Relay quote.
  *
@@ -197,15 +207,15 @@ async function normalizeQuote(quote, request, fullRequest) {
     const { details } = quote;
     const { currencyIn, currencyOut } = details;
     const { usdToFiatRate } = getFiatRates(messenger, request);
-    const dust = getFiatValueFromUsd(calculateDustUsd(quote, request), usdToFiatRate);
+    const dust = (0, amounts_1.getFiatValueFromUsd)(calculateDustUsd(quote, request), usdToFiatRate);
     const subsidizedFeeUsd = getSubsidizedFeeAmountUsd(quote);
     const appFeeUsd = new bignumber_js_1.BigNumber(quote.fees?.app?.amountUsd ?? '0');
-    const metaMaskFee = getFiatValueFromUsd(appFeeUsd, usdToFiatRate);
+    const metaMaskFee = (0, amounts_1.getFiatValueFromUsd)(appFeeUsd, usdToFiatRate);
     // Subtract app fee from provider fee since totalImpact.usd already includes it
     const providerFeeUsd = calculateProviderFee(quote).minus(appFeeUsd);
     const provider = subsidizedFeeUsd.gt(0)
         ? { usd: '0', fiat: '0' }
-        : getFiatValueFromUsd(providerFeeUsd, usdToFiatRate);
+        : (0, amounts_1.getFiatValueFromUsd)(providerFeeUsd, usdToFiatRate);
     const { gasLimits, isGasFeeToken: isSourceGasFeeToken, ...sourceNetwork } = await calculateSourceNetworkCost(quote, messenger, request, fullRequest.transaction);
     const targetNetwork = {
         usd: '0',
@@ -214,7 +224,7 @@ async function normalizeQuote(quote, request, fullRequest) {
     const sourceAmount = {
         human: currencyIn.amountFormatted,
         raw: currencyIn.amount,
-        ...getFiatValueFromUsd(new bignumber_js_1.BigNumber(currencyIn.amountUsd), usdToFiatRate),
+        ...(0, amounts_1.getFiatValueFromUsd)(new bignumber_js_1.BigNumber(currencyIn.amountUsd), usdToFiatRate),
     };
     const isTargetStablecoin = isStablecoin(request.targetChainId, request.targetTokenAddress);
     const additionalTargetAmountUsd = quote.request.tradeType === 'EXACT_INPUT'
@@ -227,7 +237,7 @@ async function normalizeQuote(quote, request, fullRequest) {
         ? new bignumber_js_1.BigNumber(currencyOut.amountFormatted)
         : new bignumber_js_1.BigNumber(currencyOut.amountUsd);
     const targetAmountUsd = baseTargetAmountUsd.plus(additionalTargetAmountUsd);
-    const targetAmount = getFiatValueFromUsd(targetAmountUsd, usdToFiatRate);
+    const targetAmount = (0, amounts_1.getFiatValueFromUsd)(targetAmountUsd, usdToFiatRate);
     const metamask = {
         gasLimits,
     };
@@ -266,20 +276,6 @@ function calculateDustUsd(quote, request) {
     const dustRaw = new bignumber_js_1.BigNumber(minimumAmount).minus(request.targetAmountMinimum);
     return dustRaw.shiftedBy(-targetDecimals).multipliedBy(targetUsdRate);
 }
-/**
- * Converts USD value to fiat value.
- *
- * @param usdValue - USD value.
- * @param usdToFiatRate - USD to fiat rate.
- * @returns Fiat value.
- */
-function getFiatValueFromUsd(usdValue, usdToFiatRate) {
-    const fiatValue = usdValue.multipliedBy(usdToFiatRate);
-    return {
-        usd: usdValue.toString(10),
-        fiat: fiatValue.toString(10),
-    };
-}
 /**
  * Calculates USD to fiat rate.
  *
@@ -320,7 +316,6 @@ async function calculateSourceNetworkCost(quote, messenger, request, transaction
     const relayParams = quote.steps
         .flatMap((step) => step.items)
         .map((item) => item.data);
-    const { relayDisabledGasStationChains } = (0, feature_flags_1.getFeatureFlags)(messenger);
     const { chainId, data, maxFeePerGas, maxPriorityFeePerGas, to, value } = relayParams[0];
     const { totalGasEstimate, totalGasLimit, gasLimits } = await calculateSourceNetworkGasLimit(relayParams, messenger, request.isPostQuote ? transaction : undefined);
     log('Gas limit', {
@@ -348,16 +343,14 @@ async function calculateSourceNetworkCost(quote, messenger, request, transaction
     if (new bignumber_js_1.BigNumber(nativeBalance).isGreaterThanOrEqualTo(max.raw)) {
         return result;
     }
-    if (relayDisabledGasStationChains.includes(sourceChainId)) {
+    const gasStationEligibility = (0, gas_station_1.getGasStationEligibility)(messenger, sourceChainId);
+    if (gasStationEligibility.isDisabledChain) {
         log('Skipping gas station as disabled chain', {
             sourceChainId,
-            disabledChainIds: relayDisabledGasStationChains,
         });
         return result;
     }
-    const supportedChains = (0, feature_flags_1.getEIP7702SupportedChains)(messenger);
-    const chainSupportsGasStation = supportedChains.some((supportedChainId) => supportedChainId.toLowerCase() === sourceChainId.toLowerCase());
-    if (!chainSupportsGasStation) {
+    if (!gasStationEligibility.chainSupportsGasStation) {
         log('Skipping gas station as chain does not support EIP-7702', {
             sourceChainId,
         });
@@ -367,40 +360,20 @@ async function calculateSourceNetworkCost(quote, messenger, request, transaction
         nativeBalance,
         max: max.raw,
     });
-    const gasFeeTokens = await messenger.call('TransactionController:getGasFeeTokens', {
-        chainId: sourceChainId,
-        data,
-        from,
-        to,
-        value: (0, controller_utils_1.toHex)(value ?? '0'),
-    });
-    log('Source gas fee tokens', { gasFeeTokens });
-    const gasFeeToken = gasFeeTokens.find((singleGasFeeToken) => singleGasFeeToken.tokenAddress.toLowerCase() ===
-        sourceTokenAddress.toLowerCase());
-    if (!gasFeeToken) {
-        log('No matching gas fee token found', {
-            sourceTokenAddress,
-            gasFeeTokens,
-        });
-        return result;
-    }
-    let finalAmount = gasFeeToken.amount;
-    const hasMultipleTransactions = relayParams.length > 1 || gasLimits.length > 1;
-    if (hasMultipleTransactions) {
-        const gasRate = new bignumber_js_1.BigNumber(gasFeeToken.amount, 16).dividedBy(gasFeeToken.gas, 16);
-        const finalAmountValue = gasRate.multipliedBy(totalGasEstimate);
-        finalAmount = (0, controller_utils_1.toHex)(finalAmountValue.toFixed(0));
-        log('Estimated gas fee token amount for batch', {
-            finalAmount: finalAmountValue.toString(10),
-            gasRate: gasRate.toString(10),
-            totalGasEstimate,
-        });
-    }
-    const finalGasFeeToken = { ...gasFeeToken, amount: finalAmount };
-    const gasFeeTokenCost = (0, gas_1.calculateGasFeeTokenCost)({
-        chainId: sourceChainId,
-        gasFeeToken: finalGasFeeToken,
+    const gasFeeTokenCost = await (0, gas_station_1.getGasStationCostInSourceTokenRaw)({
+        firstStepData: {
+            data,
+            to,
+            value,
+        },
         messenger,
+        request: {
+            from,
+            sourceChainId,
+            sourceTokenAddress,
+        },
+        totalGasEstimate,
+        totalItemCount: Math.max(relayParams.length, gasLimits.length),
     });
     if (!gasFeeTokenCost) {
         return result;