Merge pull request #46 from babs20/feature/setprototypeof

feat: add `setprototypeof`

Author: thepassle

codemods/setprototypeof/index.js

@@ -0,0 +1,45 @@
+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: 'setprototypeof',
+		transform: ({ file }) => {
+			const j = jscodeshift;
+			const root = j(file.source);
+			let dirtyFlag = false;
+
+			const { identifier } = removeImport('setprototypeof', root, j);
+
+			if (identifier) {
+				const callExpressions = root.find(j.CallExpression, {
+					callee: { name: identifier },
+				});
+
+				for (const path of callExpressions.paths()) {
+					j(path).replaceWith(
+						j.callExpression(
+							j.memberExpression(
+								j.identifier('Object'),
+								j.identifier('setPrototypeOf'),
+							),
+							path.node.arguments,
+						),
+					);
+					dirtyFlag = true;
+				}
+			}
+
+			return dirtyFlag ? root.toSource(options) : file.source;
+		},
+	};
+}

test/fixtures/setprototypeof/case-1/after.js

@@ -0,0 +1,24 @@
+var assert = require('assert');
+
+function Foo() { }
+function Bar() { }
+
+Object.setPrototypeOf(Foo.prototype, Bar.prototype);
+
+const obj = {};
+const proto = {};
+Object.setPrototypeOf(obj, proto);
+
+function testFunction() {
+	const innerObj = {};
+	const innerProto = {};
+	Object.setPrototypeOf(innerObj, innerProto);
+}
+
+const chainExample = Object.setPrototypeOf({}, {}).prop;
+
+Object.setPrototypeOf();
+Object.setPrototypeOf({}, {}, 'extra');
+
+assert.strictEqual(Object.getPrototypeOf(Foo.prototype), Bar.prototype);
+assert.strictEqual(Object.getPrototypeOf(obj), proto);

test/fixtures/setprototypeof/case-1/before.js

@@ -0,0 +1,25 @@
+var setPrototypeOf = require('setprototypeof');
+var assert = require('assert');
+
+function Foo() { }
+function Bar() { }
+
+setPrototypeOf(Foo.prototype, Bar.prototype);
+
+const obj = {};
+const proto = {};
+setPrototypeOf(obj, proto);
+
+function testFunction() {
+	const innerObj = {};
+	const innerProto = {};
+	setPrototypeOf(innerObj, innerProto);
+}
+
+const chainExample = setPrototypeOf({}, {}).prop;
+
+setPrototypeOf();
+setPrototypeOf({}, {}, 'extra');
+
+assert.strictEqual(Object.getPrototypeOf(Foo.prototype), Bar.prototype);
+assert.strictEqual(Object.getPrototypeOf(obj), proto);

test/fixtures/setprototypeof/case-1/result.js

@@ -0,0 +1,24 @@
+var assert = require('assert');
+
+function Foo() { }
+function Bar() { }
+
+Object.setPrototypeOf(Foo.prototype, Bar.prototype);
+
+const obj = {};
+const proto = {};
+Object.setPrototypeOf(obj, proto);
+
+function testFunction() {
+	const innerObj = {};
+	const innerProto = {};
+	Object.setPrototypeOf(innerObj, innerProto);
+}
+
+const chainExample = Object.setPrototypeOf({}, {}).prop;
+
+Object.setPrototypeOf();
+Object.setPrototypeOf({}, {}, 'extra');
+
+assert.strictEqual(Object.getPrototypeOf(Foo.prototype), Bar.prototype);
+assert.strictEqual(Object.getPrototypeOf(obj), proto);

test/fixtures/setprototypeof/case-2/after.js

@@ -0,0 +1,24 @@
+import assert from 'assert';
+
+function Foo() { }
+function Bar() { }
+
+Object.setPrototypeOf(Foo.prototype, Bar.prototype);
+
+const obj = {};
+const proto = {};
+Object.setPrototypeOf(obj, proto);
+
+function testFunction() {
+	const innerObj = {};
+	const innerProto = {};
+	Object.setPrototypeOf(innerObj, innerProto);
+}
+
+const chainExample = Object.setPrototypeOf({}, {}).prop;
+
+Object.setPrototypeOf();
+Object.setPrototypeOf({}, {}, 'extra');
+
+assert.strictEqual(Object.getPrototypeOf(Foo.prototype), Bar.prototype);
+assert.strictEqual(Object.getPrototypeOf(obj), proto);

test/fixtures/setprototypeof/case-2/before.js

@@ -0,0 +1,25 @@
+import setprototypeof from 'setprototypeof';
+import assert from 'assert';
+
+function Foo() { }
+function Bar() { }
+
+setprototypeof(Foo.prototype, Bar.prototype);
+
+const obj = {};
+const proto = {};
+setprototypeof(obj, proto);
+
+function testFunction() {
+	const innerObj = {};
+	const innerProto = {};
+	setprototypeof(innerObj, innerProto);
+}
+
+const chainExample = setprototypeof({}, {}).prop;
+
+setprototypeof();
+setprototypeof({}, {}, 'extra');
+
+assert.strictEqual(Object.getPrototypeOf(Foo.prototype), Bar.prototype);
+assert.strictEqual(Object.getPrototypeOf(obj), proto);

test/fixtures/setprototypeof/case-2/result.js

@@ -0,0 +1,24 @@
+import assert from 'assert';
+
+function Foo() { }
+function Bar() { }
+
+Object.setPrototypeOf(Foo.prototype, Bar.prototype);
+
+const obj = {};
+const proto = {};
+Object.setPrototypeOf(obj, proto);
+
+function testFunction() {
+	const innerObj = {};
+	const innerProto = {};
+	Object.setPrototypeOf(innerObj, innerProto);
+}
+
+const chainExample = Object.setPrototypeOf({}, {}).prop;
+
+Object.setPrototypeOf();
+Object.setPrototypeOf({}, {}, 'extra');
+
+assert.strictEqual(Object.getPrototypeOf(Foo.prototype), Bar.prototype);
+assert.strictEqual(Object.getPrototypeOf(obj), proto);