solidoracle / erc4626-rs

🤖ML-ERC46 for Arbitrum Stylus

Overview

Sample implementation of OpenZeppelin ERC4626 contract in Stylus using Rust which aims to enable heavy AI/ML computation for DeFi projects that want to enrich their smart vaults with machine learning capabilites that are historically performed offchain.

It provides a library with augmented ML capabilites thanks to the rustlearn crate, a machine learning package for Rust as seen in the rebalance function.

Smart Vault deployment address: https://stylus-testnet-explorer.arbitrum.io/address/0x6982aA8dB18E7153c614765dA4A581135e9Bf4a8

How to use

Insert your private key in an .env file using

 cargo stylus check   
 cargo stylus deploy --private-key-path=./.env

Test

You can interact with the smart contract on Arbitrum Stylus Testnet.

  cd hardhat
  yarn install
  yarn hardhat run scripts/initialize.js   

erc4626-rs's Issues

same result for if and else in deposit function

let shares = if supply == U256::ZERO { 
            amount 
        } else { 
            amount.checked_mul(supply).ok_or("Overflow")?.checked_div(self.totalAssets()?).ok_or("Divide by zero")?
        };

supply = self.totalSupply.get();
pub fn totalAssets(&self) -> Result<U256, Vec> {
Ok(self.totalSupply.get())
}

which means amount multiply totalSupply and then divide total supply, result is amount, the same in if block

self.erc20.burn(msg::sender(), shares)?; instead of self.erc20.burn(msg::sender(), amount)?;

use wrong param for burn method