read-coverage.js 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778
  1. 'use strict';
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.default = readInitialCoverage;
  6. var _constants = require('./constants');
  7. var _babylon = require('babylon');
  8. var _babelTraverse = require('babel-traverse');
  9. var _babelTraverse2 = _interopRequireDefault(_babelTraverse);
  10. var _babelTypes = require('babel-types');
  11. var t = _interopRequireWildcard(_babelTypes);
  12. 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)) newObj[key] = obj[key]; } } newObj.default = obj; return newObj; } }
  13. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  14. function readInitialCoverage(code) {
  15. if (typeof code !== 'string') {
  16. throw new Error('Code must be a string');
  17. }
  18. // Parse as leniently as possible
  19. var ast = (0, _babylon.parse)(code, {
  20. allowImportExportEverywhere: true,
  21. allowReturnOutsideFunction: true,
  22. allowSuperOutsideMethod: true,
  23. sourceType: "script", // I think ?
  24. plugins: ["*"]
  25. });
  26. var covScope = void 0;
  27. (0, _babelTraverse2.default)(ast, {
  28. ObjectProperty: function ObjectProperty(path) {
  29. var node = path.node;
  30. if (!node.computed && t.isIdentifier(node.key) && node.key.name === _constants.MAGIC_KEY) {
  31. var magicValue = path.get('value').evaluate();
  32. if (!magicValue.confident || magicValue.value !== _constants.MAGIC_VALUE) {
  33. return;
  34. }
  35. covScope = path.scope.getFunctionParent() || path.scope.getProgramParent();
  36. path.stop();
  37. }
  38. }
  39. });
  40. if (!covScope) {
  41. return null;
  42. }
  43. var result = {};
  44. var _arr = ['path', 'hash', 'gcv', 'coverageData'];
  45. for (var _i = 0; _i < _arr.length; _i++) {
  46. var key = _arr[_i];
  47. var binding = covScope.getOwnBinding(key);
  48. if (!binding) {
  49. return null;
  50. }
  51. var valuePath = binding.path.get('init');
  52. var value = valuePath.evaluate();
  53. if (!value.confident) {
  54. return null;
  55. }
  56. result[key] = value.value;
  57. }
  58. delete result.coverageData[_constants.MAGIC_KEY];
  59. return result;
  60. }