Fix filename option

src/index.ts

@@ -3,7 +3,7 @@ import { generate } from 'escodegen';
 import { parseScript } from 'esprima';
 import { visit, namedTypes as n, builders as b } from 'ast-types';
 import { Context, RunningScriptOptions } from 'vm';
-import { VM } from 'vm2';
+import { VM, VMScript } from 'vm2';
 
 /**
  * Compiles sync JavaScript code into JavaScript with async Functions.
@@ -138,7 +138,10 @@ namespace degenerator {
  ): (...args: A) => Promise<R> {
  const compiled = degenerator(code, names);
  const vm = new VM(options);
- const fn = vm.run(`${compiled};${returnName}`);
+ const script = new VMScript(`${compiled};${returnName}`, {
+ filename: options.filename,
+ });
+ const fn = vm.run(script);
  if (typeof fn !== 'function') {
  throw new Error(
  `Expected a "function" to be returned for \`${returnName}\`, but got "${typeof fn}"`

test/test.ts

@@ -159,26 +159,42 @@ describe('degenerator()', () => {
  assert.equal(val, 'foo');
  });
  });
- it('should prevent privilege escalation of untrusted code', async() => {
+ it('should prevent privilege escalation of untrusted code', async () => {
  let err;
  try {
  const fn = compile<typeof process>(
  `const f = this.constructor.constructor('return process');`,
  'f',
- [],
+ []
  );
  await fn();
- } catch(_err) {
+ } catch (_err) {
  err = _err;
  }
- assert.equal(err.message,'process is not defined')
+ assert.equal(err.message, 'process is not defined');
  });
  it('should allow to return synchronous undefined', () => {
  function u() {}
  const fn = compile(`${u}`, 'u', ['']);
- return fn().then(val => {
+ return fn().then((val) => {
  assert.strictEqual(val, undefined);
  });
  });
+ it('should support "filename" option', async () => {
+ function u() {
+ throw new Error('fail');
+ }
+ let err;
+ const fn = compile(`${u}`, 'u', [''], {
+ filename: '/foo/bar/baz.js',
+ });
+ try {
+ await fn();
+ } catch (_err) {
+ err = _err;
+ }
+ assert.strictEqual(err.message, 'fail');
+ assert(err.stack.includes('at u (/foo/bar/baz.js:'));
+ });
  });
 });