feat: Use useCanFastUnstake query (#2377)

polkadot-cloud · Jan 2, 2025 · 0d99cbb · 0d99cbb
1 parent 0c9051d
commit 0d99cbb
Show file tree

Hide file tree

Showing 20 changed files with 157 additions and 455 deletions.
diff --git a/packages/app/src/Router.tsx b/packages/app/src/Router.tsx
@@ -61,7 +61,7 @@ const RouterInner = () => {
   return (
     <ErrorBoundary FallbackComponent={ErrorFallbackApp}>
       {pluginEnabled('staking_api') && !inSetup() && activeAccount && (
-        <StakingApi activeAccount={activeAccount} />
+        <StakingApi activeAccount={activeAccount} network={network} />
       )}
       <NotificationPrompts />
       <Body>

diff --git a/packages/app/src/StakingApi/FastUnstakeApi.tsx b/packages/app/src/StakingApi/FastUnstakeApi.tsx
@@ -0,0 +1,24 @@
+// Copyright 2024 @polkadot-cloud/polkadot-staking-dashboard authors & contributors
+// SPDX-License-Identifier: GPL-3.0-only
+
+import { useFastUnstake } from 'contexts/FastUnstake'
+import { useCanFastUnstake } from 'plugin-staking-api'
+import { useEffect } from 'react'
+import type { Props } from './types'
+
+export const FastUnstakeApi = ({ activeAccount, network }: Props) => {
+  const { setFastUnstakeStatus } = useFastUnstake()
+  const { data, loading, error } = useCanFastUnstake({
+    chain: network,
+    who: activeAccount,
+  })
+
+  // Update fast unstake status on active account change. Must be bonding
+  useEffect(() => {
+    if (!loading && !error && data?.canFastUnstake) {
+      setFastUnstakeStatus(data.canFastUnstake)
+    }
+  }, [JSON.stringify(data?.canFastUnstake)])
+
+  return null
+}
diff --git a/packages/app/src/StakingApi.tsx → ...pp/src/StakingApi/UnclaimedRewardsApi.tsx b/packages/app/src/StakingApi.tsx → ...pp/src/StakingApi/UnclaimedRewardsApi.tsx
@@ -2,18 +2,13 @@
 // SPDX-License-Identifier: GPL-3.0-only
 
 import { useApi } from 'contexts/Api'
-import { useNetwork } from 'contexts/Network'
 import { usePayouts } from 'contexts/Payouts'
-import { ApolloProvider, client, useUnclaimedRewards } from 'plugin-staking-api'
+import { useUnclaimedRewards } from 'plugin-staking-api'
 import { useEffect } from 'react'
+import type { Props } from './types'
 
-interface Props {
-  activeAccount: string
-}
-
-const Inner = ({ activeAccount }: Props) => {
+export const UnclaimedRewardsApi = ({ activeAccount, network }: Props) => {
   const { activeEra } = useApi()
-  const { network } = useNetwork()
   const { setUnclaimedRewards } = usePayouts()
 
   const { data, loading, error } = useUnclaimedRewards({
@@ -22,6 +17,7 @@ const Inner = ({ activeAccount }: Props) => {
     fromEra: Math.max(activeEra.index.minus(1).toNumber(), 0),
   })
 
+  // Update unclaimed rewards on total change
   useEffect(() => {
     if (!loading && !error && data?.unclaimedRewards) {
       setUnclaimedRewards(data?.unclaimedRewards)
@@ -30,9 +26,3 @@ const Inner = ({ activeAccount }: Props) => {
 
   return null
 }
-
-export const StakingApi = (props: Props) => (
-  <ApolloProvider client={client}>
-    <Inner {...props} />
-  </ApolloProvider>
-)
diff --git a/packages/app/src/StakingApi/index.tsx b/packages/app/src/StakingApi/index.tsx
@@ -0,0 +1,28 @@
+// Copyright 2024 @polkadot-cloud/polkadot-staking-dashboard authors & contributors
+// SPDX-License-Identifier: GPL-3.0-only
+
+import { useFastUnstake } from 'contexts/FastUnstake'
+import { useStaking } from 'contexts/Staking'
+import { ApolloProvider, client } from 'plugin-staking-api'
+import { useEffect } from 'react'
+import { FastUnstakeApi } from './FastUnstakeApi'
+import type { Props } from './types'
+import { UnclaimedRewardsApi } from './UnclaimedRewardsApi'
+
+export const StakingApi = (props: Props) => {
+  const { isBonding } = useStaking()
+  const { setFastUnstakeStatus } = useFastUnstake()
+
+  useEffect(() => {
+    if (!isBonding()) {
+      setFastUnstakeStatus(null)
+    }
+  }, [isBonding()])
+
+  return (
+    <ApolloProvider client={client}>
+      <UnclaimedRewardsApi {...props} />
+      {isBonding() && <FastUnstakeApi {...props} />}
+    </ApolloProvider>
+  )
+}
diff --git a/packages/app/src/StakingApi/types.ts b/packages/app/src/StakingApi/types.ts
@@ -0,0 +1,9 @@
+// Copyright 2024 @polkadot-cloud/polkadot-staking-dashboard authors & contributors
+// SPDX-License-Identifier: GPL-3.0-only
+
+import type { NetworkId } from 'common-types'
+
+export interface Props {
+  activeAccount: string
+  network: NetworkId
+}
diff --git a/packages/app/src/contexts/FastUnstake/defaults.ts b/packages/app/src/contexts/FastUnstake/defaults.ts
@@ -2,18 +2,13 @@
 // SPDX-License-Identifier: GPL-3.0-only
 /* eslint-disable @typescript-eslint/no-unused-vars */
 
-import type { FastUnstakeContextInterface, MetaInterface } from './types'
-
-export const defaultMeta: MetaInterface = {
-  checked: [],
-}
+import type { FastUnstakeContextInterface } from './types'
 
 export const defaultFastUnstakeContext: FastUnstakeContextInterface = {
-  getLocalkey: (address) => '',
-  checking: false,
-  meta: defaultMeta,
-  isExposed: null,
+  exposed: false,
   head: undefined,
   queueDeposit: undefined,
   counterForQueue: undefined,
+  fastUnstakeStatus: null,
+  setFastUnstakeStatus: (status) => {},
 }