chore: Add tags to custom traces

app/selectors/nftController.ts

@@ -1,5 +1,5 @@
 import { createSelector } from 'reselect';
-import { NftControllerState } from '@metamask/assets-controllers';
+import { Nft, NftControllerState } from '@metamask/assets-controllers';
 import { RootState } from '../reducers';
 
 const selectNftControllerState = (state: RootState) =>
@@ -15,3 +15,14 @@ export const selectAllNfts = createSelector(
   selectNftControllerState,
   (nftControllerState: NftControllerState) => nftControllerState.allNfts,
 );
+
+export const selectAllNftsFlat = createSelector(
+  selectAllNfts,
+  (nftsByChainByAccount) => {
+    const nftsByChainArray = Object.values(nftsByChainByAccount);
+    return nftsByChainArray.reduce((acc, nftsByChain) => {
+      const nftsArrays = Object.values(nftsByChain);
+      return acc.concat(...nftsArrays);
+    }, [] as Nft[]);
+  },
+);

app/selectors/tokensController.ts

@@ -37,3 +37,24 @@ export const selectDetectedTokens = createSelector(
   (tokensControllerState: TokensControllerState) =>
     tokensControllerState?.detectedTokens,
 );
+
+const selectAllTokens = createSelector(
+  selectTokensControllerState,
+  (tokensControllerState: TokensControllerState) =>
+    tokensControllerState?.allTokens,
+);
+
+export const selectAllTokensFlat = createSelector(
+  selectAllTokens,
+  (tokensByAccountByChain) => {
+    if (Object.values(tokensByAccountByChain).length === 0) {
+      return [];
+    }
+    const tokensByAccountArray = Object.values(tokensByAccountByChain);
+
+    return tokensByAccountArray.reduce((acc, tokensByAccount) => {
+      const tokensArray = Object.values(tokensByAccount);
+      return acc.concat(...tokensArray);
+    }, [] as Token[]);
+  },
+);

app/store/index.ts

@@ -16,6 +16,7 @@ import thunk from 'redux-thunk';
 
 import persistConfig from './persistConfig';
 import { AppStateEventProcessor } from '../core/AppStateEventListener';
+import { getTraceTags } from '../util/sentry/tags';
 
 // TODO: Improve type safety by using real Action types instead of `any`
 // TODO: Replace "any" with type
@@ -119,6 +120,7 @@ const createStoreAndPersistor = async (appStartTime: number) => {
       {
         name: TraceName.EngineInitialization,
         op: TraceOperation.EngineInitialization,
+        tags: getTraceTags(store.getState?.()),
       },
       () => {
         EngineService.initalizeEngine(store);

app/util/sentry/tags/index.test.ts

@@ -0,0 +1,229 @@
+import { RootState } from '../../../reducers';
+import { getTraceTags } from './';
+import initialRootState, {
+  backgroundState,
+} from '../../../util/test/initial-root-state';
+import { userInitialState } from '../../../reducers/user';
+import { createMockAccountsControllerState } from '../../../util/test/accountsControllerTestUtils';
+
+describe('Tags Utils', () => {
+  beforeEach(() => {
+    jest.resetAllMocks();
+  });
+
+  describe('getTraceTags', () => {
+    it('includes if unlocked', () => {
+      const state = {
+        ...initialRootState,
+        user: { ...userInitialState, userLoggedIn: true },
+      };
+
+      const tags = getTraceTags(state);
+
+      expect(tags?.['wallet.unlocked']).toStrictEqual(true);
+    });
+
+    it('includes if not unlocked', () => {
+      const state = {
+        ...initialRootState,
+        user: { ...userInitialState, userLoggedIn: false },
+      };
+
+      const tags = getTraceTags(state);
+
+      expect(tags?.['wallet.unlocked']).toStrictEqual(false);
+    });
+
+    it('includes pending approval type', () => {
+      const state = {
+        ...initialRootState,
+        engine: {
+          backgroundState: {
+            ...backgroundState,
+            ApprovalController: {
+              ...backgroundState.ApprovalController,
+              pendingApprovals: {
+                1: {
+                  type: 'eth_sendTransaction',
+                },
+              },
+            },
+          },
+        },
+      } as unknown as RootState;
+
+      const tags = getTraceTags(state);
+
+      expect(tags?.['wallet.pending_approval']).toStrictEqual(
+        'eth_sendTransaction',
+      );
+    });
+
+    it('includes first pending approval type if multiple', () => {
+      const state = {
+        ...initialRootState,
+        engine: {
+          backgroundState: {
+            ...backgroundState,
+
+            ApprovalController: {
+              ...backgroundState.ApprovalController,
+              pendingApprovals: {
+                1: {
+                  type: 'eth_sendTransaction',
+                },
+                2: {
+                  type: 'personal_sign',
+                },
+              },
+            },
+          },
+        },
+      } as unknown as RootState;
+
+      const tags = getTraceTags(state);
+
+      expect(tags?.['wallet.pending_approval']).toStrictEqual(
+        'eth_sendTransaction',
+      );
+    });
+
+    it('includes account count', () => {
+      const state = {
+        ...initialRootState,
+        engine: {
+          backgroundState: {
+            ...backgroundState,
+            AccountsController: createMockAccountsControllerState([
+              '0x1234',
+              '0x4321',
+            ]),
+          },
+        },
+      } as unknown as RootState;
+
+      const tags = getTraceTags(state);
+
+      expect(tags?.['wallet.account_count']).toStrictEqual(2);
+    });
+
+    it('includes nft count', () => {
+      const state = {
+        ...initialRootState,
+        engine: {
+          backgroundState: {
+            ...backgroundState,
+            NftController: {
+              ...backgroundState.NftController,
+              allNfts: {
+                '0x1234': {
+                  '0x1': [
+                    {
+                      tokenId: '1',
+                    },
+                    {
+                      tokenId: '2',
+                    },
+                  ],
+                  '0x2': [
+                    {
+                      tokenId: '3',
+                    },
+                    {
+                      tokenId: '4',
+                    },
+                  ],
+                },
+                '0x4321': {
+                  '0x3': [
+                    {
+                      tokenId: '5',
+                    },
+                  ],
+                },
+              },
+            },
+          },
+        },
+      } as unknown as RootState;
+
+      const tags = getTraceTags(state);
+
+      expect(tags?.['wallet.nft_count']).toStrictEqual(5);
+    });
+
+    it('includes notification count', () => {
+      const state = {
+        ...initialRootState,
+        engine: {
+          backgroundState: {
+            ...backgroundState,
+            NotificationServicesController: {
+              metamaskNotificationsList: [{}, {}, {}],
+            },
+          },
+        },
+      } as unknown as RootState;
+
+      const tags = getTraceTags(state);
+
+      expect(tags?.['wallet.notification_count']).toStrictEqual(3);
+    });
+
+    it('includes token count', () => {
+      const state = {
+        ...initialRootState,
+        engine: {
+          backgroundState: {
+            ...backgroundState,
+            TokensController: {
+              allTokens: {
+                '0x1': {
+                  '0x1234': [{}, {}],
+                  '0x4321': [{}],
+                },
+                '0x2': {
+                  '0x5678': [{}],
+                },
+              },
+            },
+          },
+        },
+      } as unknown as RootState;
+
+      const tags = getTraceTags(state);
+
+      expect(tags?.['wallet.token_count']).toStrictEqual(4);
+    });
+
+    it('includes transaction count', () => {
+      const state = {
+        ...initialRootState,
+        engine: {
+          backgroundState: {
+            ...backgroundState,
+            TransactionController: {
+              transactions: [
+                {
+                  id: 1,
+                  chainId: '0x1',
+                },
+                {
+                  id: 2,
+                  chainId: '0x1',
+                },
+                {
+                  id: 3,
+                  chainId: '0x2',
+                },
+              ],
+            },
+          },
+        },
+      } as unknown as RootState;
+      const tags = getTraceTags(state);
+
+      expect(tags?.['wallet.transaction_count']).toStrictEqual(3);
+    });
+  });
+});

app/util/sentry/tags/index.ts

@@ -0,0 +1,30 @@
+import { RootState } from '../../../reducers';
+import { selectAllNftsFlat } from '../../../selectors/nftController';
+import { selectInternalAccounts } from '../../../selectors/accountsController';
+import { selectAllTokensFlat } from '../../../selectors/tokensController';
+import { getNotificationsList } from '../../../selectors/notifications';
+import { selectTransactions } from '../../../selectors/transactionController';
+import { selectPendingApprovals } from '../../../selectors/approvalController';
+
+export function getTraceTags(state: RootState) {
+  if (!Object.keys(state?.engine?.backgroundState).length) return;
+  const unlocked = state.user.userLoggedIn;
+  const accountCount = selectInternalAccounts(state).length;
+  const nftCount = selectAllNftsFlat(state).length;
+  const notificationCount = getNotificationsList(state).length;
+  const tokenCount = selectAllTokensFlat(state).length;
+  const transactionCount = selectTransactions(state).length;
+  const pendingApprovals = Object.values(selectPendingApprovals(state));
+
+  const firstApprovalType = pendingApprovals?.[0]?.type;
+
+  return {
+    'wallet.account_count': accountCount,
+    'wallet.nft_count': nftCount,
+    'wallet.notification_count': notificationCount,
+    'wallet.pending_approval': firstApprovalType,
+    'wallet.token_count': tokenCount,
+    'wallet.transaction_count': transactionCount,
+    'wallet.unlocked': unlocked,
+  };
+}