@gearbox-protocol/sdk 8.28.4 → 8.29.0

package/dist/cjs/dev/AccountOpener.js

@@ -49,6 +49,7 @@ class AccountOpener extends import_sdk.SDKConstruct {
   #faucet;
   #poolDepositMultiplier;
   #minDebtMultiplier;
+  #allowMint;
   constructor(service, options_ = {}) {
     super(service.sdk);
     const {
@@ -56,9 +57,11 @@ class AccountOpener extends import_sdk.SDKConstruct {
       depositorKey,
       faucet,
       poolDepositMultiplier = 30000n,
-      minDebtMultiplier = 10100n
+      minDebtMultiplier = 10100n,
+      allowMint = false
     } = options_;
     this.#service = service;
+    this.#allowMint = allowMint;
     this.#logger = (0, import_sdk.childLogger)("AccountOpener", service.sdk.logger);
     this.#anvil = (0, import_createAnvilClient.createAnvilClient)({
       chain: service.sdk.provider.chain,
@@ -244,6 +247,27 @@ class AccountOpener extends import_sdk.SDKConstruct {
       },
       "looking for open strategy"
     );
+    let borrowerBalance = await this.#anvil.readContract({
+      address: underlying,
+      abi: import_iERC20.ierc20Abi,
+      functionName: "balanceOf",
+      args: [borrower.address]
+    });
+    if (borrowerBalance < minDebt) {
+      this.#logger?.warn(
+        `borrower balance in underlying: ${this.sdk.tokensMeta.formatBN(underlying, borrowerBalance)} is less than the minimum debt: ${this.sdk.tokensMeta.formatBN(underlying, minDebt)}`
+      );
+      if (this.#allowMint) {
+        borrowerBalance = await this.#tryMint(
+          underlying,
+          borrower,
+          minDebt - borrowerBalance
+        );
+      }
+    }
+    this.#logger?.debug(
+      `borrower balance in underlying: ${this.sdk.tokensMeta.formatBN(underlying, borrowerBalance)} ${this.sdk.tokensMeta.symbol(underlying)}`
+    );
     const strategy = await this.sdk.routerFor(cm).findOpenStrategyPath({
       creditManager: cm.creditManager,
       expectedBalances,
@@ -359,12 +383,24 @@ class AccountOpener extends import_sdk.SDKConstruct {
     try {
       const amnt = this.sdk.tokensMeta.formatBN(pool.underlying, amount) + " " + this.sdk.tokensMeta.symbol(pool.underlying);
       this.#logger?.debug(`depositing ${amnt} into pool ${poolName}`);
-      const allowance = await this.#anvil.readContract({
+      let allowance = await this.#anvil.readContract({
         address: underlying,
         abi: import_iERC20.ierc20Abi,
         functionName: "balanceOf",
         args: [depositor.address]
       });
+      if (allowance < amount) {
+        this.#logger?.warn(
+          `depositor balance in underlying: ${this.sdk.tokensMeta.formatBN(pool.underlying, allowance)} is less than the amount to deposit: ${amnt}`
+        );
+        if (this.#allowMint) {
+          allowance = await this.#tryMint(
+            underlying,
+            depositor,
+            amount - allowance
+          );
+        }
+      }
       this.#logger?.debug(
         `depositor balance in underlying: ${this.sdk.tokensMeta.formatBN(pool.underlying, allowance)}`
       );
@@ -417,6 +453,78 @@ class AccountOpener extends import_sdk.SDKConstruct {
       };
     }
   }
+  /**
+   * Tries to mint tokens, returns the balance after attempt (success or not)
+   * @param token
+   * @param dest
+   * @param amount
+   */
+  async #tryMint(token, dest, amount) {
+    const symbol = this.sdk.tokensMeta.symbol(token);
+    const amnt = this.sdk.tokensMeta.formatBN(token, amount);
+    try {
+      await this.#mint(token, dest, amount);
+    } catch (e) {
+      this.#logger?.warn(
+        `failed to mint ${amnt} ${symbol} to ${dest.address}: ${e}`
+      );
+    }
+    return await this.#anvil.readContract({
+      address: token,
+      abi: import_iERC20.ierc20Abi,
+      functionName: "balanceOf",
+      args: [dest.address]
+    });
+  }
+  async #mint(token, dest, amount) {
+    const symbol = this.sdk.tokensMeta.symbol(token);
+    const amnt = this.sdk.tokensMeta.formatBN(token, amount);
+    this.#logger?.warn(`minting ${amnt} ${symbol} to ${dest.address}`);
+    const owner = await this.#anvil.readContract({
+      address: token,
+      abi: import_abi.iOwnableAbi,
+      functionName: "owner"
+    });
+    this.#logger?.warn(`owner of ${symbol}: ${owner}`);
+    await this.#anvil.impersonateAccount({ address: owner });
+    await this.#anvil.setBalance({
+      address: owner,
+      value: (0, import_viem.parseEther)("100")
+    });
+    let hash = await this.#anvil.writeContract({
+      account: owner,
+      address: token,
+      abi: (0, import_viem.parseAbi)(["function mint(uint256 amount) external returns (bool)"]),
+      functionName: "mint",
+      args: [amount],
+      chain: this.#anvil.chain
+    });
+    this.#logger?.debug(`mint ${amnt} ${symbol} to ${owner} in tx ${hash}`);
+    let receipt = await this.#anvil.waitForTransactionReceipt({ hash });
+    if (receipt.status === "reverted") {
+      throw new Error(
+        `failed to mint ${amnt} ${symbol} to ${owner} in tx ${hash}: reverted`
+      );
+    }
+    hash = await this.#anvil.writeContract({
+      account: owner,
+      address: token,
+      abi: import_iERC20.ierc20Abi,
+      functionName: "transfer",
+      args: [dest.address, amount],
+      chain: this.#anvil.chain
+    });
+    this.#logger?.debug(
+      `transfer ${amnt} ${symbol} from ${owner} to ${dest.address} in tx ${hash}`
+    );
+    receipt = await this.#anvil.waitForTransactionReceipt({ hash });
+    if (receipt.status === "reverted") {
+      throw new Error(
+        `failed to transfer ${amnt} ${symbol} from ${owner} to ${dest.address} in tx ${hash}: reverted`
+      );
+    }
+    await this.#anvil.stopImpersonatingAccount({ address: owner });
+  }
   /**
    * Creates borrower wallet,
    * Sets ETH balance,

package/dist/esm/dev/AccountOpener.js

@@ -2,6 +2,7 @@ import {
   BaseError,
   erc20Abi,
   isAddress,
+  parseAbi,
   parseEther,
   parseEventLogs
 } from "viem";
@@ -19,7 +20,7 @@ import {
   SDKConstruct,
   sendRawTx
 } from "../sdk/index.js";
-import { iDegenNftv2Abi } from "./abi.js";
+import { iDegenNftv2Abi, iOwnableAbi } from "./abi.js";
 import { claimFromFaucet } from "./claimFromFaucet.js";
 import { createAnvilClient } from "./createAnvilClient.js";
 const DIRECT_TRANSFERS_QUOTA = 10000n;
@@ -41,6 +42,7 @@ class AccountOpener extends SDKConstruct {
   #faucet;
   #poolDepositMultiplier;
   #minDebtMultiplier;
+  #allowMint;
   constructor(service, options_ = {}) {
     super(service.sdk);
     const {
@@ -48,9 +50,11 @@ class AccountOpener extends SDKConstruct {
       depositorKey,
       faucet,
       poolDepositMultiplier = 30000n,
-      minDebtMultiplier = 10100n
+      minDebtMultiplier = 10100n,
+      allowMint = false
     } = options_;
     this.#service = service;
+    this.#allowMint = allowMint;
     this.#logger = childLogger("AccountOpener", service.sdk.logger);
     this.#anvil = createAnvilClient({
       chain: service.sdk.provider.chain,
@@ -236,6 +240,27 @@ class AccountOpener extends SDKConstruct {
       },
       "looking for open strategy"
     );
+    let borrowerBalance = await this.#anvil.readContract({
+      address: underlying,
+      abi: ierc20Abi,
+      functionName: "balanceOf",
+      args: [borrower.address]
+    });
+    if (borrowerBalance < minDebt) {
+      this.#logger?.warn(
+        `borrower balance in underlying: ${this.sdk.tokensMeta.formatBN(underlying, borrowerBalance)} is less than the minimum debt: ${this.sdk.tokensMeta.formatBN(underlying, minDebt)}`
+      );
+      if (this.#allowMint) {
+        borrowerBalance = await this.#tryMint(
+          underlying,
+          borrower,
+          minDebt - borrowerBalance
+        );
+      }
+    }
+    this.#logger?.debug(
+      `borrower balance in underlying: ${this.sdk.tokensMeta.formatBN(underlying, borrowerBalance)} ${this.sdk.tokensMeta.symbol(underlying)}`
+    );
     const strategy = await this.sdk.routerFor(cm).findOpenStrategyPath({
       creditManager: cm.creditManager,
       expectedBalances,
@@ -351,12 +376,24 @@ class AccountOpener extends SDKConstruct {
     try {
       const amnt = this.sdk.tokensMeta.formatBN(pool.underlying, amount) + " " + this.sdk.tokensMeta.symbol(pool.underlying);
       this.#logger?.debug(`depositing ${amnt} into pool ${poolName}`);
-      const allowance = await this.#anvil.readContract({
+      let allowance = await this.#anvil.readContract({
         address: underlying,
         abi: ierc20Abi,
         functionName: "balanceOf",
         args: [depositor.address]
       });
+      if (allowance < amount) {
+        this.#logger?.warn(
+          `depositor balance in underlying: ${this.sdk.tokensMeta.formatBN(pool.underlying, allowance)} is less than the amount to deposit: ${amnt}`
+        );
+        if (this.#allowMint) {
+          allowance = await this.#tryMint(
+            underlying,
+            depositor,
+            amount - allowance
+          );
+        }
+      }
       this.#logger?.debug(
         `depositor balance in underlying: ${this.sdk.tokensMeta.formatBN(pool.underlying, allowance)}`
       );
@@ -409,6 +446,78 @@ class AccountOpener extends SDKConstruct {
       };
     }
   }
+  /**
+   * Tries to mint tokens, returns the balance after attempt (success or not)
+   * @param token
+   * @param dest
+   * @param amount
+   */
+  async #tryMint(token, dest, amount) {
+    const symbol = this.sdk.tokensMeta.symbol(token);
+    const amnt = this.sdk.tokensMeta.formatBN(token, amount);
+    try {
+      await this.#mint(token, dest, amount);
+    } catch (e) {
+      this.#logger?.warn(
+        `failed to mint ${amnt} ${symbol} to ${dest.address}: ${e}`
+      );
+    }
+    return await this.#anvil.readContract({
+      address: token,
+      abi: ierc20Abi,
+      functionName: "balanceOf",
+      args: [dest.address]
+    });
+  }
+  async #mint(token, dest, amount) {
+    const symbol = this.sdk.tokensMeta.symbol(token);
+    const amnt = this.sdk.tokensMeta.formatBN(token, amount);
+    this.#logger?.warn(`minting ${amnt} ${symbol} to ${dest.address}`);
+    const owner = await this.#anvil.readContract({
+      address: token,
+      abi: iOwnableAbi,
+      functionName: "owner"
+    });
+    this.#logger?.warn(`owner of ${symbol}: ${owner}`);
+    await this.#anvil.impersonateAccount({ address: owner });
+    await this.#anvil.setBalance({
+      address: owner,
+      value: parseEther("100")
+    });
+    let hash = await this.#anvil.writeContract({
+      account: owner,
+      address: token,
+      abi: parseAbi(["function mint(uint256 amount) external returns (bool)"]),
+      functionName: "mint",
+      args: [amount],
+      chain: this.#anvil.chain
+    });
+    this.#logger?.debug(`mint ${amnt} ${symbol} to ${owner} in tx ${hash}`);
+    let receipt = await this.#anvil.waitForTransactionReceipt({ hash });
+    if (receipt.status === "reverted") {
+      throw new Error(
+        `failed to mint ${amnt} ${symbol} to ${owner} in tx ${hash}: reverted`
+      );
+    }
+    hash = await this.#anvil.writeContract({
+      account: owner,
+      address: token,
+      abi: ierc20Abi,
+      functionName: "transfer",
+      args: [dest.address, amount],
+      chain: this.#anvil.chain
+    });
+    this.#logger?.debug(
+      `transfer ${amnt} ${symbol} from ${owner} to ${dest.address} in tx ${hash}`
+    );
+    receipt = await this.#anvil.waitForTransactionReceipt({ hash });
+    if (receipt.status === "reverted") {
+      throw new Error(
+        `failed to transfer ${amnt} ${symbol} from ${owner} to ${dest.address} in tx ${hash}: reverted`
+      );
+    }
+    await this.#anvil.stopImpersonatingAccount({ address: owner });
+  }
   /**
    * Creates borrower wallet,
    * Sets ETH balance,

package/dist/types/dev/AccountOpener.d.ts

@@ -12,6 +12,7 @@ export interface AccountOpenerOptions {
     depositorKey?: Hex;
     poolDepositMultiplier?: bigint | string | number;
     minDebtMultiplier?: bigint | string | number;
+    allowMint?: boolean;
 }
 export interface TargetAccount {
     creditManager: Address;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gearbox-protocol/sdk",
-  "version": "8.28.4",
+  "version": "8.29.0",
   "description": "Gearbox SDK",
   "license": "MIT",
   "main": "./dist/cjs/sdk/index.js",