@metamask/transaction-controller 52.3.0 → 54.0.0

package/dist/TransactionController.mjs

@@ -9,7 +9,7 @@ var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (
     if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
     return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
 };
-var _TransactionController_instances, _TransactionController_internalEvents, _TransactionController_methodDataHelper, _TransactionController_incomingTransactionHelper, _TransactionController_incomingTransactionOptions, _TransactionController_pendingTransactionOptions, _TransactionController_publishBatchHook, _TransactionController_publicKeyEIP7702, _TransactionController_trace, _TransactionController_transactionHistoryLimit, _TransactionController_isFirstTimeInteractionEnabled, _TransactionController_isSimulationEnabled, _TransactionController_testGasFeeFlows, _TransactionController_multichainTrackingHelper, _TransactionController_retryTransaction, _TransactionController_rejectTransaction, _TransactionController_getChainId, _TransactionController_getNetworkClientId, _TransactionController_getEthQuery, _TransactionController_getProvider, _TransactionController_createNonceTracker, _TransactionController_createPendingTransactionTracker, _TransactionController_checkForPendingTransactionAndStartPolling, _TransactionController_stopAllTracking, _TransactionController_addIncomingTransactionHelperListeners, _TransactionController_removePendingTransactionTrackerListeners, _TransactionController_addPendingTransactionTrackerListeners, _TransactionController_getNonceTrackerPendingTransactions, _TransactionController_getGasFeeFlows, _TransactionController_getLayer1GasFeeFlows, _TransactionController_updateTransactionInternal, _TransactionController_updateFirstTimeInteraction, _TransactionController_updateSimulationData, _TransactionController_onGasFeePollerTransactionUpdate, _TransactionController_getSelectedAccount, _TransactionController_getInternalAccounts, _TransactionController_updateSubmitHistory, _TransactionController_updateGasEstimate, _TransactionController_deleteTransaction, _TransactionController_isRejectError, _TransactionController_rejectTransactionAndThrow;
+var _TransactionController_instances, _TransactionController_internalEvents, _TransactionController_methodDataHelper, _TransactionController_incomingTransactionHelper, _TransactionController_incomingTransactionOptions, _TransactionController_pendingTransactionOptions, _TransactionController_publishBatchHook, _TransactionController_publicKeyEIP7702, _TransactionController_trace, _TransactionController_transactionHistoryLimit, _TransactionController_isFirstTimeInteractionEnabled, _TransactionController_isSimulationEnabled, _TransactionController_testGasFeeFlows, _TransactionController_multichainTrackingHelper, _TransactionController_retryTransaction, _TransactionController_rejectTransaction, _TransactionController_getChainId, _TransactionController_getNetworkClientId, _TransactionController_getEthQuery, _TransactionController_getProvider, _TransactionController_markNonceDuplicatesDropped, _TransactionController_signTransaction, _TransactionController_onConfirmedTransaction, _TransactionController_createNonceTracker, _TransactionController_createPendingTransactionTracker, _TransactionController_checkForPendingTransactionAndStartPolling, _TransactionController_stopAllTracking, _TransactionController_addIncomingTransactionHelperListeners, _TransactionController_removePendingTransactionTrackerListeners, _TransactionController_addPendingTransactionTrackerListeners, _TransactionController_getNonceTrackerPendingTransactions, _TransactionController_getGasFeeFlows, _TransactionController_getLayer1GasFeeFlows, _TransactionController_updateTransactionInternal, _TransactionController_updateFirstTimeInteraction, _TransactionController_updateSimulationData, _TransactionController_onGasFeePollerTransactionUpdate, _TransactionController_getSelectedAccount, _TransactionController_getInternalAccounts, _TransactionController_updateSubmitHistory, _TransactionController_updateGasEstimate, _TransactionController_registerActionHandlers, _TransactionController_deleteTransaction, _TransactionController_isRejectError, _TransactionController_rejectTransactionAndThrow;
 function $importDefault(module) {
     if (module?.__esModule) {
         return module.default;
@@ -159,7 +159,7 @@ export class TransactionController extends BaseController {
      * @param options - The controller options.
      */
     constructor(options) {
-        const { disableHistory, disableSendFlowHistory, disableSwaps, enableTxParamsGasFeeUpdates, getCurrentAccountEIP1559Compatibility, getCurrentNetworkEIP1559Compatibility, getExternalPendingTransactions, getGasFeeEstimates, getNetworkClientRegistry, getNetworkState, getPermittedAccounts, getSavedGasFees, incomingTransactions = {}, isFirstTimeInteractionEnabled, isSimulationEnabled, messenger, pendingTransactions = {}, publicKeyEIP7702, securityProviderRequest, sign, state, testGasFeeFlows, trace, transactionHistoryLimit = 40, hooks, } = options;
+        const { disableHistory, disableSendFlowHistory, disableSwaps, isAutomaticGasFeeUpdateEnabled, getCurrentAccountEIP1559Compatibility, getCurrentNetworkEIP1559Compatibility, getExternalPendingTransactions, getGasFeeEstimates, getNetworkClientRegistry, getNetworkState, getPermittedAccounts, getSavedGasFees, incomingTransactions = {}, isFirstTimeInteractionEnabled, isSimulationEnabled, messenger, pendingTransactions = {}, publicKeyEIP7702, securityProviderRequest, sign, state, testGasFeeFlows, trace, transactionHistoryLimit = 40, hooks, } = options;
         super({
             name: controllerName,
             metadata,
@@ -189,7 +189,8 @@ export class TransactionController extends BaseController {
             __classPrivateFieldGet(this, _TransactionController_multichainTrackingHelper, "f").checkForPendingTransactionAndStartPolling();
         });
         this.messagingSystem = messenger;
-        this.isTxParamsGasFeeUpdatesEnabled = enableTxParamsGasFeeUpdates ?? false;
+        this.isTxParamsGasFeeUpdatesEnabled =
+            isAutomaticGasFeeUpdateEnabled ?? ((_txMeta) => false);
         this.getNetworkState = getNetworkState;
         this.isSendFlowHistoryDisabled = disableSendFlowHistory ?? false;
         this.isHistoryDisabled = disableHistory ?? false;
@@ -217,10 +218,9 @@ export class TransactionController extends BaseController {
         __classPrivateFieldSet(this, _TransactionController_trace, trace ?? ((_request, fn) => fn?.()), "f");
         this.afterSign = hooks?.afterSign ?? (() => true);
         this.beforeCheckPendingTransaction =
-            hooks?.beforeCheckPendingTransaction ??
-                /* istanbul ignore next */
-                (() => true);
-        this.beforePublish = hooks?.beforePublish ?? (() => true);
+            /* istanbul ignore next */
+            hooks?.beforeCheckPendingTransaction ?? (() => Promise.resolve(true));
+        this.beforePublish = hooks?.beforePublish ?? (() => Promise.resolve(true));
         this.getAdditionalSignArguments =
             hooks?.getAdditionalSignArguments ?? (() => []);
         this.publish =
@@ -296,6 +296,7 @@ export class TransactionController extends BaseController {
         });
         this.onBootCleanup();
         __classPrivateFieldGet(this, _TransactionController_checkForPendingTransactionAndStartPolling, "f").call(this);
+        __classPrivateFieldGet(this, _TransactionController_instances, "m", _TransactionController_registerActionHandlers).call(this);
     }
     /**
      * Stops polling and removes listeners to prepare the controller for garbage collection.
@@ -336,12 +337,12 @@ export class TransactionController extends BaseController {
     /**
      * Determine which chains support atomic batch transactions with the given account address.
      *
-     * @param address - The address of the account to check.
-     * @returns  The supported chain IDs.
+     * @param request - Request object containing the account address and other parameters.
+     * @returns  Result object containing the supported chains and related information.
      */
-    async isAtomicBatchSupported(address) {
+    async isAtomicBatchSupported(request) {
         return isAtomicBatchSupported({
-            address,
+            ...request,
             getEthQuery: (chainId) => __classPrivateFieldGet(this, _TransactionController_instances, "m", _TransactionController_getEthQuery).call(this, { chainId }),
             messenger: this.messagingSystem,
             publicKeyEIP7702: __classPrivateFieldGet(this, _TransactionController_publicKeyEIP7702, "f"),
@@ -401,7 +402,7 @@ export class TransactionController extends BaseController {
         });
         const delegationAddressPromise = getDelegationAddress(txParams.from, ethQuery).catch(() => undefined);
         const isEIP1559Compatible = await this.getEIP1559Compatibility(networkClientId);
-        validateTxParams(txParams, isEIP1559Compatible);
+        validateTxParams(txParams, isEIP1559Compatible, chainId);
         if (!txParams.type) {
             // Determine transaction type based on transaction parameters and network compatibility
             setEnvelopeType(txParams, isEIP1559Compatible);
@@ -570,6 +571,7 @@ export class TransactionController extends BaseController {
             chainId: __classPrivateFieldGet(this, _TransactionController_instances, "m", _TransactionController_getChainId).call(this, networkClientId),
             ethQuery,
             isSimulationEnabled: __classPrivateFieldGet(this, _TransactionController_isSimulationEnabled, "f").call(this),
+            messenger: this.messagingSystem,
             txParams: transaction,
         });
         return { gas: estimatedGas, simulationFails };
@@ -590,6 +592,7 @@ export class TransactionController extends BaseController {
             chainId: __classPrivateFieldGet(this, _TransactionController_instances, "m", _TransactionController_getChainId).call(this, networkClientId),
             ethQuery,
             isSimulationEnabled: __classPrivateFieldGet(this, _TransactionController_isSimulationEnabled, "f").call(this),
+            messenger: this.messagingSystem,
             txParams: transaction,
         });
         const gas = addGasBuffer(estimatedGas, blockGasLimit, multiplier);
@@ -678,12 +681,13 @@ export class TransactionController extends BaseController {
                 updatedTransactionMeta.baseFeePerGas = baseFeePerGas;
             }
             // Update same nonce local transactions as dropped and define replacedBy properties.
-            this.markNonceDuplicatesDropped(transactionId);
+            __classPrivateFieldGet(this, _TransactionController_instances, "m", _TransactionController_markNonceDuplicatesDropped).call(this, transactionId);
             // Update external provided transaction with updated gas values and confirmed status.
             this.updateTransaction(updatedTransactionMeta, `${controllerName}:confirmExternalTransaction - Add external transaction`);
             this.onTransactionStatusChange(updatedTransactionMeta);
             // Intentional given potential duration of process.
             this.updatePostBalance(updatedTransactionMeta).catch((error) => {
+                /* istanbul ignore next */
                 log('Error while updating post balance', error);
                 throw error;
             });
@@ -932,20 +936,16 @@ export class TransactionController extends BaseController {
     /**
      * Update a custodial transaction.
      *
-     * @param transactionId - The ID of the transaction to update.
-     * @param options - The custodial transaction options to update.
-     * @param options.errorMessage - The error message to be assigned in case transaction status update to failed.
-     * @param options.hash - The new hash value to be assigned.
-     * @param options.status - The new status value to be assigned.
+     * @param request - The custodial transaction update request.
+     *
+     * @returns The updated transaction metadata.
      */
-    updateCustodialTransaction(transactionId, { errorMessage, hash, status, }) {
+    updateCustodialTransaction(request) {
+        const { transactionId, errorMessage, hash, status, gasLimit, gasPrice, maxFeePerGas, maxPriorityFeePerGas, nonce, type, } = request;
         const transactionMeta = this.getTransaction(transactionId);
         if (!transactionMeta) {
             throw new Error(`Cannot update custodial transaction as no transaction metadata found`);
         }
-        if (!transactionMeta.custodyId) {
-            throw new Error('Transaction must be a custodian transaction');
-        }
         if (status &&
             ![
                 TransactionStatus.submitted,
@@ -961,11 +961,27 @@ export class TransactionController extends BaseController {
         if (updatedTransactionMeta.status === TransactionStatus.failed) {
             updatedTransactionMeta.error = normalizeTxError(new Error(errorMessage));
         }
+        // Update txParams properties with a single pickBy operation
+        updatedTransactionMeta.txParams = merge({}, updatedTransactionMeta.txParams, pickBy({
+            gasLimit,
+            gasPrice,
+            maxFeePerGas,
+            maxPriorityFeePerGas,
+            nonce,
+            type,
+        }));
+        // Special case for type change to legacy
+        if (type === TransactionEnvelopeType.legacy) {
+            delete updatedTransactionMeta.txParams.maxFeePerGas;
+            delete updatedTransactionMeta.txParams.maxPriorityFeePerGas;
+        }
         this.updateTransaction(updatedTransactionMeta, `${controllerName}:updateCustodialTransaction - Custodial transaction updated`);
-        if ([TransactionStatus.submitted, TransactionStatus.failed].includes(status)) {
+        if (status &&
+            [TransactionStatus.submitted, TransactionStatus.failed].includes(status)) {
             this.messagingSystem.publish(`${controllerName}:transactionFinished`, updatedTransactionMeta);
             __classPrivateFieldGet(this, _TransactionController_internalEvents, "f").emit(`${updatedTransactionMeta.id}:finished`, updatedTransactionMeta);
         }
+        return updatedTransactionMeta;
     }
     /**
      * Search transaction metadata for matching entries.
@@ -1381,13 +1397,13 @@ export class TransactionController extends BaseController {
                 }
             });
             this.onTransactionStatusChange(transactionMeta);
-            const rawTx = await __classPrivateFieldGet(this, _TransactionController_trace, "f").call(this, { name: 'Sign', parentContext: traceContext }, () => this.signTransaction(transactionMeta));
-            if (!this.beforePublish(transactionMeta)) {
+            const rawTx = await __classPrivateFieldGet(this, _TransactionController_trace, "f").call(this, { name: 'Sign', parentContext: traceContext }, () => __classPrivateFieldGet(this, _TransactionController_instances, "m", _TransactionController_signTransaction).call(this, transactionMeta));
+            if (!(await this.beforePublish(transactionMeta))) {
                 log('Skipping publishing transaction based on hook');
                 this.messagingSystem.publish(`${controllerName}:transactionPublishingSkipped`, transactionMeta);
                 return ApprovalState.SkippedViaBeforePublishHook;
             }
-            if (!rawTx) {
+            if (!rawTx && !transactionMeta.isExternalSign) {
                 return ApprovalState.NotApproved;
             }
             const { networkClientId } = transactionMeta;
@@ -1414,7 +1430,7 @@ export class TransactionController extends BaseController {
             }
             await __classPrivateFieldGet(this, _TransactionController_trace, "f").call(this, { name: 'Publish', parentContext: traceContext }, async () => {
                 const publishHook = publishHookOverride ?? this.publish;
-                ({ transactionHash: hash } = await publishHook(transactionMeta, rawTx));
+                ({ transactionHash: hash } = await publishHook(transactionMeta, rawTx ?? '0x'));
                 if (hash === undefined) {
                     hash = await this.publishTransaction(ethQuery, {
                         ...transactionMeta,
@@ -1638,44 +1654,6 @@ export class TransactionController extends BaseController {
         });
         return newTransactionMeta;
     }
-    /**
-     * Sets other txMeta statuses to dropped if the txMeta that has been confirmed has other transactions
-     * in the transactions have the same nonce.
-     *
-     * @param transactionId - Used to identify original transaction.
-     */
-    markNonceDuplicatesDropped(transactionId) {
-        const transactionMeta = this.getTransaction(transactionId);
-        if (!transactionMeta) {
-            return;
-        }
-        const nonce = transactionMeta.txParams?.nonce;
-        const from = transactionMeta.txParams?.from;
-        const { chainId } = transactionMeta;
-        const sameNonceTransactions = this.state.transactions.filter((transaction) => transaction.id !== transactionId &&
-            transaction.txParams.from === from &&
-            transaction.txParams.nonce === nonce &&
-            transaction.chainId === chainId &&
-            transaction.type !== TransactionType.incoming);
-        const sameNonceTransactionIds = sameNonceTransactions.map((transaction) => transaction.id);
-        if (sameNonceTransactions.length === 0) {
-            return;
-        }
-        this.update((state) => {
-            for (const transaction of state.transactions) {
-                if (sameNonceTransactionIds.includes(transaction.id)) {
-                    transaction.replacedBy = transactionMeta?.hash;
-                    transaction.replacedById = transactionMeta?.id;
-                }
-            }
-        });
-        for (const transaction of this.state.transactions) {
-            if (sameNonceTransactionIds.includes(transaction.id) &&
-                transaction.status !== TransactionStatus.failed) {
-                this.setTransactionStatusDropped(transaction);
-            }
-        }
-    }
     /**
      * Method to set transaction status to dropped.
      *
@@ -1732,47 +1710,6 @@ export class TransactionController extends BaseController {
         const currentAccountIsEIP1559Compatible = await this.getCurrentAccountEIP1559Compatibility();
         return (currentNetworkIsEIP1559Compatible && currentAccountIsEIP1559Compatible);
     }
-    async signTransaction(transactionMeta) {
-        const { txParams } = transactionMeta;
-        log('Signing transaction', txParams);
-        const { authorizationList, from } = txParams;
-        const finalTxParams = { ...txParams };
-        finalTxParams.authorizationList = await signAuthorizationList({
-            authorizationList,
-            messenger: this.messagingSystem,
-            transactionMeta,
-        });
-        const unsignedEthTx = prepareTransaction(transactionMeta.chainId, finalTxParams);
-        this.approvingTransactionIds.add(transactionMeta.id);
-        const signedTx = await new Promise((resolve, reject) => {
-            this.sign?.(unsignedEthTx, from, ...this.getAdditionalSignArguments(transactionMeta)).then(resolve, reject);
-            this.signAbortCallbacks.set(transactionMeta.id, () => reject(new Error('Signing aborted by user')));
-        });
-        this.signAbortCallbacks.delete(transactionMeta.id);
-        if (!signedTx) {
-            log('Skipping signed status as no signed transaction');
-            return undefined;
-        }
-        const transactionMetaFromHook = cloneDeep(transactionMeta);
-        if (!this.afterSign(transactionMetaFromHook, signedTx)) {
-            this.updateTransaction(transactionMetaFromHook, 'TransactionController#signTransaction - Update after sign');
-            log('Skipping signed status based on hook');
-            return undefined;
-        }
-        const transactionMetaWithRsv = {
-            ...this.updateTransactionMetaRSV(transactionMetaFromHook, signedTx),
-            status: TransactionStatus.signed,
-            txParams: finalTxParams,
-        };
-        this.updateTransaction(transactionMetaWithRsv, 'TransactionController#approveTransaction - Transaction signed');
-        this.onTransactionStatusChange(transactionMetaWithRsv);
-        const rawTx = serializeTransaction(signedTx);
-        const transactionMetaWithRawTx = merge({}, transactionMetaWithRsv, {
-            rawTx,
-        });
-        this.updateTransaction(transactionMetaWithRawTx, 'TransactionController#approveTransaction - RawTransaction added');
-        return rawTx;
-    }
     onTransactionStatusChange(transactionMeta) {
         this.messagingSystem.publish(`${controllerName}:transactionStatusUpdated`, {
             transactionMeta,
@@ -1781,17 +1718,6 @@ export class TransactionController extends BaseController {
     getNonceTrackerTransactions(statuses, address, chainId) {
         return getAndFormatTransactionsForNonceTracker(chainId, address, statuses, this.state.transactions);
     }
-    onConfirmedTransaction(transactionMeta) {
-        log('Processing confirmed transaction', transactionMeta.id);
-        this.markNonceDuplicatesDropped(transactionMeta.id);
-        this.messagingSystem.publish(`${controllerName}:transactionConfirmed`, transactionMeta);
-        this.onTransactionStatusChange(transactionMeta);
-        // Intentional given potential duration of process.
-        this.updatePostBalance(transactionMeta).catch((error) => {
-            log('Error while updating post balance', error);
-            throw error;
-        });
-    }
     async updatePostBalance(transactionMeta) {
         try {
             const { networkClientId, type } = transactionMeta;
@@ -1939,6 +1865,92 @@ _TransactionController_internalEvents = new WeakMap(), _TransactionController_me
         chainId,
         networkClientId,
     }).provider;
+}, _TransactionController_markNonceDuplicatesDropped = function _TransactionController_markNonceDuplicatesDropped(transactionId) {
+    const transactionMeta = this.getTransaction(transactionId);
+    if (!transactionMeta) {
+        return;
+    }
+    const nonce = transactionMeta.txParams?.nonce;
+    const from = transactionMeta.txParams?.from;
+    const { chainId } = transactionMeta;
+    const sameNonceTransactions = this.state.transactions.filter((transaction) => transaction.id !== transactionId &&
+        transaction.txParams.from === from &&
+        nonce &&
+        transaction.txParams.nonce === nonce &&
+        transaction.chainId === chainId &&
+        transaction.type !== TransactionType.incoming);
+    const sameNonceTransactionIds = sameNonceTransactions.map((transaction) => transaction.id);
+    if (sameNonceTransactions.length === 0) {
+        return;
+    }
+    this.update((state) => {
+        for (const transaction of state.transactions) {
+            if (sameNonceTransactionIds.includes(transaction.id)) {
+                transaction.replacedBy = transactionMeta?.hash;
+                transaction.replacedById = transactionMeta?.id;
+            }
+        }
+    });
+    for (const transaction of this.state.transactions) {
+        if (sameNonceTransactionIds.includes(transaction.id) &&
+            transaction.status !== TransactionStatus.failed) {
+            this.setTransactionStatusDropped(transaction);
+        }
+    }
+}, _TransactionController_signTransaction = async function _TransactionController_signTransaction(transactionMeta) {
+    const { isExternalSign, txParams } = transactionMeta;
+    if (isExternalSign) {
+        log('Skipping sign as signed externally');
+        return undefined;
+    }
+    log('Signing transaction', txParams);
+    const { authorizationList, from } = txParams;
+    const finalTxParams = { ...txParams };
+    finalTxParams.authorizationList = await signAuthorizationList({
+        authorizationList,
+        messenger: this.messagingSystem,
+        transactionMeta,
+    });
+    const unsignedEthTx = prepareTransaction(transactionMeta.chainId, finalTxParams);
+    this.approvingTransactionIds.add(transactionMeta.id);
+    const signedTx = await new Promise((resolve, reject) => {
+        this.sign?.(unsignedEthTx, from, ...this.getAdditionalSignArguments(transactionMeta)).then(resolve, reject);
+        this.signAbortCallbacks.set(transactionMeta.id, () => reject(new Error('Signing aborted by user')));
+    });
+    this.signAbortCallbacks.delete(transactionMeta.id);
+    if (!signedTx) {
+        log('Skipping signed status as no signed transaction');
+        return undefined;
+    }
+    const transactionMetaFromHook = cloneDeep(transactionMeta);
+    if (!this.afterSign(transactionMetaFromHook, signedTx)) {
+        this.updateTransaction(transactionMetaFromHook, 'TransactionController#signTransaction - Update after sign');
+        log('Skipping signed status based on hook');
+        return undefined;
+    }
+    const transactionMetaWithRsv = {
+        ...this.updateTransactionMetaRSV(transactionMetaFromHook, signedTx),
+        status: TransactionStatus.signed,
+        txParams: finalTxParams,
+    };
+    this.updateTransaction(transactionMetaWithRsv, 'TransactionController#approveTransaction - Transaction signed');
+    this.onTransactionStatusChange(transactionMetaWithRsv);
+    const rawTx = serializeTransaction(signedTx);
+    const transactionMetaWithRawTx = merge({}, transactionMetaWithRsv, {
+        rawTx,
+    });
+    this.updateTransaction(transactionMetaWithRawTx, 'TransactionController#approveTransaction - RawTransaction added');
+    return rawTx;
+}, _TransactionController_onConfirmedTransaction = function _TransactionController_onConfirmedTransaction(transactionMeta) {
+    log('Processing confirmed transaction', transactionMeta.id);
+    __classPrivateFieldGet(this, _TransactionController_instances, "m", _TransactionController_markNonceDuplicatesDropped).call(this, transactionMeta.id);
+    this.messagingSystem.publish(`${controllerName}:transactionConfirmed`, transactionMeta);
+    this.onTransactionStatusChange(transactionMeta);
+    // Intentional given potential duration of process.
+    this.updatePostBalance(transactionMeta).catch((error) => {
+        log('Error while updating post balance', error);
+        throw error;
+    });
 }, _TransactionController_createNonceTracker = function _TransactionController_createNonceTracker({ provider, blockTracker, chainId, }) {
     return new NonceTracker({
         // TODO: Fix types
@@ -1967,7 +1979,6 @@ _TransactionController_internalEvents = new WeakMap(), _TransactionController_me
         }),
         hooks: {
             beforeCheckPendingTransaction: this.beforeCheckPendingTransaction.bind(this),
-            beforePublish: this.beforePublish.bind(this),
         },
     });
     __classPrivateFieldGet(this, _TransactionController_instances, "m", _TransactionController_addPendingTransactionTrackerListeners).call(this, pendingTransactionTracker);
@@ -1982,7 +1993,7 @@ _TransactionController_internalEvents = new WeakMap(), _TransactionController_me
     pendingTransactionTracker.hub.removeAllListeners('transaction-failed');
     pendingTransactionTracker.hub.removeAllListeners('transaction-updated');
 }, _TransactionController_addPendingTransactionTrackerListeners = function _TransactionController_addPendingTransactionTrackerListeners(pendingTransactionTracker) {
-    pendingTransactionTracker.hub.on('transaction-confirmed', this.onConfirmedTransaction.bind(this));
+    pendingTransactionTracker.hub.on('transaction-confirmed', __classPrivateFieldGet(this, _TransactionController_instances, "m", _TransactionController_onConfirmedTransaction).bind(this));
     pendingTransactionTracker.hub.on('transaction-dropped', this.setTransactionStatusDropped.bind(this));
     pendingTransactionTracker.hub.on('transaction-failed', this.failTransaction.bind(this));
     pendingTransactionTracker.hub.on('transaction-updated', this.updateTransaction.bind(this));
@@ -2183,8 +2194,11 @@ _TransactionController_internalEvents = new WeakMap(), _TransactionController_me
         ethQuery,
         isCustomNetwork,
         isSimulationEnabled: __classPrivateFieldGet(this, _TransactionController_isSimulationEnabled, "f").call(this),
+        messenger: this.messagingSystem,
         txMeta: transactionMeta,
     });
+}, _TransactionController_registerActionHandlers = function _TransactionController_registerActionHandlers() {
+    this.messagingSystem.registerActionHandler(`${controllerName}:updateCustodialTransaction`, this.updateCustodialTransaction.bind(this));
 }, _TransactionController_deleteTransaction = function _TransactionController_deleteTransaction(transactionId) {
     this.update((state) => {
         const transactions = state.transactions.filter(({ id }) => id !== transactionId);