Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- "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 _getRequireWildcardCache() { if (typeof WeakMap !== "function") return null; var cache = new WeakMap(); _getRequireWildcardCache = function () { return cache; }; return cache; }
- function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } var cache = _getRequireWildcardCache(); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; if (obj != null) { var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } 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', 'v8intrinsic']
- };
- }
- /**
- * 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 = {}) {
- this.opts = { ...defaultOpts(),
- ...opts
- };
- this.fileCoverage = null;
- this.sourceMap = null;
- }
- /**
- * 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;
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement