Merge pull request #1099 from ionicprotocol/development

reset button

Author: jourdanDunkley

packages/ui/components/veion/MyVeionTable.tsx

@@ -13,6 +13,7 @@ import type {
 import TokenPair from '@ui/components/TokenPair';
 import { pools } from '@ui/constants';
 import { useVeIONContext } from '@ui/context/VeIonContext';
+import { useVeIONVote } from '@ui/hooks/veion/useVeIONVote';
 import type { MyVeionData } from '@ui/types/veION';
 
 import ExtendVeion from './ExtendVeion';
@@ -32,8 +33,10 @@ function MyVeionTable() {
   const router = useRouter();
   const {
     setSelectedManagePosition,
-    locks: { myLocks, isLoading }
+    locks: { myLocks, isLoading },
+    currentChain
   } = useVeIONContext();
+  const { handleReset, isResetting } = useVeIONVote(currentChain);
 
   const hasLockExpired = (lockExpiryDate: string, isPermanent: boolean) => {
     if (isPermanent) return false;
@@ -134,6 +137,17 @@ function MyVeionTable() {
           data.lockExpires.isPermanent
         );
 
+        const handleResetClick = async () => {
+          try {
+            await switchChain({
+              chainId: data.chainId
+            });
+            await handleReset(data.id);
+          } catch (error) {
+            console.error('Reset failed:', error);
+          }
+        };
+
         return (
           <div className="flex gap-2 w-full">
             {/* {isExpired ? (
@@ -178,6 +192,14 @@ function MyVeionTable() {
                 bg="bg-white/10"
                 className="bg-accent"
               />
+              <ActionButton
+                half
+                action={handleResetClick}
+                label={isResetting ? 'Resetting...' : 'Reset'}
+                bg="bg-white/10"
+                className="bg-red-500 hover:bg-red-600 text-white"
+                disabled={isResetting}
+              />
             </>
             {/* )} */}
           </div>

packages/ui/hooks/veion/useVeIONVote.ts

@@ -22,6 +22,7 @@ interface VoteParams {
 
 interface VoteState {
   isVoting: boolean;
+  isResetting: boolean;
   error: string | null;
   pendingVotes: Map<
     string,
@@ -55,6 +56,7 @@ export function useVeIONVote(chain: number) {
 
   const [state, setState] = useState<VoteState>({
     isVoting: false,
+    isResetting: false,
     error: null,
     pendingVotes: new Map()
   });
@@ -160,12 +162,60 @@ export function useVeIONVote(chain: number) {
     }
   }
 
+  const handleReset = async (tokenId: number) => {
+    if (!isConnected || !address) {
+      throw new Error('Wallet not connected');
+    }
+
+    if (!voterContract || !publicClient || !walletClient) {
+      throw new Error('Contract not initialized');
+    }
+
+    try {
+      // Validate chain
+      const isSwitched = await handleSwitchOriginChain(chain, chainId);
+      if (!isSwitched) throw new Error('Failed to switch chain');
+
+      setState((prev) => ({ ...prev, isResetting: true }));
+
+      await write(getContractConfig('reset', [tokenId]), {
+        successMessage: 'Successfully reset votes',
+        errorMessage: 'Failed to reset votes, last voted this epoch'
+      });
+
+      // Refetch data after successful reset
+      await Promise.all([
+        votingPeriod.refetch?.(),
+        selectedPoolRows.refetch?.()
+      ]);
+
+      setState((prev) => ({
+        ...prev,
+        isResetting: false,
+        pendingVotes: new Map()
+      }));
+
+      return true;
+    } catch (error) {
+      console.error('Vote reset failed:', error);
+      setState((prev) => ({
+        ...prev,
+        error:
+          error instanceof Error ? error.message : 'Unknown error occurred',
+        isResetting: false
+      }));
+      return false;
+    }
+  };
+
   return {
     handleVote,
+    handleReset,
     simulateVote,
     resetState: () =>
       setState({
         isVoting: false,
+        isResetting: false,
         error: null,
         pendingVotes: new Map()
       }),