Add metavm.NODE_CONTEXT

PR-URL: #109
metarhia · Jul 31, 2023 · d0f7cd7 · d0f7cd7
1 parent 1073e2b
commit d0f7cd7
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 25 deletions.
diff --git a/metavm.js b/metavm.js
@@ -21,31 +21,32 @@ const MODULE_TYPE = {
   COMMONJS: 2,
 };
 
-const EMPTY_CONTEXT = vm.createContext(Object.freeze({}), CONTEXT_OPTIONS);
+const DEFAULT = {
+  AbortController,
+  Event,
+  EventTarget,
+  MessageChannel,
+  MessageEvent,
+  MessagePort,
+  Buffer,
+  URL,
+  URLSearchParams,
+  TextDecoder,
+  TextEncoder,
+  queueMicrotask,
+  setTimeout,
+  setImmediate,
+  setInterval,
+  clearTimeout,
+  clearImmediate,
+  clearInterval,
+};
 
-const COMMON_CONTEXT = vm.createContext(
-  Object.freeze({
-    AbortController,
-    Event,
-    EventTarget,
-    MessageChannel,
-    MessageEvent,
-    MessagePort,
-    Buffer,
-    URL,
-    URLSearchParams,
-    TextDecoder,
-    TextEncoder,
-    console,
-    queueMicrotask,
-    setTimeout,
-    setImmediate,
-    setInterval,
-    clearTimeout,
-    clearImmediate,
-    clearInterval,
-  }),
-);
+const NODE = { global, console, process };
+
+const EMPTY_CONTEXT = vm.createContext(Object.freeze({}), CONTEXT_OPTIONS);
+const COMMON_CONTEXT = vm.createContext(Object.freeze({ ...DEFAULT }));
+const NODE_CONTEXT = vm.createContext(Object.freeze({ ...DEFAULT, ...NODE }));
 
 class MetavmError extends Error {}
 
@@ -163,6 +164,7 @@ module.exports = {
   createScript,
   EMPTY_CONTEXT,
   COMMON_CONTEXT,
+  NODE_CONTEXT,
   MODULE_TYPE,
   readScript,
 };
diff --git a/test/unit.js b/test/unit.js
@@ -218,8 +218,9 @@ metatests.test('Create default context', async (test) => {
 metatests.test('Create common context', async (test) => {
   const context = metavm.createContext(metavm.COMMON_CONTEXT);
   test.strictSame(typeof context, 'object');
-  test.strictSame(context.console, console);
   test.strictSame(context.global, undefined);
+  test.strictSame(context.console, undefined);
+  test.strictSame(context.process, undefined);
   test.strictSame(context.AbortController, AbortController);
   test.strictSame(context.Buffer, Buffer);
   test.strictSame(context.Event, Event);
@@ -241,6 +242,15 @@ metatests.test('Create common context', async (test) => {
   test.end();
 });
 
+metatests.test('Create nodejs context', async (test) => {
+  const context = metavm.createContext(metavm.NODE_CONTEXT);
+  test.strictSame(typeof context, 'object');
+  test.strictSame(context.global, global);
+  test.strictSame(context.console, console);
+  test.strictSame(context.process, process);
+  test.end();
+});
+
 metatests.test('Create custom context', async (test) => {
   const sandbox = { field: 'value' };
   sandbox.global = sandbox;