@gearbox-protocol/sdk 8.9.0 → 8.10.0

package/dist/cjs/dev/AccountOpener.js

@@ -117,6 +117,12 @@ class AccountOpener extends import_sdk.SDKConstruct {
       }
     }
     this.#logger?.info(`opened ${success}/${targets.length} accounts`);
+    try {
+      await this.#directlyDistributeTokens(results);
+      this.#logger?.info("distributed direct transfer tokens");
+    } catch (e) {
+      this.#logger?.error(`failed to distribute tokens: ${e}`);
+    }
     return results;
   }
   async #openAccount(input, index, total) {
@@ -503,6 +509,87 @@ class AccountOpener extends import_sdk.SDKConstruct {
     });
     return acc;
   }
+  /**
+   * Claims tokens from faucet, uniformly distributes them to accounts
+   * @param targets
+   * @returns
+   */
+  async #directlyDistributeTokens(targets) {
+    const tokens = new import_sdk.AddressSet(
+      targets.flatMap((t) => t.input.directTransfer ?? [])
+    );
+    if (tokens.size === 0) {
+      return;
+    }
+    const distributorKey = (0, import_accounts.generatePrivateKey)();
+    const distributor = (0, import_accounts.privateKeyToAccount)(distributorKey);
+    await this.#anvil.setBalance({
+      address: distributor.address,
+      value: (0, import_viem.parseEther)("100")
+    });
+    await (0, import_claimFromFaucet.claimFromFaucet)({
+      publicClient: this.#anvil,
+      wallet: this.#anvil,
+      faucet: this.faucet,
+      amountUSD: (minAmountUSD) => minAmountUSD * BigInt(targets.length),
+      claimer: distributor,
+      role: "reward token distributor",
+      logger: this.#logger
+    });
+    const actualBalances = await this.#anvil.multicall({
+      contracts: tokens.map(
+        (token) => ({
+          address: token,
+          abi: import_viem.erc20Abi,
+          functionName: "balanceOf",
+          args: [distributor.address]
+        })
+      ),
+      allowFailure: false
+    });
+    const tokensArr = tokens.asArray();
+    for (let i = 0; i < tokensArr.length; i++) {
+      const token = tokensArr[i];
+      const balance = actualBalances[i];
+      this.#logger?.debug(`${token} balance: ${balance}`);
+    }
+    for (const { account, input } of targets) {
+      if (!account) {
+        continue;
+      }
+      const directTransfer = new import_sdk.AddressSet(input.directTransfer);
+      for (let i = 0; i < tokensArr.length; i++) {
+        const token = tokensArr[i];
+        if (!directTransfer.has(token)) {
+          continue;
+        }
+        const balance = actualBalances[i] / BigInt(targets.length);
+        this.#logger?.debug(
+          `sending ${balance} ${token} to ${account.creditAccount}`
+        );
+        const hash = await this.#anvil.writeContract({
+          account: distributor,
+          address: token,
+          abi: import_viem.erc20Abi,
+          functionName: "transfer",
+          args: [account.creditAccount, balance],
+          chain: this.#anvil.chain
+        });
+        const receipt = await this.#anvil.waitForTransactionReceipt({
+          hash
+        });
+        if (receipt.status === "reverted") {
+          this.#logger?.error(
+            `failed to send ${token} to ${account.creditAccount}, tx ${hash} reverted`
+          );
+        } else {
+          this.#logger?.debug(
+            `sent ${token} to ${account.creditAccount}, tx: ${hash}`
+          );
+        }
+      }
+    }
+  }
   #getCollateralQuota(cm, collateral, collateralAmount, debt, logger) {
     const {
       underlying,

package/dist/cjs/sdk/utils/AddressSet.js

@@ -0,0 +1,48 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var AddressSet_exports = {};
+__export(AddressSet_exports, {
+  AddressSet: () => AddressSet
+});
+module.exports = __toCommonJS(AddressSet_exports);
+var import_viem = require("viem");
+class AddressSet extends Set {
+  constructor(entries) {
+    super(Array.from(entries ?? []).map((a) => (0, import_viem.getAddress)(a)));
+  }
+  add(value) {
+    return super.add((0, import_viem.getAddress)(value));
+  }
+  delete(value) {
+    return super.delete((0, import_viem.getAddress)(value));
+  }
+  has(value) {
+    return super.has((0, import_viem.getAddress)(value));
+  }
+  asArray() {
+    return Array.from(this);
+  }
+  map(fn) {
+    return this.asArray().map(fn);
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  AddressSet
+});

package/dist/cjs/sdk/utils/index.js

@@ -16,6 +16,7 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var utils_exports = {};
 module.exports = __toCommonJS(utils_exports);
 __reExport(utils_exports, require("./AddressMap.js"), module.exports);
+__reExport(utils_exports, require("./AddressSet.js"), module.exports);
 __reExport(utils_exports, require("./bytes32ToString.js"), module.exports);
 __reExport(utils_exports, require("./childLogger.js"), module.exports);
 __reExport(utils_exports, require("./createRawTx.js"), module.exports);
@@ -32,6 +33,7 @@ __reExport(utils_exports, require("./zod.js"), module.exports);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ...require("./AddressMap.js"),
+  ...require("./AddressSet.js"),
   ...require("./bytes32ToString.js"),
   ...require("./childLogger.js"),
   ...require("./createRawTx.js"),

package/dist/esm/dev/AccountOpener.js

@@ -1,10 +1,17 @@
-import { BaseError, isAddress, parseEther, parseEventLogs } from "viem";
+import {
+  BaseError,
+  erc20Abi,
+  isAddress,
+  parseEther,
+  parseEventLogs
+} from "viem";
 import { generatePrivateKey, privateKeyToAccount } from "viem/accounts";
 import { ierc20Abi } from "../abi/iERC20.js";
 import { iCreditFacadeV300Abi, iPoolV300Abi } from "../abi/v300.js";
 import {
   ADDRESS_0X0,
   AddressMap,
+  AddressSet,
   childLogger,
   formatBN,
   MAX_UINT256,
@@ -102,6 +109,12 @@ class AccountOpener extends SDKConstruct {
       }
     }
     this.#logger?.info(`opened ${success}/${targets.length} accounts`);
+    try {
+      await this.#directlyDistributeTokens(results);
+      this.#logger?.info("distributed direct transfer tokens");
+    } catch (e) {
+      this.#logger?.error(`failed to distribute tokens: ${e}`);
+    }
     return results;
   }
   async #openAccount(input, index, total) {
@@ -488,6 +501,87 @@ class AccountOpener extends SDKConstruct {
     });
     return acc;
   }
+  /**
+   * Claims tokens from faucet, uniformly distributes them to accounts
+   * @param targets
+   * @returns
+   */
+  async #directlyDistributeTokens(targets) {
+    const tokens = new AddressSet(
+      targets.flatMap((t) => t.input.directTransfer ?? [])
+    );
+    if (tokens.size === 0) {
+      return;
+    }
+    const distributorKey = generatePrivateKey();
+    const distributor = privateKeyToAccount(distributorKey);
+    await this.#anvil.setBalance({
+      address: distributor.address,
+      value: parseEther("100")
+    });
+    await claimFromFaucet({
+      publicClient: this.#anvil,
+      wallet: this.#anvil,
+      faucet: this.faucet,
+      amountUSD: (minAmountUSD) => minAmountUSD * BigInt(targets.length),
+      claimer: distributor,
+      role: "reward token distributor",
+      logger: this.#logger
+    });
+    const actualBalances = await this.#anvil.multicall({
+      contracts: tokens.map(
+        (token) => ({
+          address: token,
+          abi: erc20Abi,
+          functionName: "balanceOf",
+          args: [distributor.address]
+        })
+      ),
+      allowFailure: false
+    });
+    const tokensArr = tokens.asArray();
+    for (let i = 0; i < tokensArr.length; i++) {
+      const token = tokensArr[i];
+      const balance = actualBalances[i];
+      this.#logger?.debug(`${token} balance: ${balance}`);
+    }
+    for (const { account, input } of targets) {
+      if (!account) {
+        continue;
+      }
+      const directTransfer = new AddressSet(input.directTransfer);
+      for (let i = 0; i < tokensArr.length; i++) {
+        const token = tokensArr[i];
+        if (!directTransfer.has(token)) {
+          continue;
+        }
+        const balance = actualBalances[i] / BigInt(targets.length);
+        this.#logger?.debug(
+          `sending ${balance} ${token} to ${account.creditAccount}`
+        );
+        const hash = await this.#anvil.writeContract({
+          account: distributor,
+          address: token,
+          abi: erc20Abi,
+          functionName: "transfer",
+          args: [account.creditAccount, balance],
+          chain: this.#anvil.chain
+        });
+        const receipt = await this.#anvil.waitForTransactionReceipt({
+          hash
+        });
+        if (receipt.status === "reverted") {
+          this.#logger?.error(
+            `failed to send ${token} to ${account.creditAccount}, tx ${hash} reverted`
+          );
+        } else {
+          this.#logger?.debug(
+            `sent ${token} to ${account.creditAccount}, tx: ${hash}`
+          );
+        }
+      }
+    }
+  }
   #getCollateralQuota(cm, collateral, collateralAmount, debt, logger) {
     const {
       underlying,

package/dist/esm/sdk/utils/AddressSet.js

@@ -0,0 +1,24 @@
+import { getAddress } from "viem";
+class AddressSet extends Set {
+  constructor(entries) {
+    super(Array.from(entries ?? []).map((a) => getAddress(a)));
+  }
+  add(value) {
+    return super.add(getAddress(value));
+  }
+  delete(value) {
+    return super.delete(getAddress(value));
+  }
+  has(value) {
+    return super.has(getAddress(value));
+  }
+  asArray() {
+    return Array.from(this);
+  }
+  map(fn) {
+    return this.asArray().map(fn);
+  }
+}
+export {
+  AddressSet
+};

package/dist/esm/sdk/utils/index.js

@@ -1,4 +1,5 @@
 export * from "./AddressMap.js";
+export * from "./AddressSet.js";
 export * from "./bytes32ToString.js";
 export * from "./childLogger.js";
 export * from "./createRawTx.js";

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

@@ -25,6 +25,10 @@ export interface TargetAccount {
      * TODO: not implemented
      */
     collateral?: Address;
+    /**
+     * These tokens will be transferred directly from faucet to credit account
+     */
+    directTransfer?: Address[];
     leverage?: number;
     slippage?: number;
 }

package/dist/types/sdk/utils/AddressSet.d.ts

@@ -0,0 +1,9 @@
+import { type Address } from "viem";
+export declare class AddressSet extends Set<Address> {
+    constructor(entries?: Iterable<Address>);
+    add(value: Address): this;
+    delete(value: Address): boolean;
+    has(value: Address): boolean;
+    asArray(): Address[];
+    map<T>(fn: (address: Address) => T): T[];
+}

package/dist/types/sdk/utils/index.d.ts

@@ -1,4 +1,5 @@
 export * from "./AddressMap.js";
+export * from "./AddressSet.js";
 export * from "./bytes32ToString.js";
 export * from "./childLogger.js";
 export * from "./createRawTx.js";

package/package.json

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