[echelon xlpt] include price conversion susde_usdc_xlpt

Author: 0xwml

constants/echelon.py

@@ -1,5 +1,7 @@
 LENDING_CONTRACT_ADDRESS = "0xc6bc659f1649553c1a3fa05d9727433dc03843baac29473c817d06d39e7621ba"
 
+XLPT_ORACLE_CONTRACT_ADDRESS = "0xff9c14f0ab345b7804388d93401f5f0651f29dff7082bf386eea9b7a695c769e"
+
 SUSDE_MARKET_ADDRESS = "0x778362f04f7904ba0b76913ec7c0c5cc04e469b0b96929c6998b34910690a740"
 
 SUSDE_TOKEN_ADDRESS = "0xb30a694a344edee467d9f82330bbe7c3b89f440a1ecd2da1f3bca266560fce69"

integrations/echelon_xlpt_integration.py

@@ -9,7 +9,7 @@
 from constants.example_integrations import (
     ECHELON_SUSDE_USDC_XLPT_COLLATERAL_START_BLOCK
 )
-from constants.echelon import LENDING_CONTRACT_ADDRESS, SUSDE_USDC_XLPT_MARKET_ADDRESS, SUSDE_USDC_TOKEN_ADDRESS, ETHENA_SUSDE_USDC_XLPT_ADDRESS_API_URL
+from constants.echelon import LENDING_CONTRACT_ADDRESS, XLPT_ORACLE_CONTRACT_ADDRESS, SUSDE_USDC_XLPT_MARKET_ADDRESS, SUSDE_USDC_TOKEN_ADDRESS, ETHENA_SUSDE_USDC_XLPT_ADDRESS_API_URL
 from constants.chains import Chain
 from integrations.integration_ids import IntegrationID as IntID
 from integrations.l2_delegation_integration import L2DelegationIntegration
@@ -37,7 +37,7 @@ def __init__(
         self.token_address = token_address
         self.market_address = market_address
         self.decimals = str(decimals)
-        self.echelon_ts_location = "ts/echelon_balances.ts"
+        self.echelon_ts_location = "ts/echelon_xlpt_balances.ts"
 
     def get_l2_block_balances(
         self, cached_data: Dict[int, Dict[str, float]], blocks: List[int]
@@ -69,6 +69,8 @@ def get_participants_data(self, block, user_addresses=[]):
                     str(self.decimals),
                     str(block),
                     json.dumps(user_addresses),
+                    XLPT_ORACLE_CONTRACT_ADDRESS,
+                    SUSDE_USDC_TOKEN_ADDRESS,
                 ],
                 capture_output=True,
                 text=True,

ts/echelon_xlpt_balances.ts

@@ -0,0 +1,89 @@
+import * as dotenv from "dotenv";
+import { Aptos, AptosConfig, Network } from "@aptos-labs/ts-sdk";
+
+dotenv.config();
+
+const config = new AptosConfig({ network: Network.MAINNET });
+// Aptos is the main entrypoint for all functions
+const client = new Aptos(config);
+
+const args = process.argv.slice(2);
+const LENDING_CONTRACT_ADDRESS = args[0];
+const market_address = args[1];
+const decimals = Number(args[2]);
+const block = Number(args[3]);
+const user_addresses: string[] = JSON.parse(args[4]);
+const XLPT_ORACLE_ADDRESS = args[5];
+const SUSDE_USDC_TOKEN_ADDRESS = args[6];
+
+async function getStrategy() {
+  // iterate over all users and get their susde balance
+  const user_balances: Record<string, number> = {};
+  for (const address of user_addresses) {
+    const susde_usdc_xlpt_balance = await client.view({
+      payload: {
+        function: `${LENDING_CONTRACT_ADDRESS}::lending::account_coins`,
+        functionArguments: [address, market_address],
+      },
+      options: { ledgerVersion: block },
+    });
+
+    const susde_usdc_xlpt_price = await client.view({
+      payload: {
+        function: `${XLPT_ORACLE_ADDRESS}::oracle::get_price`,
+        functionArguments: [SUSDE_USDC_TOKEN_ADDRESS],
+      },
+      options: { ledgerVersion: block },
+    });
+
+    const susde_usdc_value = Number(susde_usdc_xlpt_balance) * fp64ToFloat(BigInt((susde_usdc_xlpt_price[0] as { v: string }).v));;
+
+    user_balances[address] = scaleDownByDecimals(
+      susde_usdc_value,
+      decimals
+    );
+  }
+
+  console.log(JSON.stringify(user_balances));
+}
+
+function scaleDownByDecimals(value: number, decimals: number) {
+  return value / 10 ** decimals;
+}
+
+const ZERO = BigInt(0);
+const ONE = BigInt(1);
+
+export const fp64ToFloat = (a: bigint): number => {
+  // avoid large number
+  let mask = BigInt("0xffffffff000000000000000000000000");
+  if ((a & mask) != ZERO) {
+    throw new Error("too large");
+  }
+
+  // integer part
+  mask = BigInt("0x10000000000000000");
+  let base = 1;
+  let result = 0;
+  for (let i = 0; i < 32; ++i) {
+    if ((a & mask) != ZERO) {
+      result += base;
+    }
+    base *= 2;
+    mask = mask << ONE;
+  }
+
+  // fractional part
+  mask = BigInt("0x8000000000000000");
+  base = 0.5;
+  for (let i = 0; i < 32; ++i) {
+    if ((a & mask) != ZERO) {
+      result += base;
+    }
+    base /= 2;
+    mask = mask >> ONE;
+  }
+  return result;
+};
+
+const strategy = getStrategy().catch(console.error);