Internal tweaks

index.js

@@ -1,15 +1,14 @@
 import vm from 'node:vm';
 
-export default function functionTimeout(function_, {timeout} = {}) {
-	const script = new vm.Script('returnValue = function_()');
+const script = new vm.Script('returnValue = functionToRun()');
 
-	const wrappedFunction = (...arguments_) => {
-		const context = {
-			function_: () => function_(...arguments_),
-		};
+// TODO: Document the `context` option and add to types when I know it's something I want to keep.
 
+// If you use the `context` option, you do it at your own risk.
+export default function functionTimeout(function_, {timeout, context = vm.createContext()} = {}) {
+	const wrappedFunction = (...arguments_) => {
+		context.functionToRun = () => function_(...arguments_);
 		script.runInNewContext(context, {timeout});
-
 		return context.returnValue;
 	};
 

package.json

@@ -16,6 +16,7 @@
 		"node": "./index.js",
 		"default": "./browser.js"
 	},
+	"sideEffects": false,
 	"engines": {
 		"node": ">=18"
 	},

test.js

@@ -26,3 +26,11 @@ test('no timeout', t => {
 	const fixtureFunction = (a, b) => [a, b];
 	t.deepEqual(functionTimeout(fixtureFunction)(1, 2), [1, 2]);
 });
+
+test('multiple', t => {
+	const run = functionTimeout((a, b) => [a, b], {timeout: 100});
+	t.deepEqual(run(1, 2), [1, 2]);
+	t.deepEqual(run(1, 3), [1, 3]);
+	t.deepEqual(run(1, 4), [1, 4]);
+	t.deepEqual(run(1, 5), [1, 5]);
+});