es.string.match-all.js 4.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  1. 'use strict';
  2. /* eslint-disable es-x/no-string-prototype-matchall -- safe */
  3. var $ = require('../internals/export');
  4. var global = require('../internals/global');
  5. var call = require('../internals/function-call');
  6. var uncurryThis = require('../internals/function-uncurry-this');
  7. var createIteratorConstructor = require('../internals/create-iterator-constructor');
  8. var requireObjectCoercible = require('../internals/require-object-coercible');
  9. var toLength = require('../internals/to-length');
  10. var toString = require('../internals/to-string');
  11. var anObject = require('../internals/an-object');
  12. var classof = require('../internals/classof-raw');
  13. var isRegExp = require('../internals/is-regexp');
  14. var getRegExpFlags = require('../internals/regexp-get-flags');
  15. var getMethod = require('../internals/get-method');
  16. var redefine = require('../internals/redefine');
  17. var fails = require('../internals/fails');
  18. var wellKnownSymbol = require('../internals/well-known-symbol');
  19. var speciesConstructor = require('../internals/species-constructor');
  20. var advanceStringIndex = require('../internals/advance-string-index');
  21. var regExpExec = require('../internals/regexp-exec-abstract');
  22. var InternalStateModule = require('../internals/internal-state');
  23. var IS_PURE = require('../internals/is-pure');
  24. var MATCH_ALL = wellKnownSymbol('matchAll');
  25. var REGEXP_STRING = 'RegExp String';
  26. var REGEXP_STRING_ITERATOR = REGEXP_STRING + ' Iterator';
  27. var setInternalState = InternalStateModule.set;
  28. var getInternalState = InternalStateModule.getterFor(REGEXP_STRING_ITERATOR);
  29. var RegExpPrototype = RegExp.prototype;
  30. var TypeError = global.TypeError;
  31. var stringIndexOf = uncurryThis(''.indexOf);
  32. var un$MatchAll = uncurryThis(''.matchAll);
  33. var WORKS_WITH_NON_GLOBAL_REGEX = !!un$MatchAll && !fails(function () {
  34. un$MatchAll('a', /./);
  35. });
  36. var $RegExpStringIterator = createIteratorConstructor(function RegExpStringIterator(regexp, string, $global, fullUnicode) {
  37. setInternalState(this, {
  38. type: REGEXP_STRING_ITERATOR,
  39. regexp: regexp,
  40. string: string,
  41. global: $global,
  42. unicode: fullUnicode,
  43. done: false
  44. });
  45. }, REGEXP_STRING, function next() {
  46. var state = getInternalState(this);
  47. if (state.done) return { value: undefined, done: true };
  48. var R = state.regexp;
  49. var S = state.string;
  50. var match = regExpExec(R, S);
  51. if (match === null) return { value: undefined, done: state.done = true };
  52. if (state.global) {
  53. if (toString(match[0]) === '') R.lastIndex = advanceStringIndex(S, toLength(R.lastIndex), state.unicode);
  54. return { value: match, done: false };
  55. }
  56. state.done = true;
  57. return { value: match, done: false };
  58. });
  59. var $matchAll = function (string) {
  60. var R = anObject(this);
  61. var S = toString(string);
  62. var C = speciesConstructor(R, RegExp);
  63. var flags = toString(getRegExpFlags(R));
  64. var matcher, $global, fullUnicode;
  65. matcher = new C(C === RegExp ? R.source : R, flags);
  66. $global = !!~stringIndexOf(flags, 'g');
  67. fullUnicode = !!~stringIndexOf(flags, 'u');
  68. matcher.lastIndex = toLength(R.lastIndex);
  69. return new $RegExpStringIterator(matcher, S, $global, fullUnicode);
  70. };
  71. // `String.prototype.matchAll` method
  72. // https://tc39.es/ecma262/#sec-string.prototype.matchall
  73. $({ target: 'String', proto: true, forced: WORKS_WITH_NON_GLOBAL_REGEX }, {
  74. matchAll: function matchAll(regexp) {
  75. var O = requireObjectCoercible(this);
  76. var flags, S, matcher, rx;
  77. if (regexp != null) {
  78. if (isRegExp(regexp)) {
  79. flags = toString(requireObjectCoercible(getRegExpFlags(regexp)));
  80. if (!~stringIndexOf(flags, 'g')) throw TypeError('`.matchAll` does not allow non-global regexes');
  81. }
  82. if (WORKS_WITH_NON_GLOBAL_REGEX) return un$MatchAll(O, regexp);
  83. matcher = getMethod(regexp, MATCH_ALL);
  84. if (matcher === undefined && IS_PURE && classof(regexp) == 'RegExp') matcher = $matchAll;
  85. if (matcher) return call(matcher, regexp, O);
  86. } else if (WORKS_WITH_NON_GLOBAL_REGEX) return un$MatchAll(O, regexp);
  87. S = toString(O);
  88. rx = new RegExp(regexp, 'g');
  89. return IS_PURE ? call($matchAll, rx, S) : rx[MATCH_ALL](S);
  90. }
  91. });
  92. IS_PURE || MATCH_ALL in RegExpPrototype || redefine(RegExpPrototype, MATCH_ALL, $matchAll);