index.js 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.createRegExpFeaturePlugin = createRegExpFeaturePlugin;
  6. var _regexpuCore = _interopRequireDefault(require("regexpu-core"));
  7. var _features = require("./features");
  8. var _util = require("./util");
  9. var _package = _interopRequireDefault(require("../package.json"));
  10. var _core = require("@babel/core");
  11. var _helperAnnotateAsPure = _interopRequireDefault(require("@babel/helper-annotate-as-pure"));
  12. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  13. function pullFlag(node, flag) {
  14. node.flags = node.flags.replace(flag, "");
  15. }
  16. const version = _package.default.version.split(".").reduce((v, x) => v * 1e5 + +x, 0);
  17. const versionKey = "@babel/plugin-regexp-features/version";
  18. function createRegExpFeaturePlugin({
  19. name,
  20. feature,
  21. options = {}
  22. }) {
  23. return {
  24. name,
  25. pre() {
  26. var _file$get;
  27. const {
  28. file
  29. } = this;
  30. const features = (_file$get = file.get(_features.featuresKey)) != null ? _file$get : 0;
  31. let newFeatures = (0, _features.enableFeature)(features, _features.FEATURES[feature]);
  32. const {
  33. useUnicodeFlag,
  34. runtime = true
  35. } = options;
  36. if (useUnicodeFlag === false) {
  37. newFeatures = (0, _features.enableFeature)(newFeatures, _features.FEATURES.unicodeFlag);
  38. }
  39. if (newFeatures !== features) {
  40. file.set(_features.featuresKey, newFeatures);
  41. }
  42. if (!runtime) {
  43. file.set(_features.runtimeKey, false);
  44. }
  45. if (!file.has(versionKey) || file.get(versionKey) < version) {
  46. file.set(versionKey, version);
  47. }
  48. },
  49. visitor: {
  50. RegExpLiteral(path) {
  51. var _file$get2;
  52. const {
  53. node
  54. } = path;
  55. const {
  56. file
  57. } = this;
  58. const features = file.get(_features.featuresKey);
  59. const runtime = (_file$get2 = file.get(_features.runtimeKey)) != null ? _file$get2 : true;
  60. const regexpuOptions = (0, _util.generateRegexpuOptions)(node, features);
  61. if (regexpuOptions === null) {
  62. return;
  63. }
  64. const namedCaptureGroups = {};
  65. if (regexpuOptions.namedGroup) {
  66. regexpuOptions.onNamedGroup = (name, index) => {
  67. namedCaptureGroups[name] = index;
  68. };
  69. }
  70. node.pattern = (0, _regexpuCore.default)(node.pattern, node.flags, regexpuOptions);
  71. if (regexpuOptions.namedGroup && Object.keys(namedCaptureGroups).length > 0 && runtime && !isRegExpTest(path)) {
  72. const call = _core.types.callExpression(this.addHelper("wrapRegExp"), [node, _core.types.valueToNode(namedCaptureGroups)]);
  73. (0, _helperAnnotateAsPure.default)(call);
  74. path.replaceWith(call);
  75. }
  76. if ((0, _features.hasFeature)(features, _features.FEATURES.unicodeFlag)) {
  77. pullFlag(node, "u");
  78. }
  79. if ((0, _features.hasFeature)(features, _features.FEATURES.dotAllFlag)) {
  80. pullFlag(node, "s");
  81. }
  82. }
  83. }
  84. };
  85. }
  86. function isRegExpTest(path) {
  87. return path.parentPath.isMemberExpression({
  88. object: path.node,
  89. computed: false
  90. }) && path.parentPath.get("property").isIdentifier({
  91. name: "test"
  92. });
  93. }