From dd6fff7297132af2f6c61ecba75187ec056e8506 Mon Sep 17 00:00:00 2001
From: refi93 <rafael.korbas@gmail.com>
Date: Tue, 7 Nov 2023 17:54:25 +0100
Subject: [PATCH] feat(extension): track manual re-sync and hd wallet discovery

---
 .../settings/components/SettingsWalletBase.tsx   | 16 ++++++++++++++--
 .../components/HardwareWalletFlow.tsx            |  9 +++++++++
 .../components/WalletSetupWizard.tsx             | 10 ++++++++++
 packages/common/src/analytics/types.ts           |  3 +++
 4 files changed, 36 insertions(+), 2 deletions(-)

diff --git a/apps/browser-extension-wallet/src/views/browser-view/features/settings/components/SettingsWalletBase.tsx b/apps/browser-extension-wallet/src/views/browser-view/features/settings/components/SettingsWalletBase.tsx
index c634abb085..1dfe5636a3 100644
--- a/apps/browser-extension-wallet/src/views/browser-view/features/settings/components/SettingsWalletBase.tsx
+++ b/apps/browser-extension-wallet/src/views/browser-view/features/settings/components/SettingsWalletBase.tsx
@@ -15,6 +15,7 @@ import { useAnalyticsContext, useBackgroundServiceAPIContext } from '@providers'
 import { useSearchParams, useObservable, Button } from '@lace/common';
 import { walletRoutePaths } from '@routes/wallet-paths';
 import { PostHogAction } from '@providers/AnalyticsProvider/analyticsTracker';
+import uniq from 'lodash/uniq';
 
 const { Title } = Typography;
 
@@ -58,7 +59,7 @@ export const SettingsWalletBase = <AdditionalDrawers extends string>({
 
   const { t } = useTranslation();
   const { addressesDiscoverer } = useAddressesDiscoverer();
-  const { environmentName, inMemoryWallet } = useWalletStore();
+  const { environmentName, inMemoryWallet, walletInfo } = useWalletStore();
   const { AVAILABLE_CHAINS } = config();
   const unspendable = useObservable(inMemoryWallet.balance.utxo.unspendable$);
 
@@ -116,7 +117,18 @@ export const SettingsWalletBase = <AdditionalDrawers extends string>({
     <Button
       size="medium"
       className={styles.settingsButton}
-      onClick={() => addressesDiscoverer.discover()}
+      onClick={async () => {
+        analytics.sendEventToPostHog(PostHogAction.SettingsWalletHdWalletSyncSyncClick);
+
+        const oldHdAddressesCount = uniq((walletInfo?.addresses ?? []).map(({ index }) => index)).length;
+        const newAddresses = await addressesDiscoverer.discover();
+        const newHdAddressesCount = uniq(newAddresses.map(({ index }) => index)).length;
+        const newHdWalletAddressesDiscovered = newHdAddressesCount > oldHdAddressesCount;
+
+        if (newHdWalletAddressesDiscovered) {
+          analytics.sendEventToPostHog(PostHogAction.SettingsWalletHdWalletSyncSyncNewAddresses);
+        }
+      }}
       block={popupView}
       data-testid="settings-wallet-wallet-sync-cta"
     >
diff --git a/apps/browser-extension-wallet/src/views/browser-view/features/wallet-setup/components/HardwareWalletFlow.tsx b/apps/browser-extension-wallet/src/views/browser-view/features/wallet-setup/components/HardwareWalletFlow.tsx
index 045f000b0b..f8b24bc34c 100644
--- a/apps/browser-extension-wallet/src/views/browser-view/features/wallet-setup/components/HardwareWalletFlow.tsx
+++ b/apps/browser-extension-wallet/src/views/browser-view/features/wallet-setup/components/HardwareWalletFlow.tsx
@@ -22,6 +22,7 @@ import { useTranslation } from 'react-i18next';
 import {
   AnalyticsEventNames,
   EnhancedAnalyticsOptInStatus,
+  PostHogAction,
   postHogOnboardingActions
 } from '@providers/AnalyticsProvider/analyticsTracker';
 import { config } from '@src/config';
@@ -174,6 +175,14 @@ export const HardwareWalletFlow = ({
       setDoesUserAllowAnalytics(
         isAnalyticsAccepted ? EnhancedAnalyticsOptInStatus.OptedIn : EnhancedAnalyticsOptInStatus.OptedOut
       );
+      const addressDiscoverySubscriber = wallet.asyncKeyAgent.knownAddresses$.subscribe((addresses) => {
+        if (addresses.length === 0) return;
+        const hdWalletDiscovered = addresses.some((addr) => addr.index > 0);
+        if (hdWalletDiscovered) {
+          analytics.sendEventToPostHog(PostHogAction.OnboardingRestoreHdWallet);
+        }
+        addressDiscoverySubscriber.unsubscribe();
+      });
       await analytics.setOptedInForEnhancedAnalytics(
         isAnalyticsAccepted ? EnhancedAnalyticsOptInStatus.OptedIn : EnhancedAnalyticsOptInStatus.OptedOut
       );
diff --git a/apps/browser-extension-wallet/src/views/browser-view/features/wallet-setup/components/WalletSetupWizard.tsx b/apps/browser-extension-wallet/src/views/browser-view/features/wallet-setup/components/WalletSetupWizard.tsx
index ccd21cc641..0f130ebf31 100644
--- a/apps/browser-extension-wallet/src/views/browser-view/features/wallet-setup/components/WalletSetupWizard.tsx
+++ b/apps/browser-extension-wallet/src/views/browser-view/features/wallet-setup/components/WalletSetupWizard.tsx
@@ -23,6 +23,7 @@ import { WarningModal } from '@src/views/browser-view/components/WarningModal';
 import {
   AnalyticsEventNames,
   EnhancedAnalyticsOptInStatus,
+  PostHogAction,
   postHogOnboardingActions,
   UserTrackingType
 } from '@providers/AnalyticsProvider/analyticsTracker';
@@ -290,6 +291,15 @@ export const WalletSetupWizard = ({
       await analytics.setOptedInForEnhancedAnalytics(
         isAnalyticsAccepted ? EnhancedAnalyticsOptInStatus.OptedIn : EnhancedAnalyticsOptInStatus.OptedOut
       );
+      const addressDiscoverySubscriber = wallet.wallet.asyncKeyAgent.knownAddresses$.subscribe((addresses) => {
+        if (addresses.length === 0) return;
+        const hdWalletDiscovered = addresses.some((addr) => addr.index > 0);
+        if (setupType === SetupType.RESTORE && hdWalletDiscovered) {
+          analytics.sendEventToPostHog(PostHogAction.OnboardingRestoreHdWallet);
+        }
+        addressDiscoverySubscriber.unsubscribe();
+      });
+
       if (setupType === SetupType.FORGOT_PASSWORD) {
         deleteFromLocalStorage('isForgotPasswordFlow');
         goToMyWallet(wallet);
diff --git a/packages/common/src/analytics/types.ts b/packages/common/src/analytics/types.ts
index e0cac05cd9..2cae01f100 100644
--- a/packages/common/src/analytics/types.ts
+++ b/packages/common/src/analytics/types.ts
@@ -23,6 +23,7 @@ export enum PostHogAction {
   OnboardingRestoreEnterPassphrase09NextClick = 'onboarding | restore wallet | enter passphrase #09 | next | click',
   OnboardingRestoreEnterPassphrase17NextClick = 'onboarding | restore wallet | enter passphrase #17 | next | click',
   OnboardingRestoreWalletNamePasswordNextClick = 'onboarding | restore wallet | wallet name & password | next | click',
+  OnboardingRestoreHdWallet = 'onboarding | restore wallet | hd wallet',
   // Create new wallet
   OnboardingCreateDoneGoToWallet = 'onboarding | new wallet | all done | go to my wallet | click',
   OnboardingCreateAnalyticsAgreeClick = 'onboarding | new wallet | analytics | agree | click',
@@ -159,6 +160,8 @@ export enum PostHogAction {
   SettingsAnalyticsAgreeClick = 'settings | analytics | agree | click',
   SettingsAnalyticsSkipClick = 'settings | analytics | skip | click',
   SettingsFaqsClick = 'settings | faqs | click',
+  SettingsWalletHdWalletSyncSyncClick = 'settings | wallet | hd wallet sync | sync | click',
+  SettingsWalletHdWalletSyncSyncNewAddresses = 'settings | wallet | hd wallet sync | sync | new addresses',
   // Recieve section
   ReceiveClick = 'receive | receive | click',
   ReceiveCopyAddressIconClick = 'receive | receive | copy address icon | click',