From 77f81cbae6a168f5bbd4716bf0354a9b2d7527f6 Mon Sep 17 00:00:00 2001
From: Ignacio <ignacio@forbole.com>
Date: Wed, 28 Feb 2024 14:31:17 +0800
Subject: [PATCH] feat: support claiming rewards

---
 src/features/staking/components/logged-in.tsx | 33 +++++++++++----
 src/features/staking/context/actions.ts       | 26 +++++++++---
 src/features/staking/lib/core/base.ts         | 42 ++++++++++++++-----
 3 files changed, 77 insertions(+), 24 deletions(-)

diff --git a/src/features/staking/components/logged-in.tsx b/src/features/staking/components/logged-in.tsx
index 79229f7..a2bfa36 100644
--- a/src/features/staking/components/logged-in.tsx
+++ b/src/features/staking/components/logged-in.tsx
@@ -7,7 +7,11 @@ import Link from "next/link";
 import { memo, useState } from "react";
 
 import type { StakingState } from "../context";
-import { stakeValidator, unstakeValidator } from "../context/actions";
+import {
+  claimRewardsAction,
+  stakeValidator,
+  unstakeValidator,
+} from "../context/actions";
 import { useStaking } from "../context/hooks";
 import type { StakeAddresses } from "../lib/core/base";
 import { formatCoin } from "../lib/core/coins";
@@ -99,8 +103,6 @@ function StakingPage() {
                   <Button
                     disabled={isLoading}
                     onClick={() => {
-                      setIsLoading(true);
-
                       if (!client || !validator) return;
 
                       setIsLoading(true);
@@ -110,9 +112,11 @@ function StakingPage() {
                         validator: validator.operatorAddress,
                       };
 
-                      unstakeValidator(addresses, client, staking).then(() => {
-                        setIsLoading(false);
-                      });
+                      unstakeValidator(addresses, client, staking).finally(
+                        () => {
+                          setIsLoading(false);
+                        },
+                      );
                     }}
                   >
                     Undelegate
@@ -121,7 +125,20 @@ function StakingPage() {
                     <Button
                       disabled={isLoading || delegation.rewards.amount === "0"}
                       onClick={() => {
-                        // @TODO
+                        if (!client) return;
+
+                        setIsLoading(true);
+
+                        const addresses: StakeAddresses = {
+                          delegator: account.bech32Address,
+                          validator: delegation.validatorAddress,
+                        };
+
+                        claimRewardsAction(addresses, client, staking).finally(
+                          () => {
+                            setIsLoading(false);
+                          },
+                        );
                       }}
                     >
                       Claim rewards
@@ -191,7 +208,7 @@ function StakingPage() {
                     validator: validator.operatorAddress,
                   };
 
-                  stakeValidator(addresses, client, staking).then(() => {
+                  stakeValidator(addresses, client, staking).finally(() => {
                     setIsLoading(false);
                   });
                 }}
diff --git a/src/features/staking/context/actions.ts b/src/features/staking/context/actions.ts
index 988e31b..ad8b721 100644
--- a/src/features/staking/context/actions.ts
+++ b/src/features/staking/context/actions.ts
@@ -1,6 +1,7 @@
 import type { StakingContextType, Unbonding } from ".";
 import type { SigningClient, StakeAddresses } from "../lib/core/base";
 import {
+  claimRewards,
   getBalance,
   getDelegations,
   getRewards,
@@ -105,24 +106,24 @@ export const fetchStakingData = async (
 };
 
 export const stakeValidator = async (
-  addressess: StakeAddresses,
+  addresses: StakeAddresses,
   client: SigningClient,
   staking: StakingContextType,
 ) => {
-  await stakeAmount(addressess, client, {
+  await stakeAmount(addresses, client, {
     amount: "1000",
     denom: "uxion",
   });
 
-  await fetchStakingData(addressess.delegator, staking);
+  await fetchStakingData(addresses.delegator, staking);
 };
 
 export const unstakeValidator = async (
-  addressess: StakeAddresses,
+  addresses: StakeAddresses,
   client: SigningClient,
   staking: StakingContextType,
 ) => {
-  const result = await unstakeAmount(addressess, client, {
+  const result = await unstakeAmount(addresses, client, {
     amount: "1000",
     denom: "uxion",
   });
@@ -130,5 +131,18 @@ export const unstakeValidator = async (
   // eslint-disable-next-line no-console
   console.log("debug: actions.ts: result", result);
 
-  await fetchStakingData(addressess.delegator, staking);
+  await fetchStakingData(addresses.delegator, staking);
+};
+
+export const claimRewardsAction = async (
+  addresses: StakeAddresses,
+  client: SigningClient,
+  staking: StakingContextType,
+) => {
+  const result = await claimRewards(addresses, client);
+
+  // eslint-disable-next-line no-console
+  console.log("debug: actions.ts: result", result);
+
+  await fetchStakingData(addresses.delegator, staking);
 };
diff --git a/src/features/staking/lib/core/base.ts b/src/features/staking/lib/core/base.ts
index 45af7a1..d31a278 100644
--- a/src/features/staking/lib/core/base.ts
+++ b/src/features/staking/lib/core/base.ts
@@ -4,6 +4,7 @@ import type {
   Coin,
   MsgDelegateEncodeObject,
   MsgUndelegateEncodeObject,
+  MsgWithdrawDelegatorRewardEncodeObject,
   StdFee,
 } from "@cosmjs/stargate";
 import {
@@ -15,6 +16,7 @@ import {
 } from "@cosmjs/stargate";
 import { Tendermint34Client } from "@cosmjs/tendermint-rpc";
 import BigNumber from "bignumber.js";
+import { MsgWithdrawDelegatorReward } from "cosmjs-types/cosmos/distribution/v1beta1/tx";
 import {
   MsgDelegate,
   MsgUndelegate,
@@ -40,19 +42,20 @@ const getStakingQueryClient = async () => {
 
 type FeeOpts = {
   address: string;
-  amount: Coin[];
+  amount?: Coin[];
   client: SigningClient;
-  gasLimit: string;
-  memo: string;
+  gasLimit?: string;
+  memo?: string;
   msgs: EncodeObject[];
 };
 
 const getCosmosFee = async ({
   address,
-  amount,
+  // @TODO: review
+  amount = [{ amount: "1000", denom: "uxion" }],
   client,
-  gasLimit,
-  memo,
+  gasLimit = "400000",
+  memo = "",
   msgs,
 }: FeeOpts) => {
   // @TODO: create signer from local account
@@ -188,11 +191,7 @@ export const unstakeAmount = async (
 
   const fee: StdFee = await getCosmosFee({
     address: addresses.delegator,
-    // @TODO: review
-    amount: [{ amount: "1000", denom: "uxion" }],
     client,
-    gasLimit: "400000",
-    memo: "",
     msgs: [msgAny],
   });
 
@@ -207,3 +206,26 @@ export const getUnbonding = async (
 
   return queryClient.staking.unbondingDelegation(address, validatorAddress);
 };
+
+export const claimRewards = async (
+  addresses: StakeAddresses,
+  client: NonNullable<SigningClient>,
+) => {
+  const msg = MsgWithdrawDelegatorReward.fromPartial({
+    delegatorAddress: addresses.delegator,
+    validatorAddress: addresses.validator,
+  });
+
+  const msgAny: MsgWithdrawDelegatorRewardEncodeObject = {
+    typeUrl: "/cosmos.distribution.v1beta1.MsgWithdrawDelegatorReward",
+    value: msg,
+  };
+
+  const fee: StdFee = await getCosmosFee({
+    address: addresses.delegator,
+    client,
+    msgs: [msgAny],
+  });
+
+  return await client.signAndBroadcast(addresses.delegator, [msgAny], fee);
+};