feat(packages): co-staking service (#247)

* feat(packages): co-staking service

* chore(packages): remove unused

* chore(packages): minimize ubbn baby sat btc conversions

* chore(packages): move to baby.ts

* chore(packages): format

* chore(packages): format simple-staking

Author: gbarkhatov

packages/babylon-proto-ts/src/lib/lcd/baby.ts

@@ -1,6 +1,4 @@
-import type {
-  DelegationDelegatorReward
-} from "cosmjs-types/cosmos/distribution/v1beta1/distribution";
+import type { DelegationDelegatorReward } from "cosmjs-types/cosmos/distribution/v1beta1/distribution";
 import type {
   DelegationResponse,
   Validator,
@@ -19,7 +17,7 @@ const createBabylonClient = ({ request }: Dependencies) => ({
       return await fetchAllPages(
         request,
         `/cosmos/staking/v1beta1/delegations/${address}`,
-        "delegationResponses"
+        "delegationResponses",
       );
     } catch (error) {
       throw new Error(`Failed to fetch delegations for ${address}`, {
@@ -46,7 +44,11 @@ const createBabylonClient = ({ request }: Dependencies) => ({
 
   async getValidators(): Promise<Validator[]> {
     try {
-      return await fetchAllPages(request, "/cosmos/staking/v1beta1/validators", "validators");
+      return await fetchAllPages(
+        request,
+        "/cosmos/staking/v1beta1/validators",
+        "validators",
+      );
     } catch (error) {
       throw new Error(`Failed to fetch validators`, {
         cause: error,
@@ -92,31 +94,100 @@ const createBabylonClient = ({ request }: Dependencies) => ({
     const btcStakingPortion = Number(params?.btcStakingPortion ?? 0);
     const fpPortion = Number(params?.fpPortion ?? 0);
     return {
-      btcStakingPortion: Number.isFinite(btcStakingPortion) ? btcStakingPortion : 0,
+      btcStakingPortion: Number.isFinite(btcStakingPortion)
+        ? btcStakingPortion
+        : 0,
       fpPortion: Number.isFinite(fpPortion) ? fpPortion : 0,
     };
   },
 
   async getCostakingParams(): Promise<{
     costakingPortion: number;
     validatorsPortion: number;
+    scoreRatioBtcByBaby: string;
   }> {
     const response = await request("/babylon/costaking/v1/params");
     const params = response?.params ?? response;
-    const costakingPortion = Number(params?.costakingPortion ?? 0);
-    const validatorsPortion = Number(params?.validatorsPortion ?? 0);
+    const costakingPortion = Number(
+      params?.costakingPortion ?? params?.costaking_portion ?? 0,
+    );
+    const validatorsPortion = Number(
+      params?.validatorsPortion ?? params?.validators_portion ?? 0,
+    );
+    const scoreRatioBtcByBaby =
+      params?.scoreRatioBtcByBaby ?? params?.score_ratio_btc_by_baby ?? "50";
     return {
-      costakingPortion: Number.isFinite(costakingPortion) ? costakingPortion : 0,
-      validatorsPortion: Number.isFinite(validatorsPortion) ? validatorsPortion : 0,
+      costakingPortion: Number.isFinite(costakingPortion)
+        ? costakingPortion
+        : 0,
+      validatorsPortion: Number.isFinite(validatorsPortion)
+        ? validatorsPortion
+        : 0,
+      scoreRatioBtcByBaby: scoreRatioBtcByBaby,
     };
   },
 
-  async getAnnualProvisions(): Promise<number> {
+  async getCoStakerRewardsTracker(costakerAddress: string): Promise<{
+    startPeriodCumulativeReward: number;
+    activeSatoshis: string;
+    activeBaby: string;
+    totalScore: string;
+  } | null> {
+    if (!costakerAddress) {
+      return null;
+    }
+
     try {
       const response = await request(
-        "/cosmos/mint/v1beta1/annual_provisions",
+        `/babylon/costaking/v1/costakers/${costakerAddress}/rewards_tracker`,
+      );
+
+      return {
+        startPeriodCumulativeReward:
+          response?.start_period_cumulative_reward ?? 0,
+        activeSatoshis: response?.active_satoshis ?? "0",
+        activeBaby: response?.active_baby ?? "0",
+        totalScore: response?.total_score ?? "0",
+      };
+    } catch (error: any) {
+      // Return null for 404 errors (user has not co-staked yet)
+      if (error?.message?.includes("404") || error?.status === 404) {
+        return null;
+      }
+      throw new Error(
+        `Failed to fetch co-staker rewards tracker for ${costakerAddress}`,
+        {
+          cause: error,
+        },
       );
-      const annualProvisions = response?.annualProvisions ?? response?.annual_provisions ?? response;
+    }
+  },
+
+  async getCurrentCoStakingRewards(): Promise<{
+    rewards: Array<{ denom: string; amount: string }>;
+    period: number;
+    totalScore: string;
+  }> {
+    try {
+      const response = await request("/babylon/costaking/v1/current_rewards");
+
+      return {
+        rewards: response?.rewards ?? [],
+        period: response?.period ?? 0,
+        totalScore: response?.total_score ?? "0",
+      };
+    } catch (error) {
+      throw new Error("Failed to fetch current co-staking rewards", {
+        cause: error,
+      });
+    }
+  },
+
+  async getAnnualProvisions(): Promise<number> {
+    try {
+      const response = await request("/cosmos/mint/v1beta1/annual_provisions");
+      const annualProvisions =
+        response?.annualProvisions ?? response?.annual_provisions ?? response;
       const result = Number(annualProvisions);
       return result;
     } catch (error) {
@@ -126,13 +197,11 @@ const createBabylonClient = ({ request }: Dependencies) => ({
     }
   },
 
-
   async getSupply(denom: string = "ubbn"): Promise<bigint> {
     try {
-      const response = await request(
-        "/cosmos/bank/v1beta1/supply/by_denom",
-        { denom },
-      );
+      const response = await request("/cosmos/bank/v1beta1/supply/by_denom", {
+        denom,
+      });
       const amount = response?.amount?.amount ?? 0;
       return BigInt(amount);
     } catch (error: any) {
@@ -144,9 +213,9 @@ const createBabylonClient = ({ request }: Dependencies) => ({
 
   async getCurrentEpoch() {
     try {
-      const {
-        current_epoch, epoch_boundary
-      } = await request("/babylon/epoching/v1/current_epoch");
+      const { current_epoch, epoch_boundary } = await request(
+        "/babylon/epoching/v1/current_epoch",
+      );
 
       return {
         epochBoundary: parseInt(epoch_boundary, 10),

packages/babylon-proto-ts/src/lib/rpc/btc.ts

@@ -1,4 +1,4 @@
-import type{ Coin } from "@cosmjs/stargate";
+import type { Coin } from "@cosmjs/stargate";
 
 import { REWARD_GAUGE_KEY_BTC_DELEGATION } from "../../constants";
 import * as btclightclientquery from "../../generated/babylon/btclightclient/v1/query";
@@ -67,7 +67,7 @@ const createBTCClient = ({ incentive, btcLight }: Dependencies) => ({
         cause: error,
       });
     }
-  }
-})
+  },
+});
 
-export default createBTCClient;
+export default createBTCClient;

packages/babylon-proto-ts/src/lib/txs/baby.ts

@@ -16,15 +16,19 @@ export interface StakeParams {
   amount: bigint;
 }
 
-const createStakeMsg = ({ delegatorAddress, validatorAddress, amount }: StakeParams) => {
+const createStakeMsg = ({
+  delegatorAddress,
+  validatorAddress,
+  amount,
+}: StakeParams) => {
   const wrappedDelegateMsg = epochingtx.MsgWrappedDelegate.fromPartial({
     msg: {
       delegatorAddress,
       validatorAddress,
       amount: {
-        denom: 'ubbn',
-        amount: amount.toString()
-      }
+        denom: "ubbn",
+        amount: amount.toString(),
+      },
     },
   });
 
@@ -47,16 +51,20 @@ export interface UnstakeParams {
   amount: bigint;
 }
 
-const createUnstakeMsg = ({ delegatorAddress, validatorAddress, amount }: UnstakeParams) => {
-  const wrappedUndelegateMsg = epochingtx.MsgWrappedUndelegate.fromPartial({ 
+const createUnstakeMsg = ({
+  delegatorAddress,
+  validatorAddress,
+  amount,
+}: UnstakeParams) => {
+  const wrappedUndelegateMsg = epochingtx.MsgWrappedUndelegate.fromPartial({
     msg: {
       delegatorAddress,
       validatorAddress,
       amount: {
-        denom: 'ubbn',
-        amount: amount.toString()
-      }
-    }
+        denom: "ubbn",
+        amount: amount.toString(),
+      },
+    },
   });
 
   return {

packages/babylon-proto-ts/src/lib/txs/btc.ts

@@ -29,9 +29,7 @@ const createClaimRewardMsg = ({ address }: ClaimRewardParams) => {
  * @param params - The parameters for the stake expansion
  * @returns The BTC stake expansion message
  */
-const createExpandMsg = (
-  params: Partial<btcstakingtx.MsgBtcStakeExpand>,
-) => {
+const createExpandMsg = (params: Partial<btcstakingtx.MsgBtcStakeExpand>) => {
   const stakeExpandMsg = btcstakingtx.MsgBtcStakeExpand.fromPartial(params);
 
   return {

packages/babylon-proto-ts/src/lib/txs/index.ts

@@ -1,7 +1,7 @@
-import btc from './btc';
-import baby from './baby';
+import baby from "./baby";
+import btc from "./btc";
 
 export default {
   baby,
-  btc
-}
+  btc,
+};

packages/babylon-proto-ts/src/lib/utils/baby.ts

@@ -17,4 +17,4 @@ export function babyToUbbn(bbn: number): bigint {
  */
 export function ubbnToBaby(ubbn: bigint): number {
   return Number(ubbn) / BABY_IN_UBBN;
-}
+}

packages/babylon-proto-ts/src/lib/utils/index.ts

@@ -1,11 +1,8 @@
 import { createAminoTypes } from "./amino";
+import { babyToUbbn, ubbnToBaby } from "./baby";
 import { normalizeCosmjsAmount, normalizeRewardResponse } from "./normalize";
+import { buildPaginationParams, fetchAllPages } from "./pagination";
 import { createRegistry } from "./registry";
-import { babyToUbbn, ubbnToBaby } from './baby';
-import { 
-  fetchAllPages, 
-  buildPaginationParams
-} from "./pagination";
 
 export default {
   ubbnToBaby,

packages/babylon-proto-ts/src/lib/utils/pagination.ts

@@ -18,17 +18,19 @@ export interface PaginatedResult<T> {
 }
 
 // Helper to convert pagination options to query parameters
-export function buildPaginationParams(options: PaginationOptions = {}): Record<string, string> {
+export function buildPaginationParams(
+  options: PaginationOptions = {},
+): Record<string, string> {
   const params: Record<string, string> = {};
-  
+
   if (options.limit !== undefined) {
-    params['pagination.limit'] = options.limit.toString();
+    params["pagination.limit"] = options.limit.toString();
   }
-  
-  if (options.key !== undefined && options.key !== '') {
-    params['pagination.key'] = options.key;
+
+  if (options.key !== undefined && options.key !== "") {
+    params["pagination.key"] = options.key;
   }
-  
+
   return params;
 }
 
@@ -37,34 +39,34 @@ export async function fetchAllPages<T>(
   request: RequestFn,
   endpoint: string,
   dataKey: string,
-  options: PaginationOptions = {}
+  options: PaginationOptions = {},
 ): Promise<T[]> {
   const allData: T[] = [];
   let nextKey: string | null = options.key || null;
   const limit = options.limit || DEFAULT_PAGINATION_LIMIT;
-  
+
   do {
     const params = buildPaginationParams({
       ...options,
       limit,
       key: nextKey || undefined,
     });
-    
+
     const response = await request(endpoint, params);
     const data = response[dataKey];
-    
+
     if (data && Array.isArray(data)) {
       allData.push(...data);
     }
 
     const newNextKey = response.pagination?.nextKey;
-    nextKey = newNextKey && newNextKey !== '' ? newNextKey : null;
+    nextKey = newNextKey && newNextKey !== "" ? newNextKey : null;
 
-    if (data && data.length === 0 && nextKey === null) {  
+    if (data && data.length === 0 && nextKey === null) {
       break;
-    } 
+    }
   } while (nextKey !== null);
-  
+
   return allData;
 }
 
@@ -73,13 +75,13 @@ export async function fetchPage<T>(
   request: RequestFn,
   endpoint: string,
   dataKey: string,
-  options: PaginationOptions = {}
+  options: PaginationOptions = {},
 ): Promise<PaginatedResult<T>> {
   const params = buildPaginationParams(options);
-  
+
   const response = await request(endpoint, params);
   const data = response[dataKey];
-  
+
   return {
     data: data || [],
     pagination: {