| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677 |
- const { parseSync, traverse } = require('@babel/core');
- const { defaults } = require('@istanbuljs/schema');
- const { MAGIC_KEY, MAGIC_VALUE } = require('./constants');
- function getAst(code) {
- if (typeof code === 'object' && typeof code.type === 'string') {
- // Assume code is already a babel ast.
- return code;
- }
- if (typeof code !== 'string') {
- throw new Error('Code must be a string');
- }
- // Parse as leniently as possible
- return parseSync(code, {
- babelrc: false,
- configFile: false,
- parserOpts: {
- allowAwaitOutsideFunction: true,
- allowImportExportEverywhere: true,
- allowReturnOutsideFunction: true,
- allowSuperOutsideMethod: true,
- sourceType: 'script',
- plugins: defaults.instrumenter.parserPlugins
- }
- });
- }
- module.exports = function readInitialCoverage(code) {
- const ast = getAst(code);
- let covScope;
- traverse(ast, {
- ObjectProperty(path) {
- const { node } = path;
- if (
- !node.computed &&
- path.get('key').isIdentifier() &&
- node.key.name === MAGIC_KEY
- ) {
- const magicValue = path.get('value').evaluate();
- if (!magicValue.confident || magicValue.value !== MAGIC_VALUE) {
- return;
- }
- covScope =
- path.scope.getFunctionParent() ||
- path.scope.getProgramParent();
- path.stop();
- }
- }
- });
- if (!covScope) {
- return null;
- }
- const result = {};
- for (const key of ['path', 'hash', 'gcv', 'coverageData']) {
- const binding = covScope.getOwnBinding(key);
- if (!binding) {
- return null;
- }
- const valuePath = binding.path.get('init');
- const value = valuePath.evaluate();
- if (!value.confident) {
- return null;
- }
- result[key] = value.value;
- }
- delete result.coverageData[MAGIC_KEY];
- delete result.coverageData.hash;
- return result;
- };
|