From 8efbaf61635a8b239b26b7c7bdf6cc2e6000403f Mon Sep 17 00:00:00 2001
From: Guillermo Bescos <g.bescos@yahoo.com>
Date: Wed, 20 Dec 2023 07:06:32 +0700
Subject: [PATCH] Request

---
 frontend/pages/request.tsx | 190 +++++++++++++++++++++++++++++++++++++
 1 file changed, 190 insertions(+)
 create mode 100644 frontend/pages/request.tsx

diff --git a/frontend/pages/request.tsx b/frontend/pages/request.tsx
new file mode 100644
index 00000000..1bfd208c
--- /dev/null
+++ b/frontend/pages/request.tsx
@@ -0,0 +1,190 @@
+import type { NextPage } from 'next'
+import Layout from '../components/Layout'
+import SEO from '../components/SEO'
+import { useAnchorWallet, useConnection } from '@solana/wallet-adapter-react'
+import {
+  PythBalance,
+  StakeAccount,
+  StakeConnection,
+  STAKING_ADDRESS,
+} from '@pythnetwork/staking'
+import { useEffect, useState } from 'react'
+import { Wallet } from '@coral-xyz/anchor'
+import { PublicKey } from '@solana/web3.js'
+import toast from 'react-hot-toast'
+import { capitalizeFirstLetter } from 'utils/capitalizeFirstLetter'
+
+const RequestSplit: NextPage = () => {
+  const { connection } = useConnection()
+  const anchorWallet = useAnchorWallet()
+
+  const [owner, setOwner] = useState<PublicKey>()
+  const [amount, setAmount] = useState<PythBalance>()
+
+  const handleSetOwner = (event: any) => {
+    try {
+      setOwner(new PublicKey(event.target.value))
+    } catch (e) {
+      setOwner(undefined)
+    }
+  }
+  const handleSetAmount = (event: any) => {
+    try {
+      setAmount(PythBalance.fromString(event.target.value))
+    } catch (e) {
+      setAmount(undefined)
+    }
+  }
+
+  const [stakeConnection, setStakeConnection] = useState<StakeConnection>()
+  const [stakeAccounts, setStakeAccounts] = useState<StakeAccount[]>()
+  const [selectedStakeAccount, setSelectStakeAccount] = useState<StakeAccount>()
+
+  const handleSelectStakeAccount = (event: any) => {
+    for (const stakeAccount of stakeAccounts!) {
+      if (stakeAccount.address.toString() === event.target.value) {
+        setSelectStakeAccount(stakeAccount)
+        break
+      }
+    }
+  }
+
+  useEffect(() => {
+    const initialize = async () => {
+      const stakeConnection = await StakeConnection.createStakeConnection(
+        connection,
+        anchorWallet as Wallet,
+        STAKING_ADDRESS
+      )
+      setStakeConnection(stakeConnection)
+    }
+
+    if (!anchorWallet) {
+      setStakeConnection(undefined)
+    } else {
+      initialize()
+    }
+  }, [anchorWallet])
+
+  useEffect(() => {
+    const loadStakeAccounts = async () => {
+      if (stakeConnection && anchorWallet) {
+        const stakeAccounts = await stakeConnection.getStakeAccounts(
+          anchorWallet.publicKey
+        )
+        setStakeAccounts(stakeAccounts)
+      } else {
+        setStakeAccounts(undefined)
+      }
+    }
+    loadStakeAccounts()
+  }, [stakeConnection])
+
+  useEffect(() => {
+    const loadCurrentRequest = async () => {
+      if (stakeConnection && selectedStakeAccount) {
+        const request = await stakeConnection.getSplitRequest(
+          selectedStakeAccount
+        )
+        if (request) {
+          setAmount(request.balance)
+          setOwner(request.recipient)
+        }
+      }
+      loadCurrentRequest()
+    }
+  }, [selectedStakeAccount])
+
+  useEffect(() => {
+    if (stakeAccounts && stakeAccounts.length > 0)
+      setSelectStakeAccount(stakeAccounts[0])
+  }, [stakeAccounts])
+
+  const requestSplit = async () => {
+    if (stakeConnection && selectedStakeAccount && owner && amount)
+      try {
+        await stakeConnection.requestSplit(selectedStakeAccount, amount, owner)
+        toast.success('Successfully created transfer request')
+      } catch (err) {
+        toast.error(capitalizeFirstLetter(err.message))
+      }
+  }
+
+  return (
+    <Layout>
+      <SEO title={'Request split'} />
+
+      {stakeConnection &&
+        stakeAccounts !== undefined &&
+        stakeAccounts.length > 0 && (
+          <div>
+            <p className=" p-2 ">
+              Request a transfer of locked tokens to a new account
+            </p>
+
+            <select
+              style={{ color: 'black' }}
+              value={selectedStakeAccount?.address.toString()}
+              onChange={handleSelectStakeAccount}
+            >
+              {stakeAccounts!.map((option, index) => (
+                <option key={index} value={option.address.toBase58()}>
+                  {option.address.toString()}
+                </option>
+              ))}
+            </select>
+            {selectedStakeAccount && (
+              <p>
+                This account has{' '}
+                {new PythBalance(selectedStakeAccount!.tokenBalance).toString()}{' '}
+                tokens
+              </p>
+            )}
+
+            <p className=" text-sm ">New owner</p>
+            <input
+              type="text"
+              style={{ color: 'black' }}
+              value={owner ? owner.toString() : ''}
+              onChange={handleSetOwner}
+            />
+            <p className=" text-sm ">Amount</p>
+            <input
+              type="text"
+              style={{ color: 'black' }}
+              value={amount ? amount.toString() : ''}
+              onChange={handleSetAmount}
+            />
+            <p className=" text-sm ">
+              Owner : {owner ? owner.toString() : 'Invalid new owner'}
+            </p>
+            <p className=" text-sm ">
+              Amount to be transferred:{' '}
+              {amount ? amount.toString() : 'Invalid amount to transfer'}
+            </p>
+          </div>
+        )}
+
+      {stakeConnection && owner && amount ? (
+        <p>
+          <button
+            className="rounded-full p-2 hover:bg-hoverGray"
+            onClick={() => requestSplit()}
+          >
+            Click to approve
+          </button>
+        </p>
+      ) : !stakeConnection ? (
+        <p className="p-2 hover:bg-hoverGray"> Please connect wallet</p>
+      ) : !owner ? (
+        <p className="p-2 hover:bg-hoverGray ">Please insert valid new owner</p>
+      ) : (
+        <p className="p-2 hover:bg-hoverGray ">
+          Please insert valid amount to be transferred
+        </p>
+      )}
+    </Layout>
+  )
+}
+
+export default RequestSplit