204 lines
7.1 KiB
JavaScript
204 lines
7.1 KiB
JavaScript
|
"use strict";
|
||
|
|
||
|
Object.defineProperty(exports, "__esModule", {
|
||
|
value: true
|
||
|
});
|
||
|
exports.defaultOpts = defaultOpts;
|
||
|
exports.default = void 0;
|
||
|
|
||
|
var parser = _interopRequireWildcard(require("@babel/parser"));
|
||
|
|
||
|
var t = _interopRequireWildcard(require("@babel/types"));
|
||
|
|
||
|
var _traverse = _interopRequireDefault(require("@babel/traverse"));
|
||
|
|
||
|
var _generator = _interopRequireDefault(require("@babel/generator"));
|
||
|
|
||
|
var _visitor = _interopRequireDefault(require("./visitor"));
|
||
|
|
||
|
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
|
||
|
|
||
|
function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = Object.defineProperty && Object.getOwnPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : {}; if (desc.get || desc.set) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; return newObj; } }
|
||
|
|
||
|
/*
|
||
|
Copyright 2012-2015, Yahoo Inc.
|
||
|
Copyrights licensed under the New BSD License. See the accompanying LICENSE file for terms.
|
||
|
*/
|
||
|
function defaultOpts() {
|
||
|
return {
|
||
|
coverageVariable: '__coverage__',
|
||
|
coverageGlobalScope: 'this',
|
||
|
coverageGlobalScopeFunc: true,
|
||
|
preserveComments: false,
|
||
|
compact: true,
|
||
|
esModules: false,
|
||
|
autoWrap: false,
|
||
|
produceSourceMap: false,
|
||
|
ignoreClassMethods: [],
|
||
|
sourceMapUrlCallback: null,
|
||
|
debug: false,
|
||
|
|
||
|
/* babel parser plugins are to be enabled when the feature is stage 3 and
|
||
|
* implemented in a released version of node.js */
|
||
|
plugins: ['asyncGenerators', 'bigInt', 'classProperties', 'classPrivateProperties', 'dynamicImport', 'importMeta', 'objectRestSpread', 'optionalCatchBinding', 'flow', 'jsx']
|
||
|
};
|
||
|
}
|
||
|
/**
|
||
|
* Instrumenter is the public API for the instrument library.
|
||
|
* It is typically used for ES5 code. For ES6 code that you
|
||
|
* are already running under `babel` use the coverage plugin
|
||
|
* instead.
|
||
|
* @param {Object} opts optional.
|
||
|
* @param {string} [opts.coverageVariable=__coverage__] name of global coverage variable.
|
||
|
* @param {boolean} [opts.preserveComments=false] preserve comments in output
|
||
|
* @param {boolean} [opts.compact=true] generate compact code.
|
||
|
* @param {boolean} [opts.esModules=false] set to true to instrument ES6 modules.
|
||
|
* @param {boolean} [opts.autoWrap=false] set to true to allow `return` statements outside of functions.
|
||
|
* @param {boolean} [opts.produceSourceMap=false] set to true to produce a source map for the instrumented code.
|
||
|
* @param {Array} [opts.ignoreClassMethods=[]] set to array of class method names to ignore for coverage.
|
||
|
* @param {Function} [opts.sourceMapUrlCallback=null] a callback function that is called when a source map URL
|
||
|
* is found in the original code. This function is called with the source file name and the source map URL.
|
||
|
* @param {boolean} [opts.debug=false] - turn debugging on
|
||
|
* @param {array} [opts.plugins=['asyncGenerators','dynamicImport','objectRestSpread','optionalCatchBinding','flow','jsx']] - set plugins
|
||
|
*/
|
||
|
|
||
|
|
||
|
class Instrumenter {
|
||
|
constructor(opts = defaultOpts()) {
|
||
|
this.opts = this.normalizeOpts(opts);
|
||
|
this.fileCoverage = null;
|
||
|
this.sourceMap = null;
|
||
|
}
|
||
|
/**
|
||
|
* normalize options passed in and assign defaults.
|
||
|
* @param opts
|
||
|
* @private
|
||
|
*/
|
||
|
|
||
|
|
||
|
normalizeOpts(opts) {
|
||
|
const normalize = (name, defaultValue) => {
|
||
|
if (!opts.hasOwnProperty(name)) {
|
||
|
opts[name] = defaultValue;
|
||
|
}
|
||
|
};
|
||
|
|
||
|
const defOpts = defaultOpts();
|
||
|
Object.keys(defOpts).forEach(k => {
|
||
|
normalize(k, defOpts[k]);
|
||
|
});
|
||
|
return opts;
|
||
|
}
|
||
|
/**
|
||
|
* instrument the supplied code and track coverage against the supplied
|
||
|
* filename. It throws if invalid code is passed to it. ES5 and ES6 syntax
|
||
|
* is supported. To instrument ES6 modules, make sure that you set the
|
||
|
* `esModules` property to `true` when creating the instrumenter.
|
||
|
*
|
||
|
* @param {string} code - the code to instrument
|
||
|
* @param {string} filename - the filename against which to track coverage.
|
||
|
* @param {object} [inputSourceMap] - the source map that maps the not instrumented code back to it's original form.
|
||
|
* Is assigned to the coverage object and therefore, is available in the json output and can be used to remap the
|
||
|
* coverage to the untranspiled source.
|
||
|
* @returns {string} the instrumented code.
|
||
|
*/
|
||
|
|
||
|
|
||
|
instrumentSync(code, filename, inputSourceMap) {
|
||
|
if (typeof code !== 'string') {
|
||
|
throw new Error('Code must be a string');
|
||
|
}
|
||
|
|
||
|
filename = filename || String(new Date().getTime()) + '.js';
|
||
|
const opts = this.opts;
|
||
|
const ast = parser.parse(code, {
|
||
|
allowReturnOutsideFunction: opts.autoWrap,
|
||
|
sourceType: opts.esModules ? 'module' : 'script',
|
||
|
plugins: opts.plugins
|
||
|
});
|
||
|
const ee = (0, _visitor.default)(t, filename, {
|
||
|
coverageVariable: opts.coverageVariable,
|
||
|
coverageGlobalScope: opts.coverageGlobalScope,
|
||
|
coverageGlobalScopeFunc: opts.coverageGlobalScopeFunc,
|
||
|
ignoreClassMethods: opts.ignoreClassMethods,
|
||
|
inputSourceMap
|
||
|
});
|
||
|
let output = {};
|
||
|
const visitor = {
|
||
|
Program: {
|
||
|
enter: ee.enter,
|
||
|
|
||
|
exit(path) {
|
||
|
output = ee.exit(path);
|
||
|
}
|
||
|
|
||
|
}
|
||
|
};
|
||
|
(0, _traverse.default)(ast, visitor);
|
||
|
const generateOptions = {
|
||
|
compact: opts.compact,
|
||
|
comments: opts.preserveComments,
|
||
|
sourceMaps: opts.produceSourceMap,
|
||
|
sourceFileName: filename
|
||
|
};
|
||
|
const codeMap = (0, _generator.default)(ast, generateOptions, code);
|
||
|
this.fileCoverage = output.fileCoverage;
|
||
|
this.sourceMap = codeMap.map;
|
||
|
const cb = this.opts.sourceMapUrlCallback;
|
||
|
|
||
|
if (cb && output.sourceMappingURL) {
|
||
|
cb(filename, output.sourceMappingURL);
|
||
|
}
|
||
|
|
||
|
return codeMap.code;
|
||
|
}
|
||
|
/**
|
||
|
* callback-style instrument method that calls back with an error
|
||
|
* as opposed to throwing one. Note that in the current implementation,
|
||
|
* the callback will be called in the same process tick and is not asynchronous.
|
||
|
*
|
||
|
* @param {string} code - the code to instrument
|
||
|
* @param {string} filename - the filename against which to track coverage.
|
||
|
* @param {Function} callback - the callback
|
||
|
* @param {Object} inputSourceMap - the source map that maps the not instrumented code back to it's original form.
|
||
|
* Is assigned to the coverage object and therefore, is available in the json output and can be used to remap the
|
||
|
* coverage to the untranspiled source.
|
||
|
*/
|
||
|
|
||
|
|
||
|
instrument(code, filename, callback, inputSourceMap) {
|
||
|
if (!callback && typeof filename === 'function') {
|
||
|
callback = filename;
|
||
|
filename = null;
|
||
|
}
|
||
|
|
||
|
try {
|
||
|
const out = this.instrumentSync(code, filename, inputSourceMap);
|
||
|
callback(null, out);
|
||
|
} catch (ex) {
|
||
|
callback(ex);
|
||
|
}
|
||
|
}
|
||
|
/**
|
||
|
* returns the file coverage object for the last file instrumented.
|
||
|
* @returns {Object} the file coverage object.
|
||
|
*/
|
||
|
|
||
|
|
||
|
lastFileCoverage() {
|
||
|
return this.fileCoverage;
|
||
|
}
|
||
|
/**
|
||
|
* returns the source map produced for the last file instrumented.
|
||
|
* @returns {null|Object} the source map object.
|
||
|
*/
|
||
|
|
||
|
|
||
|
lastSourceMap() {
|
||
|
return this.sourceMap;
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
var _default = Instrumenter;
|
||
|
exports.default = _default;
|