Eaglefi (DefiLlama#15380)

Author: ayoubbuoya

projects/eaglefi/index.js

@@ -0,0 +1,66 @@
+const { queryKey, getAddresssDataStoreKeys, bytesToStr, bytesToBigInt } = require('../helper/chain/massa.js');
+const { sumTokens2 } = require('../helper/unwrapLPs.js');
+
+
+const registryAddress = "AS1NYihs2Wy4D4P68JGY2hYSDDaqZ5YxhM2nDRsJVFZUykEEdSAW";
+
+
+async function getAllPoolsAddresses(registryAddress) {
+
+  const registryKeys = await getAddresssDataStoreKeys(registryAddress, "", true);
+  const poolsAddresses = [];
+
+  const transform = val => {
+    const arg = new Uint8Array(val);
+
+    const start_offset = 4;
+    let end_offset = start_offset + 62;
+    const p = arg.slice(start_offset, end_offset);
+    let poolAddressPlus = bytesToStr(p);
+    poolAddressPlus = "AS" + poolAddressPlus.split("AS")[1];
+    const poolAddress = poolAddressPlus.slice(0, poolAddressPlus.length - 1);
+
+    return poolAddress;
+  }
+
+  for (const serializedKey of registryKeys) {
+    const key = bytesToStr(serializedKey);
+
+    if (!key.startsWith("pools")) {
+      continue;
+    }
+
+    // Query that key to get the pools address
+    let poolAddress = await queryKey([registryAddress], key, transform);
+    poolsAddresses.push(poolAddress.toString());
+  }
+
+
+  return poolsAddresses;
+}
+
+
+async function tvl(
+  api
+) {
+  const poolsAddresses = await getAllPoolsAddresses(registryAddress);
+
+
+  for (const poolAddress of poolsAddresses) {
+    const aReserve = await queryKey([poolAddress], "aTokenReserve", val => bytesToBigInt(val));
+    const bReserve = await queryKey([poolAddress], "bTokenReserve", val => bytesToBigInt(val));
+
+    const aToken = await queryKey([poolAddress], "tokenA");
+    const bToken = await queryKey([poolAddress], "tokenB");
+
+    api.add(aToken, aReserve)
+    api.add(bToken, bReserve)
+  }
+
+  return sumTokens2({ api })
+}
+
+module.exports = {
+  timetravel: false,
+  massa: { tvl, }
+};

projects/helper/chain/massa.js

@@ -39,6 +39,10 @@ function strToBytes(str) {
   return new Uint8Array(Buffer.from(str, 'utf-8'));
 }
 
+function bytesToStr(bytes) {
+  return u8ArrayToString(bytes).replace(/\0/g, '');
+}
+
 
 /**
  * Converts a Uint8Array into an unsigned 64-bit integer (u64) BigInt.
@@ -99,8 +103,27 @@ async function getTokenBalances(tokenAddresses, ownerAddresses) {
   return res.map((entry) => bytesToBigInt(entry.candidate_value).toString());
 }
 
+function convertUnit8ArrayToNumberArray(unit8Array) {
+  return Array.from(unit8Array, (byte) => byte)
+}
+
+
+async function getAddresssDataStoreKeys(address, prefix, is_final) {
+  const res = await request('get_addresses_datastore_keys', [{
+    address: address,
+    prefix: convertUnit8ArrayToNumberArray(prefix),
+    is_final: is_final,
+  }]);
+
+  return res[0].keys;
+
+}
+
 module.exports = {
   queryKey,
   u8ArrayToString,
   getTokenBalances,
+  getAddresssDataStoreKeys,
+  bytesToStr,
+  bytesToBigInt,
 };