fix: vote delegation form for wallets that have active pool and vote …

…delegation
input-output-hk · Jan 27, 2025 · 295cf77 · 295cf77
1 parent dd63902
commit 295cf77
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/source/renderer/app/api/wallets/types.ts b/source/renderer/app/api/wallets/types.ts
@@ -77,7 +77,11 @@ export type SyncStateStatus =
   | 'syncing'
   | 'not_responding';
 export type Discovery = 'random' | 'sequential';
-export type DelegationStatus = 'delegating' | 'not_delegating';
+export type DelegationStatus =
+  | 'delegating'
+  | 'not_delegating'
+  | 'voting'
+  | 'voting_and_delegating';
 export type WalletSyncStateProgress = {
   quantity: number;
   unit: 'percentage';

diff --git a/source/renderer/app/domains/Wallet.ts b/source/renderer/app/domains/Wallet.ts
@@ -33,9 +33,13 @@ export const WalletSyncStateStatuses: {
 export const WalletDelegationStatuses: {
   DELEGATING: DelegationStatus;
   NOT_DELEGATING: DelegationStatus;
+  VOTING: DelegationStatus;
+  VOTING_AND_DELEGATING: DelegationStatus;
 } = {
   DELEGATING: 'delegating',
   NOT_DELEGATING: 'not_delegating',
+  VOTING: 'voting',
+  VOTING_AND_DELEGATING: 'voting_and_delegating',
 };
 export type HwDeviceStatus =
   | 'connecting'
@@ -234,10 +238,12 @@ export default class Wallet {
 
   @computed
   get isDelegating(): boolean {
-    return this.lastDelegationStakePoolStatus
-      ? this.lastDelegationStakePoolStatus ===
-          WalletDelegationStatuses.DELEGATING
-      : this.delegationStakePoolStatus === WalletDelegationStatuses.DELEGATING;
+    const statusToCheck = (this.lastDelegationStakePoolStatus ||
+      this.delegationStakePoolStatus) as DelegationStatus;
+    return [
+      WalletDelegationStatuses.DELEGATING,
+      WalletDelegationStatuses.VOTING_AND_DELEGATING,
+    ].includes(statusToCheck);
   }
 
   @computed