feat: add array.prototype.forEach

codemods/array.prototype.forEach/index.js

@@ -0,0 +1,30 @@
+import jscodeshift from 'jscodeshift';
+import { transformArrayMethod } from '../shared.js';
+
+/**
+ * @typedef {import('../../types.js').Codemod} Codemod
+ * @typedef {import('../../types.js').CodemodOptions} CodemodOptions
+ */
+
+/**
+ * @param {CodemodOptions} [options]
+ * @returns {Codemod}
+ */
+export default function (options) {
+	return {
+		name: 'array.prototype.forEach',
+		transform: ({ file }) => {
+			const j = jscodeshift;
+			const root = j(file.source);
+
+      const dirty = transformArrayMethod(
+        'array.prototype.forEach',
+        'forEach',
+        root,
+        j,
+      );
+
+			return dirty ? root.toSource(options) : file.source;
+		},
+	};
+}

codemods/index.js

@@ -68,6 +68,8 @@ import arrayPrototypeFlat from './array.prototype.flat/index.js';
 
 import arrayPrototypeFlatMap from './array.prototype.flatMap/index.js';
 
+import arrayPrototypeForEach from './array.prototype.forEach/index.js';
+
 export const codemods = {
     'is-whitespace': isWhitespace,
     'is-array-buffer': isArrayBuffer,
@@ -104,4 +106,5 @@ export const codemods = {
     'array.prototype.findIndex': arrayPrototypeFindIndex,
     'array.prototype.flat': arrayPrototypeFlat,
     'array.prototype.flatMap': arrayPrototypeFlatMap,
+    'array.prototype.forEach': arrayPrototypeForEach,
 };

test/fixtures/array.prototype.forEach/case-1/after.js

@@ -0,0 +1,14 @@
+var assert = require('assert');
+
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x >= 2;
+  }),
+  [2, 3],
+);
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x <= 2;
+  }),
+  [1, 2],
+);

test/fixtures/array.prototype.forEach/case-1/before.js

@@ -0,0 +1,15 @@
+var forEach = require('array.prototype.forEach');
+var assert = require('assert');
+
+assert.deepEqual(
+  forEach([1, 2, 3], function (x) {
+    return x >= 2;
+  }),
+  [2, 3],
+);
+assert.deepEqual(
+  forEach([1, 2, 3], function (x) {
+    return x <= 2;
+  }),
+  [1, 2],
+);

test/fixtures/array.prototype.forEach/case-1/result.js

@@ -0,0 +1,14 @@
+var assert = require('assert');
+
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x >= 2;
+  }),
+  [2, 3],
+);
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x <= 2;
+  }),
+  [1, 2],
+);

test/fixtures/array.prototype.forEach/case-2/after.js

@@ -0,0 +1,14 @@
+var assert = require('assert');
+
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x >= 2;
+  }),
+  [2, 3],
+);
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x <= 2;
+  }),
+  [1, 2],
+);

test/fixtures/array.prototype.forEach/case-2/before.js

@@ -0,0 +1,15 @@
+import forEach from 'array.prototype.forEach';
+var assert = require('assert');
+
+assert.deepEqual(
+  forEach([1, 2, 3], function (x) {
+    return x >= 2;
+  }),
+  [2, 3],
+);
+assert.deepEqual(
+  forEach([1, 2, 3], function (x) {
+    return x <= 2;
+  }),
+  [1, 2],
+);

test/fixtures/array.prototype.forEach/case-2/result.js

@@ -0,0 +1,14 @@
+var assert = require('assert');
+
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x >= 2;
+  }),
+  [2, 3],
+);
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x <= 2;
+  }),
+  [1, 2],
+);

test/fixtures/array.prototype.forEach/case-3/after.js

@@ -0,0 +1,14 @@
+var assert = require('assert');
+
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x >= 2;
+  }),
+  [2, 3],
+);
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x <= 2;
+  }),
+  [1, 2],
+);

test/fixtures/array.prototype.forEach/case-3/before.js

@@ -0,0 +1,15 @@
+var banana = require('array.prototype.forEach');
+var assert = require('assert');
+
+assert.deepEqual(
+  banana([1, 2, 3], function (x) {
+    return x >= 2;
+  }),
+  [2, 3],
+);
+assert.deepEqual(
+  banana([1, 2, 3], function (x) {
+    return x <= 2;
+  }),
+  [1, 2],
+);

test/fixtures/array.prototype.forEach/case-3/result.js

@@ -0,0 +1,14 @@
+var assert = require('assert');
+
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x >= 2;
+  }),
+  [2, 3],
+);
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x <= 2;
+  }),
+  [1, 2],
+);

test/fixtures/array.prototype.forEach/case-4/after.js

@@ -0,0 +1,14 @@
+var assert = require('assert');
+
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x >= 2;
+  }),
+  [2, 3],
+);
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x <= 2;
+  }),
+  [1, 2],
+);

test/fixtures/array.prototype.forEach/case-4/before.js

@@ -0,0 +1,15 @@
+import banana from 'array.prototype.forEach';
+var assert = require('assert');
+
+assert.deepEqual(
+  banana([1, 2, 3], function (x) {
+    return x >= 2;
+  }),
+  [2, 3],
+);
+assert.deepEqual(
+  banana([1, 2, 3], function (x) {
+    return x <= 2;
+  }),
+  [1, 2],
+);

test/fixtures/array.prototype.forEach/case-4/result.js

@@ -0,0 +1,14 @@
+var assert = require('assert');
+
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x >= 2;
+  }),
+  [2, 3],
+);
+assert.deepEqual(
+  [1, 2, 3].forEach(function (x) {
+    return x <= 2;
+  }),
+  [1, 2],
+);