feat: pop upgrade (#1303)

* feat: pop upgrade

Author: jeremy-babylonlabs

package-lock.json

@@ -30,7 +30,7 @@
   },
   "dependencies": {
     "@babylonlabs-io/babylon-proto-ts": "1.7.1",
-    "@babylonlabs-io/btc-staking-ts": "2.3.4",
+    "@babylonlabs-io/btc-staking-ts": "2.4.1",
     "@babylonlabs-io/core-ui": "1.10.1",
     "@babylonlabs-io/wallet-connector": "1.10.0",
     "@bitcoin-js/tiny-secp256k1-asmjs": "2.2.3",

package.json

@@ -2,7 +2,7 @@ import { getPublicKeyNoCoord } from "@babylonlabs-io/btc-staking-ts";
 
 import { ClientError, ERROR_CODES } from "@/ui/common/errors";
 
-import { NetworkInfo } from "../types/networkInfo";
+import { NetworkInfo, NetworkUpgradeConfig } from "../types/networkInfo";
 
 import { apiWrapper } from "./apiWrapper";
 
@@ -25,6 +25,7 @@ interface NetworkInfoAPI {
     bbn: BbnParams[];
     btc: BtcCheckpointParams[];
   };
+  network_upgrade?: NetworkUpgradeConfig;
 }
 
 export interface BbnParams {
@@ -55,7 +56,7 @@ export const getNetworkInfo = async (): Promise<NetworkInfo> => {
     "Error getting network info",
   );
   const { data } = response;
-  const { params, staking_status } = data.data;
+  const { params, staking_status, network_upgrade } = data.data;
 
   const stakingVersions = (params.bbn || [])
     .sort((a, b) => a.version - b.version)
@@ -159,5 +160,6 @@ export const getNetworkInfo = async (): Promise<NetworkInfo> => {
         genesisParam: genesisEpochCheckParam,
       },
     },
+    networkUpgrade: network_upgrade,
   };
 };

src/ui/common/api/getNetworkInfo.ts

@@ -13,29 +13,33 @@ import { ClientError, ERROR_CODES } from "@/ui/common/errors";
 
 interface BbnRpcContextType {
   queryClient: QueryClient | undefined;
+  tmClient: Tendermint34Client | undefined;
   isLoading: boolean;
   error: Error | null;
   reconnect: () => Promise<void>;
 }
 
 const BbnRpcContext = createContext<BbnRpcContextType>({
   queryClient: undefined,
+  tmClient: undefined,
   isLoading: true,
   error: null,
   reconnect: async () => {},
 });
 
 export function BbnRpcProvider({ children }: { children: React.ReactNode }) {
   const [queryClient, setQueryClient] = useState<QueryClient>();
+  const [tmClient, setTmClient] = useState<Tendermint34Client>();
   const [isLoading, setIsLoading] = useState(true);
   const [error, setError] = useState<Error | null>(null);
   const { rpc } = getNetworkConfigBBN();
 
   const connect = useCallback(async () => {
     try {
-      const tmClient = await Tendermint34Client.connect(rpc);
-      const client = QueryClient.withExtensions(tmClient);
+      const tmClientInstance = await Tendermint34Client.connect(rpc);
+      const client = QueryClient.withExtensions(tmClientInstance);
       setQueryClient(client);
+      setTmClient(tmClientInstance);
       setIsLoading(false);
       setError(null);
     } catch (err) {
@@ -73,7 +77,7 @@ export function BbnRpcProvider({ children }: { children: React.ReactNode }) {
 
   return (
     <BbnRpcContext.Provider
-      value={{ queryClient, isLoading, error, reconnect }}
+      value={{ queryClient, tmClient, isLoading, error, reconnect }}
     >
       {children}
     </BbnRpcContext.Provider>

src/ui/common/context/rpc/BbnRpcProvider.tsx

@@ -8,7 +8,7 @@ import {
   setupBankExtension,
 } from "@cosmjs/stargate";
 
-import { ONE_MINUTE } from "@/ui/common/constants";
+import { ONE_MINUTE, ONE_SECOND } from "@/ui/common/constants";
 import { useBbnRpc } from "@/ui/common/context/rpc/BbnRpcProvider";
 import { useCosmosWallet } from "@/ui/common/context/wallet/CosmosWalletProvider";
 import { ClientError } from "@/ui/common/errors";
@@ -21,6 +21,7 @@ import { useRpcErrorHandler } from "../useRpcErrorHandler";
 const BBN_BTCLIGHTCLIENT_TIP_KEY = "BBN_BTCLIGHTCLIENT_TIP";
 const BBN_BALANCE_KEY = "BBN_BALANCE";
 const BBN_REWARDS_KEY = "BBN_REWARDS";
+const BBN_HEIGHT_KEY = "BBN_HEIGHT";
 const REWARD_GAUGE_KEY_BTC_DELEGATION = "BTC_STAKER";
 
 /**
@@ -30,7 +31,7 @@ const REWARD_GAUGE_KEY_BTC_DELEGATION = "BTC_STAKER";
 export const useBbnQuery = () => {
   const { isGeoBlocked, isLoading: isHealthcheckLoading } = useHealthCheck();
   const { bech32Address, connected } = useCosmosWallet();
-  const { queryClient } = useBbnRpc();
+  const { queryClient, tmClient } = useBbnRpc();
   const { hasRpcError, reconnect } = useRpcErrorHandler();
 
   /**
@@ -143,10 +144,37 @@ export const useBbnQuery = () => {
     refetchInterval: false, // Disable automatic periodic refetching
   });
 
+  /**
+   * Gets the current height of the Babylon Genesis chain.
+   * @returns {Promise<number>} - The current height of the Babylon Genesis chain.
+   */
+  const babyTipQuery = useClientQuery({
+    queryKey: [BBN_HEIGHT_KEY],
+    queryFn: async () => {
+      if (!tmClient) {
+        return 0;
+      }
+      try {
+        const status = await tmClient.status();
+        return status.syncInfo.latestBlockHeight;
+      } catch (error) {
+        throw new ClientError(
+          ERROR_CODES.EXTERNAL_SERVICE_UNAVAILABLE,
+          "Error getting Babylon chain height",
+          { cause: error as Error },
+        );
+      }
+    },
+    enabled: Boolean(tmClient && connected),
+    staleTime: ONE_SECOND * 10,
+    refetchInterval: false, // Disable automatic periodic refetching
+  });
+
   return {
     rewardsQuery,
     balanceQuery,
     btcTipQuery,
+    babyTipQuery,
     hasRpcError,
     reconnectRpc: reconnect,
     queryClient,

src/ui/common/hooks/client/rpc/queries/useBbnQuery.ts

@@ -1,16 +1,21 @@
 import { BabylonBtcStakingManager } from "@babylonlabs-io/btc-staking-ts";
-import { useCallback } from "react";
+import { useCallback, useMemo } from "react";
 
+import { getNetworkConfigBBN } from "@/ui/common/config/network/bbn";
 import { useBTCWallet } from "@/ui/common/context/wallet/BTCWalletProvider";
 import { useCosmosWallet } from "@/ui/common/context/wallet/CosmosWalletProvider";
 import { useBbnTransaction } from "@/ui/common/hooks/client/rpc/mutation/useBbnTransaction";
+import { useBbnQuery } from "@/ui/common/hooks/client/rpc/queries/useBbnQuery";
 import { useEventBus } from "@/ui/common/hooks/useEventBus";
 import { useLogger } from "@/ui/common/hooks/useLogger";
 import { useAppState } from "@/ui/common/state";
 
 export const useStakingManagerService = () => {
   const { networkInfo } = useAppState();
   const { signBbnTx } = useBbnTransaction();
+  const {
+    babyTipQuery: { data: bbnHeight = 0 },
+  } = useBbnQuery();
   const logger = useLogger();
   const eventBus = useEventBus();
 
@@ -24,6 +29,18 @@ export const useStakingManagerService = () => {
 
   const versionedParams = networkInfo?.params.bbnStakingParams?.versions;
 
+  const networkUpgrade = useMemo(
+    () => ({
+      pop: {
+        upgradeHeight: networkInfo?.networkUpgrade?.pop?.[0]?.height ?? 0,
+        version: networkInfo?.networkUpgrade?.pop?.[0]?.version ?? 0,
+      },
+    }),
+    [networkInfo?.networkUpgrade],
+  );
+
+  const { chainId } = getNetworkConfigBBN();
+
   const isLoading =
     !btcNetwork ||
     !cosmosConnected ||
@@ -56,6 +73,10 @@ export const useStakingManagerService = () => {
 
     const bbnProvider = {
       signTransaction: signBbnTx,
+      getCurrentHeight: async () => {
+        return bbnHeight;
+      },
+      getChainId: async () => chainId,
     };
 
     return new BabylonBtcStakingManager(
@@ -64,18 +85,22 @@ export const useStakingManagerService = () => {
       btcProvider,
       bbnProvider,
       eventBus,
+      networkUpgrade,
     );
   }, [
     isLoading,
     btcNetwork,
     versionedParams,
+    networkUpgrade,
+    bbnHeight,
     logger,
     cosmosConnected,
     btcConnected,
     eventBus,
     signPsbt,
     signMessage,
     signBbnTx,
+    chainId,
   ]);
 
   return {

src/ui/common/hooks/services/useStakingManagerService.ts

@@ -1,5 +1,14 @@
 import { StakingParams } from "@babylonlabs-io/btc-staking-ts";
 
+export interface PopUpgradeConfig {
+  height: number;
+  version: number;
+}
+
+export interface NetworkUpgradeConfig {
+  pop?: PopUpgradeConfig[];
+}
+
 export interface BbnStakingParamsVersion extends StakingParams {
   version: number;
   minCommissionRate: string;
@@ -28,7 +37,6 @@ export interface BtcEpochCheckParams {
   latestParam: BtcEpochCheckParamsVersion;
   versions: BtcEpochCheckParamsVersion[];
 }
-
 export interface StakingStatus {
   isStakingOpen: boolean;
 }
@@ -41,4 +49,5 @@ export interface Params {
 export interface NetworkInfo {
   stakingStatus: StakingStatus;
   params: Params;
+  networkUpgrade?: NetworkUpgradeConfig;
 }