track typus treasury DefiLlama#12421

Author: g1nt0ki

projects/helper/chain/sui.js

@@ -134,13 +134,23 @@ function dexExport({
 }
 
 
-async function sumTokens({ balances = {}, owners = [], blacklistedTokens = [], tokens = [], api }) {
+async function sumTokens({ owners = [], blacklistedTokens = [], api, tokens = [], }) {
   owners = getUniqueAddresses(owners, true)
   const bals = await call('suix_getAllBalances', owners)
-  bals.forEach(i => api.add(i.coinType, i.totalBalance))
+  const blacklistSet = new Set(blacklistedTokens)
+  const tokenSet = new Set(tokens)
+  bals.forEach(i => {
+    if (blacklistSet.has(i.coinType)) return;
+    if (tokenSet.size > 0 && !tokenSet.has(i.coinType)) return;
+    api.add(i.coinType, i.totalBalance)
+  })
   return api.getBalances()
 }
 
+function sumTokensExport(config) {
+  return (api) => sumTokens({ ...config, api })
+}
+
 async function queryEventsByType({ eventType, transform = i => i }) {
   const query = `query GetEvents($after: String, $eventType: String!) {
   events(first: 50, after: $after, filter: { eventType: $eventType }) {
@@ -174,5 +184,6 @@ module.exports = {
   getDynamicFieldObjects,
   dexExport,
   sumTokens,
+  sumTokensExport,
   queryEventsByType,
 };

projects/helper/sumTokens.js

@@ -4,6 +4,7 @@ const { sumTokens2: sumTokensEVM, nullAddress, } = require('./unwrapLPs')
 const sdk = require('@defillama/sdk')
 
 const helpers = {
+  "sui": require("./chain/sui"),
   "eos": require("./chain/eos"),
   "ton": require("./chain/ton"),
   "ergo": require("./chain/ergo"),

projects/treasury/typus-finance.js

@@ -0,0 +1,8 @@
+const { treasuryExports } = require("../helper/treasury");
+
+module.exports = treasuryExports({
+  sui: {
+    owners: ['0xb9a09efd534d29cc9f990db26b2dab00289f32de0cdcefa68c6808de208bc9cb'],
+    ownTokens: ['0xf82dc05634970553615eef6112a1ac4fb7bf10272bf6cbe0f80ef44a6c489385::typus::TYPUS'],
+  },
+})