UNPKG

2.87 kBJavaScriptView Raw
1'use strict';
2var $ = require('../internals/export');
3var call = require('../internals/function-call');
4var uncurryThis = require('../internals/function-uncurry-this');
5var requireObjectCoercible = require('../internals/require-object-coercible');
6var isCallable = require('../internals/is-callable');
7var isNullOrUndefined = require('../internals/is-null-or-undefined');
8var isRegExp = require('../internals/is-regexp');
9var toString = require('../internals/to-string');
10var getMethod = require('../internals/get-method');
11var getRegExpFlags = require('../internals/regexp-get-flags');
12var getSubstitution = require('../internals/get-substitution');
13var wellKnownSymbol = require('../internals/well-known-symbol');
14var IS_PURE = require('../internals/is-pure');
15
16var REPLACE = wellKnownSymbol('replace');
17var $TypeError = TypeError;
18var indexOf = uncurryThis(''.indexOf);
19var replace = uncurryThis(''.replace);
20var stringSlice = uncurryThis(''.slice);
21var max = Math.max;
22
23// `String.prototype.replaceAll` method
24// https://tc39.es/ecma262/#sec-string.prototype.replaceall
25$({ target: 'String', proto: true }, {
26 replaceAll: function replaceAll(searchValue, replaceValue) {
27 var O = requireObjectCoercible(this);
28 var IS_REG_EXP, flags, replacer, string, searchString, functionalReplace, searchLength, advanceBy, replacement;
29 var position = 0;
30 var endOfLastMatch = 0;
31 var result = '';
32 if (!isNullOrUndefined(searchValue)) {
33 IS_REG_EXP = isRegExp(searchValue);
34 if (IS_REG_EXP) {
35 flags = toString(requireObjectCoercible(getRegExpFlags(searchValue)));
36 if (!~indexOf(flags, 'g')) throw new $TypeError('`.replaceAll` does not allow non-global regexes');
37 }
38 replacer = getMethod(searchValue, REPLACE);
39 if (replacer) {
40 return call(replacer, searchValue, O, replaceValue);
41 } else if (IS_PURE && IS_REG_EXP) {
42 return replace(toString(O), searchValue, replaceValue);
43 }
44 }
45 string = toString(O);
46 searchString = toString(searchValue);
47 functionalReplace = isCallable(replaceValue);
48 if (!functionalReplace) replaceValue = toString(replaceValue);
49 searchLength = searchString.length;
50 advanceBy = max(1, searchLength);
51 position = indexOf(string, searchString);
52 while (position !== -1) {
53 replacement = functionalReplace
54 ? toString(replaceValue(searchString, position, string))
55 : getSubstitution(searchString, string, position, [], undefined, replaceValue);
56 result += stringSlice(string, endOfLastMatch, position) + replacement;
57 endOfLastMatch = position + searchLength;
58 position = position + advanceBy > string.length ? -1 : indexOf(string, searchString, position + advanceBy);
59 }
60 if (endOfLastMatch < string.length) {
61 result += stringSlice(string, endOfLastMatch);
62 }
63 return result;
64 }
65});