es.string.match-all.js 4.1 KB

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