Merge pull request #3 from ronanru/main

feat: add array.prototype.filter

codemods/array.prototype.filter/index.js

@@ -0,0 +1,48 @@
+import jscodeshift from 'jscodeshift';
+import { removeImport } 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.filter',
+		transform: ({ file }) => {
+			const j = jscodeshift;
+			const root = j(file.source);
+			let dirtyFlag = false;
+
+			removeImport('array.prototype.filter', root, j);
+
+			root
+				.find(j.CallExpression, {
+					callee: {
+						type: 'Identifier',
+						name: 'filter',
+					},
+				})
+				.forEach((path) => {
+					const args = path.value.arguments;
+					if (args.length === 2) {
+						const [array, callback] = args;
+
+						const newExpression = j.callExpression(
+							//@ts-ignore
+							j.memberExpression(array, j.identifier('filter')),
+							[callback],
+						);
+						j(path).replaceWith(newExpression);
+						dirtyFlag = true;
+					}
+				});
+
+			return dirtyFlag ? root.toSource(options) : file.source;
+		},
+	};
+}

codemods/index.js

@@ -14,6 +14,8 @@ import isRegexp from './is-regexp/index.js';
 
 import arrayPrototypeMap from './array.prototype.map/index.js';
 
+import arrayPrototypeFilter from './array.prototype.filter/index.js';
+
 export const codemods = {
 	'is-whitespace': isWhitespace,
 	'is-array-buffer': isArrayBuffer,
@@ -23,4 +25,5 @@ export const codemods = {
 	'is-string': isString,
 	'is-regexp': isRegexp,
 	'array.prototype.map': arrayPrototypeMap,
+	'array.prototype.filter': arrayPrototypeFilter,
 };

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

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

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

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

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

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