Merge pull request #77 from invariant-labs/add-supported-tokens-eclipse

add supported tokens to eclipse
invariant-labs · Oct 31, 2024 · 10c7154 · 10c7154
2 parents bb5dda0 + adbf398
commit 10c7154
Show file tree

Hide file tree

Showing 4 changed files with 65 additions and 5 deletions.
diff --git a/svm/package.json b/svm/package.json
@@ -11,7 +11,7 @@
   "license": "ISC",
   "dependencies": {
     "@invariant-labs/sdk": "0.9.67",
-    "@invariant-labs/sdk-eclipse": "^0.0.21",
+    "@invariant-labs/sdk-eclipse": "^0.0.28",
     "@invariant-labs/staker-sdk": "0.2.11",
     "@project-serum/anchor": "0.21.0",
     "@solana/spl-token": "0.1.6",

diff --git a/svm/src/eclipse/full-snap.ts b/svm/src/eclipse/full-snap.ts
@@ -11,6 +11,7 @@ import {
   PoolStatsDataWithString,
   PoolWithAddress,
   printBN,
+  supportedTokens,
   TimeData,
   TokenStatsDataWithString,
 } from "../utils";
@@ -248,6 +249,20 @@ export const createSnapshotForNetwork = async (network: Network) => {
   tvl24.change = ((tvl24.value - prevTvl24) / prevTvl24) * 100;
   fees24.change = ((fees24.value - prevFees24) / prevFees24) * 100;
 
+  if (network === Network.MAIN) {
+    for (const supportedToken of Object.keys(supportedTokens)) {
+      const result = await market.getCurrentTokenStats(
+        supportedToken,
+        "So11111111111111111111111111111111111111112",
+        tokensPricesData["7vfCXTUXx5WJV5JADk17DUJ4ksgau7utNKj4b963voxs"].price
+      );
+
+      if (!("error" in result) && tokensDataObject[supportedToken]) {
+        tokensDataObject[supportedToken].price = +result.priceUsd;
+      }
+    }
+  }
+
   fs.writeFileSync(
     fileName,
     JSON.stringify({

diff --git a/svm/src/eclipse/snap.ts b/svm/src/eclipse/snap.ts
@@ -19,6 +19,7 @@ import {
   getUsdValue24,
   PoolSnapshot,
   PoolStatsData,
+  supportedTokens,
   TokenData,
 } from "../utils";
 
@@ -79,6 +80,24 @@ export const createSnapshotForNetwork = async (network: Network) => {
 
   let poolsData: any[] = [];
 
+  const supportedTokensWithPrices = {};
+
+  if (network === Network.MAIN) {
+    for (const supportedToken of Object.keys(supportedTokens)) {
+      const result = await market.getCurrentTokenStats(
+        supportedToken,
+        "So11111111111111111111111111111111111111112",
+        coingeckoPrices["ethereum"]
+      );
+
+      if (!("error" in result)) {
+        supportedTokensWithPrices[supportedToken] = +result.priceUsd;
+      }
+
+      tokensData[supportedToken] = supportedTokens[supportedToken];
+    }
+  }
+
   for (let pool of allPools) {
     const pair = new Pair(pool.tokenX, pool.tokenY, {
       fee: pool.fee.v,
@@ -94,19 +113,27 @@ export const createSnapshotForNetwork = async (network: Network) => {
     poolsDict[address.toString()] = pool;
 
     let lastSnapshot: PoolSnapshot | undefined;
-    const tokenXData = tokensData?.[pool.tokenX.toString()] ?? {
+    let tokenXData = tokensData?.[pool.tokenX.toString()] ?? {
       decimals: 0,
     };
-    const tokenYData = tokensData?.[pool.tokenY.toString()] ?? {
+    let tokenYData = tokensData?.[pool.tokenY.toString()] ?? {
       decimals: 0,
     };
-    const tokenXPrice = tokenXData.coingeckoId
+    let tokenXPrice = tokenXData.coingeckoId
       ? coingeckoPrices[tokenXData.coingeckoId] ?? 0
       : 0;
-    const tokenYPrice = tokenYData.coingeckoId
+    let tokenYPrice = tokenYData.coingeckoId
       ? coingeckoPrices[tokenYData.coingeckoId] ?? 0
       : 0;
 
+    if (Object.keys(supportedTokens).includes(pool.tokenX.toString())) {
+      tokenXPrice = supportedTokensWithPrices[pool.tokenX.toString()];
+    }
+
+    if (Object.keys(supportedTokens).includes(pool.tokenY.toString())) {
+      tokenYPrice = supportedTokensWithPrices[pool.tokenY.toString()];
+    }
+
     if (snaps?.[address.toString()]) {
       lastSnapshot =
         snaps[address.toString()].snapshots[
@@ -231,6 +258,15 @@ export const createSnapshotForNetwork = async (network: Network) => {
       };
     }
 
+    snaps[address].tokenX = {
+      address: xAddress,
+      decimals: tokensData?.[xAddress]?.decimals ?? 0,
+    };
+    snaps[address].tokenY = {
+      address: yAddress,
+      decimals: tokensData?.[yAddress]?.decimals ?? 0,
+    };
+
     snaps[address].snapshots.push({
       timestamp,
       ...stats,

diff --git a/svm/src/utils.ts b/svm/src/utils.ts
@@ -546,3 +546,12 @@ export const getEclipseTokensData = (
       return {};
   }
 };
+
+export const supportedTokens = {
+  LaihKXA47apnS599tyEyasY2REfEzBNe4heunANhsMx: {
+    decimals: 5,
+  },
+  trbts2EsWyMdnCjsHUFBKLtgudmBD7Rfbz8zCg1s4EK: {
+    decimals: 9,
+  },
+};