feat: migrate MetaMetricsController to BaseControllerV2 (#28113)

<!--
Please submit this PR as a draft initially.
Do not mark it as "Ready for review" until the template has been
completely filled out, and PR status checks have passed at least once.
-->

## **Description**

This PR migrates MetaMetricsController to inherit from BaseController V2

<!--
Write a short description of the changes included in this pull request,
also include relevant motivation and context. Have in mind the following
questions:
1. What is the reason for the change?
2. What is the improvement/solution?
-->

[![Open in GitHub
Codespaces](https://github.com/codespaces/badge.svg)](https://codespaces.new/MetaMask/metamask-extension/pull/28113?quickstart=1)

## **Related issues**

Fixes: #25925 

## **Manual testing steps**

## **Screenshots/Recordings**

<!-- If applicable, add screenshots and/or recordings to visualize the
before and after of your change. -->

### **Before**

<!-- [screenshots/recordings] -->

### **After**

<!-- [screenshots/recordings] -->

## **Pre-merge author checklist**

- [x] I've followed [MetaMask Contributor
Docs](https://github.com/MetaMask/contributor-docs) and [MetaMask
Extension Coding
Standards](https://github.com/MetaMask/metamask-extension/blob/develop/.github/guidelines/CODING_GUIDELINES.md).
- [x] I've completed the PR template to the best of my ability
- [x] I’ve included tests if applicable
- [x] I’ve documented my code using [JSDoc](https://jsdoc.app/) format
if applicable
- [x] I’ve applied the right labels on the PR (see [labeling
guidelines](https://github.com/MetaMask/metamask-extension/blob/develop/.github/guidelines/LABELING_GUIDELINES.md)).
Not required for external contributors.

## **Pre-merge reviewer checklist**

- [ ] I've manually tested the PR (e.g. pull and build branch, run the
app, test code being changed).
- [ ] I confirm that this PR addresses all acceptance criteria described
in the ticket it closes and includes the necessary testing evidence such
as recordings and or screenshots.

app/scripts/controllers/metametrics-data-deletion/metametrics-data-deletion.test.ts

@@ -1,5 +1,6 @@
 import { ControllerMessenger } from '@metamask/base-controller';
 import {
+  AllowedActions,
   MetaMetricsDataDeletionController,
   type MetaMetricsDataDeletionControllerMessengerActions,
 } from './metametrics-data-deletion';
@@ -10,8 +11,8 @@ describe('MetaMetricsDataDeletionController', () => {
       const mockMetaMetricsId = 'mockId';
       const mockTaskId = 'mockTaskId';
       const { controller, dataDeletionService } = setupController({
+        metaMetricsId: mockMetaMetricsId,
         options: {
-          getMetaMetricsId: jest.fn().mockReturnValue(mockMetaMetricsId),
           dataDeletionService: {
             createDataDeletionRegulationTask: jest
               .fn()
@@ -43,8 +44,8 @@ describe('MetaMetricsDataDeletionController', () => {
       const mockMetaMetricsId = 'mockId';
       const mockTaskId = 'mockTaskId';
       const { controller, dataDeletionService } = setupController({
+        metaMetricsId: mockMetaMetricsId,
         options: {
-          getMetaMetricsId: jest.fn().mockReturnValue(mockMetaMetricsId),
           dataDeletionService: {
             createDataDeletionRegulationTask: jest
               .fn()
@@ -76,9 +77,7 @@ describe('MetaMetricsDataDeletionController', () => {
 
     it('fails to creates a data deletion task when user has never participating in metrics tracking', async () => {
       const { controller } = setupController({
-        options: {
-          getMetaMetricsId: jest.fn().mockReturnValue(null),
-        },
+        metaMetricsId: null,
       });
       await expect(
         controller.createMetaMetricsDataDeletionTask(),
@@ -94,8 +93,8 @@ describe('MetaMetricsDataDeletionController', () => {
       const mockMetaMetricsId = 'mockId';
       const mockTaskId = 'mockTaskId';
       const { controller, dataDeletionService } = setupController({
+        metaMetricsId: mockMetaMetricsId,
         options: {
-          getMetaMetricsId: jest.fn().mockReturnValue(mockMetaMetricsId),
           dataDeletionService: {
             createDataDeletionRegulationTask: jest
               .fn()
@@ -128,28 +127,35 @@ describe('MetaMetricsDataDeletionController', () => {
  *
  * @param options - Setup options.
  * @param options.options - Controller constructor options.
+ * @param options.metaMetricsId - The MetaMetrics ID to use.
  * @returns The test controller, a messenger instance, and related mocks.
  */
 function setupController({
   options,
+  metaMetricsId,
 }: {
   options?: Partial<
     ConstructorParameters<typeof MetaMetricsDataDeletionController>[0]
   >;
+  metaMetricsId?: string | null;
 } = {}): {
   controller: MetaMetricsDataDeletionController;
   dataDeletionService: ConstructorParameters<
     typeof MetaMetricsDataDeletionController
   >[0]['dataDeletionService'];
   messenger: ControllerMessenger<
-    MetaMetricsDataDeletionControllerMessengerActions,
+    MetaMetricsDataDeletionControllerMessengerActions | AllowedActions,
     never
   >;
 } {
-  const messenger = new ControllerMessenger<
-    MetaMetricsDataDeletionControllerMessengerActions,
+  const controllerMessenger = new ControllerMessenger<
+    MetaMetricsDataDeletionControllerMessengerActions | AllowedActions,
     never
   >();
+  controllerMessenger.registerActionHandler(
+    'MetaMetricsController:getState',
+    jest.fn().mockReturnValue({ metaMetricsId }),
+  );
   const mockCreateDataDeletionRegulationTaskResponse = 'mockRegulateId';
   const mockFetchDeletionRegulationStatusResponse = 'UNKNOWN';
   const mockDataDeletionService = {
@@ -164,9 +170,9 @@ function setupController({
   const constructorOptions = {
     dataDeletionService: mockDataDeletionService,
     getMetaMetricsId: jest.fn().mockReturnValue('mockMetaMetricsId'),
-    messenger: messenger.getRestricted({
+    messenger: controllerMessenger.getRestricted({
       name: 'MetaMetricsDataDeletionController',
-      allowedActions: [],
+      allowedActions: ['MetaMetricsController:getState'],
       allowedEvents: [],
     }),
     ...options,
@@ -176,6 +182,6 @@ function setupController({
   return {
     controller,
     dataDeletionService: constructorOptions.dataDeletionService,
-    messenger,
+    messenger: controllerMessenger,
   };
 }

app/scripts/controllers/metametrics-data-deletion/metametrics-data-deletion.ts

@@ -5,6 +5,7 @@ import {
 import { PublicInterface } from '@metamask/utils';
 import type { DataDeletionService } from '../../services/data-deletion-service';
 import { DeleteRegulationStatus } from '../../../../shared/constants/metametrics';
+import { MetaMetricsControllerGetStateAction } from '../metametrics-controller';
 
 // Unique name for the controller
 const controllerName = 'MetaMetricsDataDeletionController';
@@ -70,14 +71,24 @@ export type MetaMetricsDataDeletionControllerMessengerActions =
   | CreateMetaMetricsDataDeletionTaskAction
   | UpdateDataDeletionTaskStatusAction;
 
+/**
+ * Actions that this controller is allowed to call.
+ */
+export type AllowedActions = MetaMetricsControllerGetStateAction;
+
+/**
+ * Events that this controller is allowed to subscribe.
+ */
+export type AllowedEvents = never;
+
 // Type for the messenger of MetaMetricsDataDeletionController
 export type MetaMetricsDataDeletionControllerMessenger =
   RestrictedControllerMessenger<
     typeof controllerName,
-    MetaMetricsDataDeletionControllerMessengerActions,
-    never,
-    never,
-    never
+    MetaMetricsDataDeletionControllerMessengerActions | AllowedActions,
+    AllowedEvents,
+    AllowedActions['type'],
+    AllowedEvents['type']
   >;
 
 /**
@@ -91,27 +102,22 @@ export class MetaMetricsDataDeletionController extends BaseController<
 > {
   #dataDeletionService: PublicInterface<DataDeletionService>;
 
-  #getMetaMetricsId: () => string | null;
-
   /**
    * Creates a MetaMetricsDataDeletionController instance.
    *
    * @param args - The arguments to this function.
    * @param args.dataDeletionService - The service used for deleting data.
    * @param args.messenger - Messenger used to communicate with BaseV2 controller.
    * @param args.state - Initial state to set on this controller.
-   * @param args.getMetaMetricsId - A function that returns the current MetaMetrics ID.
    */
   constructor({
     dataDeletionService,
     messenger,
     state,
-    getMetaMetricsId,
   }: {
     dataDeletionService: PublicInterface<DataDeletionService>;
     messenger: MetaMetricsDataDeletionControllerMessenger;
     state?: Partial<MetaMetricsDataDeletionState>;
-    getMetaMetricsId: () => string | null;
   }) {
     // Call the constructor of BaseControllerV2
     super({
@@ -120,7 +126,6 @@ export class MetaMetricsDataDeletionController extends BaseController<
       name: controllerName,
       state: { ...getDefaultState(), ...state },
     });
-    this.#getMetaMetricsId = getMetaMetricsId;
     this.#dataDeletionService = dataDeletionService;
     this.#registerMessageHandlers();
   }
@@ -146,7 +151,9 @@ export class MetaMetricsDataDeletionController extends BaseController<
    *
    */
   async createMetaMetricsDataDeletionTask(): Promise<void> {
-    const metaMetricsId = this.#getMetaMetricsId();
+    const { metaMetricsId } = this.messagingSystem.call(
+      'MetaMetricsController:getState',
+    );
     if (!metaMetricsId) {
       throw new Error('MetaMetrics ID not found');
     }