@metamask/transaction-controller 17.0.0 → 18.0.0

package/dist/TransactionController.js

@@ -24,7 +24,7 @@ const eth_method_registry_1 = __importDefault(require("eth-method-registry"));
 const ethereumjs_util_1 = require("ethereumjs-util");
 const events_1 = require("events");
 const lodash_1 = require("lodash");
-const nonce_tracker_1 = __importDefault(require("nonce-tracker"));
+const nonce_tracker_1 = require("nonce-tracker");
 const uuid_1 = require("uuid");
 const EtherscanRemoteTransactionSource_1 = require("./helpers/EtherscanRemoteTransactionSource");
 const IncomingTransactionHelper_1 = require("./helpers/IncomingTransactionHelper");
@@ -55,7 +55,7 @@ const controllerName = 'TransactionController';
 /**
  * Controller responsible for submitting and managing transactions.
  */
-class TransactionController extends base_controller_1.BaseController {
+class TransactionController extends base_controller_1.BaseControllerV1 {
     /**
      * Creates a TransactionController instance.
      *
@@ -123,7 +123,7 @@ class TransactionController extends base_controller_1.BaseController {
         this.registry = new eth_method_registry_1.default({ provider });
         this.getSavedGasFees = getSavedGasFees !== null && getSavedGasFees !== void 0 ? getSavedGasFees : ((_chainId) => undefined);
         this.getCurrentAccountEIP1559Compatibility =
-            getCurrentAccountEIP1559Compatibility;
+            getCurrentAccountEIP1559Compatibility !== null && getCurrentAccountEIP1559Compatibility !== void 0 ? getCurrentAccountEIP1559Compatibility : (() => Promise.resolve(true));
         this.getCurrentNetworkEIP1559Compatibility =
             getCurrentNetworkEIP1559Compatibility;
         this.getGasFeeEstimates =
@@ -140,11 +140,12 @@ class TransactionController extends base_controller_1.BaseController {
         this.beforePublish = (_d = hooks === null || hooks === void 0 ? void 0 : hooks.beforePublish) !== null && _d !== void 0 ? _d : (() => true);
         this.getAdditionalSignArguments =
             (_e = hooks === null || hooks === void 0 ? void 0 : hooks.getAdditionalSignArguments) !== null && _e !== void 0 ? _e : (() => []);
-        this.nonceTracker = new nonce_tracker_1.default({
+        this.nonceTracker = new nonce_tracker_1.NonceTracker({
+            // @ts-expect-error provider types misaligned: SafeEventEmitterProvider vs Record<string,string>
             provider,
             blockTracker,
-            getPendingTransactions: (address) => (0, utils_1.getAndFormatTransactionsForNonceTracker)(address, types_1.TransactionStatus.submitted, this.state.transactions),
-            getConfirmedTransactions: (address) => (0, utils_1.getAndFormatTransactionsForNonceTracker)(address, types_1.TransactionStatus.confirmed, this.state.transactions),
+            getPendingTransactions: this.getNonceTrackerTransactions.bind(this, types_1.TransactionStatus.submitted),
+            getConfirmedTransactions: this.getNonceTrackerTransactions.bind(this, types_1.TransactionStatus.confirmed),
         });
         this.incomingTransactionHelper = new IncomingTransactionHelper_1.IncomingTransactionHelper({
             blockTracker,
@@ -192,6 +193,7 @@ class TransactionController extends base_controller_1.BaseController {
             transactionMeta: newTransactionMeta,
         });
         this.updateTransaction(newTransactionMeta, 'TransactionController#failTransaction - Add error message and set status to failed');
+        this.onTransactionStatusChange(newTransactionMeta);
         this.hub.emit(`${transactionMeta.id}:finished`, newTransactionMeta);
     }
     registryLookup(fourBytePrefix) {
@@ -249,7 +251,7 @@ class TransactionController extends base_controller_1.BaseController {
      */
     addTransaction(txParams, { actionId, deviceConfirmedOn, method, origin, requireApproval, securityAlertResponse, sendFlowHistory, swaps = {}, type, } = {}) {
         return __awaiter(this, void 0, void 0, function* () {
-            const chainId = this.getChainId();
+            (0, logger_1.projectLogger)('Adding transaction', txParams);
             txParams = (0, utils_1.normalizeTxParams)(txParams);
             const isEIP1559Compatible = yield this.getEIP1559Compatibility();
             (0, validation_1.validateTxParams)(txParams, isEIP1559Compatible);
@@ -259,6 +261,7 @@ class TransactionController extends base_controller_1.BaseController {
             const dappSuggestedGasFees = this.generateDappSuggestedGasFees(txParams, origin);
             const transactionType = type !== null && type !== void 0 ? type : (yield (0, transaction_type_1.determineTransactionType)(txParams, this.ethQuery)).type;
             const existingTransactionMeta = this.getTransactionWithActionId(actionId);
+            const chainId = this.getChainId();
             // If a request to add a transaction with the same actionId is submitted again, a new transaction will not be created for it.
             const transactionMeta = existingTransactionMeta || {
                 // Add actionId to txMeta to check if same actionId is seen again
@@ -338,6 +341,8 @@ class TransactionController extends base_controller_1.BaseController {
                 return;
             }
             if (gasValues) {
+                // Not good practice to reassign a parameter but temporarily avoiding a larger refactor.
+                gasValues = (0, utils_1.normalizeGasFeeValues)(gasValues);
                 (0, utils_1.validateGasValues)(gasValues);
             }
             const transactionMeta = this.getTransaction(transactionId);
@@ -373,7 +378,7 @@ class TransactionController extends base_controller_1.BaseController {
                     gasLimit: transactionMeta.txParams.gas,
                     maxFeePerGas: newMaxFeePerGas,
                     maxPriorityFeePerGas: newMaxPriorityFeePerGas,
-                    type: '2',
+                    type: types_1.TransactionEnvelopeType.feeMarket,
                     nonce: transactionMeta.txParams.nonce,
                     to: transactionMeta.txParams.from,
                     value: '0x0',
@@ -432,6 +437,8 @@ class TransactionController extends base_controller_1.BaseController {
                 return;
             }
             if (gasValues) {
+                // Not good practice to reassign a parameter but temporarily avoiding a larger refactor.
+                gasValues = (0, utils_1.normalizeGasFeeValues)(gasValues);
                 (0, utils_1.validateGasValues)(gasValues);
             }
             const transactionMeta = this.state.transactions.find(({ id }) => id === transactionId);
@@ -464,7 +471,7 @@ class TransactionController extends base_controller_1.BaseController {
                 (0, utils_1.validateMinimumIncrease)(maxPriorityFeePerGasValues, minMaxPriorityFeePerGas)) ||
                 (existingMaxPriorityFeePerGas && minMaxPriorityFeePerGas);
             const txParams = newMaxFeePerGas && newMaxPriorityFeePerGas
-                ? Object.assign(Object.assign({}, transactionMeta.txParams), { gasLimit: transactionMeta.txParams.gas, maxFeePerGas: newMaxFeePerGas, maxPriorityFeePerGas: newMaxPriorityFeePerGas, type: '2' }) : Object.assign(Object.assign({}, transactionMeta.txParams), { gasLimit: transactionMeta.txParams.gas, gasPrice: newGasPrice });
+                ? Object.assign(Object.assign({}, transactionMeta.txParams), { gasLimit: transactionMeta.txParams.gas, maxFeePerGas: newMaxFeePerGas, maxPriorityFeePerGas: newMaxPriorityFeePerGas, type: types_1.TransactionEnvelopeType.feeMarket }) : Object.assign(Object.assign({}, transactionMeta.txParams), { gasLimit: transactionMeta.txParams.gas, gasPrice: newGasPrice });
             const unsignedEthTx = this.prepareUnsignedEthTx(txParams);
             const signedTx = yield this.sign(unsignedEthTx, transactionMeta.txParams.from);
             yield this.updateTransactionMetaRSV(transactionMeta, signedTx);
@@ -606,29 +613,16 @@ class TransactionController extends base_controller_1.BaseController {
                 this.markNonceDuplicatesDropped(transactionId);
                 // Update external provided transaction with updated gas values and confirmed status.
                 this.updateTransaction(transactionMeta, 'TransactionController:confirmExternalTransaction - Add external transaction');
-                if (transactionMeta.type === types_1.TransactionType.swap) {
-                    (0, swaps_1.updatePostTransactionBalance)(transactionMeta, {
-                        ethQuery: this.ethQuery,
-                        getTransaction: this.getTransaction.bind(this),
-                        updateTransaction: this.updateTransaction.bind(this),
-                    })
-                        .then(({ updatedTransactionMeta, approvalTransactionMeta }) => {
-                        this.hub.emit('post-transaction-balance-updated', {
-                            transactionMeta: updatedTransactionMeta,
-                            approvalTransactionMeta,
-                        });
-                    })
-                        .catch((error) => {
-                        /* istanbul ignore next */
-                        (0, logger_1.projectLogger)('Error while updating post transaction balance', error);
-                    });
-                }
+                this.onTransactionStatusChange(transactionMeta);
+                // Intentional given potential duration of process.
+                // eslint-disable-next-line @typescript-eslint/no-floating-promises
+                this.updatePostBalance(transactionMeta);
                 this.hub.emit('transaction-confirmed', {
                     transactionMeta,
                 });
             }
             catch (error) {
-                console.error(error);
+                console.error('Failed to confirm external transaction', error);
             }
         });
     }
@@ -749,6 +743,44 @@ class TransactionController extends base_controller_1.BaseController {
             return this.nonceTracker.getNonceLock(address);
         });
     }
+    /**
+     * Updates the editable parameters of a transaction.
+     *
+     * @param txId - The ID of the transaction to update.
+     * @param params - The editable parameters to update.
+     * @param params.data - Data to pass with the transaction.
+     * @param params.gas - Maximum number of units of gas to use for the transaction.
+     * @param params.gasPrice - Price per gas for legacy transactions.
+     * @param params.from - Address to send the transaction from.
+     * @param params.to - Address to send the transaction to.
+     * @param params.value - Value associated with the transaction.
+     * @returns The updated transaction metadata.
+     */
+    updateEditableParams(txId, { data, gas, gasPrice, from, to, value, }) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const transactionMeta = this.getTransaction(txId);
+            if (!transactionMeta) {
+                throw new Error(`Cannot update editable params as no transaction metadata found`);
+            }
+            (0, utils_1.validateIfTransactionUnapproved)(transactionMeta, 'updateEditableParams');
+            const editableParams = {
+                txParams: {
+                    data,
+                    from,
+                    to,
+                    value,
+                    gas,
+                    gasPrice,
+                },
+            };
+            editableParams.txParams = (0, lodash_1.pickBy)(editableParams.txParams);
+            const updatedTransaction = (0, lodash_1.merge)(transactionMeta, editableParams);
+            const { type } = yield (0, transaction_type_1.determineTransactionType)(updatedTransaction.txParams, this.ethQuery);
+            updatedTransaction.type = type;
+            this.updateTransaction(updatedTransaction, `Update Editable Params for ${txId}`);
+            return this.getTransaction(txId);
+        });
+    }
     /**
      * Signs and returns the raw transaction data for provided transaction params list.
      *
@@ -801,14 +833,12 @@ class TransactionController extends base_controller_1.BaseController {
      *
      * @param transactionId - The ID of the transaction to update.
      * @param options - The custodial transaction options to update.
-     * @param options.custodyStatus - The new custody status value to be assigned.
      * @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.
      */
-    updateCustodialTransaction(transactionId, { custodyStatus, errorMessage, hash, status, }) {
-        let transactionMeta;
-        transactionMeta = this.getTransaction(transactionId);
+    updateCustodialTransaction(transactionId, { errorMessage, hash, status, }) {
+        const transactionMeta = this.getTransaction(transactionId);
         if (!transactionMeta) {
             throw new Error(`Cannot update custodial transaction as no transaction metadata found`);
         }
@@ -823,23 +853,112 @@ class TransactionController extends base_controller_1.BaseController {
             ].includes(status)) {
             throw new Error(`Cannot update custodial transaction with status: ${status}`);
         }
-        if (status === types_1.TransactionStatus.signed) {
-            transactionMeta.status = status;
-        }
+        const updatedTransactionMeta = (0, lodash_1.merge)(transactionMeta, (0, lodash_1.pickBy)({ hash, status }));
         if (status === types_1.TransactionStatus.submitted) {
-            transactionMeta.submittedTime = new Date().getTime();
-            transactionMeta.status = status;
+            updatedTransactionMeta.submittedTime = new Date().getTime();
         }
         if (status === types_1.TransactionStatus.failed) {
-            transactionMeta = Object.assign(Object.assign({}, transactionMeta), { error: (0, utils_1.normalizeTxError)(new Error(errorMessage)), status: types_1.TransactionStatus.failed });
+            updatedTransactionMeta.error = (0, utils_1.normalizeTxError)(new Error(errorMessage));
         }
-        if (custodyStatus) {
-            transactionMeta.custodyStatus = custodyStatus;
+        this.updateTransaction(updatedTransactionMeta, `TransactionController:updateCustodialTransaction - Custodial transaction updated`);
+    }
+    /**
+     * Creates approvals for all unapproved transactions persisted.
+     */
+    initApprovals() {
+        const chainId = this.getChainId();
+        const unapprovedTxs = this.state.transactions.filter((transaction) => transaction.status === types_1.TransactionStatus.unapproved &&
+            transaction.chainId === chainId);
+        for (const txMeta of unapprovedTxs) {
+            this.processApproval(txMeta, {
+                shouldShowRequest: false,
+            }).catch((error) => {
+                if ((error === null || error === void 0 ? void 0 : error.code) === rpc_errors_1.errorCodes.provider.userRejectedRequest) {
+                    return;
+                }
+                console.error('Error during persisted transaction approval', error);
+            });
         }
-        if (hash) {
-            transactionMeta.hash = hash;
+    }
+    /**
+     * Search transaction metadata for matching entries.
+     *
+     * @param opts - Options bag.
+     * @param opts.searchCriteria - An object containing values or functions for transaction properties to filter transactions with.
+     * @param opts.initialList - The transactions to search. Defaults to the current state.
+     * @param opts.filterToCurrentNetwork - Whether to filter the results to the current network. Defaults to true.
+     * @param opts.limit - The maximum number of transactions to return. No limit by default.
+     * @returns An array of transactions matching the provided options.
+     */
+    getTransactions({ searchCriteria = {}, initialList, filterToCurrentNetwork = true, limit, } = {}) {
+        const chainId = this.getChainId();
+        // searchCriteria is an object that might have values that aren't predicate
+        // methods. When providing any other value type (string, number, etc), we
+        // consider this shorthand for "check the value at key for strict equality
+        // with the provided value". To conform this object to be only methods, we
+        // mapValues (lodash) such that every value on the object is a method that
+        // returns a boolean.
+        const predicateMethods = (0, lodash_1.mapValues)(searchCriteria, (predicate) => {
+            return typeof predicate === 'function'
+                ? predicate
+                : (v) => v === predicate;
+        });
+        const transactionsToFilter = initialList !== null && initialList !== void 0 ? initialList : this.state.transactions;
+        // Combine sortBy and pickBy to transform our state object into an array of
+        // matching transactions that are sorted by time.
+        const filteredTransactions = (0, lodash_1.sortBy)((0, lodash_1.pickBy)(transactionsToFilter, (transaction) => {
+            if (filterToCurrentNetwork && transaction.chainId !== chainId) {
+                return false;
+            }
+            // iterate over the predicateMethods keys to check if the transaction
+            // matches the searchCriteria
+            for (const [key, predicate] of Object.entries(predicateMethods)) {
+                // We return false early as soon as we know that one of the specified
+                // search criteria do not match the transaction. This prevents
+                // needlessly checking all criteria when we already know the criteria
+                // are not fully satisfied. We check both txParams and the base
+                // object as predicate keys can be either.
+                if (key in transaction.txParams) {
+                    if (predicate(transaction.txParams[key]) === false) {
+                        return false;
+                    }
+                }
+                else if (predicate(transaction[key]) === false) {
+                    return false;
+                }
+            }
+            return true;
+        }), 'time');
+        if (limit !== undefined) {
+            // We need to have all transactions of a given nonce in order to display
+            // necessary details in the UI. We use the size of this set to determine
+            // whether we have reached the limit provided, thus ensuring that all
+            // transactions of nonces we include will be sent to the UI.
+            const nonces = new Set();
+            const txs = [];
+            // By default, the transaction list we filter from is sorted by time ASC.
+            // To ensure that filtered results prefers the newest transactions we
+            // iterate from right to left, inserting transactions into front of a new
+            // array. The original order is preserved, but we ensure that newest txs
+            // are preferred.
+            for (let i = filteredTransactions.length - 1; i > -1; i--) {
+                const txMeta = filteredTransactions[i];
+                const { nonce } = txMeta.txParams;
+                if (!nonces.has(nonce)) {
+                    if (nonces.size < limit) {
+                        nonces.add(nonce);
+                    }
+                    else {
+                        continue;
+                    }
+                }
+                // Push transaction into the beginning of our array to ensure the
+                // original order is preserved.
+                txs.unshift(txMeta);
+            }
+            return txs;
         }
-        this.updateTransaction(transactionMeta, `TransactionController:updateCustodialTransaction - Custodial transaction updated`);
+        return filteredTransactions;
     }
     signExternalTransaction(transactionParams) {
         return __awaiter(this, void 0, void 0, function* () {
@@ -874,7 +993,8 @@ class TransactionController extends base_controller_1.BaseController {
     }
     updateGasProperties(transactionMeta) {
         return __awaiter(this, void 0, void 0, function* () {
-            const isEIP1559Compatible = yield this.getEIP1559Compatibility();
+            const isEIP1559Compatible = (yield this.getEIP1559Compatibility()) &&
+                transactionMeta.txParams.type !== types_1.TransactionEnvelopeType.legacy;
             const chainId = this.getChainId();
             yield (0, gas_1.updateGas)({
                 ethQuery: this.ethQuery,
@@ -895,8 +1015,6 @@ class TransactionController extends base_controller_1.BaseController {
         return this.state.transactions.filter((transaction) => transaction.status === status && transaction.chainId === chainId);
     }
     onBootCleanup() {
-        this.createApprovalsForUnapprovedTransactions();
-        this.loadGasValuesForUnapprovedTransactions();
         this.submitApprovedTransactions();
     }
     /**
@@ -916,26 +1034,6 @@ class TransactionController extends base_controller_1.BaseController {
             });
         }
     }
-    /**
-     * Update the gas values of all unapproved transactions on current chain.
-     */
-    loadGasValuesForUnapprovedTransactions() {
-        return __awaiter(this, void 0, void 0, function* () {
-            const unapprovedTransactions = this.getCurrentChainTransactionsByStatus(types_1.TransactionStatus.unapproved);
-            const results = yield Promise.allSettled(unapprovedTransactions.map((transactionMeta) => __awaiter(this, void 0, void 0, function* () {
-                yield this.updateGasProperties(transactionMeta);
-                this.updateTransaction(transactionMeta, 'TransactionController:loadGasValuesForUnapprovedTransactions - Gas values updated');
-            })));
-            for (const [index, result] of results.entries()) {
-                if (result.status === 'rejected') {
-                    const transactionMeta = unapprovedTransactions[index];
-                    this.failTransaction(transactionMeta, result.reason);
-                    /* istanbul ignore next */
-                    console.error('Error while loading gas values for persisted transaction id: ', transactionMeta.id, result.reason);
-                }
-            }
-        });
-    }
     /**
      * Force to submit approved transactions on current chain.
      */
@@ -965,6 +1063,13 @@ class TransactionController extends base_controller_1.BaseController {
                             shouldShowRequest,
                         });
                         resultCallbacks = acceptResult.resultCallbacks;
+                        if (resultCallbacks) {
+                            this.hub.once(`${transactionId}:publish-skip`, () => {
+                                resultCallbacks === null || resultCallbacks === void 0 ? void 0 : resultCallbacks.success();
+                                // Remove the reference to prevent additional reports once submitted.
+                                resultCallbacks = undefined;
+                            });
+                        }
                     }
                     const { isCompleted: isTxCompleted } = this.isTransactionCompleted(transactionId);
                     if (!isTxCompleted) {
@@ -1036,7 +1141,6 @@ class TransactionController extends base_controller_1.BaseController {
                     (0, logger_1.projectLogger)('Skipping approval as signing in progress', transactionId);
                     return;
                 }
-                const { approved: status } = types_1.TransactionStatus;
                 let nonceToUse = nonce;
                 // if a nonce already exists on the transactionMeta it means this is a speedup or cancel transaction
                 // so we want to reuse that nonce and hope that it beats the previous attempt to chain. Otherwise use a new locked nonce
@@ -1044,23 +1148,19 @@ class TransactionController extends base_controller_1.BaseController {
                     nonceLock = yield this.nonceTracker.getNonceLock(from);
                     nonceToUse = (0, ethereumjs_util_1.addHexPrefix)(nonceLock.nextNonce.toString(16));
                 }
-                transactionMeta.status = status;
+                transactionMeta.status = types_1.TransactionStatus.approved;
                 transactionMeta.txParams.nonce = nonceToUse;
                 transactionMeta.txParams.chainId = chainId;
                 const baseTxParams = Object.assign(Object.assign({}, transactionMeta.txParams), { gasLimit: transactionMeta.txParams.gas });
                 this.updateTransaction(transactionMeta, 'TransactionController#approveTransaction - Transaction approved');
+                this.onTransactionStatusChange(transactionMeta);
                 const isEIP1559 = (0, utils_1.isEIP1559Transaction)(transactionMeta.txParams);
                 const txParams = isEIP1559
-                    ? Object.assign(Object.assign({}, baseTxParams), { maxFeePerGas: transactionMeta.txParams.maxFeePerGas, maxPriorityFeePerGas: transactionMeta.txParams.maxPriorityFeePerGas, estimatedBaseFee: transactionMeta.txParams.estimatedBaseFee, 
-                        // specify type 2 if maxFeePerGas and maxPriorityFeePerGas are set
-                        type: '2' }) : baseTxParams;
-                // delete gasPrice if maxFeePerGas and maxPriorityFeePerGas are set
-                if (isEIP1559) {
-                    delete txParams.gasPrice;
-                }
-                const rawTx = yield this.signTransaction(transactionMeta);
+                    ? Object.assign(Object.assign({}, baseTxParams), { estimatedBaseFee: transactionMeta.txParams.estimatedBaseFee, type: types_1.TransactionEnvelopeType.feeMarket }) : baseTxParams;
+                const rawTx = yield this.signTransaction(transactionMeta, txParams);
                 if (!this.beforePublish(transactionMeta)) {
                     (0, logger_1.projectLogger)('Skipping publishing transaction based on hook');
+                    this.hub.emit(`${transactionMeta.id}:publish-skip`, transactionMeta);
                     return;
                 }
                 if (!rawTx) {
@@ -1070,11 +1170,12 @@ class TransactionController extends base_controller_1.BaseController {
                 transactionMeta.hash = hash;
                 transactionMeta.status = types_1.TransactionStatus.submitted;
                 transactionMeta.submittedTime = new Date().getTime();
+                this.updateTransaction(transactionMeta, 'TransactionController#approveTransaction - Transaction submitted');
                 this.hub.emit('transaction-submitted', {
                     transactionMeta,
                 });
-                this.updateTransaction(transactionMeta, 'TransactionController#approveTransaction - Transaction submitted');
                 this.hub.emit(`${transactionMeta.id}:finished`, transactionMeta);
+                this.onTransactionStatusChange(transactionMeta);
             }
             catch (error) {
                 this.failTransaction(transactionMeta, error);
@@ -1107,13 +1208,14 @@ class TransactionController extends base_controller_1.BaseController {
             return;
         }
         transactionMeta.status = types_1.TransactionStatus.rejected;
+        const transactions = this.state.transactions.filter(({ id }) => id !== transactionId);
+        this.update({ transactions: this.trimTransactionsForState(transactions) });
         this.hub.emit(`${transactionMeta.id}:finished`, transactionMeta);
         this.hub.emit('transaction-rejected', {
             transactionMeta,
             actionId,
         });
-        const transactions = this.state.transactions.filter(({ id }) => id !== transactionId);
-        this.update({ transactions: this.trimTransactionsForState(transactions) });
+        this.onTransactionStatusChange(transactionMeta);
     }
     /**
      * Trim the amount of transactions that are set on the state. Checks
@@ -1124,7 +1226,7 @@ class TransactionController extends base_controller_1.BaseController {
      * representation, this function will not break apart transactions with the
      * same nonce, created on the same day, per network. Not accounting for transactions of the same
      * nonce, same day and network combo can result in confusing or broken experiences
-     * in the UI. The transactions are then updated using the BaseController update.
+     * in the UI. The transactions are then updated using the BaseControllerV1 update.
      *
      * @param transactions - The transactions to be applied to the state.
      * @returns The trimmed list of transactions.
@@ -1288,25 +1390,23 @@ class TransactionController extends base_controller_1.BaseController {
      */
     addExternalTransaction(transactionMeta) {
         var _a, _b;
-        return __awaiter(this, void 0, void 0, function* () {
-            const chainId = this.getChainId();
-            const { transactions } = this.state;
-            const fromAddress = (_a = transactionMeta === null || transactionMeta === void 0 ? void 0 : transactionMeta.txParams) === null || _a === void 0 ? void 0 : _a.from;
-            const sameFromAndNetworkTransactions = transactions.filter((transaction) => transaction.txParams.from === fromAddress &&
-                transaction.chainId === chainId);
-            const confirmedTxs = sameFromAndNetworkTransactions.filter((transaction) => transaction.status === types_1.TransactionStatus.confirmed);
-            const pendingTxs = sameFromAndNetworkTransactions.filter((transaction) => transaction.status === types_1.TransactionStatus.submitted);
-            (0, external_transactions_1.validateConfirmedExternalTransaction)(transactionMeta, confirmedTxs, pendingTxs);
-            // Make sure provided external transaction has non empty history array
-            if (!((_b = transactionMeta.history) !== null && _b !== void 0 ? _b : []).length) {
-                if (!this.isHistoryDisabled) {
-                    (0, history_1.addInitialHistorySnapshot)(transactionMeta);
-                }
+        const chainId = this.getChainId();
+        const { transactions } = this.state;
+        const fromAddress = (_a = transactionMeta === null || transactionMeta === void 0 ? void 0 : transactionMeta.txParams) === null || _a === void 0 ? void 0 : _a.from;
+        const sameFromAndNetworkTransactions = transactions.filter((transaction) => transaction.txParams.from === fromAddress &&
+            transaction.chainId === chainId);
+        const confirmedTxs = sameFromAndNetworkTransactions.filter((transaction) => transaction.status === types_1.TransactionStatus.confirmed);
+        const pendingTxs = sameFromAndNetworkTransactions.filter((transaction) => transaction.status === types_1.TransactionStatus.submitted);
+        (0, external_transactions_1.validateConfirmedExternalTransaction)(transactionMeta, confirmedTxs, pendingTxs);
+        // Make sure provided external transaction has non empty history array
+        if (!((_b = transactionMeta.history) !== null && _b !== void 0 ? _b : []).length) {
+            if (!this.isHistoryDisabled) {
+                (0, history_1.addInitialHistorySnapshot)(transactionMeta);
             }
-            const updatedTransactions = [...transactions, transactionMeta];
-            this.update({
-                transactions: this.trimTransactionsForState(updatedTransactions),
-            });
+        }
+        const updatedTransactions = [...transactions, transactionMeta];
+        this.update({
+            transactions: this.trimTransactionsForState(updatedTransactions),
         });
     }
     /**
@@ -1351,6 +1451,7 @@ class TransactionController extends base_controller_1.BaseController {
             transactionMeta,
         });
         this.updateTransaction(transactionMeta, 'TransactionController#setTransactionStatusDropped - Transaction dropped');
+        this.onTransactionStatusChange(transactionMeta);
     }
     /**
      * Get transaction with provided actionId.
@@ -1391,26 +1492,22 @@ class TransactionController extends base_controller_1.BaseController {
         });
     }
     getEIP1559Compatibility() {
-        var _a, _b;
         return __awaiter(this, void 0, void 0, function* () {
             const currentNetworkIsEIP1559Compatible = yield this.getCurrentNetworkEIP1559Compatibility();
-            const currentAccountIsEIP1559Compatible = (_b = (_a = this.getCurrentAccountEIP1559Compatibility) === null || _a === void 0 ? void 0 : _a.call(this)) !== null && _b !== void 0 ? _b : true;
+            const currentAccountIsEIP1559Compatible = yield this.getCurrentAccountEIP1559Compatibility();
             return (currentNetworkIsEIP1559Compatible && currentAccountIsEIP1559Compatible);
         });
     }
     addPendingTransactionTrackerListeners() {
-        this.pendingTransactionTracker.hub.on('transaction-confirmed', (transactionMeta) => {
-            this.hub.emit('transaction-confirmed', { transactionMeta });
-            this.hub.emit(`${transactionMeta.id}:confirmed`, transactionMeta);
-        });
+        this.pendingTransactionTracker.hub.on('transaction-confirmed', this.onConfirmedTransaction.bind(this));
         this.pendingTransactionTracker.hub.on('transaction-dropped', this.setTransactionStatusDropped.bind(this));
         this.pendingTransactionTracker.hub.on('transaction-failed', this.failTransaction.bind(this));
         this.pendingTransactionTracker.hub.on('transaction-updated', this.updateTransaction.bind(this));
     }
-    signTransaction(transactionMeta) {
+    signTransaction(transactionMeta, txParams) {
         var _a;
         return __awaiter(this, void 0, void 0, function* () {
-            const { txParams } = transactionMeta;
+            (0, logger_1.projectLogger)('Signing transaction', txParams);
             const unsignedEthTx = this.prepareUnsignedEthTx(txParams);
             this.inProcessOfSigning.add(transactionMeta.id);
             const signedTx = yield ((_a = this.sign) === null || _a === void 0 ? void 0 : _a.call(this, unsignedEthTx, txParams.from, ...this.getAdditionalSignArguments(transactionMeta)));
@@ -1419,19 +1516,58 @@ class TransactionController extends base_controller_1.BaseController {
                 return undefined;
             }
             if (!this.afterSign(transactionMeta, signedTx)) {
+                this.updateTransaction(transactionMeta, 'TransactionController#signTransaction - Update after sign');
                 (0, logger_1.projectLogger)('Skipping signed status based on hook');
                 return undefined;
             }
             yield this.updateTransactionMetaRSV(transactionMeta, signedTx);
             transactionMeta.status = types_1.TransactionStatus.signed;
             this.updateTransaction(transactionMeta, 'TransactionController#approveTransaction - Transaction signed');
+            this.onTransactionStatusChange(transactionMeta);
             const rawTx = (0, ethereumjs_util_1.bufferToHex)(signedTx.serialize());
             transactionMeta.rawTx = rawTx;
             this.updateTransaction(transactionMeta, 'TransactionController#approveTransaction - RawTransaction added');
             return rawTx;
         });
     }
+    onTransactionStatusChange(transactionMeta) {
+        this.hub.emit('transaction-status-update', { transactionMeta });
+    }
+    getNonceTrackerTransactions(status, address) {
+        const currentChainId = this.getChainId();
+        return (0, utils_1.getAndFormatTransactionsForNonceTracker)(currentChainId, address, status, this.state.transactions);
+    }
+    onConfirmedTransaction(transactionMeta) {
+        (0, logger_1.projectLogger)('Processing confirmed transaction', transactionMeta.id);
+        this.hub.emit('transaction-confirmed', { transactionMeta });
+        this.hub.emit(`${transactionMeta.id}:confirmed`, transactionMeta);
+        this.onTransactionStatusChange(transactionMeta);
+        // Intentional given potential duration of process.
+        // eslint-disable-next-line @typescript-eslint/no-floating-promises
+        this.updatePostBalance(transactionMeta);
+    }
+    updatePostBalance(transactionMeta) {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                if (transactionMeta.type !== types_1.TransactionType.swap) {
+                    return;
+                }
+                const { updatedTransactionMeta, approvalTransactionMeta } = yield (0, swaps_1.updatePostTransactionBalance)(transactionMeta, {
+                    ethQuery: this.ethQuery,
+                    getTransaction: this.getTransaction.bind(this),
+                    updateTransaction: this.updateTransaction.bind(this),
+                });
+                this.hub.emit('post-transaction-balance-updated', {
+                    transactionMeta: updatedTransactionMeta,
+                    approvalTransactionMeta,
+                });
+            }
+            catch (error) {
+                /* istanbul ignore next */
+                (0, logger_1.projectLogger)('Error while updating post transaction balance', error);
+            }
+        });
+    }
 }
 exports.TransactionController = TransactionController;
-exports.default = TransactionController;
 //# sourceMappingURL=TransactionController.js.map