fix withdraw bug (solana-labs#2163)

pyth-network · Feb 25, 2024 · 8f8427e · 8f8427e
1 parent 472c479
commit 8f8427e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/VoteStakeRegistry/components/TokenBalance/WithdrawCommunityTokensBtn.tsx b/VoteStakeRegistry/components/TokenBalance/WithdrawCommunityTokensBtn.tsx
@@ -91,7 +91,10 @@ const WithDrawCommunityTokens = () => {
               )
             ).result
             if (!governance) throw new Error('failed to fetch governance')
-            if (proposal.account.getTimeToVoteEnd(governance.account) > 0) {
+            if (
+              proposal.account.getTimeToVoteEnd(governance.account) > 0 &&
+              governance.account.realm.equals(realm!.pubkey)
+            ) {
               setIsLoading(false)
               // Note: It's technically possible to withdraw the vote here but I think it would be confusing and people would end up unconsciously withdrawing their votes
               notify({

diff --git a/components/TokenBalance/VanillaWithdrawTokensButton.tsx b/components/TokenBalance/VanillaWithdrawTokensButton.tsx
@@ -119,7 +119,10 @@ const VanillaWithdrawTokensButton = ({
               )
             ).result
             if (!governance) throw new Error('failed to fetch governance')
-            if (proposal.account.getTimeToVoteEnd(governance.account) > 0) {
+            if (
+              proposal.account.getTimeToVoteEnd(governance.account) > 0 &&
+              governance.account.realm.equals(realm!.pubkey)
+            ) {
               // Note: It's technically possible to withdraw the vote here but I think it would be confusing and people would end up unconsciously withdrawing their votes
               notify({
                 type: 'error',