feat(BABY): add delegation service

Author: 0xDazzer

src/ui/baby/hooks/api/useDelegations.ts

@@ -3,13 +3,13 @@ import { useClientQuery } from "@/ui/common/hooks/client/useClient";
 
 const BABY_DELEGATIONS_KEY = "BABY_DELEGATIONS_KEY";
 
-export function useValidators(
+export function useDelegations(
   address: string,
-  { enabled }: { enabled: boolean },
+  { enabled = true }: { enabled?: boolean } = {},
 ) {
   return useClientQuery({
     queryKey: [BABY_DELEGATIONS_KEY, address],
     queryFn: () => babylon.client.baby.getDelegations(address),
-    enabled,
+    enabled: Boolean(address) && enabled,
   });
 }

src/ui/baby/hooks/services/useDelegationService.ts

@@ -0,0 +1,118 @@
+import { useCallback } from "react";
+
+import babylon from "@/infrastructure/babylon";
+import { useDelegations } from "@/ui/baby/hooks/api/useDelegations";
+import { useError } from "@/ui/common/context/Error/ErrorProvider";
+import { useCosmosWallet } from "@/ui/common/context/wallet/CosmosWalletProvider";
+import { useBbnTransaction } from "@/ui/common/hooks/client/rpc/mutation/useBbnTransaction";
+import { useLogger } from "@/ui/common/hooks/useLogger";
+
+interface StakingParams {
+  validatorAddress: string;
+  amount: string;
+}
+
+export function useDelegationService() {
+  const { bech32Address } = useCosmosWallet();
+  const { data: delegations = [], refetch: refetchDelegations } =
+    useDelegations(bech32Address);
+  const { signBbnTx, sendBbnTx, estimateBbnGasFee } = useBbnTransaction();
+  const { handleError } = useError();
+  const logger = useLogger();
+
+  const stake = useCallback(
+    async ({ validatorAddress, amount }: StakingParams) => {
+      try {
+        if (!bech32Address) throw Error("Babylon Wallet is not connected");
+
+        const stakeMsg = babylon.txs.baby.createStakeMsg({
+          validatorAddress,
+          delegatorAddress: bech32Address,
+          amount: babylon.utils.babyToUbbn(Number(amount)),
+        });
+        const signedTx = await signBbnTx(stakeMsg);
+        const result = await sendBbnTx(signedTx);
+
+        logger.info("Baby Staking: stake", {
+          txHash: result?.txHash,
+        });
+        await refetchDelegations();
+      } catch (error: any) {
+        handleError({ error });
+        logger.error(error);
+      }
+    },
+    [
+      bech32Address,
+      signBbnTx,
+      sendBbnTx,
+      handleError,
+      refetchDelegations,
+      logger,
+    ],
+  );
+
+  const unstake = useCallback(
+    async ({ validatorAddress, amount }: StakingParams) => {
+      try {
+        if (!bech32Address) throw Error("Babylon Wallet is not connected");
+
+        const unstakeMsg = babylon.txs.baby.createUnstakeMsg({
+          validatorAddress,
+          delegatorAddress: bech32Address,
+          amount: babylon.utils.babyToUbbn(Number(amount)),
+        });
+        const signedTx = await signBbnTx(unstakeMsg);
+        const result = await sendBbnTx(signedTx);
+
+        logger.info("Baby Staking: unstake", {
+          txHash: result?.txHash,
+        });
+        await refetchDelegations();
+      } catch (error: any) {
+        handleError({ error });
+        logger.error(error);
+      }
+    },
+    [
+      bech32Address,
+      signBbnTx,
+      sendBbnTx,
+      handleError,
+      refetchDelegations,
+      logger,
+    ],
+  );
+
+  const estimateStakingFee = useCallback(
+    async ({ validatorAddress, amount }: StakingParams) => {
+      try {
+        if (!bech32Address) throw Error("Babylon Wallet is not connected");
+
+        const stakeMsg = babylon.txs.baby.createStakeMsg({
+          validatorAddress,
+          delegatorAddress: bech32Address,
+          amount: babylon.utils.babyToUbbn(Number(amount)),
+        });
+        const result = await estimateBbnGasFee(stakeMsg);
+
+        return result.amount.reduce(
+          (sum, { amount }) => sum + Number(amount),
+          0,
+        );
+      } catch (error: any) {
+        handleError({ error });
+        logger.error(error);
+        return 0;
+      }
+    },
+    [bech32Address, estimateBbnGasFee, handleError, logger],
+  );
+
+  return {
+    delegations,
+    stake,
+    unstake,
+    estimateStakingFee,
+  };
+}

src/ui/baby/hooks/services/useRewardService.ts

@@ -27,7 +27,7 @@ export function useRewardService() {
         const signedTx = await signBbnTx(claimRewardMsg);
         const result = await sendBbnTx(signedTx);
 
-        await logger.info("Baby Staking: claim reward", {
+        logger.info("Baby Staking: claim reward", {
           txHash: result?.txHash,
         });
         await refetchRewards();