From f6ec12db94140f3f3c92e0a705f161bb156f1410 Mon Sep 17 00:00:00 2001
From: fmtabbara <fmtabbara@hotmail.co.uk>
Date: Wed, 9 Mar 2022 12:15:29 +0000
Subject: [PATCH] allow delegation of vested tokens

---
 nym-wallet/src/index.tsx                      |  4 +-
 nym-wallet/src/pages/bond/BondForm.tsx        |  5 +-
 .../src/pages/delegate/DelegateForm.tsx       | 59 +++++++++++--------
 .../src/pages/delegate/validationSchema.ts    | 24 +++++++-
 nym-wallet/src/types/global.ts                |  6 ++
 5 files changed, 66 insertions(+), 32 deletions(-)

diff --git a/nym-wallet/src/index.tsx b/nym-wallet/src/index.tsx
index 0ac5588621..13e782df80 100644
--- a/nym-wallet/src/index.tsx
+++ b/nym-wallet/src/index.tsx
@@ -1,4 +1,4 @@
-import React, { useContext, useEffect, useLayoutEffect } from 'react'
+import React, { useContext, useEffect } from 'react'
 import ReactDOM from 'react-dom'
 import { ErrorBoundary } from 'react-error-boundary'
 import { BrowserRouter as Router } from 'react-router-dom'
@@ -15,7 +15,7 @@ import { maximizeWindow } from './utils'
 const App = () => {
   const { clientDetails } = useContext(ClientContext)
 
-  useLayoutEffect(() => {
+  useEffect(() => {
     maximizeWindow()
   }, [])
 
diff --git a/nym-wallet/src/pages/bond/BondForm.tsx b/nym-wallet/src/pages/bond/BondForm.tsx
index d5a9bf0177..13f0b3622e 100644
--- a/nym-wallet/src/pages/bond/BondForm.tsx
+++ b/nym-wallet/src/pages/bond/BondForm.tsx
@@ -12,11 +12,10 @@ import {
 } from '@mui/material'
 import { yupResolver } from '@hookform/resolvers/yup'
 import { useForm } from 'react-hook-form'
-import { EnumNodeType } from '../../types/global'
 import { NodeTypeSelector } from '../../components/NodeTypeSelector'
 import { bond, vestingBond, majorToMinor } from '../../requests'
 import { validationSchema } from './validationSchema'
-import { Gateway, MixNode, TBondArgs } from '../../types'
+import { Gateway, MixNode, TBondArgs, EnumNodeType } from '../../types'
 import { ClientContext } from '../../context/main'
 import { Fee, TokenPoolSelector } from '../../components'
 
@@ -107,7 +106,7 @@ export const BondForm = ({
   const watchNodeType = watch('nodeType', defaultValues.nodeType)
   const watchAdvancedOptions = watch('withAdvancedOptions', defaultValues.withAdvancedOptions)
 
-  const onSubmit = async (data: TBondFormFields, cb: (data: TBondArgs) => Promise<any>) => {
+  const onSubmit = async (data: TBondFormFields, cb: (data: TBondArgs) => Promise<void>) => {
     const formattedData = formatData(data)
     const pledge = await majorToMinor(data.amount)
 
diff --git a/nym-wallet/src/pages/delegate/DelegateForm.tsx b/nym-wallet/src/pages/delegate/DelegateForm.tsx
index 469e885ccb..bfe934a3bb 100644
--- a/nym-wallet/src/pages/delegate/DelegateForm.tsx
+++ b/nym-wallet/src/pages/delegate/DelegateForm.tsx
@@ -1,24 +1,26 @@
-import React, { useContext } from 'react'
+import React, { useEffect, useContext } from 'react'
 import { Box, Button, CircularProgress, FormControl, Grid, InputAdornment, TextField, Typography } from '@mui/material'
 import { yupResolver } from '@hookform/resolvers/yup'
 import { useForm } from 'react-hook-form'
-import { EnumNodeType } from '../../types'
+import { DelegationResult, EnumNodeType, TDelegateArgs } from '../../types'
 import { validationSchema } from './validationSchema'
 import { ClientContext } from '../../context/main'
-import { delegate, majorToMinor } from '../../requests'
+import { delegate, majorToMinor, vestingDelegateToMixnode } from '../../requests'
 import { checkHasEnoughFunds } from '../../utils'
-import { Fee } from '../../components'
+import { Fee, TokenPoolSelector } from '../../components'
 
 type TDelegateForm = {
-  nodeType: EnumNodeType
   identity: string
   amount: string
+  tokenPool: string
+  type: EnumNodeType
 }
 
 const defaultValues: TDelegateForm = {
-  nodeType: EnumNodeType.mixnode,
   identity: '',
   amount: '',
+  tokenPool: 'balance',
+  type: EnumNodeType.mixnode,
 }
 
 export const DelegateForm = ({
@@ -30,37 +32,36 @@ export const DelegateForm = ({
 }) => {
   const {
     register,
-    watch,
     handleSubmit,
-    setError,
+    setValue,
+    reset,
     formState: { errors, isSubmitting },
   } = useForm<TDelegateForm>({
     defaultValues,
     resolver: yupResolver(validationSchema),
   })
 
-  const watchNodeType = watch('nodeType', defaultValues.nodeType)
+  const { userBalance, currency, clientDetails } = useContext(ClientContext)
 
-  const { userBalance, currency } = useContext(ClientContext)
-
-  const onSubmit = async (data: TDelegateForm) => {
-    const hasEnoughFunds = await checkHasEnoughFunds(data.amount)
-    if (!hasEnoughFunds) {
-      return setError('amount', {
-        message: 'Not enough funds in wallet',
-      })
-    }
+  useEffect(() => {
+    reset()
+  }, [clientDetails])
 
+  const onSubmit = async (data: TDelegateForm, cb: (data: TDelegateArgs) => Promise<DelegationResult>) => {
     const amount = await majorToMinor(data.amount)
 
-    await delegate({
-      type: data.nodeType,
+    await cb({
+      type: data.type,
       identity: data.identity,
       amount,
     })
-      .then((res) => {
+      .then(async (res) => {
+        if (data.tokenPool === 'balance') {
+          await userBalance.fetchBalance()
+        } else {
+          await userBalance.fetchTokenAllocation()
+        }
         onSuccess({ amount: data.amount, address: res.target_address })
-        userBalance.fetchBalance()
       })
       .catch((e) => {
         console.log(e)
@@ -86,7 +87,13 @@ export const DelegateForm = ({
             />
           </Grid>
 
-          <Grid item xs={12}>
+          {userBalance.originalVesting && (
+            <Grid item xs={6}>
+              <TokenPoolSelector onSelect={(pool) => setValue('tokenPool', pool)} />
+            </Grid>
+          )}
+
+          <Grid item xs={6}>
             <TextField
               {...register('amount')}
               required
@@ -102,7 +109,7 @@ export const DelegateForm = ({
               }}
             />
           </Grid>
-          <Grid item>
+          <Grid item xs={12}>
             <Fee feeType="DelegateToMixnode" />
           </Grid>
         </Grid>
@@ -117,7 +124,9 @@ export const DelegateForm = ({
         }}
       >
         <Button
-          onClick={handleSubmit(onSubmit)}
+          onClick={handleSubmit((data) =>
+            onSubmit(data, data.tokenPool === 'balance' ? delegate : vestingDelegateToMixnode),
+          )}
           disabled={isSubmitting}
           data-testid="delegate-button"
           variant="contained"
diff --git a/nym-wallet/src/pages/delegate/validationSchema.ts b/nym-wallet/src/pages/delegate/validationSchema.ts
index 4ac48cc248..6841d9aae6 100644
--- a/nym-wallet/src/pages/delegate/validationSchema.ts
+++ b/nym-wallet/src/pages/delegate/validationSchema.ts
@@ -1,5 +1,5 @@
 import * as Yup from 'yup'
-import { validateAmount, validateKey } from '../../utils'
+import { checkHasEnoughFunds, checkHasEnoughLockedTokens, validateAmount, validateKey } from '../../utils'
 
 export const validationSchema = Yup.object().shape({
   identity: Yup.string()
@@ -9,7 +9,27 @@ export const validationSchema = Yup.object().shape({
       'A valid identity key is required e.g. 824WyExLUWvLE2mpSHBatN4AoByuLzfnHFeHWiBYzg4z',
       (value) => (!!value ? validateKey(value, 32) : false),
     ),
+
   amount: Yup.string()
     .required()
-    .test('valid-amount-key', 'A valid amount is required', (value) => (!!value ? validateAmount(value, '0') : false)),
+    .test('valid-amount', 'A valid amount is required', async function (value) {
+      const isValid = await validateAmount(value || '', '0')
+      const hasEnoughBalance = await checkHasEnoughFunds(value || '')
+      const hasEnoughLocked = await checkHasEnoughLockedTokens(value || '')
+
+      if (!isValid) {
+        return this.createError({ message: `A valid amount is required` })
+      }
+
+      if (this.parent.tokenPool === 'balance' && !hasEnoughBalance) {
+        console.log('here')
+        return this.createError({ message: 'Not enough funds in wallet' })
+      }
+
+      if (this.parent.tokenPool === 'locked' && !hasEnoughLocked) {
+        return this.createError({ message: 'Not enough locked tokens' })
+      }
+
+      return true
+    }),
 })
diff --git a/nym-wallet/src/types/global.ts b/nym-wallet/src/types/global.ts
index fecaee0423..f08653cb0d 100644
--- a/nym-wallet/src/types/global.ts
+++ b/nym-wallet/src/types/global.ts
@@ -62,6 +62,12 @@ export type TBondArgs = {
   ownerSignature: string
 }
 
+export type TDelegateArgs = {
+  type: EnumNodeType
+  identity: string
+  amount: Coin
+}
+
 export type TCurrency = {
   minor: 'UNYM' | 'UNYMT'
   major: 'NYM' | 'NYMT'