@gearbox-protocol/sdk 11.0.0-next.2 → 11.0.0-next.3

package/dist/cjs/dev/index.js

@@ -27,6 +27,7 @@ __reExport(dev_exports, require("./ltUtils.js"), module.exports);
 __reExport(dev_exports, require("./migrateFaucet.js"), module.exports);
 __reExport(dev_exports, require("./mint/index.js"), module.exports);
 __reExport(dev_exports, require("./RevolverTransport.js"), module.exports);
+__reExport(dev_exports, require("./replaceStorage.js"), module.exports);
 __reExport(dev_exports, require("./types.js"), module.exports);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
@@ -42,5 +43,6 @@ __reExport(dev_exports, require("./types.js"), module.exports);
   ...require("./migrateFaucet.js"),
   ...require("./mint/index.js"),
   ...require("./RevolverTransport.js"),
+  ...require("./replaceStorage.js"),
   ...require("./types.js")
 });

package/dist/cjs/dev/replaceStorage.js

@@ -0,0 +1,77 @@
+"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 replaceStorage_exports = {};
+__export(replaceStorage_exports, {
+  replaceStorage: () => replaceStorage
+});
+module.exports = __toCommonJS(replaceStorage_exports);
+var import_viem = require("viem");
+var import_actions = require("viem/actions");
+async function replaceStorage(client, params) {
+  const { value, slotMatch, address, abi, functionName, args } = params;
+  const newValHex = (0, import_viem.numberToHex)(params.value, { size: 32 });
+  const { accessList } = await client.createAccessList({
+    to: address,
+    data: (0, import_viem.encodeFunctionData)({
+      abi,
+      functionName,
+      args
+    })
+  });
+  for (const { address: addr_, storageKeys } of accessList) {
+    const addr = addr_.toLowerCase();
+    for (const slot of storageKeys) {
+      try {
+        const result = await (0, import_actions.readContract)(client, {
+          abi,
+          address,
+          functionName,
+          args,
+          stateOverride: [
+            {
+              address: addr,
+              stateDiff: [
+                {
+                  slot,
+                  value: newValHex
+                }
+              ]
+            }
+          ]
+        });
+        if (slotMatch(
+          result,
+          value
+        )) {
+          await client.setStorageAt({
+            address,
+            index: slot,
+            value: newValHex
+          });
+          return;
+        }
+      } catch {
+      }
+    }
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  replaceStorage
+});

package/dist/esm/dev/createAnvilClient.js

@@ -1,9 +1,4 @@
-import {
-  createPublicClient,
-  testActions,
-  toHex,
-  walletActions
-} from "viem";
+import { createPublicClient, testActions, toHex, walletActions } from "viem";
 function extendAnvilClient(client) {
   return client.extend(testActions({ mode: "anvil" })).extend(walletActions).extend((c) => ({
     anvilNodeInfo: () => anvilNodeInfo(c),

package/dist/esm/dev/index.js

@@ -10,4 +10,5 @@ export * from "./ltUtils.js";
 export * from "./migrateFaucet.js";
 export * from "./mint/index.js";
 export * from "./RevolverTransport.js";
+export * from "./replaceStorage.js";
 export * from "./types.js";

package/dist/esm/dev/replaceStorage.js

@@ -0,0 +1,56 @@
+import {
+  encodeFunctionData,
+  numberToHex
+} from "viem";
+import { readContract } from "viem/actions";
+async function replaceStorage(client, params) {
+  const { value, slotMatch, address, abi, functionName, args } = params;
+  const newValHex = numberToHex(params.value, { size: 32 });
+  const { accessList } = await client.createAccessList({
+    to: address,
+    data: encodeFunctionData({
+      abi,
+      functionName,
+      args
+    })
+  });
+  for (const { address: addr_, storageKeys } of accessList) {
+    const addr = addr_.toLowerCase();
+    for (const slot of storageKeys) {
+      try {
+        const result = await readContract(client, {
+          abi,
+          address,
+          functionName,
+          args,
+          stateOverride: [
+            {
+              address: addr,
+              stateDiff: [
+                {
+                  slot,
+                  value: newValHex
+                }
+              ]
+            }
+          ]
+        });
+        if (slotMatch(
+          result,
+          value
+        )) {
+          await client.setStorageAt({
+            address,
+            index: slot,
+            value: newValHex
+          });
+          return;
+        }
+      } catch {
+      }
+    }
+  }
+}
+export {
+  replaceStorage
+};

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

@@ -10,4 +10,5 @@ export * from "./ltUtils.js";
 export * from "./migrateFaucet.js";
 export * from "./mint/index.js";
 export * from "./RevolverTransport.js";
+export * from "./replaceStorage.js";
 export * from "./types.js";

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

@@ -0,0 +1,24 @@
+import type { Abi } from "abitype";
+import { type ContractFunctionArgs, type ContractFunctionName, type ContractFunctionParameters, type ReadContractReturnType } from "viem";
+import type { AnvilClient } from "./createAnvilClient.js";
+export type ReplaceStorageParams<abi extends Abi | readonly unknown[] = Abi, functionName extends ContractFunctionName<abi, "pure" | "view"> = ContractFunctionName<abi, "pure" | "view">, args extends ContractFunctionArgs<abi, "pure" | "view", functionName> = ContractFunctionArgs<abi, "pure" | "view", functionName>> = ContractFunctionParameters<abi, "pure" | "view", functionName, args, false> & {
+    /**
+     * New value to set
+     */
+    value: bigint;
+    /**
+     * Comparison function, that returns true if value read from contract matches new value
+     * @param readVal - value read by contract function, can be tuple, struct, etc...
+     * @param value - new value to be set
+     * @returns
+     */
+    slotMatch: (readVal: ReadContractReturnType<abi, functionName, args>, value: bigint) => boolean;
+};
+/**
+ * Replaces bigint value in contract storage
+ * Success is checked by reading contract function and asserting that its return contains new value
+ * @param client
+ * @param params
+ * @returns
+ */
+export declare function replaceStorage<const abi extends Abi | readonly unknown[], functionName extends ContractFunctionName<abi, "pure" | "view">, const args extends ContractFunctionArgs<abi, "pure" | "view", functionName>>(client: AnvilClient, params: ReplaceStorageParams<abi, functionName, args>): Promise<void>;

package/package.json

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