rocketh 0.11.21 → 0.11.22-testing.1

package/src/utils/curry.test.ts

@@ -0,0 +1,42 @@
+import type {Environment} from '../environment/types.js';
+import {withEnvironment} from './curry.js';
+
+// Mock environment for testing
+const mockEnv = {} as Environment;
+
+// Example functions that take environment as first parameter
+const exampleFunctions = {
+	deploy: (env: Environment, contractName: string, args: any[]): Promise<void> => {
+		return Promise.resolve();
+	},
+
+	verify: (env: Environment, address: string): Promise<boolean> => {
+		return Promise.resolve(true);
+	},
+
+	getBalance: (env: Environment, address: string): Promise<bigint> => {
+		return Promise.resolve(BigInt(0));
+	},
+
+	syncFunction: (env: Environment, value: number): number => {
+		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};

package/src/utils/curry.ts

@@ -0,0 +1,60 @@
+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 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> {
+	const result = {} as CurriedFunctions<T>;
+
+	for (const [key, func] of Object.entries(functions)) {
+		// Create a new function that automatically passes the environment as the first argument
+		(result as any)[key] = (...args: any[]) => func(env, ...args);
+	}
+
+	return result;
+}