chore: add selector to get account groups by addresses (#35547)

<!--
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**

<!--
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/35547?quickstart=1)

Adds a new selector (`getAccountGroupsByAddress`) to get an arrays of
account groups given an array of addresses as input. This change can't
be tested since the selector is not consumed anywhere yet

## **Changelog**

<!--
If this PR is not End-User-Facing and should not show up in the
CHANGELOG, you can choose to either:
1. Write `CHANGELOG entry: null`
2. Label with `no-changelog`

If this PR is End-User-Facing, please write a short User-Facing
description in the past tense like:
`CHANGELOG entry: Added a new tab for users to see their NFTs`
`CHANGELOG entry: Fixed a bug that was causing some NFTs to flicker`

(This helps the Release Engineer do their job more quickly and
accurately)
-->

CHANGELOG entry: null

## **Related issues**

Fixes: https://consensyssoftware.atlassian.net/browse/MUL-723

## **Manual testing steps**

None

## **Screenshots/Recordings**

None

## **Pre-merge author checklist**

- [ ] I've followed [MetaMask Contributor
Docs](https://github.com/MetaMask/contributor-docs) and [MetaMask
Extension Coding
Standards](https://github.com/MetaMask/metamask-extension/blob/main/.github/guidelines/CODING_GUIDELINES.md).
- [ ] I've completed the PR template to the best of my ability
- [ ] I’ve included tests if applicable
- [ ] I’ve documented my code using [JSDoc](https://jsdoc.app/) format
if applicable
- [ ] I’ve applied the right labels on the PR (see [labeling
guidelines](https://github.com/MetaMask/metamask-extension/blob/main/.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.

---------

Co-authored-by: Hassan Malik <41640681+hmalik88@users.noreply.github.com>

Author: gantunesr

ui/selectors/multichain-accounts/account-tree.test.ts

@@ -30,6 +30,7 @@ import {
   getWalletsWithAccounts,
   getNetworkAddressCount,
   getWallet,
+  getAccountGroupsByAddress,
 } from './account-tree';
 import { MultichainAccountsState } from './account-tree.types';
 import {
@@ -1189,4 +1190,47 @@ describe('Multichain Accounts Selectors', () => {
       expect(result[2].id).toBe('account-2');
     });
   });
+
+  describe('getAccountGroupsByAddress', () => {
+    it('returns the correct account groups without duplicated values', () => {
+      const result = getAccountGroupsByAddress(typedMockState, [
+        '0x0dcd5d886577d5081b0c52e242ef29e70be3e7bc',
+        '0xc42edfcc21ed14dda456aa0756c153f7985d8813',
+        '0xeb9e64b93097bc15f01f13eae97015c57ab64823',
+        '0xec1adf982415d2ef5ec55899b9bfb8bc0f29251b',
+      ]);
+
+      expect(result.length).toBe(3);
+      expect(result[0].id).toBe(ENTROPY_GROUP_1_ID);
+      expect(result[1].id).toBe(ENTROPY_GROUP_2_ID);
+      expect(result[2].id).toBe(LEDGER_GROUP_ID);
+    });
+
+    it('returns an empty array when no addresses match', () => {
+      const result = getAccountGroupsByAddress(typedMockState, [
+        'nonExistentAddress',
+      ]);
+
+      expect(result).toEqual([]);
+    });
+
+    it('returns an empty array when given an empty address list', () => {
+      const result = getAccountGroupsByAddress(typedMockState, []);
+
+      expect(result).toEqual([]);
+    });
+
+    it('handles duplicated addresses in the input list', () => {
+      const result = getAccountGroupsByAddress(typedMockState, [
+        '0x0dcd5d886577d5081b0c52e242ef29e70be3e7bc',
+        '0x0dcd5d886577d5081b0c52e242ef29e70be3e7bc', // duplicate
+        '0xc42edfcc21ed14dda456aa0756c153f7985d8813',
+        '0xc42edfcc21ed14dda456aa0756c153f7985d8813', // duplicate
+      ]);
+
+      expect(result.length).toBe(2);
+      expect(result[0].id).toBe(ENTROPY_GROUP_1_ID);
+      expect(result[1].id).toBe(LEDGER_GROUP_ID);
+    });
+  });
 });

ui/selectors/multichain-accounts/account-tree.ts

@@ -698,3 +698,38 @@ export const getInternalAccountsFromGroupById = createSelector(
       .filter((account): account is InternalAccount => Boolean(account));
   },
 );
+
+/**
+ * Selector to get account groups by a list of addresses.
+ * Returns groups that contain at least one account matching any of the provided addresses.
+ *
+ * @param _state - Redux state.
+ * @param addresses - An array of addresses to filter account groups by.
+ * @returns An array of AccountGroupWithInternalAccounts that contain at least one matching account.
+ */
+export const getAccountGroupsByAddress = createDeepEqualSelector(
+  [
+    getAccountGroupWithInternalAccounts,
+    (_state: MultichainAccountsState, addresses: string[]) =>
+      new Set(addresses.map((address) => address.toLowerCase())),
+  ],
+  (
+    accountGroupWithInternalAccounts,
+    addressesSet: Set<string>,
+  ): AccountGroupWithInternalAccounts[] => {
+    const matchingGroups = new Set<AccountGroupWithInternalAccounts>();
+
+    accountGroupWithInternalAccounts.forEach((group) => {
+      const containsMatchingAccount = group.accounts.some((account) =>
+        addressesSet.has(account.address.toLowerCase()),
+      );
+
+      if (containsMatchingAccount) {
+        matchingGroups.add(group);
+      }
+    });
+
+    // Convert the Set of AccountGroupWithInternalAccounts to an Array
+    return [...matchingGroups];
+  },
+);