Merge pull request #7 from ajkl2533/array-prototype-unshift

Add codemod to replace `array.prototype.unshift`

codemods/array.prototype.unshift/index.js

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

codemods/index.js

@@ -20,16 +20,19 @@ import arrayIncludes from './array-includes/index.js';
 
 import objectKeys from './object-keys/index.js';
 
+import arrayPrototypeUnshift from './array.prototype.unshift/index.js';
+
 export const codemods = {
-	'is-whitespace': isWhitespace,
-	'is-array-buffer': isArrayBuffer,
-	'is-boolean-object': isBooleanObject,
-	'is-date-object': isDateObject,
-	'is-number-object': isNumberObject,
-	'is-string': isString,
-	'is-regexp': isRegexp,
-	'array.prototype.map': arrayPrototypeMap,
-	'array.prototype.filter': arrayPrototypeFilter,
-	'array-includes': arrayIncludes,
-	'object-keys': objectKeys,
+    'is-whitespace': isWhitespace,
+    'is-array-buffer': isArrayBuffer,
+    'is-boolean-object': isBooleanObject,
+    'is-date-object': isDateObject,
+    'is-number-object': isNumberObject,
+    'is-string': isString,
+    'is-regexp': isRegexp,
+    'array.prototype.map': arrayPrototypeMap,
+    'array.prototype.filter': arrayPrototypeFilter,
+    'array-includes': arrayIncludes,
+    'object-keys': objectKeys,
+    'array.prototype.unshift': arrayPrototypeUnshift
 };

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

@@ -0,0 +1,5 @@
+var assert = require('assert');
+
+var a = [1, 1, 1];
+assert.deepEqual(a.unshift(1, 2), 5);
+assert.deepEqual(a, [1, 2, 1, 1, 1]);

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

@@ -0,0 +1,6 @@
+var unshift = require('array.prototype.unshift');
+var assert = require('assert');
+
+var a = [1, 1, 1];
+assert.deepEqual(unshift(a, 1, 2), 5);
+assert.deepEqual(a, [1, 2, 1, 1, 1]);

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

@@ -0,0 +1,5 @@
+var assert = require('assert');
+
+var a = [1, 1, 1];
+assert.deepEqual(a.unshift(1, 2), 5);
+assert.deepEqual(a, [1, 2, 1, 1, 1]);