123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204 |
- "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;
|