50 lines
1.7 KiB
JavaScript
50 lines
1.7 KiB
JavaScript
|
import _typeof from "./typeof.js";
|
||
|
|
import setPrototypeOf from "./setPrototypeOf.js";
|
||
|
|
import inherits from "./inherits.js";
|
||
|
|
export default function _wrapRegExp() {
|
||
|
|
_wrapRegExp = function _wrapRegExp(e, r) {
|
||
|
|
return new BabelRegExp(e, void 0, r);
|
||
|
|
};
|
||
|
|
var e = RegExp.prototype,
|
||
|
|
r = new WeakMap();
|
||
|
|
function BabelRegExp(e, t, p) {
|
||
|
|
var o = new RegExp(e, t);
|
||
|
|
return r.set(o, p || r.get(e)), setPrototypeOf(o, BabelRegExp.prototype);
|
||
|
|
}
|
||
|
|
function buildGroups(e, t) {
|
||
|
|
var p = r.get(t);
|
||
|
|
return Object.keys(p).reduce(function (r, t) {
|
||
|
|
var o = p[t];
|
||
|
|
if ("number" == typeof o) r[t] = e[o];else {
|
||
|
|
for (var i = 0; void 0 === e[o[i]] && i + 1 < o.length;) i++;
|
||
|
|
r[t] = e[o[i]];
|
||
|
|
}
|
||
|
|
return r;
|
||
|
|
}, Object.create(null));
|
||
|
|
}
|
||
|
|
return inherits(BabelRegExp, RegExp), BabelRegExp.prototype.exec = function (r) {
|
||
|
|
var t = e.exec.call(this, r);
|
||
|
|
if (t) {
|
||
|
|
t.groups = buildGroups(t, this);
|
||
|
|
var p = t.indices;
|
||
|
|
p && (p.groups = buildGroups(p, this));
|
||
|
|
}
|
||
|
|
return t;
|
||
|
|
}, BabelRegExp.prototype[Symbol.replace] = function (t, p) {
|
||
|
|
if ("string" == typeof p) {
|
||
|
|
var o = r.get(this);
|
||
|
|
return e[Symbol.replace].call(this, t, p.replace(/\$<([^>]+)>/g, function (e, r) {
|
||
|
|
var t = o[r];
|
||
|
|
return "$" + (Array.isArray(t) ? t.join("$") : t);
|
||
|
|
}));
|
||
|
|
}
|
||
|
|
if ("function" == typeof p) {
|
||
|
|
var i = this;
|
||
|
|
return e[Symbol.replace].call(this, t, function () {
|
||
|
|
var e = arguments;
|
||
|
|
return "object" != _typeof(e[e.length - 1]) && (e = [].slice.call(e)).push(buildGroups(e, i)), p.apply(this, e);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
return e[Symbol.replace].call(this, t, p);
|
||
|
|
}, _wrapRegExp.apply(this, arguments);
|
||
|
|
}
|