feat: add additional array prototype methods (#4309)

* feat: add additional array prototype methods * add string prototype toString, toLocalString (not inherited from object) Co-authored-by: Lukas Taegert-Atkinson <lukastaegert@users.noreply.github.com>
rollup · Dec 23, 2021 · 473568c · 473568c
1 parent e17a379
commit 473568c
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 2 deletions.
diff --git a/src/ast/nodes/shared/ArrayPrototype.ts b/src/ast/nodes/shared/ArrayPrototype.ts
@@ -128,6 +128,8 @@ export const ARRAY_PROTOTYPE = new ObjectEntity(
 		filter: METHOD_CALLS_ARG_DEOPTS_SELF_RETURNS_NEW_ARRAY,
 		find: METHOD_CALLS_ARG_DEOPTS_SELF_RETURNS_UNKNOWN,
 		findIndex: METHOD_CALLS_ARG_DEOPTS_SELF_RETURNS_NUMBER,
+		flat: METHOD_DEOPTS_SELF_RETURNS_NEW_ARRAY,
+		flatMap: METHOD_CALLS_ARG_DEOPTS_SELF_RETURNS_NEW_ARRAY,
 		forEach: METHOD_CALLS_ARG_DEOPTS_SELF_RETURNS_UNKNOWN,
 		includes: METHOD_RETURNS_BOOLEAN,
 		indexOf: METHOD_RETURNS_NUMBER,
@@ -145,6 +147,8 @@ export const ARRAY_PROTOTYPE = new ObjectEntity(
 		some: METHOD_CALLS_ARG_DEOPTS_SELF_RETURNS_BOOLEAN,
 		sort: METHOD_CALLS_ARG_MUTATES_SELF_RETURNS_SELF,
 		splice: METHOD_MUTATES_SELF_RETURNS_NEW_ARRAY,
+		toLocaleString: METHOD_RETURNS_STRING,
+		toString: METHOD_RETURNS_STRING,
 		unshift: METHOD_MUTATES_SELF_RETURNS_NUMBER,
 		values: METHOD_DEOPTS_SELF_RETURNS_UNKNOWN
 	} as unknown as PropertyMap,

diff --git a/test/form/samples/builtin-prototypes/array-expression/_expected.js b/test/form/samples/builtin-prototypes/array-expression/_expected.js
@@ -39,6 +39,10 @@ _findArray[0].effect();
 const _findIndexArray = [{ effect() {} }];
 _findIndexArray.findIndex(element => (element.effect = () => console.log(1)));
 _findIndexArray[0].effect();
+[1].flatMap(() => console.log(1) || 1);
+const _flatMapArray = [{ effect() {} }];
+_flatMapArray.flatMap(element => (element.effect = () => console.log(1)));
+_flatMapArray[0].effect();
 [1].forEach(() => console.log(1) || true);
 const _forEachArray = [{ effect() {} }];
 _forEachArray.forEach(element => (element.effect = () => console.log(1)));

diff --git a/test/form/samples/builtin-prototypes/array-expression/main.js b/test/form/samples/builtin-prototypes/array-expression/main.js
@@ -37,6 +37,8 @@ const _entriesArray = [{ effect() {} }];
 [..._entriesArray.entries()][0][1].effect = () => console.log(1);
 _entriesArray[0].effect();
 
+const _flat = [].flat().join(',').trim();
+const _flat2 = [].flat(1).join(',').trim();
 const _includes = [].includes(1).valueOf();
 const _indexOf = [].indexOf(1).toPrecision(1);
 const _join = [].join(',').trim();
@@ -78,6 +80,12 @@ const _findIndexArray = [{ effect() {} }];
 _findIndexArray.findIndex(element => (element.effect = () => console.log(1)));
 _findIndexArray[0].effect();
 
+const _flatMap = [1].flatMap(() => 1).join(',');
+const _flatMapEffect = [1].flatMap(() => console.log(1) || 1);
+const _flatMapArray = [{ effect() {} }];
+_flatMapArray.flatMap(element => (element.effect = () => console.log(1)));
+_flatMapArray[0].effect();
+
 const _forEach = [1].forEach(() => {});
 const _forEachEffect = [1].forEach(() => console.log(1) || true);
 const _forEachArray = [{ effect() {} }];
@@ -131,10 +139,11 @@ exported.splice(0);
 const _unshift = [1].unshift(0).toPrecision(1);
 exported.unshift(0);
 
+const _toLocaleString = [1].toLocaleString().trim();
+const _toString = [1].toString().trim();
+
 // inherited
 const _hasOwnProperty = [1].hasOwnProperty('toString').valueOf();
 const _isPrototypeOf = [1].isPrototypeOf([]).valueOf();
 const _propertyIsEnumerable = [1].propertyIsEnumerable('toString').valueOf();
-const _toLocaleString = [1].toLocaleString().trim();
-const _toString = [1].toString().trim();
 const _valueOf = [1].valueOf();