feat(packages): wallet menu component (#46) (#50)

* feat(packages): wallet menu component, icon components and util components

Co-authored-by: Jeremy <[email protected]>

Author: 0xDazzer

packages/babylon-proto-ts/src/lib/client.ts

@@ -0,0 +1,68 @@
+import type { DistributionExtension, StakingExtension } from "@cosmjs/stargate";
+import type { DelegationDelegatorReward } from "cosmjs-types/cosmos/distribution/v1beta1/distribution";
+import type { DelegationResponse, Validator } from "cosmjs-types/cosmos/staking/v1beta1/staking";
+
+interface Dependencies {
+  staking: StakingExtension["staking"];
+  distribution: DistributionExtension["distribution"];
+}
+
+const createBabylonClient = ({ staking, distribution }: Dependencies) => ({
+  /**
+   * Gets all delegations of the user's account.
+   * @param {string} address - The address to get the delegations of.
+   * @returns {Promise<DelegationResponse[]>} - The delegations of the address.
+   */
+  async getDelegations(
+    address: string,
+  ): Promise<DelegationResponse[]> {
+    try {
+      const response =
+        await staking.delegatorDelegations(address);
+      return response.delegationResponses || [];
+    } catch (error) {
+      throw new Error(`Failed to fetch delegations for ${address}`, {
+        cause: error,
+      });
+    }
+  },
+
+  /**
+   * Gets all delegation rewards of the user's account.
+   * @param {string} address - The address to get the delegation rewards of.
+   * @returns {Promise<DelegationDelegatorReward[]>} - The delegation rewards of the address.
+   */
+  async getRewards(
+    address: string,
+  ): Promise<DelegationDelegatorReward[]> {
+    try {
+      const response =
+        await distribution.delegationTotalRewards(address);
+      return response.rewards || [];
+    } catch (error) {
+      if (error instanceof Error && error.message.includes("no delegation")) {
+        return [];
+      }
+      throw new Error(`Failed to fetch delegation rewards for ${address}`, {
+        cause: error,
+      });
+    }
+  },
+
+  /**
+   * Gets all the validators.
+   * @returns {Promise<Validator[]>} - All validators.
+   */
+  async getValidators(): Promise<Validator[]> {
+    try {
+      const response = await staking.validators("");
+      return response.validators || [];
+    } catch (error) {
+      throw new Error(`Failed to fetch validators`, {
+        cause: error,
+      });
+    }
+  }
+})
+
+export default createBabylonClient;

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

@@ -0,0 +1,100 @@
+import type{ BankExtension, Coin } from "@cosmjs/stargate";
+
+import { REWARD_GAUGE_KEY_BTC_DELEGATION } from "../../constants";
+import * as btclightclientquery from "../../generated/babylon/btclightclient/v1/query";
+import * as incentivequery from "../../generated/babylon/incentive/query";
+
+interface Dependencies {
+  incentive: incentivequery.QueryClientImpl;
+  btcLight: btclightclientquery.QueryClientImpl;
+  bank: BankExtension["bank"];
+}
+
+const createBTCClient = ({ incentive, bank, btcLight }: Dependencies) => ({
+/**
+   * Gets the rewards of the user's account.
+   * @param {string} address - The address to get the rewards of.
+   * @returns {Promise<number>} - The rewards of the address.
+   */
+  async getRewards(address: string): Promise<number> {
+    try {
+      const req = incentivequery.QueryRewardGaugesRequest.fromPartial({
+        address,
+      });
+
+      const rewards = await incentive.RewardGauges(req);
+      if (!rewards || !rewards.rewardGauges) {
+        return 0;
+      }
+
+      const coins =
+        rewards.rewardGauges[REWARD_GAUGE_KEY_BTC_DELEGATION]?.coins;
+      if (!coins) {
+        return 0;
+      }
+
+      const withdrawnCoins =
+        rewards.rewardGauges[
+          REWARD_GAUGE_KEY_BTC_DELEGATION
+        ]?.withdrawnCoins.reduce(
+          (acc: number, coin: Coin) => acc + Number(coin.amount),
+          0,
+        ) || 0;
+
+      return (
+        coins.reduce(
+          (acc: number, coin: Coin) => acc + Number(coin.amount),
+          0,
+        ) - withdrawnCoins
+      );
+    } catch (error) {
+      // If error message contains "reward gauge not found", silently return 0
+      // This is to handle the case where the user has no rewards, meaning
+      // they have not staked
+      if (
+        error instanceof Error &&
+        error.message.includes("reward gauge not found")
+      ) {
+        return 0;
+      }
+      throw new Error(`Failed to fetch rewards for ${address}`, {
+        cause: error,
+      });
+    }
+  },
+
+  /**
+   * Gets the balance of an address in the Babylon chain.
+   * @param {string} address - The address to get the balance of.
+   * @param {string} denom - The denom of the balance to get.
+   * @returns {Promise<number>} - The balance of the address.
+   */
+  async getBalance(address: string, denom: string = "ubbn"): Promise<number> {
+    try {
+      const balance = await bank.balance(address, denom);
+      return Number(balance?.amount ?? 0);
+    } catch (error) {
+      throw new Error(`Failed to fetch balance for ${address}`, {
+        cause: error,
+      });
+    }
+  },
+
+  /**
+   * Gets the tip of the Bitcoin blockchain.
+   * @returns {Promise<BTCHeaderInfoResponse>} - The tip of the Bitcoin blockchain.
+   */
+  async getBTCTipHeight(): Promise<number> {
+    try {
+      const req = btclightclientquery.QueryTipRequest.fromPartial({});
+      const { header } = await btcLight.Tip(req);
+      return Number(header?.height ?? 0);
+    } catch (error) {
+      throw new Error(`Failed to fetch BTC tip height`, {
+        cause: error,
+      });
+    }
+  }
+})
+
+export default createBTCClient;