From 330866a5042ed74b065a1d68d678c360deb20d0e Mon Sep 17 00:00:00 2001
From: Julien Delaigues <0xju@pm.me>
Date: Tue, 2 Apr 2024 10:36:10 +0800
Subject: [PATCH] feat: track wallet's address when staking, unstaking, claming
 reward

---
 .../components/staking_section/claim_rewards_modal.tsx      | 5 ++++-
 .../staking/components/staking_section/staking_modal.tsx    | 1 +
 .../staking/components/staking_section/unstaking_modal.tsx  | 6 +++++-
 3 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/screens/staking/components/staking_section/claim_rewards_modal.tsx b/src/screens/staking/components/staking_section/claim_rewards_modal.tsx
index 77255c80..792ab69d 100644
--- a/src/screens/staking/components/staking_section/claim_rewards_modal.tsx
+++ b/src/screens/staking/components/staking_section/claim_rewards_modal.tsx
@@ -121,7 +121,10 @@ const ClaimRewardsModal = () => {
 
                   stakingRef.current.postHog?.capture(
                     PostHogCustomEvent.ClaimedRewards,
-                    claimableRewardsForNetwork,
+                    {
+                      ...claimableRewardsForNetwork,
+                      walletAddress: selectedAccount.address,
+                    },
                   );
 
                   toastSuccess({
diff --git a/src/screens/staking/components/staking_section/staking_modal.tsx b/src/screens/staking/components/staking_section/staking_modal.tsx
index 4aef207a..5afe6741 100644
--- a/src/screens/staking/components/staking_section/staking_modal.tsx
+++ b/src/screens/staking/components/staking_section/staking_modal.tsx
@@ -151,6 +151,7 @@ const StakingModal = () => {
           stakingRef.current.postHog?.capture(PostHogCustomEvent.StakedTokens, {
             amount,
             denom: mainNetworkDenom[selectedAccount.networkId],
+            walletAddress: selectedAccount.address,
           });
 
           toastSuccess({
diff --git a/src/screens/staking/components/staking_section/unstaking_modal.tsx b/src/screens/staking/components/staking_section/unstaking_modal.tsx
index 6b6fd3b9..bda5ad37 100644
--- a/src/screens/staking/components/staking_section/unstaking_modal.tsx
+++ b/src/screens/staking/components/staking_section/unstaking_modal.tsx
@@ -152,7 +152,11 @@ const UnstakingModal = () => {
 
           stakingRef.current.postHog?.capture(
             PostHogCustomEvent.UnstakedTokens,
-            { amount, denom: mainNetworkDenom[selectedAccount.networkId] },
+            {
+              amount,
+              denom: mainNetworkDenom[selectedAccount.networkId],
+              walletAddress: selectedAccount.address,
+            },
           );
 
           toastSuccess({