@strkfarm/sdk 1.0.17 → 1.0.19

package/src/strategies/vesu-rebalance.ts

@@ -1,5 +1,5 @@
 import { ContractAddr, Web3Number } from "@/dataTypes";
-import { IConfig, IInvestmentFlow, IProtocol, IStrategyMetadata, RiskFactor, RiskType } from "@/interfaces";
+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';
@@ -7,6 +7,8 @@ import { Global } from "@/global";
 import { assert } from "@/utils";
 import axios from "axios";
 import { PricerBase } from "@/modules/pricerBase";
+import { BaseStrategy, SingleActionAmount, SingleTokenInfo } from "./base-strategy";
+import { getAPIUsingHeadlessBrowser } from "@/node/headless";
 
 interface PoolProps {
     pool_id: ContractAddr;
@@ -21,6 +23,10 @@ interface Change {
     isDeposit: boolean;
 }
 
+interface VesuRebalanceSettings {
+    feeBps: number;
+}
+
 interface PoolInfoFull {
     pool_id: ContractAddr;
     pool_name: string | undefined;
@@ -42,15 +48,13 @@ interface PoolInfoFull {
  * This class implements an automated rebalancing strategy for Vesu pools,
  * managing deposits and withdrawals while optimizing yield through STRK rewards.
  */
-export class VesuRebalance {
-    /** Configuration object for the strategy */
-    readonly config: IConfig;
+export class VesuRebalance extends BaseStrategy<SingleTokenInfo, SingleActionAmount> {
     /** Contract address of the strategy */
     readonly address: ContractAddr;
     /** Pricer instance for token price calculations */
     readonly pricer: PricerBase;
     /** Metadata containing strategy information */
-    readonly metadata: IStrategyMetadata;
+    readonly metadata: IStrategyMetadata<VesuRebalanceSettings>;
     /** Contract instance for interacting with the strategy */
     readonly contract: Contract;
     readonly BASE_WEIGHT = 10000; // 10000 bps = 100%
@@ -62,8 +66,8 @@ export class VesuRebalance {
      * @param metadata - Strategy metadata including deposit tokens and address
      * @throws {Error} If more than one deposit token is specified
      */
-    constructor(config: IConfig, pricer: PricerBase, metadata: IStrategyMetadata) {
-        this.config = config;
+    constructor(config: IConfig, pricer: PricerBase, metadata: IStrategyMetadata<VesuRebalanceSettings>) {
+        super(config);
         this.pricer = pricer;
         
         assert(metadata.depositTokens.length === 1, 'VesuRebalance only supports 1 deposit token');
@@ -79,12 +83,13 @@ export class VesuRebalance {
      * @param receiver - Address that will receive the strategy tokens
      * @returns Populated contract call for deposit
      */
-    depositCall(assets: Web3Number, receiver: ContractAddr) {
+    depositCall(amountInfo: SingleActionAmount, receiver: ContractAddr) {
         // Technically its not erc4626 abi, but we just need approve call
         // so, its ok to use it
-        const assetContract = new Contract(VesuRebalanceAbi, this.metadata.depositTokens[0].address, this.config.provider);
-        const call1 = assetContract.populate('approve', [this.address.address, uint256.bnToUint256(assets.toWei())]);
-        const call2 = this.contract.populate('deposit', [uint256.bnToUint256(assets.toWei()), receiver.address]);
+        assert(amountInfo.tokenInfo.address.eq(this.asset().address), 'Deposit token mismatch');
+        const assetContract = new Contract(VesuRebalanceAbi, this.asset().address.address, this.config.provider);
+        const call1 = assetContract.populate('approve', [this.address.address, uint256.bnToUint256(amountInfo.amount.toWei())]);
+        const call2 = this.contract.populate('deposit', [uint256.bnToUint256(amountInfo.amount.toWei()), receiver.address]);
         return [call1, call2];
     }
 
@@ -95,8 +100,8 @@ export class VesuRebalance {
      * @param owner - Address that owns the strategy tokens
      * @returns Populated contract call for withdrawal
      */
-    withdrawCall(assets: Web3Number, receiver: ContractAddr, owner: ContractAddr) {
-        return [this.contract.populate('withdraw', [uint256.bnToUint256(assets.toWei()), receiver.address, owner.address])];
+    withdrawCall(amountInfo: SingleActionAmount, receiver: ContractAddr, owner: ContractAddr) {
+        return [this.contract.populate('withdraw', [uint256.bnToUint256(amountInfo.amount.toWei()), receiver.address, owner.address])];
     }
 
     /**
@@ -127,6 +132,7 @@ export class VesuRebalance {
         let price = await this.pricer.getPrice(this.metadata.depositTokens[0].symbol);
         const usdValue = Number(amount.toFixed(6)) * price.price;
         return {
+            tokenInfo: this.asset(),
             amount,
             usdValue
         }
@@ -142,11 +148,30 @@ export class VesuRebalance {
         let price = await this.pricer.getPrice(this.metadata.depositTokens[0].symbol);
         const usdValue = Number(amount.toFixed(6)) * price.price;
         return {
+            tokenInfo: this.asset(),
             amount,
             usdValue
         }
     }
 
+    static async getAllPossibleVerifiedPools(asset: ContractAddr) {
+        const data = await getAPIUsingHeadlessBrowser('https://api.vesu.xyz/pools');
+        const verifiedPools =data.data.filter((d: any) => d.isVerified);
+        const pools = verifiedPools.map((p: any) => {
+            const hasMyAsset = p.assets.find((a: any) => asset.eqString(a.address));
+            if (hasMyAsset) {
+                return {
+                    pool_id: ContractAddr.from(p.id),
+                    max_weight: 10000,
+                    v_token: ContractAddr.from(hasMyAsset.vToken.address),
+                    name: p.name,
+                }
+            }
+            return null;
+        }).filter((p: PoolProps | null) => p !== null);
+        return pools;
+    }
+
     /**
      * Retrieves the list of allowed pools and their detailed information from multiple sources:
      * 1. Contract's allowed pools
@@ -217,11 +242,10 @@ export class VesuRebalance {
         let isErrorPositionsAPI = false;
         let vesuPositions: any[] = [];
         try {
-            const res = await axios.get(`https://api.vesu.xyz/positions?walletAddress=${this.address.address}`)
-            const data = await res.data;
+            const data = await getAPIUsingHeadlessBrowser(`https://api.vesu.xyz/positions?walletAddress=${this.address.address}`)
             vesuPositions = data.data;
         } catch (e) {
-            console.error(`${VesuRebalance.name}: Error fetching pools for ${this.address.address}`, e);
+            console.error(`${VesuRebalance.name}: Error fetching positions for ${this.address.address}`, e);
             isErrorPositionsAPI = true;
         }
         
@@ -229,8 +253,7 @@ export class VesuRebalance {
         let isErrorPoolsAPI = false;
         let pools: any[] = [];
         try {
-            const res = await axios.get(`https://api.vesu.xyz/pools`);
-            const data = await res.data;
+            const data = await getAPIUsingHeadlessBrowser('https://api.vesu.xyz/pools');
             pools = data.data;
         } catch (e) {
             console.error(`${VesuRebalance.name}: Error fetching pools for ${this.address.address}`, e);
@@ -242,13 +265,13 @@ export class VesuRebalance {
         const info = allowedPools.map(async (p) => {
             const vesuPosition = vesuPositions.find((d: any) => d.pool.id.toString() === num.getDecimalString(p.pool_id.address.toString()));
             const pool = pools.find((d: any) => d.id == num.getDecimalString(p.pool_id.address));
-            const assetInfo = pool?.assets.find((d: any) => ContractAddr.from(this.asset().address).eqString(d.address));
+            const assetInfo = pool?.assets.find((d: any) => this.asset().address.eqString(d.address));
             let vTokenContract = new Contract(VesuRebalanceAbi, p.v_token.address, this.config.provider);
             const bal = await vTokenContract.balanceOf(this.address.address);
             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,
@@ -297,7 +320,7 @@ export class VesuRebalance {
             return acc + (curr.APY.netApy * weight);
         }, 0);
         
-        return weightedApy;
+        return weightedApy * (1 - (this.metadata.additionalInfo.feeBps / 10000));
     }
 
     /**
@@ -418,24 +441,26 @@ export class VesuRebalance {
         const netYield = this.netAPYGivenPools(pools);
        
         const baseFlow: IInvestmentFlow = {
-            title: "Deposit $1000",
-            subItems: [`Net yield: ${(netYield * 100).toFixed(2)}%`],
+            title: "Your Deposit",
+            subItems: [{key: `Net yield`, value: `${(netYield * 100).toFixed(2)}%`}],
             linkedFlows: [],
+            style: {backgroundColor: FlowChartColors.Purple.valueOf()},
         };
 
-        pools.forEach((p) => {
-            if (p.amount.eq(0)) return;
+        let _pools = [...pools];
+        _pools = _pools.sort((a, b) => Number(b.amount.toString()) - Number(a.amount.toString()));
+        _pools.forEach((p) => {
             const flow: IInvestmentFlow = {
-                title: `${p.pool_name} - $${(p.current_weight * 1000).toFixed(2)}`,
+                title: `Pool name: ${p.pool_name}`,
                 subItems: [
-                    `APY: ${(p.APY.netApy * 100).toFixed(2)}%`,
-                    `Weight: ${(p.current_weight * 100).toFixed(2)}% / ${(p.max_weight * 100).toFixed(2)}%`,
+                    {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];
     }
 }
@@ -445,22 +470,85 @@ const _protocol: IProtocol = {name: 'Vesu', logo: 'https://static-assets-8zct.on
 // 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},
+    {type: RiskType.ORACLE_RISK, value: 0.5, weight: 25},
 ]
 /**
  * Represents the Vesu Rebalance Strategies.
  */
-export const VesuRebalanceStrategies: IStrategyMetadata[] = [{
-    name: 'Vesu STRK',
+export const VesuRebalanceStrategies: IStrategyMetadata<VesuRebalanceSettings>[] = [{
+    name: 'Vesu Fusion STRK',
     description: _description.replace('{{TOKEN}}', 'STRK'),
-    address: ContractAddr.from('0xeeb729d554ae486387147b13a9c8871bc7991d454e8b5ff570d4bf94de71e1'),
+    address: ContractAddr.from('0x778007f8136a5b827325d21613803e796bda4d676fbe1e34aeab0b2a2ec027f'),
     type: 'ERC4626',
     depositTokens: [Global.getDefaultTokens().find(t => t.symbol === 'STRK')!],
     protocols: [_protocol],
     maxTVL: Web3Number.fromWei('0', 18),
     risk: {
         riskFactor: _riskFactor,
-        netRisk: _riskFactor.reduce((acc, curr) => acc + curr.value * curr.weight, 0) / 100,
-    }
+        netRisk: _riskFactor.reduce((acc, curr) => acc + curr.value * curr.weight, 0) / _riskFactor.reduce((acc, curr) => acc + curr.weight, 0),
+    },
+    additionalInfo: {
+        feeBps: 1000,
+    },
+}, {
+    name: 'Vesu Fusion ETH',
+    description: _description.replace('{{TOKEN}}', 'ETH'),
+    address: ContractAddr.from('0x26ea414fdf74ace1df5bc5ac72cbac670d438ef19b31edf9d59f98718fc0ab2'),
+    type: 'ERC4626',
+    depositTokens: [Global.getDefaultTokens().find(t => t.symbol === 'ETH')!],
+    protocols: [_protocol],
+    maxTVL: Web3Number.fromWei('0', 18),
+    risk: {
+        riskFactor: _riskFactor,
+        netRisk: _riskFactor.reduce((acc, curr) => acc + curr.value * curr.weight, 0) / _riskFactor.reduce((acc, curr) => acc + curr.weight, 0),
+    },
+    additionalInfo: {
+        feeBps: 1000,
+    },
+}, {
+    name: 'Vesu Fusion USDC',
+    description: _description.replace('{{TOKEN}}', 'USDC'),
+    address: ContractAddr.from('0x3a69adeb993cddb266962d9c995e3d0641dab627df22b825fa31bda460c3c14'),
+    type: 'ERC4626',
+    depositTokens: [Global.getDefaultTokens().find(t => t.symbol === 'USDC')!],
+    protocols: [_protocol],
+    maxTVL: Web3Number.fromWei('0', 6),
+    risk: {
+        riskFactor: _riskFactor,
+        netRisk: _riskFactor.reduce((acc, curr) => acc + curr.value * curr.weight, 0) / _riskFactor.reduce((acc, curr) => acc + curr.weight, 0),
+    },
+    additionalInfo: {
+        feeBps: 1000,
+    },
+// }, {
+//     name: 'Vesu Fusion USDT',
+//     description: _description.replace('{{TOKEN}}', 'USDT'),
+//     address: ContractAddr.from('0x778007f8136a5b827325d21613803e796bda4d676fbe1e34aeab0b2a2ec027f'),
+//     type: 'ERC4626',
+//     depositTokens: [Global.getDefaultTokens().find(t => t.symbol === 'USDT')!],
+//     protocols: [_protocol],
+//     maxTVL: Web3Number.fromWei('0', 6),
+//     risk: {
+//         riskFactor: _riskFactor,
+//         netRisk: _riskFactor.reduce((acc, curr) => acc + curr.value * curr.weight, 0) / _riskFactor.reduce((acc, curr) => acc + curr.weight, 0),
+//     },
+//     additionalInfo: {
+//         feeBps: 1000,
+//     },
+// }, {
+//     name: 'Vesu Fusion WBTC',
+//     description: _description.replace('{{TOKEN}}', 'WBTC'),
+//     address: ContractAddr.from('0x778007f8136a5b827325d21613803e796bda4d676fbe1e34aeab0b2a2ec027f'),
+//     type: 'ERC4626',
+//     depositTokens: [Global.getDefaultTokens().find(t => t.symbol === 'WBTC')!],
+//     protocols: [_protocol],
+//     maxTVL: Web3Number.fromWei('0', 8),
+//     risk: {
+//         riskFactor: _riskFactor,
+//         netRisk: _riskFactor.reduce((acc, curr) => acc + curr.value * curr.weight, 0) / _riskFactor.reduce((acc, curr) => acc + curr.weight, 0),
+//     },
+//     additionalInfo: {
+//         feeBps: 1000,
+//     },
 }]