@strkfarm/sdk 1.0.16 → 1.0.18

package/dist/index.mjs

@@ -1,10 +1,3 @@
-var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require : typeof Proxy !== "undefined" ? new Proxy(x, {
-  get: (a, b) => (typeof require !== "undefined" ? require : a)[b]
-}) : x)(function(x) {
-  if (typeof require !== "undefined") return require.apply(this, arguments);
-  throw Error('Dynamic require of "' + x + '" is not supported');
-});
-
 // src/modules/pricer.ts
 import axios2 from "axios";
 
@@ -28,6 +21,7 @@ var FatalError = class extends Error {
 var tokens = [{
   name: "Starknet",
   symbol: "STRK",
+  logo: "https://assets.coingecko.com/coins/images/26433/small/starknet.png",
   address: "0x4718f5a0fc34cc1af16a1cdee98ffb20c31f5cd61d6ab07201858f4287c938d",
   decimals: 18,
   coingeckId: "starknet"
@@ -60,6 +54,7 @@ var Global = class {
         symbol: token.symbol,
         address: token.address,
         decimals: token.decimals,
+        logo: token.logoUri,
         coingeckId: token.extensions.coingeckoId
       });
     });
@@ -139,16 +134,22 @@ var ContractAddr = class _ContractAddr {
   }
 };
 
-// src/modules/pricer.ts
-var CoinMarketCap = __require("coinmarketcap-api");
+// src/modules/pricerBase.ts
 var PricerBase = class {
+  constructor(config, tokens2) {
+    this.config = config;
+    this.tokens = tokens2;
+  }
   async getPrice(tokenSymbol) {
     throw new Error("Method not implemented");
   }
 };
-var Pricer = class {
+
+// src/modules/pricer.ts
+var Pricer = class extends PricerBase {
+  // e.g. ETH/USDC
   constructor(config, tokens2) {
-    this.tokens = [];
+    super(config, tokens2);
     this.prices = {};
     // code populates this map during runtime to determine which method to use for a given token
     // The method set will be the first one to try after first attempt
@@ -158,11 +159,6 @@ var Pricer = class {
      */
     this.PRICE_API = `https://api.coinbase.com/v2/prices/{{PRICER_KEY}}/buy`;
     this.EKUBO_API = "https://quoter-mainnet-api.ekubo.org/{{AMOUNT}}/{{TOKEN_ADDRESS}}/0x053c91253bc9682c04929ca02ed00b3e423f6710d2ee7e0d5ebb06f3ecf368a8";
-    // e.g. ETH/USDC
-    // backup oracle001
-    this.client = new CoinMarketCap(process.env.COINMARKETCAP_KEY);
-    this.config = config;
-    this.tokens = tokens2;
   }
   isReady() {
     const allPricesExist = Object.keys(this.prices).length === this.tokens.length;
@@ -294,10 +290,7 @@ var Pricer = class {
     return Number(data.data.amount);
   }
   async _getPriceCoinMarketCap(token) {
-    const result = await this.client.getQuotes({ symbol: token.symbol });
-    if (result.data)
-      return result.data[token.symbol].quote.USD.price;
-    throw new Error(result);
+    throw new Error("Not implemented");
   }
   async _getPriceEkubo(token, amountIn = new Web3Number(1, token.decimals), retry = 0) {
     const url = this.EKUBO_API.replace("{{TOKEN_ADDRESS}}", token.address).replace("{{AMOUNT}}", amountIn.toWei());
@@ -491,6 +484,7 @@ var _ZkLend = class _ZkLend extends ILending {
           name: pool.token.name,
           symbol: pool.token.symbol,
           address: savedTokenInfo?.address || "",
+          logo: "",
           decimals: pool.token.decimals,
           borrowFactor: Web3Number.fromWei(pool.borrow_factor.value, pool.borrow_factor.decimals),
           collareralFactor
@@ -600,14 +594,84 @@ var _ZkLend = class _ZkLend extends ILending {
 _ZkLend.POOLS_URL = "https://app.zklend.com/api/pools";
 var ZkLend = _ZkLend;
 
+// src/modules/pricer-from-api.ts
+import axios4 from "axios";
+var PricerFromApi = class extends PricerBase {
+  constructor(config, tokens2) {
+    super(config, tokens2);
+  }
+  async getPrice(tokenSymbol) {
+    try {
+      return await this.getPriceFromMyAPI(tokenSymbol);
+    } catch (e) {
+      logger.warn("getPriceFromMyAPI error", e);
+    }
+    logger.log("getPrice coinbase", tokenSymbol);
+    let retry = 0;
+    const MAX_RETRIES = 5;
+    for (retry = 1; retry < MAX_RETRIES + 1; retry++) {
+      try {
+        const priceInfo = await axios4.get(
+          `https://api.coinbase.com/v2/prices/${tokenSymbol}-USDT/spot`
+        );
+        if (!priceInfo) {
+          throw new Error("Failed to fetch price");
+        }
+        const data = await priceInfo.data;
+        const price = Number(data.data.amount);
+        return {
+          price,
+          timestamp: /* @__PURE__ */ new Date()
+        };
+      } catch (e) {
+        logger.warn("getPrice coinbase error", e, retry);
+        await new Promise((resolve) => setTimeout(resolve, retry * 1e3));
+      }
+    }
+    throw new Error(`Failed to fetch price for ${tokenSymbol}`);
+  }
+  async getPriceFromMyAPI(tokenSymbol) {
+    logger.verbose(`getPrice from redis: ${tokenSymbol}`);
+    const endpoint = "https://app.strkfarm.com";
+    const url = `${endpoint}/api/price/${tokenSymbol}`;
+    const priceInfoRes = await fetch(url);
+    const priceInfo = await priceInfoRes.json();
+    const now = /* @__PURE__ */ new Date();
+    const priceTime = new Date(priceInfo.timestamp);
+    if (now.getTime() - priceTime.getTime() > 9e5) {
+      throw new Error("Price is stale");
+    }
+    const price = Number(priceInfo.price);
+    return {
+      price,
+      timestamp: new Date(priceInfo.timestamp)
+    };
+  }
+};
+
 // src/interfaces/common.ts
 import { RpcProvider as RpcProvider2 } from "starknet";
+var RiskType = /* @__PURE__ */ ((RiskType2) => {
+  RiskType2["MARKET_RISK"] = "MARKET_RISK";
+  RiskType2["IMPERMANENT_LOSS"] = "IMPERMANENT_LOSS";
+  RiskType2["LIQUIDITY_RISK"] = "LIQUIDITY_RISK";
+  RiskType2["SMART_CONTRACT_RISK"] = "SMART_CONTRACT_RISK";
+  RiskType2["TECHNICAL_RISK"] = "TECHNICAL_RISK";
+  RiskType2["COUNTERPARTY_RISK"] = "COUNTERPARTY_RISK";
+  return RiskType2;
+})(RiskType || {});
 var Network = /* @__PURE__ */ ((Network2) => {
   Network2["mainnet"] = "mainnet";
   Network2["sepolia"] = "sepolia";
   Network2["devnet"] = "devnet";
   return Network2;
 })(Network || {});
+var FlowChartColors = /* @__PURE__ */ ((FlowChartColors2) => {
+  FlowChartColors2["Green"] = "purple";
+  FlowChartColors2["Blue"] = "#35484f";
+  FlowChartColors2["Purple"] = "#6e53dc";
+  return FlowChartColors2;
+})(FlowChartColors || {});
 function getMainnetConfig(rpcUrl = "https://starknet-mainnet.public.blastapi.io", blockIdentifier = "pending") {
   return {
     provider: new RpcProvider2({
@@ -2183,7 +2247,7 @@ function assert(condition, message) {
 }
 
 // src/strategies/vesu-rebalance.ts
-import axios4 from "axios";
+import axios5 from "axios";
 var VesuRebalance = class _VesuRebalance {
   // 10000 bps = 100%
   /**
@@ -2290,7 +2354,7 @@ var VesuRebalance = class _VesuRebalance {
     let isErrorPositionsAPI = false;
     let vesuPositions = [];
     try {
-      const res = await axios4.get(`https://api.vesu.xyz/positions?walletAddress=${this.address.address}`);
+      const res = await axios5.get(`https://api.vesu.xyz/positions?walletAddress=${this.address.address}`);
       const data2 = await res.data;
       vesuPositions = data2.data;
     } catch (e) {
@@ -2300,7 +2364,7 @@ var VesuRebalance = class _VesuRebalance {
     let isErrorPoolsAPI = false;
     let pools = [];
     try {
-      const res = await axios4.get(`https://api.vesu.xyz/pools`);
+      const res = await axios5.get(`https://api.vesu.xyz/pools`);
       const data2 = await res.data;
       pools = data2.data;
     } catch (e) {
@@ -2317,7 +2381,7 @@ var VesuRebalance = class _VesuRebalance {
       const assets = await vTokenContract.convert_to_assets(uint2562.bnToUint256(bal.toString()));
       const item = {
         pool_id: p.pool_id,
-        pool_name: vesuPosition?.pool.name,
+        pool_name: pool.name,
         max_weight: p.max_weight,
         current_weight: isErrorPositionsAPI || !vesuPosition ? 0 : Number(Web3Number.fromWei(vesuPosition.collateral.value, this.decimals()).dividedBy(totalAssets.toString()).toFixed(6)),
         v_token: p.v_token,
@@ -2460,16 +2524,50 @@ var VesuRebalance = class _VesuRebalance {
     }
     return this.contract.populate("rebalance", [actions]);
   }
+  async getInvestmentFlows(pools) {
+    const netYield = this.netAPYGivenPools(pools);
+    const baseFlow = {
+      title: "Your Deposit",
+      subItems: [{ key: `Net yield`, value: `${(netYield * 100).toFixed(2)}%` }],
+      linkedFlows: [],
+      style: { backgroundColor: "#6e53dc" /* Purple */.valueOf() }
+    };
+    let _pools = [...pools];
+    _pools = _pools.sort((a, b) => Number(b.amount.toString()) - Number(a.amount.toString()));
+    _pools.forEach((p) => {
+      const flow = {
+        title: `Pool name: ${p.pool_name}`,
+        subItems: [
+          { key: `APY`, value: `${(p.APY.netApy * 100).toFixed(2)}%` },
+          { key: "Weight", value: `${(p.current_weight * 100).toFixed(2)} / ${(p.max_weight * 100).toFixed(2)}%` }
+        ],
+        linkedFlows: [],
+        style: p.amount.greaterThan(0) ? { backgroundColor: "#35484f" /* Blue */.valueOf() } : { color: "gray" }
+      };
+      baseFlow.linkedFlows.push(flow);
+    });
+    return [baseFlow];
+  }
 };
 var _description = "Automatically diversify {{TOKEN}} holdings into different Vesu pools while reducing risk and maximizing yield. Defi spring STRK Rewards are auto-compounded as well.";
 var _protocol = { name: "Vesu", logo: "https://static-assets-8zct.onrender.com/integrations/vesu/logo.png" };
+var _riskFactor = [
+  { type: "SMART_CONTRACT_RISK" /* SMART_CONTRACT_RISK */, value: 0.5, weight: 25 },
+  { type: "TECHNICAL_RISK" /* TECHNICAL_RISK */, value: 0.5, weight: 25 },
+  { type: "COUNTERPARTY_RISK" /* COUNTERPARTY_RISK */, value: 1, weight: 50 }
+];
 var VesuRebalanceStrategies = [{
   name: "Vesu STRK",
   description: _description.replace("{{TOKEN}}", "STRK"),
   address: ContractAddr.from("0xeeb729d554ae486387147b13a9c8871bc7991d454e8b5ff570d4bf94de71e1"),
   type: "ERC4626",
   depositTokens: [Global.getDefaultTokens().find((t) => t.symbol === "STRK")],
-  protocols: [_protocol]
+  protocols: [_protocol],
+  maxTVL: Web3Number.fromWei("0", 18),
+  risk: {
+    riskFactor: _riskFactor,
+    netRisk: _riskFactor.reduce((acc, curr) => acc + curr.value * curr.weight, 0) / 100
+  }
 }];
 
 // src/notifs/telegram.ts
@@ -2737,6 +2835,7 @@ export {
   AutoCompounderSTRK,
   ContractAddr,
   FatalError,
+  FlowChartColors,
   Global,
   ILending,
   Initializable,
@@ -2745,8 +2844,9 @@ export {
   PasswordJsonCryptoUtil,
   Pragma,
   Pricer,
-  PricerBase,
+  PricerFromApi,
   PricerRedis,
+  RiskType,
   Store,
   TelegramNotif,
   VesuRebalance,

package/package.json

@@ -1,18 +1,23 @@
 {
   "name": "@strkfarm/sdk",
-  "version": "1.0.16",
+  "version": "1.0.18",
   "description": "STRKFarm TS SDK (Meant for our internal use, but feel free to use it)",
   "typings": "dist/index.d.ts",
-  "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "bin": {
     "accountsecure": "dist/cli.js"
   },
   "exports": {
     ".": {
-      "import": "./dist/index.mjs",
-      "require": "./dist/index.js",
-      "browser": "./dist/index.global.js",
+      "import": {
+        "browser": "./dist/index.browser.mjs",
+        "default": "./dist/index.mjs"
+      },
+      "require": {
+        "browser": "./dist/index.browser.global.js",
+        "default": "./dist/index.js"
+      },
+      "browser": "./dist/index.browser.global.js",
       "types": "./dist/index.d.ts"
     }
   },
@@ -22,12 +27,13 @@
   ],
   "scripts": {
     "test": "jest",
-    "build": "tsup --clean && pnpm run build:esm && npm run build:dts && npm run build:iife && npm run build-cli",
+    "build": "tsup --clean && pnpm run build:esm && npm run build:dts && npm run build:iife && npm run build-cli && pnpm run build:iife-esm",
     "build:esm": "tsup --clean false --format esm --platform node",
     "build-cli:esm": "tsup ./src/cli.ts --clean false --format esm --platform node",
     "build-cli": "tsup ./src/cli.ts --clean false && pnpm run build-cli:esm",
     "build:dts": "tsup --clean false --dts-only",
-    "build:iife": "tsup --clean false --format iife --platform browser"
+    "build:iife": "tsup --clean false --format iife --platform browser",
+    "build:iife-esm": "tsup --clean false --format iife --format esm --platform browser"
   },
   "keywords": [],
   "author": "",
@@ -48,10 +54,11 @@
     "bignumber.js": "4.0.4",
     "browser-assert": "^1.2.1",
     "chalk": "^4.1.2",
-    "coinmarketcap-api": "^3.1.1",
     "commander": "^12.1.0",
+    "form-data": "^4.0.2",
     "inquirer": "^10.1.2",
     "node-telegram-bot-api": "^0.66.0",
+    "proxy-from-env": "^1.1.0",
     "redis": "^4.7.0",
     "stacktrace-js": "^2.0.2",
     "starknet": "^6.11.0",

package/src/global.ts

@@ -48,6 +48,7 @@ export class FatalError extends Error {
 const tokens: TokenInfo[] = [{
     name: 'Starknet',
     symbol: 'STRK',
+    logo: 'https://assets.coingecko.com/coins/images/26433/small/starknet.png',
     address: '0x4718f5a0fc34cc1af16a1cdee98ffb20c31f5cd61d6ab07201858f4287c938d',
     decimals: 18,
     coingeckId: 'starknet'
@@ -110,6 +111,7 @@ export class Global {
                 symbol: token.symbol,
                 address: token.address,
                 decimals: token.decimals,
+                logo: token.logoUri,
                 coingeckId: token.extensions.coingeckoId,
             });
         });

package/src/interfaces/common.ts

@@ -1,11 +1,27 @@
-import { ContractAddr } from "@/dataTypes"
+import { ContractAddr, Web3Number } from "@/dataTypes"
 import { BlockIdentifier, RpcProvider } from "starknet"
 
+export enum RiskType {
+    MARKET_RISK = 'MARKET_RISK',
+    IMPERMANENT_LOSS = 'IMPERMANENT_LOSS',
+    LIQUIDITY_RISK = 'LIQUIDITY_RISK',
+    SMART_CONTRACT_RISK = 'SMART_CONTRACT_RISK',
+    TECHNICAL_RISK = 'TECHNICAL_RISK',
+    COUNTERPARTY_RISK = 'COUNTERPARTY_RISK', // e.g. bad debt
+}
+
+export interface RiskFactor {
+    type: RiskType,
+    value: number, // 0 to 5
+    weight: number // 0 to 100
+}
+
 export interface TokenInfo {
     name: string,
     symbol: string,
     address: string,
     decimals: number,
+    logo: string,
     coingeckId?: string,
 }
 
@@ -27,13 +43,36 @@ export interface IProtocol {
     logo: string,
 }
 
+export enum FlowChartColors {
+    Green = 'purple',
+    Blue = '#35484f',
+    Purple = '#6e53dc',
+}
+
+/**
+ * @property risk.riskFactor.factor - The risk factors that are considered for the strategy.
+ * @property risk.riskFactor.factor - The value of the risk factor from 0 to 10, 0 being the lowest and 10 being the highest.
+ */
 export interface IStrategyMetadata {
     name: string,
     description: string,
     address: ContractAddr,
     type: 'ERC4626' | 'ERC721' | 'Other',
     depositTokens: TokenInfo[],
-    protocols: IProtocol[]
+    protocols: IProtocol[],
+    auditUrl?: string,
+    maxTVL: Web3Number,
+    risk: {
+        riskFactor: RiskFactor[],
+        netRisk: number
+    }
+}
+
+export interface IInvestmentFlow {
+    title: string,
+    subItems: {key: string, value: string}[],
+    linkedFlows: IInvestmentFlow[],
+    style?: any
 }
 
 export function getMainnetConfig(rpcUrl = "https://starknet-mainnet.public.blastapi.io", blockIdentifier: BlockIdentifier = 'pending'): IConfig {

package/src/modules/index.ts

@@ -1,3 +1,4 @@
 export * from './pricer';
 export * from './pragma';
-export * from './zkLend';
+export * from './zkLend';
+export * from './pricer-from-api';

package/src/modules/pricer-from-api.ts

@@ -0,0 +1,61 @@
+import { logger } from "@/global";
+import { PriceInfo } from "./pricer";
+import axios from "axios";
+import { IConfig, TokenInfo } from "@/interfaces";
+import { PricerBase } from "./pricerBase";
+
+export class PricerFromApi extends PricerBase {
+    constructor(config: IConfig, tokens: TokenInfo[]) {
+        super(config, tokens);
+    }
+
+    async getPrice(tokenSymbol: string): Promise<PriceInfo> {
+        try {
+            return await this.getPriceFromMyAPI(tokenSymbol);
+        } catch (e) {
+            logger.warn('getPriceFromMyAPI error', e);
+        }
+        logger.log('getPrice coinbase', tokenSymbol);
+        let retry = 0;
+        const MAX_RETRIES = 5;
+        for (retry = 1; retry < MAX_RETRIES + 1; retry++) {
+            try {
+                const priceInfo = await axios.get(
+                    `https://api.coinbase.com/v2/prices/${tokenSymbol}-USDT/spot`,
+                );
+                if (!priceInfo) {
+                    throw new Error('Failed to fetch price');
+                }
+                const data = await priceInfo.data;
+                const price = Number(data.data.amount);
+                return {
+                    price,
+                    timestamp: new Date()
+                }
+            } catch (e) {
+                logger.warn('getPrice coinbase error', e, retry);
+                await new Promise((resolve) => setTimeout(resolve, retry * 1000));
+            }
+        }
+        throw new Error(`Failed to fetch price for ${tokenSymbol}`);
+    }
+
+    async getPriceFromMyAPI(tokenSymbol: string) {
+        logger.verbose(`getPrice from redis: ${tokenSymbol}`);
+        const endpoint = 'https://app.strkfarm.com'
+        const url = `${endpoint}/api/price/${tokenSymbol}`;
+        const priceInfoRes = await fetch(url);
+        const priceInfo = await priceInfoRes.json();
+        const now = new Date();
+        const priceTime = new Date(priceInfo.timestamp);
+        if (now.getTime() - priceTime.getTime() > 900000) {
+          // 15 mins
+          throw new Error('Price is stale');
+        }
+        const price = Number(priceInfo.price);
+        return {
+            price,
+            timestamp: new Date(priceInfo.timestamp)
+        }
+      }
+}

package/src/modules/pricer.ts

@@ -3,22 +3,14 @@ import { FatalError, Global, logger } from "@/global";
 import { TokenInfo } from "@/interfaces/common";
 import { IConfig } from "@/interfaces/common";
 import { Web3Number } from "@/dataTypes";
-const CoinMarketCap = require('coinmarketcap-api')
+import { PricerBase } from "./pricerBase";
 
 export interface PriceInfo {
     price: number,
     timestamp: Date
 }
 
-export abstract class PricerBase {
-    async getPrice(tokenSymbol: string): Promise<PriceInfo> {
-        throw new Error('Method not implemented');
-    }
-}
-
-export class Pricer implements PricerBase{
-    readonly config: IConfig;
-    readonly tokens: TokenInfo[] = [];
+export class Pricer extends PricerBase {
     protected prices: {
         [key: string]: PriceInfo
     } = {}
@@ -33,12 +25,8 @@ export class Pricer implements PricerBase{
     protected PRICE_API = `https://api.coinbase.com/v2/prices/{{PRICER_KEY}}/buy`;
     protected EKUBO_API = 'https://quoter-mainnet-api.ekubo.org/{{AMOUNT}}/{{TOKEN_ADDRESS}}/0x053c91253bc9682c04929ca02ed00b3e423f6710d2ee7e0d5ebb06f3ecf368a8'; // e.g. ETH/USDC
 
-    // backup oracle001
-    protected client = new CoinMarketCap(process.env.COINMARKETCAP_KEY!);
-    
     constructor(config: IConfig, tokens: TokenInfo[]) {
-        this.config = config;
-        this.tokens = tokens;
+        super(config, tokens);
     }
 
     isReady() {
@@ -187,11 +175,11 @@ export class Pricer implements PricerBase{
     }
 
     async _getPriceCoinMarketCap(token: TokenInfo): Promise<number> {
-        const result = await this.client.getQuotes({symbol: token.symbol});
-        if (result.data)
-            return result.data[token.symbol].quote.USD.price as number
+        // const result = await this.client.getQuotes({symbol: token.symbol});
+        // if (result.data)
+        //     return result.data[token.symbol].quote.USD.price as number
 
-        throw new Error(result);
+        throw new Error("Not implemented");
     }
 
     async _getPriceEkubo(token: TokenInfo, amountIn = new Web3Number(1, token.decimals), retry = 0): Promise<number> {

package/src/modules/pricerBase.ts

@@ -0,0 +1,15 @@
+import { IConfig, TokenInfo } from "@/interfaces";
+import { PriceInfo } from "./pricer";
+
+export abstract class PricerBase {
+    readonly config: IConfig;
+    readonly tokens: TokenInfo[];
+    constructor(config: IConfig, tokens: TokenInfo[]) {
+        this.config = config;
+        this.tokens = tokens;
+    }
+
+    async getPrice(tokenSymbol: string): Promise<PriceInfo> {
+        throw new Error('Method not implemented');
+    }
+}

package/src/modules/zkLend.ts

@@ -37,6 +37,7 @@ export class ZkLend extends ILending implements ILending {
                     name: pool.token.name,
                     symbol: pool.token.symbol,
                     address: savedTokenInfo?.address || '',
+                    logo: '',
                     decimals: pool.token.decimals,
                     borrowFactor: Web3Number.fromWei(pool.borrow_factor.value, pool.borrow_factor.decimals),
                     collareralFactor

package/src/node/pricer-redis.ts

@@ -6,6 +6,7 @@ import type { RedisClientType } from 'redis'
 
 export class PricerRedis extends Pricer {
     private redisClient: RedisClientType | null = null;
+    
     constructor(config: IConfig, tokens: TokenInfo[]) {
         super(config, tokens)
     }

package/src/strategies/vesu-rebalance.ts

@@ -1,11 +1,12 @@
 import { ContractAddr, Web3Number } from "@/dataTypes";
-import { IConfig, IProtocol, IStrategyMetadata } from "@/interfaces";
-import { Pricer, PricerBase } from "@/modules";
+import { FlowChartColors, IConfig, IInvestmentFlow, IProtocol, IStrategyMetadata, RiskFactor, RiskType } from "@/interfaces";
+import { Pricer } from "@/modules";
 import { CairoCustomEnum, Contract, num, uint256 } from "starknet";
 import VesuRebalanceAbi from '@/data/vesu-rebalance.abi.json';
 import { Global } from "@/global";
 import { assert } from "@/utils";
 import axios from "axios";
+import { PricerBase } from "@/modules/pricerBase";
 
 interface PoolProps {
     pool_id: ContractAddr;
@@ -247,7 +248,7 @@ export class VesuRebalance {
             const assets = await vTokenContract.convert_to_assets(uint256.bnToUint256(bal.toString()));
             const item = {
                 pool_id: p.pool_id,
-                pool_name: vesuPosition?.pool.name,
+                pool_name: pool.name,
                 max_weight: p.max_weight,
                 current_weight: isErrorPositionsAPI || !vesuPosition ? 0 : Number(Web3Number.fromWei(vesuPosition.collateral.value, this.decimals()).dividedBy(totalAssets.toString()).toFixed(6)),
                 v_token: p.v_token,
@@ -412,10 +413,43 @@ export class VesuRebalance {
         }
         return this.contract.populate('rebalance', [actions]);
     }
+
+    async getInvestmentFlows(pools: PoolInfoFull[]) {
+        const netYield = this.netAPYGivenPools(pools);
+       
+        const baseFlow: IInvestmentFlow = {
+            title: "Your Deposit",
+            subItems: [{key: `Net yield`, value: `${(netYield * 100).toFixed(2)}%`}],
+            linkedFlows: [],
+            style: {backgroundColor: FlowChartColors.Purple.valueOf()},
+        };
+
+        let _pools = [...pools];
+        _pools = _pools.sort((a, b) => Number(b.amount.toString()) - Number(a.amount.toString()));
+        _pools.forEach((p) => {
+            const flow: IInvestmentFlow = {
+                title: `Pool name: ${p.pool_name}`,
+                subItems: [
+                    {key: `APY`, value: `${(p.APY.netApy * 100).toFixed(2)}%`},
+                    {key: 'Weight', value: `${(p.current_weight * 100).toFixed(2)} / ${(p.max_weight * 100).toFixed(2)}%`}
+                ],
+                linkedFlows: [],
+                style: p.amount.greaterThan(0) ? {backgroundColor: FlowChartColors.Blue.valueOf()} : {color: 'gray'},
+            };
+            baseFlow.linkedFlows.push(flow);
+        });
+        return [baseFlow];
+    }
 }
 
 const _description = 'Automatically diversify {{TOKEN}} holdings into different Vesu pools while reducing risk and maximizing yield. Defi spring STRK Rewards are auto-compounded as well.'
 const _protocol: IProtocol = {name: 'Vesu', logo: 'https://static-assets-8zct.onrender.com/integrations/vesu/logo.png'}
+// need to fine tune better
+const _riskFactor: RiskFactor[] = [
+    {type: RiskType.SMART_CONTRACT_RISK, value: 0.5, weight: 25},
+    {type: RiskType.TECHNICAL_RISK, value: 0.5, weight: 25},
+    {type: RiskType.COUNTERPARTY_RISK, value: 1, weight: 50},
+]
 /**
  * Represents the Vesu Rebalance Strategies.
  */
@@ -425,5 +459,10 @@ export const VesuRebalanceStrategies: IStrategyMetadata[] = [{
     address: ContractAddr.from('0xeeb729d554ae486387147b13a9c8871bc7991d454e8b5ff570d4bf94de71e1'),
     type: 'ERC4626',
     depositTokens: [Global.getDefaultTokens().find(t => t.symbol === 'STRK')!],
-    protocols: [_protocol]
+    protocols: [_protocol],
+    maxTVL: Web3Number.fromWei('0', 18),
+    risk: {
+        riskFactor: _riskFactor,
+        netRisk: _riskFactor.reduce((acc, curr) => acc + curr.value * curr.weight, 0) / 100,
+    }
 }]