@metamask-previews/transaction-pay-controller 16.2.0-preview-8c978cc75 → 16.3.0-preview-15dd7d63f

package/CHANGELOG.md

@@ -7,6 +7,21 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Added
+
+- Adds a dedicated max-amount gas-station fallback flow for Relay quoting ([#7927](https://github.com/MetaMask/core/pull/7927))
+
+## [16.3.0]
+
+### Added
+
+- Add optional `refundTo` field to `TransactionConfig`, `TransactionData`, and `QuoteRequest` types, allowing callers to specify an address that receives refunds if a Relay transaction fails ([#8112](https://github.com/MetaMask/core/pull/8112))
+
+### Changed
+
+- Bump `@metamask/assets-controllers` from `^100.0.3` to `^100.1.0` ([#8107](https://github.com/MetaMask/core/pull/8107))
+- Bump `@metamask/transaction-controller` from `^62.19.0` to `^62.20.0` ([#8104](https://github.com/MetaMask/core/pull/8104))
+
 ## [16.2.0]
 
 ### Changed
@@ -465,7 +480,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 - Initial release ([#6820](https://github.com/MetaMask/core/pull/6820))
 
-[Unreleased]: https://github.com/MetaMask/core/compare/@metamask/transaction-pay-controller@16.2.0...HEAD
+[Unreleased]: https://github.com/MetaMask/core/compare/@metamask/transaction-pay-controller@16.3.0...HEAD
+[16.3.0]: https://github.com/MetaMask/core/compare/@metamask/transaction-pay-controller@16.2.0...@metamask/transaction-pay-controller@16.3.0
 [16.2.0]: https://github.com/MetaMask/core/compare/@metamask/transaction-pay-controller@16.1.2...@metamask/transaction-pay-controller@16.2.0
 [16.1.2]: https://github.com/MetaMask/core/compare/@metamask/transaction-pay-controller@16.1.1...@metamask/transaction-pay-controller@16.1.2
 [16.1.1]: https://github.com/MetaMask/core/compare/@metamask/transaction-pay-controller@16.1.0...@metamask/transaction-pay-controller@16.1.1

package/dist/TransactionPayController.cjs

@@ -62,10 +62,12 @@ class TransactionPayController extends base_controller_1.BaseController {
             const config = {
                 isMaxAmount: transactionData.isMaxAmount,
                 isPostQuote: transactionData.isPostQuote,
+                refundTo: transactionData.refundTo,
             };
             callback(config);
             transactionData.isMaxAmount = config.isMaxAmount;
             transactionData.isPostQuote = config.isPostQuote;
+            transactionData.refundTo = config.refundTo;
         });
     }
     updatePaymentToken(request) {

package/dist/TransactionPayController.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"TransactionPayController.cjs","sourceRoot":"","sources":["../src/TransactionPayController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AACA,+DAA2D;AAG3D,mCAA8B;AAE9B,6EAAoE;AACpE,+CAIqB;AACrB,iEAA0D;AAU1D,6DAAyD;AACzD,+CAA8C;AAC9C,+DAA6D;AAC7D,yDAA6D;AAE7D,MAAM,aAAa,GAAiD;IAClE,eAAe,EAAE;QACf,sBAAsB,EAAE,KAAK;QAC7B,kBAAkB,EAAE,IAAI;QACxB,OAAO,EAAE,KAAK;QACd,QAAQ,EAAE,IAAI;KACf;CACF,CAAC;AAEF,MAAM,eAAe,GAAG,GAAkC,EAAE,CAAC,CAAC;IAC5D,eAAe,EAAE,EAAE;CACpB,CAAC,CAAC;AAEH,MAAa,wBAAyB,SAAQ,gCAI7C;IAWC,YAAY,EACV,wBAAwB,EACxB,WAAW,EACX,aAAa,EACb,SAAS,EACT,KAAK,GAC2B;QAChC,KAAK,CAAC;YACJ,IAAI,EAAE,2BAAe;YACrB,QAAQ,EAAE,aAAa;YACvB,SAAS;YACT,KAAK,EAAE,EAAE,GAAG,eAAe,EAAE,EAAE,GAAG,KAAK,EAAE;SAC1C,CAAC,CAAC;;QAtBI,qEAA4D;QAE5D,wDAEmB;QAEnB,0DAEqB;QAgB5B,uBAAA,IAAI,sDAA6B,wBAAwB,MAAA,CAAC;QAC1D,uBAAA,IAAI,yCAAgB,WAAW,MAAA,CAAC;QAChC,uBAAA,IAAI,2CAAkB,aAAa,MAAA,CAAC;QAEpC,uBAAA,IAAI,6FAAwB,MAA5B,IAAI,CAA0B,CAAC;QAE/B,IAAA,oCAAsB,EACpB,SAAS,EACT,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,EACtC,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,CACvC,CAAC;QAEF,kCAAkC;QAClC,IAAI,+BAAc,CAAC;YACjB,aAAa,EAAE,uBAAA,IAAI,gGAA2B,CAAC,IAAI,CAAC,IAAI,CAAC;YACzD,SAAS;YACT,qBAAqB,EAAE,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9D,CAAC,CAAC;IACL,CAAC;IAED,oBAAoB,CAClB,aAAqB,EACrB,QAAmC;QAEnC,uBAAA,IAAI,4FAAuB,MAA3B,IAAI,EAAwB,aAAa,EAAE,CAAC,eAAe,EAAE,EAAE;YAC7D,MAAM,MAAM,GAAG;gBACb,WAAW,EAAE,eAAe,CAAC,WAAW;gBACxC,WAAW,EAAE,eAAe,CAAC,WAAW;aACzC,CAAC;YAEF,QAAQ,CAAC,MAAM,CAAC,CAAC;YAEjB,eAAe,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;YACjD,eAAe,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC;IAED,kBAAkB,CAAC,OAAkC;QACnD,IAAA,yCAAkB,EAAC,OAAO,EAAE;YAC1B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,qBAAqB,EAAE,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9D,CAAC,CAAC;IACL,CAAC;CAwGF;AA/KD,4DA+KC;oVAtGwB,aAAqB;IAC1C,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QACpB,OAAO,KAAK,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,6GAGC,aAAqB,EACrB,EAAqD;IAErD,IAAI,kBAAkB,GAAG,KAAK,CAAC;IAE/B,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QACpB,MAAM,EAAE,eAAe,EAAE,GAAG,KAAK,CAAC;QAClC,IAAI,OAAO,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;QAC7C,MAAM,oBAAoB,GAAG,OAAO,EAAE,YAAY,CAAC;QACnD,MAAM,cAAc,GAAG,OAAO,EAAE,MAAM,CAAC;QACvC,MAAM,mBAAmB,GAAG,OAAO,EAAE,WAAW,CAAC;QACjD,MAAM,mBAAmB,GAAG,OAAO,EAAE,WAAW,CAAC;QAEjD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,eAAe,CAAC,aAAa,CAAC,GAAG;gBAC/B,SAAS,EAAE,KAAK;gBAChB,MAAM,EAAE,EAAE;aACX,CAAC;YAEF,OAAO,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;QAC3C,CAAC;QAED,EAAE,CAAC,OAAO,CAAC,CAAC;QAEZ,MAAM,qBAAqB,GACzB,OAAO,CAAC,YAAY,EAAE,OAAO,EAAE,WAAW,EAAE;YAC1C,oBAAoB,EAAE,OAAO,EAAE,WAAW,EAAE;YAC9C,OAAO,CAAC,YAAY,EAAE,OAAO,KAAK,oBAAoB,EAAE,OAAO,CAAC;QAElE,MAAM,eAAe,GAAG,OAAO,CAAC,MAAM,KAAK,cAAc,CAAC;QAC1D,MAAM,cAAc,GAAG,OAAO,CAAC,WAAW,KAAK,mBAAmB,CAAC;QACnE,MAAM,kBAAkB,GAAG,OAAO,CAAC,WAAW,KAAK,mBAAmB,CAAC;QAEvE,IACE,qBAAqB;YACrB,cAAc;YACd,eAAe;YACf,kBAAkB,EAClB,CAAC;YACD,IAAA,oCAAmB,EAAC,aAAa,EAAE,OAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YAErE,kBAAkB,GAAG,IAAI,CAAC;QAC5B,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,kBAAkB,EAAE,CAAC;QACvB,IAAA,qBAAY,EAAC;YACX,aAAa,EAAE,uBAAA,IAAI,gGAA2B,CAAC,IAAI,CAAC,IAAI,CAAC;YACzD,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,eAAe,EAAE,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,aAAa,CAAC;YAC1D,aAAa;YACb,qBAAqB,EAAE,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9D,CAAC,CAAC,KAAK,CAAC,aAAI,CAAC,CAAC;IACjB,CAAC;AACH,CAAC;IAGC,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,mDAAmD,EACnD,uBAAA,IAAI,0DAA0B,CAAC,IAAI,CAAC,IAAI,CAAC,CAC1C,CAAC;IAEF,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,sCAAsC,EACtC,CAAC,WAA4B,EAA0B,EAAE,CACvD,uBAAA,IAAI,gGAA2B,MAA/B,IAAI,EAA4B,WAAW,CAAC,CAAC,CAAC,CAAC,CAClD,CAAC;IAEF,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,+CAA+C,EAC/C,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,CACrC,CAAC;IAEF,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,6CAA6C,EAC7C,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CACnC,CAAC;AACJ,CAAC,qHAGC,WAA4B;IAE5B,MAAM,kBAAkB,GACtB,uBAAA,IAAI,+CAAe,EAAE,KAArB,IAAI,EAAkB,WAAW,CAAC;QAClC,CAAC,uBAAA,IAAI,6CAAa,CAAC,CAAC,CAAC,CAAC,uBAAA,IAAI,6CAAa,MAAjB,IAAI,EAAc,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAE9D,MAAM,eAAe,GAAG,kBAAkB,CAAC,MAAM,CAC/C,CAAC,QAAQ,EAAsC,EAAE,CAC/C,IAAA,oCAAwB,EAAC,QAAQ,CAAC,CACrC,CAAC;IAEF,OAAO,eAAe,CAAC,MAAM;QAC3B,CAAC,CAAC,eAAe;QACjB,CAAC,CAAC,IAAA,gCAAgB,EAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACvC,CAAC","sourcesContent":["import type { StateMetadata } from '@metamask/base-controller';\nimport { BaseController } from '@metamask/base-controller';\nimport type { TransactionMeta } from '@metamask/transaction-controller';\nimport type { Draft } from 'immer';\nimport { noop } from 'lodash';\n\nimport { updatePaymentToken } from './actions/update-payment-token';\nimport {\n  CONTROLLER_NAME,\n  isTransactionPayStrategy,\n  TransactionPayStrategy,\n} from './constants';\nimport { QuoteRefresher } from './helpers/QuoteRefresher';\nimport type {\n  GetDelegationTransactionCallback,\n  TransactionConfigCallback,\n  TransactionData,\n  TransactionPayControllerMessenger,\n  TransactionPayControllerOptions,\n  TransactionPayControllerState,\n  UpdatePaymentTokenRequest,\n} from './types';\nimport { getStrategyOrder } from './utils/feature-flags';\nimport { updateQuotes } from './utils/quotes';\nimport { updateSourceAmounts } from './utils/source-amounts';\nimport { pollTransactionChanges } from './utils/transaction';\n\nconst stateMetadata: StateMetadata<TransactionPayControllerState> = {\n  transactionData: {\n    includeInDebugSnapshot: false,\n    includeInStateLogs: true,\n    persist: false,\n    usedInUi: true,\n  },\n};\n\nconst getDefaultState = (): TransactionPayControllerState => ({\n  transactionData: {},\n});\n\nexport class TransactionPayController extends BaseController<\n  typeof CONTROLLER_NAME,\n  TransactionPayControllerState,\n  TransactionPayControllerMessenger\n> {\n  readonly #getDelegationTransaction: GetDelegationTransactionCallback;\n\n  readonly #getStrategy?: (\n    transaction: TransactionMeta,\n  ) => TransactionPayStrategy;\n\n  readonly #getStrategies?: (\n    transaction: TransactionMeta,\n  ) => TransactionPayStrategy[];\n\n  constructor({\n    getDelegationTransaction,\n    getStrategy,\n    getStrategies,\n    messenger,\n    state,\n  }: TransactionPayControllerOptions) {\n    super({\n      name: CONTROLLER_NAME,\n      metadata: stateMetadata,\n      messenger,\n      state: { ...getDefaultState(), ...state },\n    });\n\n    this.#getDelegationTransaction = getDelegationTransaction;\n    this.#getStrategy = getStrategy;\n    this.#getStrategies = getStrategies;\n\n    this.#registerActionHandlers();\n\n    pollTransactionChanges(\n      messenger,\n      this.#updateTransactionData.bind(this),\n      this.#removeTransactionData.bind(this),\n    );\n\n    // eslint-disable-next-line no-new\n    new QuoteRefresher({\n      getStrategies: this.#getStrategiesWithFallback.bind(this),\n      messenger,\n      updateTransactionData: this.#updateTransactionData.bind(this),\n    });\n  }\n\n  setTransactionConfig(\n    transactionId: string,\n    callback: TransactionConfigCallback,\n  ): void {\n    this.#updateTransactionData(transactionId, (transactionData) => {\n      const config = {\n        isMaxAmount: transactionData.isMaxAmount,\n        isPostQuote: transactionData.isPostQuote,\n      };\n\n      callback(config);\n\n      transactionData.isMaxAmount = config.isMaxAmount;\n      transactionData.isPostQuote = config.isPostQuote;\n    });\n  }\n\n  updatePaymentToken(request: UpdatePaymentTokenRequest): void {\n    updatePaymentToken(request, {\n      messenger: this.messenger,\n      updateTransactionData: this.#updateTransactionData.bind(this),\n    });\n  }\n\n  #removeTransactionData(transactionId: string): void {\n    this.update((state) => {\n      delete state.transactionData[transactionId];\n    });\n  }\n\n  #updateTransactionData(\n    transactionId: string,\n    fn: (transactionData: Draft<TransactionData>) => void,\n  ): void {\n    let shouldUpdateQuotes = false;\n\n    this.update((state) => {\n      const { transactionData } = state;\n      let current = transactionData[transactionId];\n      const originalPaymentToken = current?.paymentToken;\n      const originalTokens = current?.tokens;\n      const originalIsMaxAmount = current?.isMaxAmount;\n      const originalIsPostQuote = current?.isPostQuote;\n\n      if (!current) {\n        transactionData[transactionId] = {\n          isLoading: false,\n          tokens: [],\n        };\n\n        current = transactionData[transactionId];\n      }\n\n      fn(current);\n\n      const isPaymentTokenUpdated =\n        current.paymentToken?.address?.toLowerCase() !==\n          originalPaymentToken?.address?.toLowerCase() ||\n        current.paymentToken?.chainId !== originalPaymentToken?.chainId;\n\n      const isTokensUpdated = current.tokens !== originalTokens;\n      const isIsMaxUpdated = current.isMaxAmount !== originalIsMaxAmount;\n      const isPostQuoteUpdated = current.isPostQuote !== originalIsPostQuote;\n\n      if (\n        isPaymentTokenUpdated ||\n        isIsMaxUpdated ||\n        isTokensUpdated ||\n        isPostQuoteUpdated\n      ) {\n        updateSourceAmounts(transactionId, current as never, this.messenger);\n\n        shouldUpdateQuotes = true;\n      }\n    });\n\n    if (shouldUpdateQuotes) {\n      updateQuotes({\n        getStrategies: this.#getStrategiesWithFallback.bind(this),\n        messenger: this.messenger,\n        transactionData: this.state.transactionData[transactionId],\n        transactionId,\n        updateTransactionData: this.#updateTransactionData.bind(this),\n      }).catch(noop);\n    }\n  }\n\n  #registerActionHandlers(): void {\n    this.messenger.registerActionHandler(\n      'TransactionPayController:getDelegationTransaction',\n      this.#getDelegationTransaction.bind(this),\n    );\n\n    this.messenger.registerActionHandler(\n      'TransactionPayController:getStrategy',\n      (transaction: TransactionMeta): TransactionPayStrategy =>\n        this.#getStrategiesWithFallback(transaction)[0],\n    );\n\n    this.messenger.registerActionHandler(\n      'TransactionPayController:setTransactionConfig',\n      this.setTransactionConfig.bind(this),\n    );\n\n    this.messenger.registerActionHandler(\n      'TransactionPayController:updatePaymentToken',\n      this.updatePaymentToken.bind(this),\n    );\n  }\n\n  #getStrategiesWithFallback(\n    transaction: TransactionMeta,\n  ): TransactionPayStrategy[] {\n    const strategyCandidates: unknown[] =\n      this.#getStrategies?.(transaction) ??\n      (this.#getStrategy ? [this.#getStrategy(transaction)] : []);\n\n    const validStrategies = strategyCandidates.filter(\n      (strategy): strategy is TransactionPayStrategy =>\n        isTransactionPayStrategy(strategy),\n    );\n\n    return validStrategies.length\n      ? validStrategies\n      : getStrategyOrder(this.messenger);\n  }\n}\n"]}
+{"version":3,"file":"TransactionPayController.cjs","sourceRoot":"","sources":["../src/TransactionPayController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AACA,+DAA2D;AAG3D,mCAA8B;AAE9B,6EAAoE;AACpE,+CAIqB;AACrB,iEAA0D;AAU1D,6DAAyD;AACzD,+CAA8C;AAC9C,+DAA6D;AAC7D,yDAA6D;AAE7D,MAAM,aAAa,GAAiD;IAClE,eAAe,EAAE;QACf,sBAAsB,EAAE,KAAK;QAC7B,kBAAkB,EAAE,IAAI;QACxB,OAAO,EAAE,KAAK;QACd,QAAQ,EAAE,IAAI;KACf;CACF,CAAC;AAEF,MAAM,eAAe,GAAG,GAAkC,EAAE,CAAC,CAAC;IAC5D,eAAe,EAAE,EAAE;CACpB,CAAC,CAAC;AAEH,MAAa,wBAAyB,SAAQ,gCAI7C;IAWC,YAAY,EACV,wBAAwB,EACxB,WAAW,EACX,aAAa,EACb,SAAS,EACT,KAAK,GAC2B;QAChC,KAAK,CAAC;YACJ,IAAI,EAAE,2BAAe;YACrB,QAAQ,EAAE,aAAa;YACvB,SAAS;YACT,KAAK,EAAE,EAAE,GAAG,eAAe,EAAE,EAAE,GAAG,KAAK,EAAE;SAC1C,CAAC,CAAC;;QAtBI,qEAA4D;QAE5D,wDAEmB;QAEnB,0DAEqB;QAgB5B,uBAAA,IAAI,sDAA6B,wBAAwB,MAAA,CAAC;QAC1D,uBAAA,IAAI,yCAAgB,WAAW,MAAA,CAAC;QAChC,uBAAA,IAAI,2CAAkB,aAAa,MAAA,CAAC;QAEpC,uBAAA,IAAI,6FAAwB,MAA5B,IAAI,CAA0B,CAAC;QAE/B,IAAA,oCAAsB,EACpB,SAAS,EACT,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,EACtC,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,CACvC,CAAC;QAEF,kCAAkC;QAClC,IAAI,+BAAc,CAAC;YACjB,aAAa,EAAE,uBAAA,IAAI,gGAA2B,CAAC,IAAI,CAAC,IAAI,CAAC;YACzD,SAAS;YACT,qBAAqB,EAAE,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9D,CAAC,CAAC;IACL,CAAC;IAED,oBAAoB,CAClB,aAAqB,EACrB,QAAmC;QAEnC,uBAAA,IAAI,4FAAuB,MAA3B,IAAI,EAAwB,aAAa,EAAE,CAAC,eAAe,EAAE,EAAE;YAC7D,MAAM,MAAM,GAAG;gBACb,WAAW,EAAE,eAAe,CAAC,WAAW;gBACxC,WAAW,EAAE,eAAe,CAAC,WAAW;gBACxC,QAAQ,EAAE,eAAe,CAAC,QAAQ;aACnC,CAAC;YAEF,QAAQ,CAAC,MAAM,CAAC,CAAC;YAEjB,eAAe,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;YACjD,eAAe,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;YACjD,eAAe,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC;IAED,kBAAkB,CAAC,OAAkC;QACnD,IAAA,yCAAkB,EAAC,OAAO,EAAE;YAC1B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,qBAAqB,EAAE,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9D,CAAC,CAAC;IACL,CAAC;CAwGF;AAjLD,4DAiLC;oVAtGwB,aAAqB;IAC1C,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QACpB,OAAO,KAAK,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,6GAGC,aAAqB,EACrB,EAAqD;IAErD,IAAI,kBAAkB,GAAG,KAAK,CAAC;IAE/B,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QACpB,MAAM,EAAE,eAAe,EAAE,GAAG,KAAK,CAAC;QAClC,IAAI,OAAO,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;QAC7C,MAAM,oBAAoB,GAAG,OAAO,EAAE,YAAY,CAAC;QACnD,MAAM,cAAc,GAAG,OAAO,EAAE,MAAM,CAAC;QACvC,MAAM,mBAAmB,GAAG,OAAO,EAAE,WAAW,CAAC;QACjD,MAAM,mBAAmB,GAAG,OAAO,EAAE,WAAW,CAAC;QAEjD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,eAAe,CAAC,aAAa,CAAC,GAAG;gBAC/B,SAAS,EAAE,KAAK;gBAChB,MAAM,EAAE,EAAE;aACX,CAAC;YAEF,OAAO,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;QAC3C,CAAC;QAED,EAAE,CAAC,OAAO,CAAC,CAAC;QAEZ,MAAM,qBAAqB,GACzB,OAAO,CAAC,YAAY,EAAE,OAAO,EAAE,WAAW,EAAE;YAC1C,oBAAoB,EAAE,OAAO,EAAE,WAAW,EAAE;YAC9C,OAAO,CAAC,YAAY,EAAE,OAAO,KAAK,oBAAoB,EAAE,OAAO,CAAC;QAElE,MAAM,eAAe,GAAG,OAAO,CAAC,MAAM,KAAK,cAAc,CAAC;QAC1D,MAAM,cAAc,GAAG,OAAO,CAAC,WAAW,KAAK,mBAAmB,CAAC;QACnE,MAAM,kBAAkB,GAAG,OAAO,CAAC,WAAW,KAAK,mBAAmB,CAAC;QAEvE,IACE,qBAAqB;YACrB,cAAc;YACd,eAAe;YACf,kBAAkB,EAClB,CAAC;YACD,IAAA,oCAAmB,EAAC,aAAa,EAAE,OAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YAErE,kBAAkB,GAAG,IAAI,CAAC;QAC5B,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,kBAAkB,EAAE,CAAC;QACvB,IAAA,qBAAY,EAAC;YACX,aAAa,EAAE,uBAAA,IAAI,gGAA2B,CAAC,IAAI,CAAC,IAAI,CAAC;YACzD,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,eAAe,EAAE,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,aAAa,CAAC;YAC1D,aAAa;YACb,qBAAqB,EAAE,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9D,CAAC,CAAC,KAAK,CAAC,aAAI,CAAC,CAAC;IACjB,CAAC;AACH,CAAC;IAGC,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,mDAAmD,EACnD,uBAAA,IAAI,0DAA0B,CAAC,IAAI,CAAC,IAAI,CAAC,CAC1C,CAAC;IAEF,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,sCAAsC,EACtC,CAAC,WAA4B,EAA0B,EAAE,CACvD,uBAAA,IAAI,gGAA2B,MAA/B,IAAI,EAA4B,WAAW,CAAC,CAAC,CAAC,CAAC,CAClD,CAAC;IAEF,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,+CAA+C,EAC/C,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,CACrC,CAAC;IAEF,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,6CAA6C,EAC7C,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CACnC,CAAC;AACJ,CAAC,qHAGC,WAA4B;IAE5B,MAAM,kBAAkB,GACtB,uBAAA,IAAI,+CAAe,EAAE,KAArB,IAAI,EAAkB,WAAW,CAAC;QAClC,CAAC,uBAAA,IAAI,6CAAa,CAAC,CAAC,CAAC,CAAC,uBAAA,IAAI,6CAAa,MAAjB,IAAI,EAAc,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAE9D,MAAM,eAAe,GAAG,kBAAkB,CAAC,MAAM,CAC/C,CAAC,QAAQ,EAAsC,EAAE,CAC/C,IAAA,oCAAwB,EAAC,QAAQ,CAAC,CACrC,CAAC;IAEF,OAAO,eAAe,CAAC,MAAM;QAC3B,CAAC,CAAC,eAAe;QACjB,CAAC,CAAC,IAAA,gCAAgB,EAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACvC,CAAC","sourcesContent":["import type { StateMetadata } from '@metamask/base-controller';\nimport { BaseController } from '@metamask/base-controller';\nimport type { TransactionMeta } from '@metamask/transaction-controller';\nimport type { Draft } from 'immer';\nimport { noop } from 'lodash';\n\nimport { updatePaymentToken } from './actions/update-payment-token';\nimport {\n  CONTROLLER_NAME,\n  isTransactionPayStrategy,\n  TransactionPayStrategy,\n} from './constants';\nimport { QuoteRefresher } from './helpers/QuoteRefresher';\nimport type {\n  GetDelegationTransactionCallback,\n  TransactionConfigCallback,\n  TransactionData,\n  TransactionPayControllerMessenger,\n  TransactionPayControllerOptions,\n  TransactionPayControllerState,\n  UpdatePaymentTokenRequest,\n} from './types';\nimport { getStrategyOrder } from './utils/feature-flags';\nimport { updateQuotes } from './utils/quotes';\nimport { updateSourceAmounts } from './utils/source-amounts';\nimport { pollTransactionChanges } from './utils/transaction';\n\nconst stateMetadata: StateMetadata<TransactionPayControllerState> = {\n  transactionData: {\n    includeInDebugSnapshot: false,\n    includeInStateLogs: true,\n    persist: false,\n    usedInUi: true,\n  },\n};\n\nconst getDefaultState = (): TransactionPayControllerState => ({\n  transactionData: {},\n});\n\nexport class TransactionPayController extends BaseController<\n  typeof CONTROLLER_NAME,\n  TransactionPayControllerState,\n  TransactionPayControllerMessenger\n> {\n  readonly #getDelegationTransaction: GetDelegationTransactionCallback;\n\n  readonly #getStrategy?: (\n    transaction: TransactionMeta,\n  ) => TransactionPayStrategy;\n\n  readonly #getStrategies?: (\n    transaction: TransactionMeta,\n  ) => TransactionPayStrategy[];\n\n  constructor({\n    getDelegationTransaction,\n    getStrategy,\n    getStrategies,\n    messenger,\n    state,\n  }: TransactionPayControllerOptions) {\n    super({\n      name: CONTROLLER_NAME,\n      metadata: stateMetadata,\n      messenger,\n      state: { ...getDefaultState(), ...state },\n    });\n\n    this.#getDelegationTransaction = getDelegationTransaction;\n    this.#getStrategy = getStrategy;\n    this.#getStrategies = getStrategies;\n\n    this.#registerActionHandlers();\n\n    pollTransactionChanges(\n      messenger,\n      this.#updateTransactionData.bind(this),\n      this.#removeTransactionData.bind(this),\n    );\n\n    // eslint-disable-next-line no-new\n    new QuoteRefresher({\n      getStrategies: this.#getStrategiesWithFallback.bind(this),\n      messenger,\n      updateTransactionData: this.#updateTransactionData.bind(this),\n    });\n  }\n\n  setTransactionConfig(\n    transactionId: string,\n    callback: TransactionConfigCallback,\n  ): void {\n    this.#updateTransactionData(transactionId, (transactionData) => {\n      const config = {\n        isMaxAmount: transactionData.isMaxAmount,\n        isPostQuote: transactionData.isPostQuote,\n        refundTo: transactionData.refundTo,\n      };\n\n      callback(config);\n\n      transactionData.isMaxAmount = config.isMaxAmount;\n      transactionData.isPostQuote = config.isPostQuote;\n      transactionData.refundTo = config.refundTo;\n    });\n  }\n\n  updatePaymentToken(request: UpdatePaymentTokenRequest): void {\n    updatePaymentToken(request, {\n      messenger: this.messenger,\n      updateTransactionData: this.#updateTransactionData.bind(this),\n    });\n  }\n\n  #removeTransactionData(transactionId: string): void {\n    this.update((state) => {\n      delete state.transactionData[transactionId];\n    });\n  }\n\n  #updateTransactionData(\n    transactionId: string,\n    fn: (transactionData: Draft<TransactionData>) => void,\n  ): void {\n    let shouldUpdateQuotes = false;\n\n    this.update((state) => {\n      const { transactionData } = state;\n      let current = transactionData[transactionId];\n      const originalPaymentToken = current?.paymentToken;\n      const originalTokens = current?.tokens;\n      const originalIsMaxAmount = current?.isMaxAmount;\n      const originalIsPostQuote = current?.isPostQuote;\n\n      if (!current) {\n        transactionData[transactionId] = {\n          isLoading: false,\n          tokens: [],\n        };\n\n        current = transactionData[transactionId];\n      }\n\n      fn(current);\n\n      const isPaymentTokenUpdated =\n        current.paymentToken?.address?.toLowerCase() !==\n          originalPaymentToken?.address?.toLowerCase() ||\n        current.paymentToken?.chainId !== originalPaymentToken?.chainId;\n\n      const isTokensUpdated = current.tokens !== originalTokens;\n      const isIsMaxUpdated = current.isMaxAmount !== originalIsMaxAmount;\n      const isPostQuoteUpdated = current.isPostQuote !== originalIsPostQuote;\n\n      if (\n        isPaymentTokenUpdated ||\n        isIsMaxUpdated ||\n        isTokensUpdated ||\n        isPostQuoteUpdated\n      ) {\n        updateSourceAmounts(transactionId, current as never, this.messenger);\n\n        shouldUpdateQuotes = true;\n      }\n    });\n\n    if (shouldUpdateQuotes) {\n      updateQuotes({\n        getStrategies: this.#getStrategiesWithFallback.bind(this),\n        messenger: this.messenger,\n        transactionData: this.state.transactionData[transactionId],\n        transactionId,\n        updateTransactionData: this.#updateTransactionData.bind(this),\n      }).catch(noop);\n    }\n  }\n\n  #registerActionHandlers(): void {\n    this.messenger.registerActionHandler(\n      'TransactionPayController:getDelegationTransaction',\n      this.#getDelegationTransaction.bind(this),\n    );\n\n    this.messenger.registerActionHandler(\n      'TransactionPayController:getStrategy',\n      (transaction: TransactionMeta): TransactionPayStrategy =>\n        this.#getStrategiesWithFallback(transaction)[0],\n    );\n\n    this.messenger.registerActionHandler(\n      'TransactionPayController:setTransactionConfig',\n      this.setTransactionConfig.bind(this),\n    );\n\n    this.messenger.registerActionHandler(\n      'TransactionPayController:updatePaymentToken',\n      this.updatePaymentToken.bind(this),\n    );\n  }\n\n  #getStrategiesWithFallback(\n    transaction: TransactionMeta,\n  ): TransactionPayStrategy[] {\n    const strategyCandidates: unknown[] =\n      this.#getStrategies?.(transaction) ??\n      (this.#getStrategy ? [this.#getStrategy(transaction)] : []);\n\n    const validStrategies = strategyCandidates.filter(\n      (strategy): strategy is TransactionPayStrategy =>\n        isTransactionPayStrategy(strategy),\n    );\n\n    return validStrategies.length\n      ? validStrategies\n      : getStrategyOrder(this.messenger);\n  }\n}\n"]}

package/dist/TransactionPayController.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"TransactionPayController.d.cts","sourceRoot":"","sources":["../src/TransactionPayController.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,kCAAkC;AAM3D,OAAO,EACL,eAAe,EAGhB,wBAAoB;AAErB,OAAO,KAAK,EAEV,yBAAyB,EAEzB,iCAAiC,EACjC,+BAA+B,EAC/B,6BAA6B,EAC7B,yBAAyB,EAC1B,oBAAgB;AAmBjB,qBAAa,wBAAyB,SAAQ,cAAc,CAC1D,OAAO,eAAe,EACtB,6BAA6B,EAC7B,iCAAiC,CAClC;;gBAWa,EACV,wBAAwB,EACxB,WAAW,EACX,aAAa,EACb,SAAS,EACT,KAAK,GACN,EAAE,+BAA+B;IA4BlC,oBAAoB,CAClB,aAAa,EAAE,MAAM,EACrB,QAAQ,EAAE,yBAAyB,GAClC,IAAI;IAcP,kBAAkB,CAAC,OAAO,EAAE,yBAAyB,GAAG,IAAI;CA6G7D"}
+{"version":3,"file":"TransactionPayController.d.cts","sourceRoot":"","sources":["../src/TransactionPayController.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,kCAAkC;AAM3D,OAAO,EACL,eAAe,EAGhB,wBAAoB;AAErB,OAAO,KAAK,EAEV,yBAAyB,EAEzB,iCAAiC,EACjC,+BAA+B,EAC/B,6BAA6B,EAC7B,yBAAyB,EAC1B,oBAAgB;AAmBjB,qBAAa,wBAAyB,SAAQ,cAAc,CAC1D,OAAO,eAAe,EACtB,6BAA6B,EAC7B,iCAAiC,CAClC;;gBAWa,EACV,wBAAwB,EACxB,WAAW,EACX,aAAa,EACb,SAAS,EACT,KAAK,GACN,EAAE,+BAA+B;IA4BlC,oBAAoB,CAClB,aAAa,EAAE,MAAM,EACrB,QAAQ,EAAE,yBAAyB,GAClC,IAAI;IAgBP,kBAAkB,CAAC,OAAO,EAAE,yBAAyB,GAAG,IAAI;CA6G7D"}

package/dist/TransactionPayController.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"TransactionPayController.d.mts","sourceRoot":"","sources":["../src/TransactionPayController.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,kCAAkC;AAM3D,OAAO,EACL,eAAe,EAGhB,wBAAoB;AAErB,OAAO,KAAK,EAEV,yBAAyB,EAEzB,iCAAiC,EACjC,+BAA+B,EAC/B,6BAA6B,EAC7B,yBAAyB,EAC1B,oBAAgB;AAmBjB,qBAAa,wBAAyB,SAAQ,cAAc,CAC1D,OAAO,eAAe,EACtB,6BAA6B,EAC7B,iCAAiC,CAClC;;gBAWa,EACV,wBAAwB,EACxB,WAAW,EACX,aAAa,EACb,SAAS,EACT,KAAK,GACN,EAAE,+BAA+B;IA4BlC,oBAAoB,CAClB,aAAa,EAAE,MAAM,EACrB,QAAQ,EAAE,yBAAyB,GAClC,IAAI;IAcP,kBAAkB,CAAC,OAAO,EAAE,yBAAyB,GAAG,IAAI;CA6G7D"}
+{"version":3,"file":"TransactionPayController.d.mts","sourceRoot":"","sources":["../src/TransactionPayController.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,kCAAkC;AAM3D,OAAO,EACL,eAAe,EAGhB,wBAAoB;AAErB,OAAO,KAAK,EAEV,yBAAyB,EAEzB,iCAAiC,EACjC,+BAA+B,EAC/B,6BAA6B,EAC7B,yBAAyB,EAC1B,oBAAgB;AAmBjB,qBAAa,wBAAyB,SAAQ,cAAc,CAC1D,OAAO,eAAe,EACtB,6BAA6B,EAC7B,iCAAiC,CAClC;;gBAWa,EACV,wBAAwB,EACxB,WAAW,EACX,aAAa,EACb,SAAS,EACT,KAAK,GACN,EAAE,+BAA+B;IA4BlC,oBAAoB,CAClB,aAAa,EAAE,MAAM,EACrB,QAAQ,EAAE,yBAAyB,GAClC,IAAI;IAgBP,kBAAkB,CAAC,OAAO,EAAE,yBAAyB,GAAG,IAAI;CA6G7D"}

package/dist/TransactionPayController.mjs

@@ -60,10 +60,12 @@ export class TransactionPayController extends BaseController {
             const config = {
                 isMaxAmount: transactionData.isMaxAmount,
                 isPostQuote: transactionData.isPostQuote,
+                refundTo: transactionData.refundTo,
             };
             callback(config);
             transactionData.isMaxAmount = config.isMaxAmount;
             transactionData.isPostQuote = config.isPostQuote;
+            transactionData.refundTo = config.refundTo;
         });
     }
     updatePaymentToken(request) {

package/dist/TransactionPayController.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"TransactionPayController.mjs","sourceRoot":"","sources":["../src/TransactionPayController.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,OAAO,EAAE,cAAc,EAAE,kCAAkC;;;AAK3D,OAAO,EAAE,kBAAkB,EAAE,2CAAuC;AACpE,OAAO,EACL,eAAe,EACf,wBAAwB,EACxB,sBAAsB,EACvB,wBAAoB;AACrB,OAAO,EAAE,cAAc,EAAE,qCAAiC;AAU1D,OAAO,EAAE,gBAAgB,EAAE,kCAA8B;AACzD,OAAO,EAAE,YAAY,EAAE,2BAAuB;AAC9C,OAAO,EAAE,mBAAmB,EAAE,mCAA+B;AAC7D,OAAO,EAAE,sBAAsB,EAAE,gCAA4B;AAE7D,MAAM,aAAa,GAAiD;IAClE,eAAe,EAAE;QACf,sBAAsB,EAAE,KAAK;QAC7B,kBAAkB,EAAE,IAAI;QACxB,OAAO,EAAE,KAAK;QACd,QAAQ,EAAE,IAAI;KACf;CACF,CAAC;AAEF,MAAM,eAAe,GAAG,GAAkC,EAAE,CAAC,CAAC;IAC5D,eAAe,EAAE,EAAE;CACpB,CAAC,CAAC;AAEH,MAAM,OAAO,wBAAyB,SAAQ,cAI7C;IAWC,YAAY,EACV,wBAAwB,EACxB,WAAW,EACX,aAAa,EACb,SAAS,EACT,KAAK,GAC2B;QAChC,KAAK,CAAC;YACJ,IAAI,EAAE,eAAe;YACrB,QAAQ,EAAE,aAAa;YACvB,SAAS;YACT,KAAK,EAAE,EAAE,GAAG,eAAe,EAAE,EAAE,GAAG,KAAK,EAAE;SAC1C,CAAC,CAAC;;QAtBI,qEAA4D;QAE5D,wDAEmB;QAEnB,0DAEqB;QAgB5B,uBAAA,IAAI,sDAA6B,wBAAwB,MAAA,CAAC;QAC1D,uBAAA,IAAI,yCAAgB,WAAW,MAAA,CAAC;QAChC,uBAAA,IAAI,2CAAkB,aAAa,MAAA,CAAC;QAEpC,uBAAA,IAAI,6FAAwB,MAA5B,IAAI,CAA0B,CAAC;QAE/B,sBAAsB,CACpB,SAAS,EACT,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,EACtC,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,CACvC,CAAC;QAEF,kCAAkC;QAClC,IAAI,cAAc,CAAC;YACjB,aAAa,EAAE,uBAAA,IAAI,gGAA2B,CAAC,IAAI,CAAC,IAAI,CAAC;YACzD,SAAS;YACT,qBAAqB,EAAE,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9D,CAAC,CAAC;IACL,CAAC;IAED,oBAAoB,CAClB,aAAqB,EACrB,QAAmC;QAEnC,uBAAA,IAAI,4FAAuB,MAA3B,IAAI,EAAwB,aAAa,EAAE,CAAC,eAAe,EAAE,EAAE;YAC7D,MAAM,MAAM,GAAG;gBACb,WAAW,EAAE,eAAe,CAAC,WAAW;gBACxC,WAAW,EAAE,eAAe,CAAC,WAAW;aACzC,CAAC;YAEF,QAAQ,CAAC,MAAM,CAAC,CAAC;YAEjB,eAAe,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;YACjD,eAAe,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC;IAED,kBAAkB,CAAC,OAAkC;QACnD,kBAAkB,CAAC,OAAO,EAAE;YAC1B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,qBAAqB,EAAE,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9D,CAAC,CAAC;IACL,CAAC;CAwGF;oVAtGwB,aAAqB;IAC1C,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QACpB,OAAO,KAAK,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,6GAGC,aAAqB,EACrB,EAAqD;IAErD,IAAI,kBAAkB,GAAG,KAAK,CAAC;IAE/B,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QACpB,MAAM,EAAE,eAAe,EAAE,GAAG,KAAK,CAAC;QAClC,IAAI,OAAO,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;QAC7C,MAAM,oBAAoB,GAAG,OAAO,EAAE,YAAY,CAAC;QACnD,MAAM,cAAc,GAAG,OAAO,EAAE,MAAM,CAAC;QACvC,MAAM,mBAAmB,GAAG,OAAO,EAAE,WAAW,CAAC;QACjD,MAAM,mBAAmB,GAAG,OAAO,EAAE,WAAW,CAAC;QAEjD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,eAAe,CAAC,aAAa,CAAC,GAAG;gBAC/B,SAAS,EAAE,KAAK;gBAChB,MAAM,EAAE,EAAE;aACX,CAAC;YAEF,OAAO,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;QAC3C,CAAC;QAED,EAAE,CAAC,OAAO,CAAC,CAAC;QAEZ,MAAM,qBAAqB,GACzB,OAAO,CAAC,YAAY,EAAE,OAAO,EAAE,WAAW,EAAE;YAC1C,oBAAoB,EAAE,OAAO,EAAE,WAAW,EAAE;YAC9C,OAAO,CAAC,YAAY,EAAE,OAAO,KAAK,oBAAoB,EAAE,OAAO,CAAC;QAElE,MAAM,eAAe,GAAG,OAAO,CAAC,MAAM,KAAK,cAAc,CAAC;QAC1D,MAAM,cAAc,GAAG,OAAO,CAAC,WAAW,KAAK,mBAAmB,CAAC;QACnE,MAAM,kBAAkB,GAAG,OAAO,CAAC,WAAW,KAAK,mBAAmB,CAAC;QAEvE,IACE,qBAAqB;YACrB,cAAc;YACd,eAAe;YACf,kBAAkB,EAClB,CAAC;YACD,mBAAmB,CAAC,aAAa,EAAE,OAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YAErE,kBAAkB,GAAG,IAAI,CAAC;QAC5B,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,kBAAkB,EAAE,CAAC;QACvB,YAAY,CAAC;YACX,aAAa,EAAE,uBAAA,IAAI,gGAA2B,CAAC,IAAI,CAAC,IAAI,CAAC;YACzD,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,eAAe,EAAE,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,aAAa,CAAC;YAC1D,aAAa;YACb,qBAAqB,EAAE,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9D,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACjB,CAAC;AACH,CAAC;IAGC,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,mDAAmD,EACnD,uBAAA,IAAI,0DAA0B,CAAC,IAAI,CAAC,IAAI,CAAC,CAC1C,CAAC;IAEF,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,sCAAsC,EACtC,CAAC,WAA4B,EAA0B,EAAE,CACvD,uBAAA,IAAI,gGAA2B,MAA/B,IAAI,EAA4B,WAAW,CAAC,CAAC,CAAC,CAAC,CAClD,CAAC;IAEF,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,+CAA+C,EAC/C,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,CACrC,CAAC;IAEF,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,6CAA6C,EAC7C,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CACnC,CAAC;AACJ,CAAC,qHAGC,WAA4B;IAE5B,MAAM,kBAAkB,GACtB,uBAAA,IAAI,+CAAe,EAAE,KAArB,IAAI,EAAkB,WAAW,CAAC;QAClC,CAAC,uBAAA,IAAI,6CAAa,CAAC,CAAC,CAAC,CAAC,uBAAA,IAAI,6CAAa,MAAjB,IAAI,EAAc,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAE9D,MAAM,eAAe,GAAG,kBAAkB,CAAC,MAAM,CAC/C,CAAC,QAAQ,EAAsC,EAAE,CAC/C,wBAAwB,CAAC,QAAQ,CAAC,CACrC,CAAC;IAEF,OAAO,eAAe,CAAC,MAAM;QAC3B,CAAC,CAAC,eAAe;QACjB,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACvC,CAAC","sourcesContent":["import type { StateMetadata } from '@metamask/base-controller';\nimport { BaseController } from '@metamask/base-controller';\nimport type { TransactionMeta } from '@metamask/transaction-controller';\nimport type { Draft } from 'immer';\nimport { noop } from 'lodash';\n\nimport { updatePaymentToken } from './actions/update-payment-token';\nimport {\n  CONTROLLER_NAME,\n  isTransactionPayStrategy,\n  TransactionPayStrategy,\n} from './constants';\nimport { QuoteRefresher } from './helpers/QuoteRefresher';\nimport type {\n  GetDelegationTransactionCallback,\n  TransactionConfigCallback,\n  TransactionData,\n  TransactionPayControllerMessenger,\n  TransactionPayControllerOptions,\n  TransactionPayControllerState,\n  UpdatePaymentTokenRequest,\n} from './types';\nimport { getStrategyOrder } from './utils/feature-flags';\nimport { updateQuotes } from './utils/quotes';\nimport { updateSourceAmounts } from './utils/source-amounts';\nimport { pollTransactionChanges } from './utils/transaction';\n\nconst stateMetadata: StateMetadata<TransactionPayControllerState> = {\n  transactionData: {\n    includeInDebugSnapshot: false,\n    includeInStateLogs: true,\n    persist: false,\n    usedInUi: true,\n  },\n};\n\nconst getDefaultState = (): TransactionPayControllerState => ({\n  transactionData: {},\n});\n\nexport class TransactionPayController extends BaseController<\n  typeof CONTROLLER_NAME,\n  TransactionPayControllerState,\n  TransactionPayControllerMessenger\n> {\n  readonly #getDelegationTransaction: GetDelegationTransactionCallback;\n\n  readonly #getStrategy?: (\n    transaction: TransactionMeta,\n  ) => TransactionPayStrategy;\n\n  readonly #getStrategies?: (\n    transaction: TransactionMeta,\n  ) => TransactionPayStrategy[];\n\n  constructor({\n    getDelegationTransaction,\n    getStrategy,\n    getStrategies,\n    messenger,\n    state,\n  }: TransactionPayControllerOptions) {\n    super({\n      name: CONTROLLER_NAME,\n      metadata: stateMetadata,\n      messenger,\n      state: { ...getDefaultState(), ...state },\n    });\n\n    this.#getDelegationTransaction = getDelegationTransaction;\n    this.#getStrategy = getStrategy;\n    this.#getStrategies = getStrategies;\n\n    this.#registerActionHandlers();\n\n    pollTransactionChanges(\n      messenger,\n      this.#updateTransactionData.bind(this),\n      this.#removeTransactionData.bind(this),\n    );\n\n    // eslint-disable-next-line no-new\n    new QuoteRefresher({\n      getStrategies: this.#getStrategiesWithFallback.bind(this),\n      messenger,\n      updateTransactionData: this.#updateTransactionData.bind(this),\n    });\n  }\n\n  setTransactionConfig(\n    transactionId: string,\n    callback: TransactionConfigCallback,\n  ): void {\n    this.#updateTransactionData(transactionId, (transactionData) => {\n      const config = {\n        isMaxAmount: transactionData.isMaxAmount,\n        isPostQuote: transactionData.isPostQuote,\n      };\n\n      callback(config);\n\n      transactionData.isMaxAmount = config.isMaxAmount;\n      transactionData.isPostQuote = config.isPostQuote;\n    });\n  }\n\n  updatePaymentToken(request: UpdatePaymentTokenRequest): void {\n    updatePaymentToken(request, {\n      messenger: this.messenger,\n      updateTransactionData: this.#updateTransactionData.bind(this),\n    });\n  }\n\n  #removeTransactionData(transactionId: string): void {\n    this.update((state) => {\n      delete state.transactionData[transactionId];\n    });\n  }\n\n  #updateTransactionData(\n    transactionId: string,\n    fn: (transactionData: Draft<TransactionData>) => void,\n  ): void {\n    let shouldUpdateQuotes = false;\n\n    this.update((state) => {\n      const { transactionData } = state;\n      let current = transactionData[transactionId];\n      const originalPaymentToken = current?.paymentToken;\n      const originalTokens = current?.tokens;\n      const originalIsMaxAmount = current?.isMaxAmount;\n      const originalIsPostQuote = current?.isPostQuote;\n\n      if (!current) {\n        transactionData[transactionId] = {\n          isLoading: false,\n          tokens: [],\n        };\n\n        current = transactionData[transactionId];\n      }\n\n      fn(current);\n\n      const isPaymentTokenUpdated =\n        current.paymentToken?.address?.toLowerCase() !==\n          originalPaymentToken?.address?.toLowerCase() ||\n        current.paymentToken?.chainId !== originalPaymentToken?.chainId;\n\n      const isTokensUpdated = current.tokens !== originalTokens;\n      const isIsMaxUpdated = current.isMaxAmount !== originalIsMaxAmount;\n      const isPostQuoteUpdated = current.isPostQuote !== originalIsPostQuote;\n\n      if (\n        isPaymentTokenUpdated ||\n        isIsMaxUpdated ||\n        isTokensUpdated ||\n        isPostQuoteUpdated\n      ) {\n        updateSourceAmounts(transactionId, current as never, this.messenger);\n\n        shouldUpdateQuotes = true;\n      }\n    });\n\n    if (shouldUpdateQuotes) {\n      updateQuotes({\n        getStrategies: this.#getStrategiesWithFallback.bind(this),\n        messenger: this.messenger,\n        transactionData: this.state.transactionData[transactionId],\n        transactionId,\n        updateTransactionData: this.#updateTransactionData.bind(this),\n      }).catch(noop);\n    }\n  }\n\n  #registerActionHandlers(): void {\n    this.messenger.registerActionHandler(\n      'TransactionPayController:getDelegationTransaction',\n      this.#getDelegationTransaction.bind(this),\n    );\n\n    this.messenger.registerActionHandler(\n      'TransactionPayController:getStrategy',\n      (transaction: TransactionMeta): TransactionPayStrategy =>\n        this.#getStrategiesWithFallback(transaction)[0],\n    );\n\n    this.messenger.registerActionHandler(\n      'TransactionPayController:setTransactionConfig',\n      this.setTransactionConfig.bind(this),\n    );\n\n    this.messenger.registerActionHandler(\n      'TransactionPayController:updatePaymentToken',\n      this.updatePaymentToken.bind(this),\n    );\n  }\n\n  #getStrategiesWithFallback(\n    transaction: TransactionMeta,\n  ): TransactionPayStrategy[] {\n    const strategyCandidates: unknown[] =\n      this.#getStrategies?.(transaction) ??\n      (this.#getStrategy ? [this.#getStrategy(transaction)] : []);\n\n    const validStrategies = strategyCandidates.filter(\n      (strategy): strategy is TransactionPayStrategy =>\n        isTransactionPayStrategy(strategy),\n    );\n\n    return validStrategies.length\n      ? validStrategies\n      : getStrategyOrder(this.messenger);\n  }\n}\n"]}
+{"version":3,"file":"TransactionPayController.mjs","sourceRoot":"","sources":["../src/TransactionPayController.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,OAAO,EAAE,cAAc,EAAE,kCAAkC;;;AAK3D,OAAO,EAAE,kBAAkB,EAAE,2CAAuC;AACpE,OAAO,EACL,eAAe,EACf,wBAAwB,EACxB,sBAAsB,EACvB,wBAAoB;AACrB,OAAO,EAAE,cAAc,EAAE,qCAAiC;AAU1D,OAAO,EAAE,gBAAgB,EAAE,kCAA8B;AACzD,OAAO,EAAE,YAAY,EAAE,2BAAuB;AAC9C,OAAO,EAAE,mBAAmB,EAAE,mCAA+B;AAC7D,OAAO,EAAE,sBAAsB,EAAE,gCAA4B;AAE7D,MAAM,aAAa,GAAiD;IAClE,eAAe,EAAE;QACf,sBAAsB,EAAE,KAAK;QAC7B,kBAAkB,EAAE,IAAI;QACxB,OAAO,EAAE,KAAK;QACd,QAAQ,EAAE,IAAI;KACf;CACF,CAAC;AAEF,MAAM,eAAe,GAAG,GAAkC,EAAE,CAAC,CAAC;IAC5D,eAAe,EAAE,EAAE;CACpB,CAAC,CAAC;AAEH,MAAM,OAAO,wBAAyB,SAAQ,cAI7C;IAWC,YAAY,EACV,wBAAwB,EACxB,WAAW,EACX,aAAa,EACb,SAAS,EACT,KAAK,GAC2B;QAChC,KAAK,CAAC;YACJ,IAAI,EAAE,eAAe;YACrB,QAAQ,EAAE,aAAa;YACvB,SAAS;YACT,KAAK,EAAE,EAAE,GAAG,eAAe,EAAE,EAAE,GAAG,KAAK,EAAE;SAC1C,CAAC,CAAC;;QAtBI,qEAA4D;QAE5D,wDAEmB;QAEnB,0DAEqB;QAgB5B,uBAAA,IAAI,sDAA6B,wBAAwB,MAAA,CAAC;QAC1D,uBAAA,IAAI,yCAAgB,WAAW,MAAA,CAAC;QAChC,uBAAA,IAAI,2CAAkB,aAAa,MAAA,CAAC;QAEpC,uBAAA,IAAI,6FAAwB,MAA5B,IAAI,CAA0B,CAAC;QAE/B,sBAAsB,CACpB,SAAS,EACT,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,EACtC,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,CACvC,CAAC;QAEF,kCAAkC;QAClC,IAAI,cAAc,CAAC;YACjB,aAAa,EAAE,uBAAA,IAAI,gGAA2B,CAAC,IAAI,CAAC,IAAI,CAAC;YACzD,SAAS;YACT,qBAAqB,EAAE,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9D,CAAC,CAAC;IACL,CAAC;IAED,oBAAoB,CAClB,aAAqB,EACrB,QAAmC;QAEnC,uBAAA,IAAI,4FAAuB,MAA3B,IAAI,EAAwB,aAAa,EAAE,CAAC,eAAe,EAAE,EAAE;YAC7D,MAAM,MAAM,GAAG;gBACb,WAAW,EAAE,eAAe,CAAC,WAAW;gBACxC,WAAW,EAAE,eAAe,CAAC,WAAW;gBACxC,QAAQ,EAAE,eAAe,CAAC,QAAQ;aACnC,CAAC;YAEF,QAAQ,CAAC,MAAM,CAAC,CAAC;YAEjB,eAAe,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;YACjD,eAAe,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;YACjD,eAAe,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC;IAED,kBAAkB,CAAC,OAAkC;QACnD,kBAAkB,CAAC,OAAO,EAAE;YAC1B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,qBAAqB,EAAE,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9D,CAAC,CAAC;IACL,CAAC;CAwGF;oVAtGwB,aAAqB;IAC1C,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QACpB,OAAO,KAAK,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,6GAGC,aAAqB,EACrB,EAAqD;IAErD,IAAI,kBAAkB,GAAG,KAAK,CAAC;IAE/B,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QACpB,MAAM,EAAE,eAAe,EAAE,GAAG,KAAK,CAAC;QAClC,IAAI,OAAO,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;QAC7C,MAAM,oBAAoB,GAAG,OAAO,EAAE,YAAY,CAAC;QACnD,MAAM,cAAc,GAAG,OAAO,EAAE,MAAM,CAAC;QACvC,MAAM,mBAAmB,GAAG,OAAO,EAAE,WAAW,CAAC;QACjD,MAAM,mBAAmB,GAAG,OAAO,EAAE,WAAW,CAAC;QAEjD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,eAAe,CAAC,aAAa,CAAC,GAAG;gBAC/B,SAAS,EAAE,KAAK;gBAChB,MAAM,EAAE,EAAE;aACX,CAAC;YAEF,OAAO,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;QAC3C,CAAC;QAED,EAAE,CAAC,OAAO,CAAC,CAAC;QAEZ,MAAM,qBAAqB,GACzB,OAAO,CAAC,YAAY,EAAE,OAAO,EAAE,WAAW,EAAE;YAC1C,oBAAoB,EAAE,OAAO,EAAE,WAAW,EAAE;YAC9C,OAAO,CAAC,YAAY,EAAE,OAAO,KAAK,oBAAoB,EAAE,OAAO,CAAC;QAElE,MAAM,eAAe,GAAG,OAAO,CAAC,MAAM,KAAK,cAAc,CAAC;QAC1D,MAAM,cAAc,GAAG,OAAO,CAAC,WAAW,KAAK,mBAAmB,CAAC;QACnE,MAAM,kBAAkB,GAAG,OAAO,CAAC,WAAW,KAAK,mBAAmB,CAAC;QAEvE,IACE,qBAAqB;YACrB,cAAc;YACd,eAAe;YACf,kBAAkB,EAClB,CAAC;YACD,mBAAmB,CAAC,aAAa,EAAE,OAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YAErE,kBAAkB,GAAG,IAAI,CAAC;QAC5B,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,kBAAkB,EAAE,CAAC;QACvB,YAAY,CAAC;YACX,aAAa,EAAE,uBAAA,IAAI,gGAA2B,CAAC,IAAI,CAAC,IAAI,CAAC;YACzD,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,eAAe,EAAE,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,aAAa,CAAC;YAC1D,aAAa;YACb,qBAAqB,EAAE,uBAAA,IAAI,4FAAuB,CAAC,IAAI,CAAC,IAAI,CAAC;SAC9D,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACjB,CAAC;AACH,CAAC;IAGC,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,mDAAmD,EACnD,uBAAA,IAAI,0DAA0B,CAAC,IAAI,CAAC,IAAI,CAAC,CAC1C,CAAC;IAEF,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,sCAAsC,EACtC,CAAC,WAA4B,EAA0B,EAAE,CACvD,uBAAA,IAAI,gGAA2B,MAA/B,IAAI,EAA4B,WAAW,CAAC,CAAC,CAAC,CAAC,CAClD,CAAC;IAEF,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,+CAA+C,EAC/C,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,CACrC,CAAC;IAEF,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,6CAA6C,EAC7C,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CACnC,CAAC;AACJ,CAAC,qHAGC,WAA4B;IAE5B,MAAM,kBAAkB,GACtB,uBAAA,IAAI,+CAAe,EAAE,KAArB,IAAI,EAAkB,WAAW,CAAC;QAClC,CAAC,uBAAA,IAAI,6CAAa,CAAC,CAAC,CAAC,CAAC,uBAAA,IAAI,6CAAa,MAAjB,IAAI,EAAc,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAE9D,MAAM,eAAe,GAAG,kBAAkB,CAAC,MAAM,CAC/C,CAAC,QAAQ,EAAsC,EAAE,CAC/C,wBAAwB,CAAC,QAAQ,CAAC,CACrC,CAAC;IAEF,OAAO,eAAe,CAAC,MAAM;QAC3B,CAAC,CAAC,eAAe;QACjB,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACvC,CAAC","sourcesContent":["import type { StateMetadata } from '@metamask/base-controller';\nimport { BaseController } from '@metamask/base-controller';\nimport type { TransactionMeta } from '@metamask/transaction-controller';\nimport type { Draft } from 'immer';\nimport { noop } from 'lodash';\n\nimport { updatePaymentToken } from './actions/update-payment-token';\nimport {\n  CONTROLLER_NAME,\n  isTransactionPayStrategy,\n  TransactionPayStrategy,\n} from './constants';\nimport { QuoteRefresher } from './helpers/QuoteRefresher';\nimport type {\n  GetDelegationTransactionCallback,\n  TransactionConfigCallback,\n  TransactionData,\n  TransactionPayControllerMessenger,\n  TransactionPayControllerOptions,\n  TransactionPayControllerState,\n  UpdatePaymentTokenRequest,\n} from './types';\nimport { getStrategyOrder } from './utils/feature-flags';\nimport { updateQuotes } from './utils/quotes';\nimport { updateSourceAmounts } from './utils/source-amounts';\nimport { pollTransactionChanges } from './utils/transaction';\n\nconst stateMetadata: StateMetadata<TransactionPayControllerState> = {\n  transactionData: {\n    includeInDebugSnapshot: false,\n    includeInStateLogs: true,\n    persist: false,\n    usedInUi: true,\n  },\n};\n\nconst getDefaultState = (): TransactionPayControllerState => ({\n  transactionData: {},\n});\n\nexport class TransactionPayController extends BaseController<\n  typeof CONTROLLER_NAME,\n  TransactionPayControllerState,\n  TransactionPayControllerMessenger\n> {\n  readonly #getDelegationTransaction: GetDelegationTransactionCallback;\n\n  readonly #getStrategy?: (\n    transaction: TransactionMeta,\n  ) => TransactionPayStrategy;\n\n  readonly #getStrategies?: (\n    transaction: TransactionMeta,\n  ) => TransactionPayStrategy[];\n\n  constructor({\n    getDelegationTransaction,\n    getStrategy,\n    getStrategies,\n    messenger,\n    state,\n  }: TransactionPayControllerOptions) {\n    super({\n      name: CONTROLLER_NAME,\n      metadata: stateMetadata,\n      messenger,\n      state: { ...getDefaultState(), ...state },\n    });\n\n    this.#getDelegationTransaction = getDelegationTransaction;\n    this.#getStrategy = getStrategy;\n    this.#getStrategies = getStrategies;\n\n    this.#registerActionHandlers();\n\n    pollTransactionChanges(\n      messenger,\n      this.#updateTransactionData.bind(this),\n      this.#removeTransactionData.bind(this),\n    );\n\n    // eslint-disable-next-line no-new\n    new QuoteRefresher({\n      getStrategies: this.#getStrategiesWithFallback.bind(this),\n      messenger,\n      updateTransactionData: this.#updateTransactionData.bind(this),\n    });\n  }\n\n  setTransactionConfig(\n    transactionId: string,\n    callback: TransactionConfigCallback,\n  ): void {\n    this.#updateTransactionData(transactionId, (transactionData) => {\n      const config = {\n        isMaxAmount: transactionData.isMaxAmount,\n        isPostQuote: transactionData.isPostQuote,\n        refundTo: transactionData.refundTo,\n      };\n\n      callback(config);\n\n      transactionData.isMaxAmount = config.isMaxAmount;\n      transactionData.isPostQuote = config.isPostQuote;\n      transactionData.refundTo = config.refundTo;\n    });\n  }\n\n  updatePaymentToken(request: UpdatePaymentTokenRequest): void {\n    updatePaymentToken(request, {\n      messenger: this.messenger,\n      updateTransactionData: this.#updateTransactionData.bind(this),\n    });\n  }\n\n  #removeTransactionData(transactionId: string): void {\n    this.update((state) => {\n      delete state.transactionData[transactionId];\n    });\n  }\n\n  #updateTransactionData(\n    transactionId: string,\n    fn: (transactionData: Draft<TransactionData>) => void,\n  ): void {\n    let shouldUpdateQuotes = false;\n\n    this.update((state) => {\n      const { transactionData } = state;\n      let current = transactionData[transactionId];\n      const originalPaymentToken = current?.paymentToken;\n      const originalTokens = current?.tokens;\n      const originalIsMaxAmount = current?.isMaxAmount;\n      const originalIsPostQuote = current?.isPostQuote;\n\n      if (!current) {\n        transactionData[transactionId] = {\n          isLoading: false,\n          tokens: [],\n        };\n\n        current = transactionData[transactionId];\n      }\n\n      fn(current);\n\n      const isPaymentTokenUpdated =\n        current.paymentToken?.address?.toLowerCase() !==\n          originalPaymentToken?.address?.toLowerCase() ||\n        current.paymentToken?.chainId !== originalPaymentToken?.chainId;\n\n      const isTokensUpdated = current.tokens !== originalTokens;\n      const isIsMaxUpdated = current.isMaxAmount !== originalIsMaxAmount;\n      const isPostQuoteUpdated = current.isPostQuote !== originalIsPostQuote;\n\n      if (\n        isPaymentTokenUpdated ||\n        isIsMaxUpdated ||\n        isTokensUpdated ||\n        isPostQuoteUpdated\n      ) {\n        updateSourceAmounts(transactionId, current as never, this.messenger);\n\n        shouldUpdateQuotes = true;\n      }\n    });\n\n    if (shouldUpdateQuotes) {\n      updateQuotes({\n        getStrategies: this.#getStrategiesWithFallback.bind(this),\n        messenger: this.messenger,\n        transactionData: this.state.transactionData[transactionId],\n        transactionId,\n        updateTransactionData: this.#updateTransactionData.bind(this),\n      }).catch(noop);\n    }\n  }\n\n  #registerActionHandlers(): void {\n    this.messenger.registerActionHandler(\n      'TransactionPayController:getDelegationTransaction',\n      this.#getDelegationTransaction.bind(this),\n    );\n\n    this.messenger.registerActionHandler(\n      'TransactionPayController:getStrategy',\n      (transaction: TransactionMeta): TransactionPayStrategy =>\n        this.#getStrategiesWithFallback(transaction)[0],\n    );\n\n    this.messenger.registerActionHandler(\n      'TransactionPayController:setTransactionConfig',\n      this.setTransactionConfig.bind(this),\n    );\n\n    this.messenger.registerActionHandler(\n      'TransactionPayController:updatePaymentToken',\n      this.updatePaymentToken.bind(this),\n    );\n  }\n\n  #getStrategiesWithFallback(\n    transaction: TransactionMeta,\n  ): TransactionPayStrategy[] {\n    const strategyCandidates: unknown[] =\n      this.#getStrategies?.(transaction) ??\n      (this.#getStrategy ? [this.#getStrategy(transaction)] : []);\n\n    const validStrategies = strategyCandidates.filter(\n      (strategy): strategy is TransactionPayStrategy =>\n        isTransactionPayStrategy(strategy),\n    );\n\n    return validStrategies.length\n      ? validStrategies\n      : getStrategyOrder(this.messenger);\n  }\n}\n"]}

package/dist/strategy/relay/gas-station.cjs

@@ -0,0 +1,92 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getGasStationCostInSourceTokenRaw = exports.getGasStationEligibility = void 0;
+const controller_utils_1 = require("@metamask/controller-utils");
+const utils_1 = require("@metamask/utils");
+const bignumber_js_1 = require("bignumber.js");
+const logger_1 = require("../../logger.cjs");
+const feature_flags_1 = require("../../utils/feature-flags.cjs");
+const gas_1 = require("../../utils/gas.cjs");
+const log = (0, utils_1.createModuleLogger)(logger_1.projectLogger, 'relay-gas-station');
+function getGasStationEligibility(messenger, sourceChainId) {
+    const { relayDisabledGasStationChains } = (0, feature_flags_1.getFeatureFlags)(messenger);
+    const supportedChains = (0, feature_flags_1.getEIP7702SupportedChains)(messenger);
+    const chainSupportsGasStation = supportedChains.some((supportedChainId) => supportedChainId.toLowerCase() === sourceChainId.toLowerCase());
+    const isDisabledChain = relayDisabledGasStationChains.includes(sourceChainId);
+    return {
+        chainSupportsGasStation,
+        isDisabledChain,
+        isEligible: !isDisabledChain && chainSupportsGasStation,
+    };
+}
+exports.getGasStationEligibility = getGasStationEligibility;
+async function getGasStationCostInSourceTokenRaw({ firstStepData, messenger, request, totalGasEstimate, totalItemCount, }) {
+    const { data, to, value } = firstStepData;
+    const { from, sourceChainId, sourceTokenAddress } = request;
+    let gasFeeTokens;
+    try {
+        gasFeeTokens = await messenger.call('TransactionController:getGasFeeTokens', {
+            chainId: sourceChainId,
+            data,
+            from,
+            to,
+            value: (0, controller_utils_1.toHex)(value ?? '0'),
+        });
+    }
+    catch (error) {
+        log('Failed to estimate gas fee tokens', {
+            error,
+            sourceChainId,
+        });
+        return undefined;
+    }
+    const gasFeeToken = gasFeeTokens.find((singleGasFeeToken) => singleGasFeeToken.tokenAddress.toLowerCase() ===
+        sourceTokenAddress.toLowerCase());
+    if (!gasFeeToken) {
+        log('No matching source token in gas fee token estimate', {
+            sourceTokenAddress,
+            sourceChainId,
+        });
+        return undefined;
+    }
+    const gasFeeTokenWithNormalizedAmount = {
+        ...gasFeeToken,
+        amount: (0, controller_utils_1.toHex)(getNormalizedGasFeeTokenAmount({
+            gasFeeToken,
+            totalGasEstimate,
+            totalItemCount,
+        })),
+    };
+    const gasFeeTokenCost = (0, gas_1.calculateGasFeeTokenCost)({
+        chainId: sourceChainId,
+        gasFeeToken: gasFeeTokenWithNormalizedAmount,
+        messenger,
+    });
+    if (!gasFeeTokenCost) {
+        log('Unable to calculate gas fee token cost using fiat rates', {
+            sourceTokenAddress,
+            sourceChainId,
+        });
+        return undefined;
+    }
+    log('Estimated gas station cost for source token', {
+        amount: gasFeeTokenCost.raw,
+        sourceTokenAddress,
+        sourceChainId,
+    });
+    return gasFeeTokenCost;
+}
+exports.getGasStationCostInSourceTokenRaw = getGasStationCostInSourceTokenRaw;
+function getNormalizedGasFeeTokenAmount({ gasFeeToken, totalGasEstimate, totalItemCount, }) {
+    let amount = new bignumber_js_1.BigNumber(gasFeeToken.amount);
+    if (totalItemCount > 1) {
+        const gas = new bignumber_js_1.BigNumber(gasFeeToken.gas);
+        const gasFeeAmount = new bignumber_js_1.BigNumber(gasFeeToken.amount);
+        if (totalGasEstimate > 0 && gas.isGreaterThan(0)) {
+            const gasRate = gasFeeAmount.dividedBy(gas);
+            amount = gasRate.multipliedBy(totalGasEstimate);
+        }
+    }
+    return amount.integerValue(bignumber_js_1.BigNumber.ROUND_CEIL).toFixed(0);
+}
+//# sourceMappingURL=gas-station.cjs.map

package/dist/strategy/relay/gas-station.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"gas-station.cjs","sourceRoot":"","sources":["../../../src/strategy/relay/gas-station.ts"],"names":[],"mappings":";;;AAAA,iEAAmD;AAGnD,2CAAqD;AACrD,+CAAyC;AAEzC,6CAA6C;AAM7C,iEAGmC;AACnC,6CAA2D;AAE3D,MAAM,GAAG,GAAG,IAAA,0BAAkB,EAAC,sBAAa,EAAE,mBAAmB,CAAC,CAAC;AAoBnE,SAAgB,wBAAwB,CACtC,SAA4C,EAC5C,aAA4C;IAE5C,MAAM,EAAE,6BAA6B,EAAE,GAAG,IAAA,+BAAe,EAAC,SAAS,CAAC,CAAC;IACrE,MAAM,eAAe,GAAG,IAAA,yCAAyB,EAAC,SAAS,CAAC,CAAC;IAC7D,MAAM,uBAAuB,GAAG,eAAe,CAAC,IAAI,CAClD,CAAC,gBAAgB,EAAE,EAAE,CACnB,gBAAgB,CAAC,WAAW,EAAE,KAAK,aAAa,CAAC,WAAW,EAAE,CACjE,CAAC;IAEF,MAAM,eAAe,GAAG,6BAA6B,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IAE9E,OAAO;QACL,uBAAuB;QACvB,eAAe;QACf,UAAU,EAAE,CAAC,eAAe,IAAI,uBAAuB;KACxD,CAAC;AACJ,CAAC;AAlBD,4DAkBC;AAEM,KAAK,UAAU,iCAAiC,CAAC,EACtD,aAAa,EACb,SAAS,EACT,OAAO,EACP,gBAAgB,EAChB,cAAc,GACO;IACrB,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,aAAa,CAAC;IAC1C,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,kBAAkB,EAAE,GAAG,OAAO,CAAC;IAE5D,IAAI,YAA2B,CAAC;IAEhC,IAAI,CAAC;QACH,YAAY,GAAG,MAAM,SAAS,CAAC,IAAI,CACjC,uCAAuC,EACvC;YACE,OAAO,EAAE,aAAa;YACtB,IAAI;YACJ,IAAI;YACJ,EAAE;YACF,KAAK,EAAE,IAAA,wBAAK,EAAC,KAAK,IAAI,GAAG,CAAC;SAC3B,CACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,mCAAmC,EAAE;YACvC,KAAK;YACL,aAAa;SACd,CAAC,CAAC;QACH,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,WAAW,GAAG,YAAY,CAAC,IAAI,CACnC,CAAC,iBAAiB,EAAE,EAAE,CACpB,iBAAiB,CAAC,YAAY,CAAC,WAAW,EAAE;QAC5C,kBAAkB,CAAC,WAAW,EAAE,CACnC,CAAC;IAEF,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,GAAG,CAAC,oDAAoD,EAAE;YACxD,kBAAkB;YAClB,aAAa;SACd,CAAC,CAAC;QACH,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,+BAA+B,GAAG;QACtC,GAAG,WAAW;QACd,MAAM,EAAE,IAAA,wBAAK,EACX,8BAA8B,CAAC;YAC7B,WAAW;YACX,gBAAgB;YAChB,cAAc;SACf,CAAC,CACH;KACF,CAAC;IAEF,MAAM,eAAe,GAAG,IAAA,8BAAwB,EAAC;QAC/C,OAAO,EAAE,aAAa;QACtB,WAAW,EAAE,+BAA+B;QAC5C,SAAS;KACV,CAAC,CAAC;IAEH,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,GAAG,CAAC,yDAAyD,EAAE;YAC7D,kBAAkB;YAClB,aAAa;SACd,CAAC,CAAC;QACH,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,GAAG,CAAC,6CAA6C,EAAE;QACjD,MAAM,EAAE,eAAe,CAAC,GAAG;QAC3B,kBAAkB;QAClB,aAAa;KACd,CAAC,CAAC;IAEH,OAAO,eAAe,CAAC;AACzB,CAAC;AA7ED,8EA6EC;AAED,SAAS,8BAA8B,CAAC,EACtC,WAAW,EACX,gBAAgB,EAChB,cAAc,GAKf;IACC,IAAI,MAAM,GAAG,IAAI,wBAAS,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAE/C,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,IAAI,wBAAS,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAC3C,MAAM,YAAY,GAAG,IAAI,wBAAS,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAEvD,IAAI,gBAAgB,GAAG,CAAC,IAAI,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;YACjD,MAAM,OAAO,GAAG,YAAY,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC,YAAY,CAAC,wBAAS,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AAC9D,CAAC","sourcesContent":["import { toHex } from '@metamask/controller-utils';\nimport type { GasFeeToken } from '@metamask/transaction-controller';\nimport type { Hex } from '@metamask/utils';\nimport { createModuleLogger } from '@metamask/utils';\nimport { BigNumber } from 'bignumber.js';\n\nimport { projectLogger } from '../../logger';\nimport type {\n  Amount,\n  QuoteRequest,\n  TransactionPayControllerMessenger,\n} from '../../types';\nimport {\n  getEIP7702SupportedChains,\n  getFeatureFlags,\n} from '../../utils/feature-flags';\nimport { calculateGasFeeTokenCost } from '../../utils/gas';\n\nconst log = createModuleLogger(projectLogger, 'relay-gas-station');\n\ntype GasStationCostParams = {\n  firstStepData: {\n    data: Hex;\n    to: Hex;\n    value?: string;\n  };\n  messenger: TransactionPayControllerMessenger;\n  request: Pick<QuoteRequest, 'from' | 'sourceChainId' | 'sourceTokenAddress'>;\n  totalGasEstimate: number;\n  totalItemCount: number;\n};\n\nexport type GasStationEligibility = {\n  chainSupportsGasStation: boolean;\n  isDisabledChain: boolean;\n  isEligible: boolean;\n};\n\nexport function getGasStationEligibility(\n  messenger: TransactionPayControllerMessenger,\n  sourceChainId: QuoteRequest['sourceChainId'],\n): GasStationEligibility {\n  const { relayDisabledGasStationChains } = getFeatureFlags(messenger);\n  const supportedChains = getEIP7702SupportedChains(messenger);\n  const chainSupportsGasStation = supportedChains.some(\n    (supportedChainId) =>\n      supportedChainId.toLowerCase() === sourceChainId.toLowerCase(),\n  );\n\n  const isDisabledChain = relayDisabledGasStationChains.includes(sourceChainId);\n\n  return {\n    chainSupportsGasStation,\n    isDisabledChain,\n    isEligible: !isDisabledChain && chainSupportsGasStation,\n  };\n}\n\nexport async function getGasStationCostInSourceTokenRaw({\n  firstStepData,\n  messenger,\n  request,\n  totalGasEstimate,\n  totalItemCount,\n}: GasStationCostParams): Promise<Amount | undefined> {\n  const { data, to, value } = firstStepData;\n  const { from, sourceChainId, sourceTokenAddress } = request;\n\n  let gasFeeTokens: GasFeeToken[];\n\n  try {\n    gasFeeTokens = await messenger.call(\n      'TransactionController:getGasFeeTokens',\n      {\n        chainId: sourceChainId,\n        data,\n        from,\n        to,\n        value: toHex(value ?? '0'),\n      },\n    );\n  } catch (error) {\n    log('Failed to estimate gas fee tokens', {\n      error,\n      sourceChainId,\n    });\n    return undefined;\n  }\n\n  const gasFeeToken = gasFeeTokens.find(\n    (singleGasFeeToken) =>\n      singleGasFeeToken.tokenAddress.toLowerCase() ===\n      sourceTokenAddress.toLowerCase(),\n  );\n\n  if (!gasFeeToken) {\n    log('No matching source token in gas fee token estimate', {\n      sourceTokenAddress,\n      sourceChainId,\n    });\n    return undefined;\n  }\n\n  const gasFeeTokenWithNormalizedAmount = {\n    ...gasFeeToken,\n    amount: toHex(\n      getNormalizedGasFeeTokenAmount({\n        gasFeeToken,\n        totalGasEstimate,\n        totalItemCount,\n      }),\n    ),\n  };\n\n  const gasFeeTokenCost = calculateGasFeeTokenCost({\n    chainId: sourceChainId,\n    gasFeeToken: gasFeeTokenWithNormalizedAmount,\n    messenger,\n  });\n\n  if (!gasFeeTokenCost) {\n    log('Unable to calculate gas fee token cost using fiat rates', {\n      sourceTokenAddress,\n      sourceChainId,\n    });\n    return undefined;\n  }\n\n  log('Estimated gas station cost for source token', {\n    amount: gasFeeTokenCost.raw,\n    sourceTokenAddress,\n    sourceChainId,\n  });\n\n  return gasFeeTokenCost;\n}\n\nfunction getNormalizedGasFeeTokenAmount({\n  gasFeeToken,\n  totalGasEstimate,\n  totalItemCount,\n}: {\n  gasFeeToken: GasFeeToken;\n  totalGasEstimate: number;\n  totalItemCount: number;\n}): string {\n  let amount = new BigNumber(gasFeeToken.amount);\n\n  if (totalItemCount > 1) {\n    const gas = new BigNumber(gasFeeToken.gas);\n    const gasFeeAmount = new BigNumber(gasFeeToken.amount);\n\n    if (totalGasEstimate > 0 && gas.isGreaterThan(0)) {\n      const gasRate = gasFeeAmount.dividedBy(gas);\n      amount = gasRate.multipliedBy(totalGasEstimate);\n    }\n  }\n\n  return amount.integerValue(BigNumber.ROUND_CEIL).toFixed(0);\n}\n"]}

package/dist/strategy/relay/gas-station.d.cts

@@ -0,0 +1,22 @@
+import type { Hex } from "@metamask/utils";
+import type { Amount, QuoteRequest, TransactionPayControllerMessenger } from "../../types.cjs";
+type GasStationCostParams = {
+    firstStepData: {
+        data: Hex;
+        to: Hex;
+        value?: string;
+    };
+    messenger: TransactionPayControllerMessenger;
+    request: Pick<QuoteRequest, 'from' | 'sourceChainId' | 'sourceTokenAddress'>;
+    totalGasEstimate: number;
+    totalItemCount: number;
+};
+export type GasStationEligibility = {
+    chainSupportsGasStation: boolean;
+    isDisabledChain: boolean;
+    isEligible: boolean;
+};
+export declare function getGasStationEligibility(messenger: TransactionPayControllerMessenger, sourceChainId: QuoteRequest['sourceChainId']): GasStationEligibility;
+export declare function getGasStationCostInSourceTokenRaw({ firstStepData, messenger, request, totalGasEstimate, totalItemCount, }: GasStationCostParams): Promise<Amount | undefined>;
+export {};
+//# sourceMappingURL=gas-station.d.cts.map

package/dist/strategy/relay/gas-station.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gas-station.d.cts","sourceRoot":"","sources":["../../../src/strategy/relay/gas-station.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAK3C,OAAO,KAAK,EACV,MAAM,EACN,YAAY,EACZ,iCAAiC,EAClC,wBAAoB;AASrB,KAAK,oBAAoB,GAAG;IAC1B,aAAa,EAAE;QACb,IAAI,EAAE,GAAG,CAAC;QACV,EAAE,EAAE,GAAG,CAAC;QACR,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,SAAS,EAAE,iCAAiC,CAAC;IAC7C,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,MAAM,GAAG,eAAe,GAAG,oBAAoB,CAAC,CAAC;IAC7E,gBAAgB,EAAE,MAAM,CAAC;IACzB,cAAc,EAAE,MAAM,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG;IAClC,uBAAuB,EAAE,OAAO,CAAC;IACjC,eAAe,EAAE,OAAO,CAAC;IACzB,UAAU,EAAE,OAAO,CAAC;CACrB,CAAC;AAEF,wBAAgB,wBAAwB,CACtC,SAAS,EAAE,iCAAiC,EAC5C,aAAa,EAAE,YAAY,CAAC,eAAe,CAAC,GAC3C,qBAAqB,CAevB;AAED,wBAAsB,iCAAiC,CAAC,EACtD,aAAa,EACb,SAAS,EACT,OAAO,EACP,gBAAgB,EAChB,cAAc,GACf,EAAE,oBAAoB,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAuEpD"}

package/dist/strategy/relay/gas-station.d.mts

@@ -0,0 +1,22 @@
+import type { Hex } from "@metamask/utils";
+import type { Amount, QuoteRequest, TransactionPayControllerMessenger } from "../../types.mjs";
+type GasStationCostParams = {
+    firstStepData: {
+        data: Hex;
+        to: Hex;
+        value?: string;
+    };
+    messenger: TransactionPayControllerMessenger;
+    request: Pick<QuoteRequest, 'from' | 'sourceChainId' | 'sourceTokenAddress'>;
+    totalGasEstimate: number;
+    totalItemCount: number;
+};
+export type GasStationEligibility = {
+    chainSupportsGasStation: boolean;
+    isDisabledChain: boolean;
+    isEligible: boolean;
+};
+export declare function getGasStationEligibility(messenger: TransactionPayControllerMessenger, sourceChainId: QuoteRequest['sourceChainId']): GasStationEligibility;
+export declare function getGasStationCostInSourceTokenRaw({ firstStepData, messenger, request, totalGasEstimate, totalItemCount, }: GasStationCostParams): Promise<Amount | undefined>;
+export {};
+//# sourceMappingURL=gas-station.d.mts.map

package/dist/strategy/relay/gas-station.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gas-station.d.mts","sourceRoot":"","sources":["../../../src/strategy/relay/gas-station.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAK3C,OAAO,KAAK,EACV,MAAM,EACN,YAAY,EACZ,iCAAiC,EAClC,wBAAoB;AASrB,KAAK,oBAAoB,GAAG;IAC1B,aAAa,EAAE;QACb,IAAI,EAAE,GAAG,CAAC;QACV,EAAE,EAAE,GAAG,CAAC;QACR,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,SAAS,EAAE,iCAAiC,CAAC;IAC7C,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,MAAM,GAAG,eAAe,GAAG,oBAAoB,CAAC,CAAC;IAC7E,gBAAgB,EAAE,MAAM,CAAC;IACzB,cAAc,EAAE,MAAM,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG;IAClC,uBAAuB,EAAE,OAAO,CAAC;IACjC,eAAe,EAAE,OAAO,CAAC;IACzB,UAAU,EAAE,OAAO,CAAC;CACrB,CAAC;AAEF,wBAAgB,wBAAwB,CACtC,SAAS,EAAE,iCAAiC,EAC5C,aAAa,EAAE,YAAY,CAAC,eAAe,CAAC,GAC3C,qBAAqB,CAevB;AAED,wBAAsB,iCAAiC,CAAC,EACtD,aAAa,EACb,SAAS,EACT,OAAO,EACP,gBAAgB,EAChB,cAAc,GACf,EAAE,oBAAoB,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAuEpD"}

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

@@ -0,0 +1,87 @@
+import { toHex } from "@metamask/controller-utils";
+import { createModuleLogger } from "@metamask/utils";
+import { BigNumber } from "bignumber.js";
+import { projectLogger } from "../../logger.mjs";
+import { getEIP7702SupportedChains, getFeatureFlags } from "../../utils/feature-flags.mjs";
+import { calculateGasFeeTokenCost } from "../../utils/gas.mjs";
+const log = createModuleLogger(projectLogger, 'relay-gas-station');
+export function getGasStationEligibility(messenger, sourceChainId) {
+    const { relayDisabledGasStationChains } = getFeatureFlags(messenger);
+    const supportedChains = getEIP7702SupportedChains(messenger);
+    const chainSupportsGasStation = supportedChains.some((supportedChainId) => supportedChainId.toLowerCase() === sourceChainId.toLowerCase());
+    const isDisabledChain = relayDisabledGasStationChains.includes(sourceChainId);
+    return {
+        chainSupportsGasStation,
+        isDisabledChain,
+        isEligible: !isDisabledChain && chainSupportsGasStation,
+    };
+}
+export async function getGasStationCostInSourceTokenRaw({ firstStepData, messenger, request, totalGasEstimate, totalItemCount, }) {
+    const { data, to, value } = firstStepData;
+    const { from, sourceChainId, sourceTokenAddress } = request;
+    let gasFeeTokens;
+    try {
+        gasFeeTokens = await messenger.call('TransactionController:getGasFeeTokens', {
+            chainId: sourceChainId,
+            data,
+            from,
+            to,
+            value: toHex(value ?? '0'),
+        });
+    }
+    catch (error) {
+        log('Failed to estimate gas fee tokens', {
+            error,
+            sourceChainId,
+        });
+        return undefined;
+    }
+    const gasFeeToken = gasFeeTokens.find((singleGasFeeToken) => singleGasFeeToken.tokenAddress.toLowerCase() ===
+        sourceTokenAddress.toLowerCase());
+    if (!gasFeeToken) {
+        log('No matching source token in gas fee token estimate', {
+            sourceTokenAddress,
+            sourceChainId,
+        });
+        return undefined;
+    }
+    const gasFeeTokenWithNormalizedAmount = {
+        ...gasFeeToken,
+        amount: toHex(getNormalizedGasFeeTokenAmount({
+            gasFeeToken,
+            totalGasEstimate,
+            totalItemCount,
+        })),
+    };
+    const gasFeeTokenCost = calculateGasFeeTokenCost({
+        chainId: sourceChainId,
+        gasFeeToken: gasFeeTokenWithNormalizedAmount,
+        messenger,
+    });
+    if (!gasFeeTokenCost) {
+        log('Unable to calculate gas fee token cost using fiat rates', {
+            sourceTokenAddress,
+            sourceChainId,
+        });
+        return undefined;
+    }
+    log('Estimated gas station cost for source token', {
+        amount: gasFeeTokenCost.raw,
+        sourceTokenAddress,
+        sourceChainId,
+    });
+    return gasFeeTokenCost;
+}
+function getNormalizedGasFeeTokenAmount({ gasFeeToken, totalGasEstimate, totalItemCount, }) {
+    let amount = new BigNumber(gasFeeToken.amount);
+    if (totalItemCount > 1) {
+        const gas = new BigNumber(gasFeeToken.gas);
+        const gasFeeAmount = new BigNumber(gasFeeToken.amount);
+        if (totalGasEstimate > 0 && gas.isGreaterThan(0)) {
+            const gasRate = gasFeeAmount.dividedBy(gas);
+            amount = gasRate.multipliedBy(totalGasEstimate);
+        }
+    }
+    return amount.integerValue(BigNumber.ROUND_CEIL).toFixed(0);
+}
+//# sourceMappingURL=gas-station.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"gas-station.mjs","sourceRoot":"","sources":["../../../src/strategy/relay/gas-station.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,mCAAmC;AAGnD,OAAO,EAAE,kBAAkB,EAAE,wBAAwB;AACrD,OAAO,EAAE,SAAS,EAAE,qBAAqB;AAEzC,OAAO,EAAE,aAAa,EAAE,yBAAqB;AAM7C,OAAO,EACL,yBAAyB,EACzB,eAAe,EAChB,sCAAkC;AACnC,OAAO,EAAE,wBAAwB,EAAE,4BAAwB;AAE3D,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,EAAE,mBAAmB,CAAC,CAAC;AAoBnE,MAAM,UAAU,wBAAwB,CACtC,SAA4C,EAC5C,aAA4C;IAE5C,MAAM,EAAE,6BAA6B,EAAE,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC;IACrE,MAAM,eAAe,GAAG,yBAAyB,CAAC,SAAS,CAAC,CAAC;IAC7D,MAAM,uBAAuB,GAAG,eAAe,CAAC,IAAI,CAClD,CAAC,gBAAgB,EAAE,EAAE,CACnB,gBAAgB,CAAC,WAAW,EAAE,KAAK,aAAa,CAAC,WAAW,EAAE,CACjE,CAAC;IAEF,MAAM,eAAe,GAAG,6BAA6B,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IAE9E,OAAO;QACL,uBAAuB;QACvB,eAAe;QACf,UAAU,EAAE,CAAC,eAAe,IAAI,uBAAuB;KACxD,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iCAAiC,CAAC,EACtD,aAAa,EACb,SAAS,EACT,OAAO,EACP,gBAAgB,EAChB,cAAc,GACO;IACrB,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,aAAa,CAAC;IAC1C,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,kBAAkB,EAAE,GAAG,OAAO,CAAC;IAE5D,IAAI,YAA2B,CAAC;IAEhC,IAAI,CAAC;QACH,YAAY,GAAG,MAAM,SAAS,CAAC,IAAI,CACjC,uCAAuC,EACvC;YACE,OAAO,EAAE,aAAa;YACtB,IAAI;YACJ,IAAI;YACJ,EAAE;YACF,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,GAAG,CAAC;SAC3B,CACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,mCAAmC,EAAE;YACvC,KAAK;YACL,aAAa;SACd,CAAC,CAAC;QACH,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,WAAW,GAAG,YAAY,CAAC,IAAI,CACnC,CAAC,iBAAiB,EAAE,EAAE,CACpB,iBAAiB,CAAC,YAAY,CAAC,WAAW,EAAE;QAC5C,kBAAkB,CAAC,WAAW,EAAE,CACnC,CAAC;IAEF,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,GAAG,CAAC,oDAAoD,EAAE;YACxD,kBAAkB;YAClB,aAAa;SACd,CAAC,CAAC;QACH,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,+BAA+B,GAAG;QACtC,GAAG,WAAW;QACd,MAAM,EAAE,KAAK,CACX,8BAA8B,CAAC;YAC7B,WAAW;YACX,gBAAgB;YAChB,cAAc;SACf,CAAC,CACH;KACF,CAAC;IAEF,MAAM,eAAe,GAAG,wBAAwB,CAAC;QAC/C,OAAO,EAAE,aAAa;QACtB,WAAW,EAAE,+BAA+B;QAC5C,SAAS;KACV,CAAC,CAAC;IAEH,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,GAAG,CAAC,yDAAyD,EAAE;YAC7D,kBAAkB;YAClB,aAAa;SACd,CAAC,CAAC;QACH,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,GAAG,CAAC,6CAA6C,EAAE;QACjD,MAAM,EAAE,eAAe,CAAC,GAAG;QAC3B,kBAAkB;QAClB,aAAa;KACd,CAAC,CAAC;IAEH,OAAO,eAAe,CAAC;AACzB,CAAC;AAED,SAAS,8BAA8B,CAAC,EACtC,WAAW,EACX,gBAAgB,EAChB,cAAc,GAKf;IACC,IAAI,MAAM,GAAG,IAAI,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAE/C,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,IAAI,SAAS,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAC3C,MAAM,YAAY,GAAG,IAAI,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAEvD,IAAI,gBAAgB,GAAG,CAAC,IAAI,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;YACjD,MAAM,OAAO,GAAG,YAAY,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AAC9D,CAAC","sourcesContent":["import { toHex } from '@metamask/controller-utils';\nimport type { GasFeeToken } from '@metamask/transaction-controller';\nimport type { Hex } from '@metamask/utils';\nimport { createModuleLogger } from '@metamask/utils';\nimport { BigNumber } from 'bignumber.js';\n\nimport { projectLogger } from '../../logger';\nimport type {\n  Amount,\n  QuoteRequest,\n  TransactionPayControllerMessenger,\n} from '../../types';\nimport {\n  getEIP7702SupportedChains,\n  getFeatureFlags,\n} from '../../utils/feature-flags';\nimport { calculateGasFeeTokenCost } from '../../utils/gas';\n\nconst log = createModuleLogger(projectLogger, 'relay-gas-station');\n\ntype GasStationCostParams = {\n  firstStepData: {\n    data: Hex;\n    to: Hex;\n    value?: string;\n  };\n  messenger: TransactionPayControllerMessenger;\n  request: Pick<QuoteRequest, 'from' | 'sourceChainId' | 'sourceTokenAddress'>;\n  totalGasEstimate: number;\n  totalItemCount: number;\n};\n\nexport type GasStationEligibility = {\n  chainSupportsGasStation: boolean;\n  isDisabledChain: boolean;\n  isEligible: boolean;\n};\n\nexport function getGasStationEligibility(\n  messenger: TransactionPayControllerMessenger,\n  sourceChainId: QuoteRequest['sourceChainId'],\n): GasStationEligibility {\n  const { relayDisabledGasStationChains } = getFeatureFlags(messenger);\n  const supportedChains = getEIP7702SupportedChains(messenger);\n  const chainSupportsGasStation = supportedChains.some(\n    (supportedChainId) =>\n      supportedChainId.toLowerCase() === sourceChainId.toLowerCase(),\n  );\n\n  const isDisabledChain = relayDisabledGasStationChains.includes(sourceChainId);\n\n  return {\n    chainSupportsGasStation,\n    isDisabledChain,\n    isEligible: !isDisabledChain && chainSupportsGasStation,\n  };\n}\n\nexport async function getGasStationCostInSourceTokenRaw({\n  firstStepData,\n  messenger,\n  request,\n  totalGasEstimate,\n  totalItemCount,\n}: GasStationCostParams): Promise<Amount | undefined> {\n  const { data, to, value } = firstStepData;\n  const { from, sourceChainId, sourceTokenAddress } = request;\n\n  let gasFeeTokens: GasFeeToken[];\n\n  try {\n    gasFeeTokens = await messenger.call(\n      'TransactionController:getGasFeeTokens',\n      {\n        chainId: sourceChainId,\n        data,\n        from,\n        to,\n        value: toHex(value ?? '0'),\n      },\n    );\n  } catch (error) {\n    log('Failed to estimate gas fee tokens', {\n      error,\n      sourceChainId,\n    });\n    return undefined;\n  }\n\n  const gasFeeToken = gasFeeTokens.find(\n    (singleGasFeeToken) =>\n      singleGasFeeToken.tokenAddress.toLowerCase() ===\n      sourceTokenAddress.toLowerCase(),\n  );\n\n  if (!gasFeeToken) {\n    log('No matching source token in gas fee token estimate', {\n      sourceTokenAddress,\n      sourceChainId,\n    });\n    return undefined;\n  }\n\n  const gasFeeTokenWithNormalizedAmount = {\n    ...gasFeeToken,\n    amount: toHex(\n      getNormalizedGasFeeTokenAmount({\n        gasFeeToken,\n        totalGasEstimate,\n        totalItemCount,\n      }),\n    ),\n  };\n\n  const gasFeeTokenCost = calculateGasFeeTokenCost({\n    chainId: sourceChainId,\n    gasFeeToken: gasFeeTokenWithNormalizedAmount,\n    messenger,\n  });\n\n  if (!gasFeeTokenCost) {\n    log('Unable to calculate gas fee token cost using fiat rates', {\n      sourceTokenAddress,\n      sourceChainId,\n    });\n    return undefined;\n  }\n\n  log('Estimated gas station cost for source token', {\n    amount: gasFeeTokenCost.raw,\n    sourceTokenAddress,\n    sourceChainId,\n  });\n\n  return gasFeeTokenCost;\n}\n\nfunction getNormalizedGasFeeTokenAmount({\n  gasFeeToken,\n  totalGasEstimate,\n  totalItemCount,\n}: {\n  gasFeeToken: GasFeeToken;\n  totalGasEstimate: number;\n  totalItemCount: number;\n}): string {\n  let amount = new BigNumber(gasFeeToken.amount);\n\n  if (totalItemCount > 1) {\n    const gas = new BigNumber(gasFeeToken.gas);\n    const gasFeeAmount = new BigNumber(gasFeeToken.amount);\n\n    if (totalGasEstimate > 0 && gas.isGreaterThan(0)) {\n      const gasRate = gasFeeAmount.dividedBy(gas);\n      amount = gasRate.multipliedBy(totalGasEstimate);\n    }\n  }\n\n  return amount.integerValue(BigNumber.ROUND_CEIL).toFixed(0);\n}\n"]}