Merge pull request #4 from ronanru/main

feat: add array-includes

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

codemods/index.js

@@ -16,6 +16,8 @@ import arrayPrototypeMap from './array.prototype.map/index.js';
 
 import arrayPrototypeFilter from './array.prototype.filter/index.js';
 
+import arrayIncludes from './array-includes/index.js';
+
 export const codemods = {
 	'is-whitespace': isWhitespace,
 	'is-array-buffer': isArrayBuffer,
@@ -26,4 +28,5 @@ export const codemods = {
 	'is-regexp': isRegexp,
 	'array.prototype.map': arrayPrototypeMap,
 	'array.prototype.filter': arrayPrototypeFilter,
+	'array-includes': arrayIncludes,
 };

test/fixtures/array-includes/case-1/after.js

@@ -0,0 +1,13 @@
+var assert = require("assert");
+var arr = ["one", "two"];
+
+var arr = [1, "foo", NaN, -0];
+
+assert.equal(arr.includes(0), true);
+assert.equal(arr.includes(-0), true);
+
+assert.equal(arr.includes(NaN), true);
+
+assert.equal(arr.includes("foo", 0), true);
+assert.equal(arr.includes("foo", 1), true);
+assert.equal(arr.includes("foo", 2), false);

test/fixtures/array-includes/case-1/before.js

@@ -0,0 +1,14 @@
+var includes = require("array-includes");
+var assert = require("assert");
+var arr = ["one", "two"];
+
+var arr = [1, "foo", NaN, -0];
+
+assert.equal(includes(arr, 0), true);
+assert.equal(includes(arr, -0), true);
+
+assert.equal(includes(arr, NaN), true);
+
+assert.equal(includes(arr, "foo", 0), true);
+assert.equal(includes(arr, "foo", 1), true);
+assert.equal(includes(arr, "foo", 2), false);

test/fixtures/array-includes/case-1/result.js

@@ -0,0 +1,13 @@
+var assert = require("assert");
+var arr = ["one", "two"];
+
+var arr = [1, "foo", NaN, -0];
+
+assert.equal(arr.includes(0), true);
+assert.equal(arr.includes(-0), true);
+
+assert.equal(arr.includes(NaN), true);
+
+assert.equal(arr.includes("foo", 0), true);
+assert.equal(arr.includes("foo", 1), true);
+assert.equal(arr.includes("foo", 2), false);