feat(BABY): add reward service

Author: 0xDazzer

package-lock.json

@@ -29,7 +29,7 @@
     "node": "24.2.0"
   },
   "dependencies": {
-    "@babylonlabs-io/babylon-proto-ts": "1.6.2",
+    "@babylonlabs-io/babylon-proto-ts": "1.7.1",
     "@babylonlabs-io/btc-staking-ts": "2.3.4",
     "@babylonlabs-io/core-ui": "1.6.0",
     "@babylonlabs-io/wallet-connector": "1.8.4",

package.json

@@ -1,7 +1,7 @@
 import babylon from "@/infrastructure/babylon";
 import { useClientQuery } from "@/ui/common/hooks/client/useClient";
 
-const BABY_DELEGATIONS_KEY = "BABY_VALIDATORS_KEY";
+const BABY_DELEGATIONS_KEY = "BABY_DELEGATIONS_KEY";
 
 export function useValidators(
   address: string,

src/ui/baby/hooks/useDelegations.ts src/ui/baby/hooks/api/useDelegations.tssrc/ui/baby/hooks/useDelegations.ts renamed to src/ui/baby/hooks/api/useDelegations.ts

@@ -1,15 +1,15 @@
 import babylon from "@/infrastructure/babylon";
 import { useClientQuery } from "@/ui/common/hooks/client/useClient";
 
-const BABY_REWARDS_KEY = "BABY_VALIDATORS_KEY";
+const BABY_REWARDS_KEY = "BABY_REWARDS_KEY";
 
-export function useValidators(
+export function useRewards(
   address: string,
   { enabled = true }: { enabled?: boolean } = {},
 ) {
   return useClientQuery({
     queryKey: [BABY_REWARDS_KEY, address],
     queryFn: () => babylon.client.baby.getRewards(address),
-    enabled,
+    enabled: Boolean(address) && enabled,
   });
 }

src/ui/baby/hooks/useRewards.ts src/ui/baby/hooks/api/useRewards.tssrc/ui/baby/hooks/useRewards.ts renamed to src/ui/baby/hooks/api/useRewards.ts

@@ -0,0 +1,46 @@
+import { useCallback } from "react";
+
+import babylon from "@/infrastructure/babylon";
+import { useRewards } from "@/ui/baby/hooks/api/useRewards";
+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";
+
+export function useRewardService() {
+  const { bech32Address } = useCosmosWallet();
+  const { data: rewards = [], refetch: refetchRewards } =
+    useRewards(bech32Address);
+  const { signBbnTx, sendBbnTx } = useBbnTransaction();
+  const { handleError } = useError();
+  const logger = useLogger();
+
+  const claimReward = useCallback(
+    async (validatorAddress: string) => {
+      try {
+        if (!bech32Address) throw Error("Babylon Wallet is not connected");
+
+        const claimRewardMsg = babylon.txs.baby.createClaimRewardMsg({
+          validatorAddress,
+          delegatorAddress: bech32Address,
+        });
+        const signedTx = await signBbnTx(claimRewardMsg);
+        const result = await sendBbnTx(signedTx);
+
+        await logger.info("Baby Staking: claim reward", {
+          txHash: result?.txHash,
+        });
+        await refetchRewards();
+      } catch (error: any) {
+        handleError({ error });
+        logger.error(error);
+      }
+    },
+    [logger, bech32Address, signBbnTx, sendBbnTx, refetchRewards, handleError],
+  );
+
+  return {
+    rewards,
+    claimReward,
+  };
+}