Helpers: Implement GetToken(Price|Decimals) for Web3 and TON

Emmet-Finance · Jun 24, 2024 · 0a26676 · 0a26676
1 parent b2b26c7
commit 0a26676
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 2 deletions.
diff --git a/src/chains/index.ts b/src/chains/index.ts
@@ -201,3 +201,11 @@ export interface GetEstimatedTime {
 export interface GetBridgeAddress {
   bridge: () => Promise<string>;
 }
+
+export interface GetTokenPrice {
+  getTokenPrice: (symbol: string) => Promise<bigint>;
+}
+
+export interface GetTokenPriceDecimals {
+  getPriceDecimals: (symbol: string) => Promise<bigint>;
+}
diff --git a/src/chains/ton.ts b/src/chains/ton.ts
@@ -19,6 +19,8 @@ import type {
   GetProtocolFeeInUSD,
   GetProvider,
   GetTokenBalance,
+  GetTokenPrice,
+  GetTokenPriceDecimals,
   GetTxFee,
   NativeCoinName,
   ProtocolFee,
@@ -49,7 +51,9 @@ export type TonHelper = GetBalance &
   TokenInfo &
   GetEstimatedTime &
   GetBridgeAddress &
-  GetProtocolFeeInUSD;
+  GetProtocolFeeInUSD &
+  GetTokenPrice &
+  GetTokenPriceDecimals;
 
 export interface TonParams {
   client: TonClient;
@@ -247,6 +251,12 @@ export function tonHandler({
 
   return {
     estimateTime: () => Promise.resolve(undefined),
+    async getPriceDecimals(_) {
+      return 0n;
+    },
+    async getTokenPrice(_) {
+      return 0n;
+    },
     async emmetHashFromtx(hash) {
       const txs = await client.getTransactions(bridge, {
         hash,

diff --git a/src/chains/web3.ts b/src/chains/web3.ts
@@ -17,6 +17,8 @@ import type {
   GetProtocolFeeInUSD,
   GetProvider,
   GetTokenBalance,
+  GetTokenPrice,
+  GetTokenPriceDecimals,
   GetTxFee,
   NativeCoinName,
   PreTransfer,
@@ -51,7 +53,9 @@ export type Web3Helper = GetBalance &
   GetEmmetHashFromTx &
   GetEstimatedTime &
   GetBridgeAddress &
-  GetProtocolFeeInUSD;
+  GetProtocolFeeInUSD &
+  GetTokenPrice &
+  GetTokenPriceDecimals;
 
 export interface Web3Params {
   provider: Provider;
@@ -207,5 +211,13 @@ export async function web3Helper({
         tx: tx,
       };
     },
+    async getTokenPrice(symbol) {
+      const tokenPrice = data.getTokenPrice(symbol);
+      return tokenPrice;
+    },
+    async getPriceDecimals(symbol) {
+      const tokenDecimals = data.getPriceDecimals(symbol);
+      return tokenDecimals;
+    },
   };
 }