refactor: migrate {AccountActivity,BackendWebSocket}Service to @metamask/messenger

README.md

@@ -201,8 +201,8 @@ linkStyle default opacity:0.5
   chain_agnostic_permission --> permission_controller;
   composable_controller --> base_controller;
   composable_controller --> json_rpc_engine;
-  core_backend --> base_controller;
   core_backend --> controller_utils;
+  core_backend --> messenger;
   core_backend --> profile_sync_controller;
   core_backend --> accounts_controller;
   core_backend --> keyring_controller;
@@ -216,6 +216,7 @@ linkStyle default opacity:0.5
   earn_controller --> transaction_controller;
   eip_5792_middleware --> transaction_controller;
   eip_5792_middleware --> keyring_controller;
+  eip_7702_internal_rpc_middleware --> controller_utils;
   eip1193_permission_middleware --> chain_agnostic_permission;
   eip1193_permission_middleware --> controller_utils;
   eip1193_permission_middleware --> json_rpc_engine;

packages/core-backend/CHANGELOG.md

@@ -9,6 +9,16 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Changed
 
+- **BREAKING:** Use new `Messenger` from `@metamask/messenger` ([#6823](https://github.com/MetaMask/core/pull/6823))
+  - Previously, `AccountActivityService` and `BackendWebSocketService` accepted a `RestrictedMessenger` instance from `@metamask/base-controller`.
+- **BREAKING:** Metadata property `anonymous` renamed to `includeInDebugSnapshot` ([#6823](https://github.com/MetaMask/core/pull/6823))
+
+### Removed
+
+- **BREAKING:** Remove exported type aliases and constants that were specific to controller messenger integration ([#6823](https://github.com/MetaMask/core/pull/6823))
+  - Removed type exports: `BackendWebSocketServiceAllowedActions`, `BackendWebSocketServiceAllowedEvents`, `AccountActivityServiceAllowedActions`, `AccountActivityServiceAllowedEvents`
+  - Removed constant exports: `ACCOUNT_ACTIVITY_SERVICE_ALLOWED_ACTIONS`, `ACCOUNT_ACTIVITY_SERVICE_ALLOWED_EVENTS`
+  - These types and constants were internal implementation details that should not have been exposed. Consumers should use the service-specific messenger types directly.
 - Bump `@metamask/profile-sync-controller` from `^25.1.1` to `^25.1.2` ([#6940](https://github.com/MetaMask/core/pull/6940))
 
 ## [3.0.0]

packages/core-backend/package.json

@@ -47,8 +47,8 @@
     "test:watch": "NODE_OPTIONS=--experimental-vm-modules jest --watch"
   },
   "dependencies": {
-    "@metamask/base-controller": "^8.4.2",
     "@metamask/controller-utils": "^11.14.1",
+    "@metamask/messenger": "^0.3.0",
     "@metamask/profile-sync-controller": "^25.1.2",
     "@metamask/utils": "^11.8.1",
     "uuid": "^8.3.2"

packages/core-backend/src/AccountActivityService.test.ts

@@ -1,24 +1,36 @@
-import { Messenger } from '@metamask/base-controller';
 import type { InternalAccount } from '@metamask/keyring-internal-api';
+import {
+  Messenger,
+  MOCK_ANY_NAMESPACE,
+  type MessengerActions,
+  type MessengerEvents,
+  type MockAnyNamespace,
+} from '@metamask/messenger';
 import type { Hex } from '@metamask/utils';
 
-import type {
-  AccountActivityServiceAllowedEvents,
-  AccountActivityServiceAllowedActions,
-} from './AccountActivityService';
 import {
   AccountActivityService,
   type AccountActivityServiceMessenger,
   type SubscriptionOptions,
-  ACCOUNT_ACTIVITY_SERVICE_ALLOWED_ACTIONS,
-  ACCOUNT_ACTIVITY_SERVICE_ALLOWED_EVENTS,
 } from './AccountActivityService';
 import type { ServerNotificationMessage } from './BackendWebSocketService';
 import { WebSocketState } from './BackendWebSocketService';
 import type { Transaction, BalanceUpdate } from './types';
 import type { AccountActivityMessage } from './types';
 import { flushPromises } from '../../../tests/helpers';
 
+type AllAccountActivityServiceActions =
+  MessengerActions<AccountActivityServiceMessenger>;
+
+type AllAccountActivityServiceEvents =
+  MessengerEvents<AccountActivityServiceMessenger>;
+
+type RootMessenger = Messenger<
+  MockAnyNamespace,
+  AllAccountActivityServiceActions,
+  AllAccountActivityServiceEvents
+>;
+
 // Helper function for completing async operations
 const completeAsyncOperations = async (timeoutMs = 0) => {
   await flushPromises();
@@ -48,25 +60,70 @@ const createMockInternalAccount = (options: {
   scopes: ['eip155:1'], // Required scopes property
 });
 
+/**
+ * Creates and returns a root messenger for testing
+ *
+ * @returns A messenger instance
+ */
+function getRootMessenger(): RootMessenger {
+  return new Messenger({
+    namespace: MOCK_ANY_NAMESPACE,
+  });
+}
+
 /**
  * Creates a real messenger with registered mock actions for testing
  * Each call creates a completely independent messenger to ensure test isolation
  *
  * @returns Object containing the messenger and mock action functions
  */
-const getMessenger = () => {
+const getMessenger = (): {
+  rootMessenger: RootMessenger;
+  messenger: AccountActivityServiceMessenger;
+  mocks: {
+    getSelectedAccount: jest.Mock;
+    connect: jest.Mock;
+    subscribe: jest.Mock;
+    channelHasSubscription: jest.Mock;
+    getSubscriptionsByChannel: jest.Mock;
+    findSubscriptionsByChannelPrefix: jest.Mock;
+    forceReconnection: jest.Mock;
+    addChannelCallback: jest.Mock;
+    removeChannelCallback: jest.Mock;
+  };
+} => {
   // Use any types for the root messenger to avoid complex type constraints in tests
   // Create a unique root messenger for each test
-  const rootMessenger = new Messenger<
-    AccountActivityServiceAllowedActions,
-    AccountActivityServiceAllowedEvents
-  >();
-  const messenger: AccountActivityServiceMessenger =
-    rootMessenger.getRestricted({
-      name: 'AccountActivityService',
-      allowedActions: [...ACCOUNT_ACTIVITY_SERVICE_ALLOWED_ACTIONS],
-      allowedEvents: [...ACCOUNT_ACTIVITY_SERVICE_ALLOWED_EVENTS],
-    });
+  const rootMessenger = getRootMessenger();
+  const messenger: AccountActivityServiceMessenger = new Messenger<
+    'AccountActivityService',
+    AllAccountActivityServiceActions,
+    AllAccountActivityServiceEvents,
+    RootMessenger
+  >({
+    namespace: 'AccountActivityService',
+    parent: rootMessenger,
+  });
+
+  rootMessenger.delegate({
+    actions: [
+      'AccountsController:getSelectedAccount',
+      'BackendWebSocketService:connect',
+      'BackendWebSocketService:forceReconnection',
+      'BackendWebSocketService:subscribe',
+      'BackendWebSocketService:getConnectionInfo',
+      'BackendWebSocketService:channelHasSubscription',
+      'BackendWebSocketService:getSubscriptionsByChannel',
+      'BackendWebSocketService:findSubscriptionsByChannelPrefix',
+      'BackendWebSocketService:addChannelCallback',
+      'BackendWebSocketService:removeChannelCallback',
+    ],
+    events: [
+      'AccountsController:selectedAccountChange',
+      'BackendWebSocketService:connectionStateChanged',
+    ],
+    messenger,
+  });
 
   // Create mock action handlers
   const mockGetSelectedAccount = jest.fn();
@@ -215,10 +272,7 @@ type WithServiceOptions = {
 type WithServiceCallback<ReturnValue> = (payload: {
   service: AccountActivityService;
   messenger: AccountActivityServiceMessenger;
-  rootMessenger: Messenger<
-    AccountActivityServiceAllowedActions,
-    AccountActivityServiceAllowedEvents
-  >;
+  rootMessenger: RootMessenger;
   mocks: {
     getSelectedAccount: jest.Mock;
     connect: jest.Mock;
@@ -633,7 +687,7 @@ describe('AccountActivityService', () => {
           });
 
           // Publish WebSocket ERROR state event - should flush tracked chains as down
-          await rootMessenger.publish(
+          rootMessenger.publish(
             'BackendWebSocketService:connectionStateChanged',
             {
               state: WebSocketState.ERROR,
@@ -667,7 +721,7 @@ describe('AccountActivityService', () => {
           mocks.getSelectedAccount.mockReturnValue(null);
 
           // Publish WebSocket ERROR state event without any tracked chains
-          await rootMessenger.publish(
+          rootMessenger.publish(
             'BackendWebSocketService:connectionStateChanged',
             {
               state: WebSocketState.ERROR,
@@ -718,7 +772,7 @@ describe('AccountActivityService', () => {
           });
 
           // Publish account change event - will be picked up by controller subscription
-          await rootMessenger.publish(
+          rootMessenger.publish(
             'AccountsController:selectedAccountChange',
             solanaAccount,
           );
@@ -748,7 +802,7 @@ describe('AccountActivityService', () => {
           });
 
           // Publish account change event - will be picked up by controller subscription
-          await rootMessenger.publish(
+          rootMessenger.publish(
             'AccountsController:selectedAccountChange',
             unknownAccount,
           );
@@ -770,7 +824,7 @@ describe('AccountActivityService', () => {
           mocks.getSelectedAccount.mockReturnValue(null);
 
           // Publish WebSocket connection event - will be picked up by controller subscription
-          await rootMessenger.publish(
+          rootMessenger.publish(
             'BackendWebSocketService:connectionStateChanged',
             {
               state: WebSocketState.CONNECTED,
@@ -810,7 +864,7 @@ describe('AccountActivityService', () => {
             });
 
             // Publish account change event on root messenger
-            await rootMessenger.publish(
+            rootMessenger.publish(
               'AccountsController:selectedAccountChange',
               newAccount,
             );
@@ -848,7 +902,7 @@ describe('AccountActivityService', () => {
             });
 
             // Publish account change event on root messenger
-            await rootMessenger.publish(
+            rootMessenger.publish(
               'AccountsController:selectedAccountChange',
               newAccount,
             );

packages/core-backend/src/AccountActivityService.ts

@@ -9,9 +9,9 @@ import type {
   AccountsControllerGetSelectedAccountAction,
   AccountsControllerSelectedAccountChangeEvent,
 } from '@metamask/accounts-controller';
-import type { RestrictedMessenger } from '@metamask/base-controller';
 import type { TraceCallback } from '@metamask/controller-utils';
 import type { InternalAccount } from '@metamask/keyring-internal-api';
+import type { Messenger } from '@metamask/messenger';
 
 import type { AccountActivityServiceMethodActions } from './AccountActivityService-method-action-types';
 import type {
@@ -96,7 +96,7 @@ export const ACCOUNT_ACTIVITY_SERVICE_ALLOWED_EVENTS = [
   'BackendWebSocketService:connectionStateChanged',
 ] as const;
 
-export type AccountActivityServiceAllowedActions =
+export type AllowedActions =
   | AccountsControllerGetSelectedAccountAction
   | BackendWebSocketServiceMethodActions;
 
@@ -134,16 +134,14 @@ export type AccountActivityServiceEvents =
   | AccountActivityServiceSubscriptionErrorEvent
   | AccountActivityServiceStatusChangedEvent;
 
-export type AccountActivityServiceAllowedEvents =
+export type AllowedEvents =
   | AccountsControllerSelectedAccountChangeEvent
   | BackendWebSocketServiceConnectionStateChangedEvent;
 
-export type AccountActivityServiceMessenger = RestrictedMessenger<
+export type AccountActivityServiceMessenger = Messenger<
   typeof SERVICE_NAME,
-  AccountActivityServiceActions | AccountActivityServiceAllowedActions,
-  AccountActivityServiceEvents | AccountActivityServiceAllowedEvents,
-  AccountActivityServiceAllowedActions['type'],
-  AccountActivityServiceAllowedEvents['type']
+  AccountActivityServiceActions | AllowedActions,
+  AccountActivityServiceEvents | AllowedEvents
 >;
 
 // =============================================================================