rocketh 0.11.21 → 0.11.22-testing.1

package/dist/executor/types.d.ts

@@ -8,4 +8,24 @@ export interface DeployScriptModule<NamedAccounts extends UnresolvedUnknownNamed
     id?: string;
 }
 export type ScriptCallback<NamedAccounts extends UnresolvedUnknownNamedAccounts = UnresolvedUnknownNamedAccounts, Data extends UnresolvedNetworkSpecificData = UnresolvedNetworkSpecificData, Deployments extends UnknownDeployments = UnknownDeployments> = (env: Environment<NamedAccounts, Data, Deployments>) => Promise<void>;
+/**
+ * Utility type to remove the first parameter from a function type
+ */
+type RemoveFirstParameter<T> = T extends (first: any, ...rest: infer R) => infer Return ? (...args: R) => Return : never;
+/**
+ * Utility type to transform an object of functions by removing their first parameter
+ */
+type CurriedFunctions<T> = {
+    [K in keyof T]: RemoveFirstParameter<T[K]>;
+};
+/**
+ * Type for the enhanced environment proxy that includes both the original environment
+ * and the curried functions
+ */
+export type EnhancedEnvironment<NamedAccounts extends UnresolvedUnknownNamedAccounts = UnresolvedUnknownNamedAccounts, Data extends UnresolvedNetworkSpecificData = UnresolvedNetworkSpecificData, Deployments extends UnknownDeployments = UnknownDeployments, Functions extends Record<string, (env: Environment<NamedAccounts, Data, Deployments>, ...args: any[]) => any> = Record<string, (env: Environment<NamedAccounts, Data, Deployments>, ...args: any[]) => any>> = Environment<NamedAccounts, Data, Deployments> & CurriedFunctions<Functions>;
+/**
+ * Type for a deploy script function that receives an enhanced environment
+ */
+export type EnhancedDeployScriptFunction<NamedAccounts extends UnresolvedUnknownNamedAccounts = UnresolvedUnknownNamedAccounts, Data extends UnresolvedNetworkSpecificData = UnresolvedNetworkSpecificData, ArgumentsTypes = undefined, Deployments extends UnknownDeployments = UnknownDeployments, Functions extends Record<string, (env: Environment<NamedAccounts, Data, Deployments>, ...args: any[]) => any> = Record<string, (env: Environment<NamedAccounts, Data, Deployments>, ...args: any[]) => any>> = (env: EnhancedEnvironment<NamedAccounts, Data, Deployments, Functions>, args?: ArgumentsTypes) => Promise<void | boolean>;
+export {};
 //# sourceMappingURL=types.d.ts.map

package/dist/executor/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/executor/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACX,WAAW,EACX,kBAAkB,EAClB,6BAA6B,EAC7B,8BAA8B,EAC9B,MAAM,yBAAyB,CAAC;AAEjC,MAAM,MAAM,oBAAoB,CAC/B,aAAa,SAAS,8BAA8B,GAAG,8BAA8B,EACrF,IAAI,SAAS,6BAA6B,GAAG,6BAA6B,EAC1E,cAAc,GAAG,SAAS,EAC1B,WAAW,SAAS,kBAAkB,GAAG,kBAAkB,IACxD,CAAC,GAAG,EAAE,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,IAAI,CAAC,EAAE,cAAc,KAAK,OAAO,CAAC,IAAI,GAAG,OAAO,CAAC,CAAC;AAE3G,MAAM,WAAW,kBAAkB,CAClC,aAAa,SAAS,8BAA8B,GAAG,8BAA8B,EACrF,IAAI,SAAS,6BAA6B,GAAG,6BAA6B,EAC1E,cAAc,GAAG,SAAS,EAC1B,WAAW,SAAS,kBAAkB,GAAG,kBAAkB;IAE3D,CAAC,GAAG,EAAE,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,IAAI,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,GAAG,OAAO,CAAC,CAAC;IACrG,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,EAAE,CAAC,EAAE,MAAM,CAAC;CACZ;AAED,MAAM,MAAM,cAAc,CACzB,aAAa,SAAS,8BAA8B,GAAG,8BAA8B,EACrF,IAAI,SAAS,6BAA6B,GAAG,6BAA6B,EAC1E,WAAW,SAAS,kBAAkB,GAAG,kBAAkB,IACxD,CAAC,GAAG,EAAE,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/executor/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACX,WAAW,EACX,kBAAkB,EAClB,6BAA6B,EAC7B,8BAA8B,EAC9B,MAAM,yBAAyB,CAAC;AAEjC,MAAM,MAAM,oBAAoB,CAC/B,aAAa,SAAS,8BAA8B,GAAG,8BAA8B,EACrF,IAAI,SAAS,6BAA6B,GAAG,6BAA6B,EAC1E,cAAc,GAAG,SAAS,EAC1B,WAAW,SAAS,kBAAkB,GAAG,kBAAkB,IACxD,CAAC,GAAG,EAAE,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,IAAI,CAAC,EAAE,cAAc,KAAK,OAAO,CAAC,IAAI,GAAG,OAAO,CAAC,CAAC;AAE3G,MAAM,WAAW,kBAAkB,CAClC,aAAa,SAAS,8BAA8B,GAAG,8BAA8B,EACrF,IAAI,SAAS,6BAA6B,GAAG,6BAA6B,EAC1E,cAAc,GAAG,SAAS,EAC1B,WAAW,SAAS,kBAAkB,GAAG,kBAAkB;IAE3D,CAAC,GAAG,EAAE,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,IAAI,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,GAAG,OAAO,CAAC,CAAC;IACrG,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,EAAE,CAAC,EAAE,MAAM,CAAC;CACZ;AAED,MAAM,MAAM,cAAc,CACzB,aAAa,SAAS,8BAA8B,GAAG,8BAA8B,EACrF,IAAI,SAAS,6BAA6B,GAAG,6BAA6B,EAC1E,WAAW,SAAS,kBAAkB,GAAG,kBAAkB,IACxD,CAAC,GAAG,EAAE,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;AAE1E;;GAEG;AACH,KAAK,oBAAoB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,IAAI,EAAE,MAAM,CAAC,KAAK,MAAM,MAAM,GACpF,CAAC,GAAG,IAAI,EAAE,CAAC,KAAK,MAAM,GACtB,KAAK,CAAC;AAET;;GAEG;AACH,KAAK,gBAAgB,CAAC,CAAC,IAAI;KACzB,CAAC,IAAI,MAAM,CAAC,GAAG,oBAAoB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;CAC1C,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,mBAAmB,CAC9B,aAAa,SAAS,8BAA8B,GAAG,8BAA8B,EACrF,IAAI,SAAS,6BAA6B,GAAG,6BAA6B,EAC1E,WAAW,SAAS,kBAAkB,GAAG,kBAAkB,EAC3D,SAAS,SAAS,MAAM,CACvB,MAAM,EACN,CAAC,GAAG,EAAE,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,CAC3E,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,CAAC,IAC5F,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,GAAG,gBAAgB,CAAC,SAAS,CAAC,CAAC;AAEhF;;GAEG;AACH,MAAM,MAAM,4BAA4B,CACvC,aAAa,SAAS,8BAA8B,GAAG,8BAA8B,EACrF,IAAI,SAAS,6BAA6B,GAAG,6BAA6B,EAC1E,cAAc,GAAG,SAAS,EAC1B,WAAW,SAAS,kBAAkB,GAAG,kBAAkB,EAC3D,SAAS,SAAS,MAAM,CACvB,MAAM,EACN,CAAC,GAAG,EAAE,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,CAC3E,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,CAAC,IAC5F,CACH,GAAG,EAAE,mBAAmB,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,CAAC,EACrE,IAAI,CAAC,EAAE,cAAc,KACjB,OAAO,CAAC,IAAI,GAAG,OAAO,CAAC,CAAC"}

package/dist/index.d.ts

@@ -1,7 +1,6 @@
 export * from './executor/index.js';
 export * from './executor/types.js';
 export * from './environment/types.js';
-export { extendEnvironment, handleSignerProtocol } from './environment/index.js';
 export { loadDeployments } from './environment/deployments.js';
 export * from './environment/utils/artifacts.js';
 export * from './environment/utils/chains.js';

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC;AACpC,cAAc,wBAAwB,CAAC;AACvC,OAAO,EAAC,iBAAiB,EAAE,oBAAoB,EAAC,MAAM,wBAAwB,CAAC;AAC/E,OAAO,EAAC,eAAe,EAAC,MAAM,8BAA8B,CAAC;AAC7D,cAAc,kCAAkC,CAAC;AACjD,cAAc,+BAA+B,CAAC;AAC9C,cAAc,gBAAgB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC;AACpC,cAAc,wBAAwB,CAAC;AACvC,OAAO,EAAC,eAAe,EAAC,MAAM,8BAA8B,CAAC;AAC7D,cAAc,kCAAkC,CAAC;AACjD,cAAc,+BAA+B,CAAC;AAC9C,cAAc,gBAAgB,CAAC"}

package/dist/index.js

@@ -1,7 +1,6 @@
 export * from './executor/index.js';
 export * from './executor/types.js';
 export * from './environment/types.js';
-export { extendEnvironment, handleSignerProtocol } from './environment/index.js';
 export { loadDeployments } from './environment/deployments.js';
 export * from './environment/utils/artifacts.js';
 export * from './environment/utils/chains.js';

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC;AACpC,cAAc,wBAAwB,CAAC;AACvC,OAAO,EAAC,iBAAiB,EAAE,oBAAoB,EAAC,MAAM,wBAAwB,CAAC;AAC/E,OAAO,EAAC,eAAe,EAAC,MAAM,8BAA8B,CAAC;AAC7D,cAAc,kCAAkC,CAAC;AACjD,cAAc,+BAA+B,CAAC;AAC9C,cAAc,gBAAgB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC;AACpC,cAAc,wBAAwB,CAAC;AACvC,OAAO,EAAC,eAAe,EAAC,MAAM,8BAA8B,CAAC;AAC7D,cAAc,kCAAkC,CAAC;AACjD,cAAc,+BAA+B,CAAC;AAC9C,cAAc,gBAAgB,CAAC"}

package/dist/internal/types.d.ts

@@ -1,6 +1,8 @@
+import { SignerProtocol } from '../environment/index.js';
 export type InternalEnvironment = {
     exportDeploymentsAsTypes(): Promise<void>;
     recoverTransactionsIfAny(): Promise<void>;
     recordMigration(id: string): void;
+    getSignerProtocol(protocol: string): SignerProtocol | undefined;
 };
 //# sourceMappingURL=types.d.ts.map

package/dist/internal/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/internal/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,mBAAmB,GAAG;IACjC,wBAAwB,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1C,wBAAwB,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1C,eAAe,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI,CAAC;CAClC,CAAC"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/internal/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,cAAc,EAAC,MAAM,yBAAyB,CAAC;AAEvD,MAAM,MAAM,mBAAmB,GAAG;IACjC,wBAAwB,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1C,wBAAwB,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1C,eAAe,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC,iBAAiB,CAAC,QAAQ,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS,CAAC;CAChE,CAAC"}

package/dist/utils/curry.d.ts

@@ -0,0 +1,35 @@
+import type { Environment, UnknownDeployments, UnresolvedNetworkSpecificData, UnresolvedUnknownNamedAccounts } from '../environment/types.js';
+/**
+ * Utility type to remove the first parameter from a function type
+ */
+type RemoveFirstParameter<T> = T extends (first: any, ...rest: infer R) => infer Return ? (...args: R) => Return : never;
+/**
+ * Utility type to transform an object of functions by removing their first parameter
+ */
+type CurriedFunctions<T> = {
+    [K in keyof T]: RemoveFirstParameter<T[K]>;
+};
+/**
+ * Creates a curried version of functions that automatically inject an environment as the first parameter.
+ *
+ * @param env - The environment object to inject as the first parameter
+ * @param functions - An object containing functions that expect the environment as their first parameter
+ * @returns An object with the same function names, but with the environment parameter removed
+ *
+ * @example
+ * ```typescript
+ * const functions = {
+ *   deploy: (env: Environment, contractName: string, args: any[]) => Promise<void>,
+ *   verify: (env: Environment, address: string) => Promise<boolean>
+ * };
+ *
+ * const curriedFunctions = withEnvironment(env, functions);
+ *
+ * // Now you can call without passing env:
+ * await curriedFunctions.deploy('MyContract', []);
+ * await curriedFunctions.verify('0x123...');
+ * ```
+ */
+export declare function withEnvironment<NamedAccounts extends UnresolvedUnknownNamedAccounts = UnresolvedUnknownNamedAccounts, Data extends UnresolvedNetworkSpecificData = UnresolvedNetworkSpecificData, Deployments extends UnknownDeployments = UnknownDeployments, T extends Record<string, (env: Environment<NamedAccounts, Data, Deployments>, ...args: any[]) => any> = Record<string, (env: Environment<NamedAccounts, Data, Deployments>, ...args: any[]) => any>>(env: Environment<NamedAccounts, Data, Deployments>, functions: T): CurriedFunctions<T>;
+export {};
+//# sourceMappingURL=curry.d.ts.map

package/dist/utils/curry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"curry.d.ts","sourceRoot":"","sources":["../../src/utils/curry.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACX,WAAW,EACX,kBAAkB,EAClB,6BAA6B,EAC7B,8BAA8B,EAC9B,MAAM,yBAAyB,CAAC;AAEjC;;GAEG;AACH,KAAK,oBAAoB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,IAAI,EAAE,MAAM,CAAC,KAAK,MAAM,MAAM,GACpF,CAAC,GAAG,IAAI,EAAE,CAAC,KAAK,MAAM,GACtB,KAAK,CAAC;AAET;;GAEG;AACH,KAAK,gBAAgB,CAAC,CAAC,IAAI;KACzB,CAAC,IAAI,MAAM,CAAC,GAAG,oBAAoB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;CAC1C,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,wBAAgB,eAAe,CAC9B,aAAa,SAAS,8BAA8B,GAAG,8BAA8B,EACrF,IAAI,SAAS,6BAA6B,GAAG,6BAA6B,EAC1E,WAAW,SAAS,kBAAkB,GAAG,kBAAkB,EAC3D,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,CAAC,GAAG,MAAM,CAC7G,MAAM,EACN,CAAC,GAAG,EAAE,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,CAC3E,EACA,GAAG,EAAE,WAAW,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,gBAAgB,CAAC,CAAC,CAAC,CASvF"}

package/dist/utils/curry.js

@@ -0,0 +1,30 @@
+/**
+ * Creates a curried version of functions that automatically inject an environment as the first parameter.
+ *
+ * @param env - The environment object to inject as the first parameter
+ * @param functions - An object containing functions that expect the environment as their first parameter
+ * @returns An object with the same function names, but with the environment parameter removed
+ *
+ * @example
+ * ```typescript
+ * const functions = {
+ *   deploy: (env: Environment, contractName: string, args: any[]) => Promise<void>,
+ *   verify: (env: Environment, address: string) => Promise<boolean>
+ * };
+ *
+ * const curriedFunctions = withEnvironment(env, functions);
+ *
+ * // Now you can call without passing env:
+ * await curriedFunctions.deploy('MyContract', []);
+ * await curriedFunctions.verify('0x123...');
+ * ```
+ */
+export function withEnvironment(env, functions) {
+    const result = {};
+    for (const [key, func] of Object.entries(functions)) {
+        // Create a new function that automatically passes the environment as the first argument
+        result[key] = (...args) => func(env, ...args);
+    }
+    return result;
+}
+//# sourceMappingURL=curry.js.map

package/dist/utils/curry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"curry.js","sourceRoot":"","sources":["../../src/utils/curry.ts"],"names":[],"mappings":"AAqBA;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,UAAU,eAAe,CAQ7B,GAAkD,EAAE,SAAY;IACjE,MAAM,MAAM,GAAG,EAAyB,CAAC;IAEzC,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;QACrD,wFAAwF;QACvF,MAAc,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,IAAW,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC"}

package/dist/utils/curry.test.d.ts

@@ -0,0 +1,16 @@
+import type { Environment } from '../environment/types.js';
+declare const exampleFunctions: {
+    deploy: (env: Environment, contractName: string, args: any[]) => Promise<void>;
+    verify: (env: Environment, address: string) => Promise<boolean>;
+    getBalance: (env: Environment, address: string) => Promise<bigint>;
+    syncFunction: (env: Environment, value: number) => number;
+};
+declare const curriedFunctions: {
+    deploy: (contractName: string, args: any[]) => Promise<void>;
+    verify: (address: string) => Promise<boolean>;
+    getBalance: (address: string) => Promise<bigint>;
+    syncFunction: (value: number) => number;
+};
+declare function testTypes(): Promise<void>;
+export { testTypes, curriedFunctions, exampleFunctions };
+//# sourceMappingURL=curry.test.d.ts.map

package/dist/utils/curry.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"curry.test.d.ts","sourceRoot":"","sources":["../../src/utils/curry.test.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,yBAAyB,CAAC;AAOzD,QAAA,MAAM,gBAAgB;kBACP,WAAW,gBAAgB,MAAM,QAAQ,GAAG,EAAE,KAAG,OAAO,CAAC,IAAI,CAAC;kBAI9D,WAAW,WAAW,MAAM,KAAG,OAAO,CAAC,OAAO,CAAC;sBAI3C,WAAW,WAAW,MAAM,KAAG,OAAO,CAAC,MAAM,CAAC;wBAI5C,WAAW,SAAS,MAAM,KAAG,MAAM;CAGvD,CAAC;AAGF,QAAA,MAAM,gBAAgB;;;;;CAA6C,CAAC;AAGpE,iBAAe,SAAS,kBASvB;AAGD,OAAO,EAAC,SAAS,EAAE,gBAAgB,EAAE,gBAAgB,EAAC,CAAC"}

package/dist/utils/curry.test.js

@@ -0,0 +1,33 @@
+import { withEnvironment } from './curry.js';
+// Mock environment for testing
+const mockEnv = {};
+// Example functions that take environment as first parameter
+const exampleFunctions = {
+    deploy: (env, contractName, args) => {
+        return Promise.resolve();
+    },
+    verify: (env, address) => {
+        return Promise.resolve(true);
+    },
+    getBalance: (env, address) => {
+        return Promise.resolve(BigInt(0));
+    },
+    syncFunction: (env, value) => {
+        return value * 2;
+    },
+};
+// Test the currying function
+const curriedFunctions = withEnvironment(mockEnv, exampleFunctions);
+// Type tests - these should compile without errors
+async function testTypes() {
+    // These calls should work without passing env
+    await curriedFunctions.deploy('MyContract', []);
+    const isVerified = await curriedFunctions.verify('0x123...');
+    const balance = await curriedFunctions.getBalance('0x456...');
+    const doubled = curriedFunctions.syncFunction(42);
+    console.log('Type tests passed!');
+    console.log({ isVerified, balance, doubled });
+}
+// Export for potential use in actual tests
+export { testTypes, curriedFunctions, exampleFunctions };
+//# sourceMappingURL=curry.test.js.map

package/dist/utils/curry.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"curry.test.js","sourceRoot":"","sources":["../../src/utils/curry.test.ts"],"names":[],"mappings":"AACA,OAAO,EAAC,eAAe,EAAC,MAAM,YAAY,CAAC;AAE3C,+BAA+B;AAC/B,MAAM,OAAO,GAAG,EAAiB,CAAC;AAElC,6DAA6D;AAC7D,MAAM,gBAAgB,GAAG;IACxB,MAAM,EAAE,CAAC,GAAgB,EAAE,YAAoB,EAAE,IAAW,EAAiB,EAAE;QAC9E,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IAC1B,CAAC;IAED,MAAM,EAAE,CAAC,GAAgB,EAAE,OAAe,EAAoB,EAAE;QAC/D,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC;IAED,UAAU,EAAE,CAAC,GAAgB,EAAE,OAAe,EAAmB,EAAE;QAClE,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC;IAED,YAAY,EAAE,CAAC,GAAgB,EAAE,KAAa,EAAU,EAAE;QACzD,OAAO,KAAK,GAAG,CAAC,CAAC;IAClB,CAAC;CACD,CAAC;AAEF,6BAA6B;AAC7B,MAAM,gBAAgB,GAAG,eAAe,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;AAEpE,mDAAmD;AACnD,KAAK,UAAU,SAAS;IACvB,8CAA8C;IAC9C,MAAM,gBAAgB,CAAC,MAAM,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;IAChD,MAAM,UAAU,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAC7D,MAAM,OAAO,GAAG,MAAM,gBAAgB,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IAC9D,MAAM,OAAO,GAAG,gBAAgB,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;IAElD,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IAClC,OAAO,CAAC,GAAG,CAAC,EAAC,UAAU,EAAE,OAAO,EAAE,OAAO,EAAC,CAAC,CAAC;AAC7C,CAAC;AAED,2CAA2C;AAC3C,OAAO,EAAC,SAAS,EAAE,gBAAgB,EAAE,gBAAgB,EAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "rocketh",
-  "version": "0.11.21",
+  "version": "0.11.22-testing.1",
   "description": "deploy smart contract on ethereum-compatible networks",
   "publishConfig": {
     "access": "public"

package/src/environment/index.ts

@@ -42,26 +42,11 @@ import {TransactionHashTracker, TransactionHashTrackerProvider} from './provider
 
 type ReceiptResult = {receipt: EIP1193TransactionReceipt; latestBlockNumber: EIP1193QUANTITY};
 
-export type EnvironmentExtenstion = (env: Environment) => Environment;
-//we store this globally so this is not lost
-(globalThis as any).extensions = [];
-export function extendEnvironment(extension: EnvironmentExtenstion): void {
-	(globalThis as any).extensions.push(extension);
-}
-
 export type SignerProtocolFunction = (protocolString: string) => Promise<Signer>;
 export type SignerProtocol = {
 	getSigner: SignerProtocolFunction;
 };
 
-//we store this globally so this is not lost
-(globalThis as any).signerProtocols = {};
-export function handleSignerProtocol(protocol: string, getSigner: SignerProtocolFunction): void {
-	(globalThis as any).signerProtocols[protocol] = {
-		getSigner,
-	};
-}
-
 function wait(numSeconds: number): Promise<void> {
 	return new Promise((resolve) => {
 		setTimeout(resolve, numSeconds * 1000);
@@ -103,6 +88,17 @@ export async function createEnvironment<
 		console.error(`failed to get genesis hash`);
 	}
 
+	const signerProtocols: {[protocol: string]: SignerProtocol} = {};
+	function getSignerProtocol(protocol: string): SignerProtocol | undefined {
+		return signerProtocols[protocol] || undefined;
+	}
+
+	function registerProtocol(protocol: string, getSigner: SignerProtocolFunction): void {
+		signerProtocols[protocol] = {
+			getSigner,
+		};
+	}
+
 	let networkName: string;
 	let saveDeployments: boolean;
 	let networkTags: {[tag: string]: boolean} = {};
@@ -157,7 +153,7 @@ export async function createEnvironment<
 		} else if (typeof accountDef === 'string') {
 			if (accountDef.startsWith('0x')) {
 				if (accountDef.length === 66) {
-					const privateKeyProtocol: SignerProtocol = (globalThis as any).signerProtocols['privateKey'];
+					const privateKeyProtocol = getSignerProtocol('privateKey');
 					if (privateKeyProtocol) {
 						const namedSigner = await privateKeyProtocol.getSigner(`privateKey:${accountDef}`);
 						const [address] = await namedSigner.signer.request({method: 'eth_accounts'});
@@ -176,9 +172,9 @@ export async function createEnvironment<
 			} else {
 				if (accountDef.indexOf(':') > 0) {
 					const [protocolID, extra] = accountDef.split(':');
-					const protocol: SignerProtocol = (globalThis as any).signerProtocols[protocolID];
+					const protocol = getSignerProtocol(protocolID);
 					if (!protocol) {
-						throw new Error(`protocol: ${protocol} is not supported`);
+						throw new Error(`protocol: ${protocolID} is not supported`);
 					}
 					const namedSigner = await protocol.getSigner(accountDef);
 					const [address] = await namedSigner.signer.request({method: 'eth_accounts'});
@@ -710,6 +706,7 @@ export async function createEnvironment<
 		showMessage,
 		showProgress,
 		hasMigrationBeenDone,
+		registerProtocol,
 	};
 	for (const extension of (globalThis as any).extensions) {
 		env = extension(env);
@@ -718,6 +715,7 @@ export async function createEnvironment<
 	return {
 		external: env,
 		internal: {
+			getSignerProtocol,
 			exportDeploymentsAsTypes,
 			recoverTransactionsIfAny,
 			recordMigration,

package/src/environment/types.ts

@@ -13,6 +13,7 @@ import type {Chain} from 'viem';
 import {ProgressIndicator} from '../internal/logging.js';
 import {TransactionHashTracker} from './providers/TransactionHashTracker.js';
 import {DeterministicDeploymentInfo} from '../executor/index.js';
+import {SignerProtocolFunction} from './index.js';
 
 export type {Abi, AbiError, AbiEvent, AbiConstructor, AbiFallback, AbiFunction, AbiReceive};
 export type Libraries = {readonly [libraryName: string]: EIP1193Account};
@@ -349,6 +350,7 @@ export interface Environment<
 	showProgress(message?: string): ProgressIndicator;
 
 	hasMigrationBeenDone(id: string): boolean;
+	registerProtocol(protocol: string, getSigner: SignerProtocolFunction): void;
 }
 
 export type DeploymentConstruction<TAbi extends Abi> = Omit<

package/src/executor/index.ts

@@ -5,13 +5,13 @@ import type {
 	Config,
 	Environment,
 	ResolvedConfig,
-	ResolvedNamedAccounts,
 	UnknownDeployments,
 	UnresolvedNetworkSpecificData,
 	UnresolvedUnknownNamedAccounts,
 } from '../environment/types.js';
-import {createEnvironment} from '../environment/index.js';
-import {DeployScriptFunction, DeployScriptModule} from './types.js';
+import {createEnvironment, SignerProtocolFunction} from '../environment/index.js';
+import {DeployScriptFunction, DeployScriptModule, EnhancedDeployScriptFunction, EnhancedEnvironment} from './types.js';
+import {withEnvironment} from '../utils/curry.js';
 import {logger, setLogLevel, spin} from '../internal/logging.js';
 import {EIP1193GenericRequestProvider, EIP1193ProviderWithoutEvents} from 'eip-1193';
 import {getRoughGasPriceEstimate} from '../utils/eth.js';
@@ -19,24 +19,70 @@ import prompts from 'prompts';
 import {formatEther} from 'viem';
 import {tsImport} from 'tsx/esm/api';
 
-export function execute<
+/**
+ * Setup function that creates the execute function for deploy scripts. It allow to specify a set of functions that will be available in the environment.
+ *
+ * @param functions - An object of utility functions that expect Environment as their first parameter
+ * @returns An execute function that provides an enhanced environment with curried functions
+ *
+ * @example
+ * ```typescript
+ * const functions = {
+ *   deploy: (env: Environment, contractName: string, args: any[]) => Promise<void>,
+ *   verify: (env: Environment, address: string) => Promise<boolean>
+ * };
+ *
+ * const execute = setup(functions);
+ *
+ * export default execute(async (env, args) => {
+ *   // env now includes both the original environment AND the curried functions
+ *   await env.deploy('MyContract', []); // No need to pass env
+ *   await env.verify('0x123...'); // No need to pass env
+ *
+ *   // Original environment properties are still available
+ *   console.log(env.network.name);
+ *   const deployment = env.get('MyContract');
+ * }, { tags: ['deploy'] });
+ * ```
+ */
+export function setup<
+	Functions extends Record<string, (env: Environment<any, any, any>, ...args: any[]) => any> = {},
 	NamedAccounts extends UnresolvedUnknownNamedAccounts = UnresolvedUnknownNamedAccounts,
 	Data extends UnresolvedNetworkSpecificData = UnresolvedNetworkSpecificData,
-	ArgumentsType = undefined,
 	Deployments extends UnknownDeployments = UnknownDeployments
->(
-	callback: DeployScriptFunction<NamedAccounts, Data, ArgumentsType, Deployments>,
-	options: {tags?: string[]; dependencies?: string[]; id?: string}
-): DeployScriptModule<NamedAccounts, Data, ArgumentsType, Deployments> {
-	const scriptModule: DeployScriptModule<NamedAccounts, Data, ArgumentsType, Deployments> = (
-		env: Environment<NamedAccounts, Data, Deployments>,
-		args?: ArgumentsType
-	) => callback(env, args);
-	scriptModule.tags = options.tags;
-	scriptModule.dependencies = options.dependencies;
-	scriptModule.id = options.id;
-	// TODO runAtTheEnd ?
-	return scriptModule;
+>(functions: Functions, signerProtocols?: Record<string, SignerProtocolFunction>) {
+	return function enhancedExecute<ArgumentsType = undefined>(
+		callback: EnhancedDeployScriptFunction<NamedAccounts, Data, ArgumentsType, Deployments, Functions>,
+		options: {tags?: string[]; dependencies?: string[]; id?: string; runAtTheEnd?: boolean}
+	): DeployScriptModule<NamedAccounts, Data, ArgumentsType, Deployments> {
+		const scriptModule: DeployScriptModule<NamedAccounts, Data, ArgumentsType, Deployments> = (
+			env: Environment<NamedAccounts, Data, Deployments>,
+			args?: ArgumentsType
+		) => {
+			if (signerProtocols) {
+				for (const key of Object.keys(signerProtocols)) {
+					env.registerProtocol(key, signerProtocols[key]);
+				}
+			}
+
+			// Create the enhanced environment by combining the original environment with curried functions
+			const curriedFunctions = withEnvironment(env, functions);
+			const enhancedEnv = Object.assign(
+				Object.create(Object.getPrototypeOf(env)),
+				env,
+				curriedFunctions
+			) as EnhancedEnvironment<NamedAccounts, Data, Deployments, Functions>;
+
+			return callback(enhancedEnv, args);
+		};
+
+		scriptModule.tags = options.tags;
+		scriptModule.dependencies = options.dependencies;
+		scriptModule.id = options.id;
+		scriptModule.runAtTheEnd = options.runAtTheEnd;
+
+		return scriptModule;
+	};
 }
 
 export type NamedAccountExecuteFunction<

package/src/executor/setup.test.ts

@@ -0,0 +1,133 @@
+import type {Environment} from '../environment/types.js';
+import {setup} from './index.js';
+
+// Mock environment for testing
+const mockEnv = {} as Environment;
+
+// Example utility functions that take Environment as first parameter
+const utilityFunctions = {
+	deployContract: async (env: Environment, contractName: string, args: any[] = []): Promise<string> => {
+		console.log(`Deploying ${contractName} with args:`, args);
+		return '0x1234567890123456789012345678901234567890';
+	},
+
+	verifyContract: async (env: Environment, address: string): Promise<boolean> => {
+		console.log(`Verifying contract at ${address}`);
+		return true;
+	},
+
+	getBalance: async (env: Environment, address: string): Promise<bigint> => {
+		console.log(`Getting balance for ${address}`);
+		return BigInt('1000000000000000000'); // 1 ETH
+	},
+
+	calculateGas: (env: Environment, operation: string): number => {
+		console.log(`Calculating gas for ${operation}`);
+		return 21000;
+	},
+
+	isDeployed: (env: Environment, contractName: string): boolean => {
+		console.log(`Checking if ${contractName} is deployed`);
+		return false;
+	},
+};
+
+// Create the enhanced execute function using setup
+const execute = setup(utilityFunctions);
+
+// Test the enhanced execute function
+const testScript = execute(
+	async (env, args) => {
+		// Type test: env should have both original Environment properties AND curried functions
+
+		// Test curried functions (no need to pass env)
+		const address = await env.deployContract('MyToken', ['TokenName', 'TKN']);
+		const isVerified = await env.verifyContract(address);
+		const balance = await env.getBalance(address);
+		const gasEstimate = env.calculateGas('transfer');
+		const deployed = env.isDeployed('MyToken');
+
+		// Test that original environment properties are still accessible
+		// (These would normally be available on a real environment)
+		// console.log('Network name:', env.network?.name);
+		// console.log('Chain ID:', env.network?.chain?.id);
+
+		console.log('Test results:', {
+			address,
+			isVerified,
+			balance: balance.toString(),
+			gasEstimate,
+			deployed,
+		});
+
+		return true; // Return true to indicate successful completion
+	},
+	{
+		tags: ['test'],
+		dependencies: [],
+		id: 'test-setup-function',
+	}
+);
+
+// Type tests - these should compile without errors
+async function testTypes() {
+	// The script should be a valid DeployScriptModule
+	console.log('Script tags:', testScript.tags);
+	console.log('Script dependencies:', testScript.dependencies);
+	console.log('Script ID:', testScript.id);
+
+	// The script should be callable with an environment
+	try {
+		const result = await testScript(mockEnv, undefined);
+		console.log('Script execution result:', result);
+	} catch (error) {
+		console.log('Script execution test completed (expected with mock env)');
+	}
+}
+
+// Example of how this would be used in a real deployment script
+export const exampleUsage = () => {
+	// Define your utility functions
+	const myFunctions = {
+		deployERC20: async (env: Environment, name: string, symbol: string): Promise<string> => {
+			// Implementation would use env.save, env.network.provider, etc.
+			return '0x...';
+		},
+
+		setupPermissions: async (env: Environment, contractAddress: string, admin: string): Promise<void> => {
+			// Implementation would interact with contracts
+		},
+
+		verifyOnEtherscan: async (env: Environment, address: string, constructorArgs: any[]): Promise<boolean> => {
+			// Implementation would call verification service
+			return true;
+		},
+	};
+
+	// Create the enhanced execute function
+	const execute = setup(myFunctions);
+
+	// Export your deployment script
+	return execute(
+		async (env, args) => {
+			// Now you can use the functions without passing env each time
+			const tokenAddress = await env.deployERC20('MyToken', 'MTK');
+			await env.setupPermissions(tokenAddress, env.namedAccounts.deployer);
+			await env.verifyOnEtherscan(tokenAddress, ['MyToken', 'MTK']);
+
+			// Original environment is still fully accessible
+			console.log(`Deployed on network: ${env.network.name}`);
+			const deployment = env.get('MyToken');
+
+			return true;
+		},
+		{
+			tags: ['token', 'deploy'],
+			dependencies: ['setup'],
+			id: 'deploy-my-token',
+		}
+	);
+};
+
+// Export for potential use in actual tests
+export {testTypes, testScript, utilityFunctions};

package/src/executor/types.ts

@@ -30,3 +30,48 @@ export type ScriptCallback<
 	Data extends UnresolvedNetworkSpecificData = UnresolvedNetworkSpecificData,
 	Deployments extends UnknownDeployments = UnknownDeployments
 > = (env: Environment<NamedAccounts, Data, Deployments>) => Promise<void>;
+
+/**
+ * Utility type to remove the first parameter from a function type
+ */
+type RemoveFirstParameter<T> = T extends (first: any, ...rest: infer R) => infer Return
+	? (...args: R) => Return
+	: never;
+
+/**
+ * Utility type to transform an object of functions by removing their first parameter
+ */
+type CurriedFunctions<T> = {
+	[K in keyof T]: RemoveFirstParameter<T[K]>;
+};
+
+/**
+ * Type for the enhanced environment proxy that includes both the original environment
+ * and the curried functions
+ */
+export type EnhancedEnvironment<
+	NamedAccounts extends UnresolvedUnknownNamedAccounts = UnresolvedUnknownNamedAccounts,
+	Data extends UnresolvedNetworkSpecificData = UnresolvedNetworkSpecificData,
+	Deployments extends UnknownDeployments = UnknownDeployments,
+	Functions extends Record<
+		string,
+		(env: Environment<NamedAccounts, Data, Deployments>, ...args: any[]) => any
+	> = Record<string, (env: Environment<NamedAccounts, Data, Deployments>, ...args: any[]) => any>
+> = Environment<NamedAccounts, Data, Deployments> & CurriedFunctions<Functions>;
+
+/**
+ * Type for a deploy script function that receives an enhanced environment
+ */
+export type EnhancedDeployScriptFunction<
+	NamedAccounts extends UnresolvedUnknownNamedAccounts = UnresolvedUnknownNamedAccounts,
+	Data extends UnresolvedNetworkSpecificData = UnresolvedNetworkSpecificData,
+	ArgumentsTypes = undefined,
+	Deployments extends UnknownDeployments = UnknownDeployments,
+	Functions extends Record<
+		string,
+		(env: Environment<NamedAccounts, Data, Deployments>, ...args: any[]) => any
+	> = Record<string, (env: Environment<NamedAccounts, Data, Deployments>, ...args: any[]) => any>
+> = (
+	env: EnhancedEnvironment<NamedAccounts, Data, Deployments, Functions>,
+	args?: ArgumentsTypes
+) => Promise<void | boolean>;

package/src/index.ts

@@ -1,7 +1,6 @@
 export * from './executor/index.js';
 export * from './executor/types.js';
 export * from './environment/types.js';
-export {extendEnvironment, handleSignerProtocol} from './environment/index.js';
 export {loadDeployments} from './environment/deployments.js';
 export * from './environment/utils/artifacts.js';
 export * from './environment/utils/chains.js';

package/src/internal/types.ts

@@ -1,5 +1,8 @@
+import {SignerProtocol} from '../environment/index.js';
+
 export type InternalEnvironment = {
 	exportDeploymentsAsTypes(): Promise<void>;
 	recoverTransactionsIfAny(): Promise<void>;
 	recordMigration(id: string): void;
+	getSignerProtocol(protocol: string): SignerProtocol | undefined;
 };