feat: Add transaction emulation actions

These will be used indirectly by the UserOperationController to emulate
adding and updating a transaction, in response to user operations.

Author: Gudahtt

packages/transaction-controller/CHANGELOG.md

@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Added
+
+- Add actions `TranasctionController:emulateNewTransaction` and `TranasctionController:emulateTransactionUpdate` ([#6935](https://github.com/MetaMask/core/pull/6935))
+
 ### Changed
 
 - Bump `@metamask/base-controller` from `^8.4.1` to `^8.4.2` ([#6917](https://github.com/MetaMask/core/pull/6917))

packages/transaction-controller/src/TransactionController.ts

@@ -357,6 +357,26 @@ export type TransactionControllerAddTransactionBatchAction = {
   handler: TransactionController['addTransactionBatch'];
 };
 
+/**
+ * Emulate a new transaction.
+ *
+ * @param transactionId - The transaction ID.
+ */
+export type TransactionControllerEmulateNewTransaction = {
+  type: `${typeof controllerName}:emulateNewTransaction`;
+  handler: TransactionController['emulateNewTransaction'];
+};
+
+/**
+ * Emmulate a transaction update.
+ *
+ * @param transactionMeta - Transaction metadata.
+ */
+export type TransactionControllerEmulateTransactionUpdate = {
+  type: `${typeof controllerName}:emulateTransactionUpdate`;
+  handler: TransactionController['emulateTransactionUpdate'];
+};
+
 /**
  * The internal actions available to the TransactionController.
  */
@@ -369,7 +389,9 @@ export type TransactionControllerActions =
   | TransactionControllerGetStateAction
   | TransactionControllerGetTransactionsAction
   | TransactionControllerUpdateCustodialTransactionAction
-  | TransactionControllerUpdateTransactionAction;
+  | TransactionControllerUpdateTransactionAction
+  | TransactionControllerEmulateNewTransaction
+  | TransactionControllerEmulateTransactionUpdate;
 
 /**
  * Configuration options for the PendingTransactionTracker
@@ -4552,4 +4574,66 @@ export class TransactionController extends BaseController<
 
     return { transactionHash };
   }
+
+  /**
+   * Emulate a new transaction.
+   *
+   * @param transactionId - The transaction ID.
+   */
+  emulateNewTransaction(transactionId: string) {
+    const transactionMeta = this.state.transactions.find(
+      (tx) => tx.id === transactionId,
+    );
+
+    if (!transactionMeta) {
+      return;
+    }
+
+    if (transactionMeta.type === TransactionType.swap) {
+      this.messagingSystem.publish('TransactionController:transactionNewSwap', {
+        transactionMeta,
+      });
+    } else if (transactionMeta.type === TransactionType.swapApproval) {
+      this.messagingSystem.publish(
+        'TransactionController:transactionNewSwapApproval',
+        { transactionMeta },
+      );
+    }
+  }
+
+  /**
+   * Emmulate a transaction update.
+   *
+   * @param transactionMeta - Transaction metadata.
+   */
+  emulateTransactionUpdate(transactionMeta: TransactionMeta) {
+    const updatedTransactionMeta = {
+      ...transactionMeta,
+      txParams: {
+        ...transactionMeta.txParams,
+        from: this.messagingSystem.call('AccountsController:getSelectedAccount')
+          .address,
+      },
+    };
+
+    const transactionExists = this.state.transactions.some(
+      (tx) => tx.id === updatedTransactionMeta.id,
+    );
+
+    if (!transactionExists) {
+      this.update((state) => {
+        state.transactions.push(updatedTransactionMeta);
+      });
+    }
+
+    this.updateTransaction(
+      updatedTransactionMeta,
+      'Generated from user operation',
+    );
+
+    this.messagingSystem.publish(
+      'TransactionController:transactionStatusUpdated',
+      { transactionMeta: updatedTransactionMeta },
+    );
+  }
 }