feat: add invariant codemod

codemods/invariant/index.js

@@ -0,0 +1,47 @@
+import jscodeshift from 'jscodeshift';
+
+/**
+ * @typedef {import('../../types.js').Codemod} Codemod
+ * @typedef {import('../../types.js').CodemodOptions} CodemodOptions
+ */
+
+/**
+ * @param {CodemodOptions} [options]
+ * @returns {Codemod}
+ */
+export default function (options) {
+	return {
+		name: 'invariant',
+		transform: ({ file }) => {
+			const j = jscodeshift;
+			const root = j(file.source);
+
+			// Transform import declarationss
+			// biome-ignore lint/complexity/noForEach: Need to iterate through statements.
+			root.find(j.ImportDeclaration).forEach((path) => {
+				if (
+					j.Literal.check(path.node.source) &&
+					path.node.source.value === 'invariant'
+				) {
+					path.node.source.value = 'tiny-invariant';
+				}
+			});
+
+			// Transform require statements
+			// biome-ignore lint/complexity/noForEach: Need to iterate through statements.
+			root
+				.find(j.CallExpression, { callee: { name: 'require' } })
+				.forEach((path) => {
+					if (
+						path.node.arguments.length === 1 &&
+						j.Literal.check(path.node.arguments[0]) &&
+						path.node.arguments[0].value === 'invariant'
+					) {
+						path.node.arguments[0].value = 'tiny-invariant';
+					}
+				});
+
+			return root.toSource();
+		},
+	};
+}

index.js

@@ -68,6 +68,7 @@ import hasSymbols from './codemods/has-symbols/index.js';
 import hasTostringtag from './codemods/has-tostringtag/index.js';
 import hasown from './codemods/hasown/index.js';
 import indexOf from './codemods/index-of/index.js';
+import invariant from './codemods/invariant/index.js';
 import isArrayBuffer from './codemods/is-array-buffer/index.js';
 import isBooleanObject from './codemods/is-boolean-object/index.js';
 import isBuiltinModule from './codemods/is-builtin-module/index.js';
@@ -235,6 +236,7 @@ export const codemods = {
   "has-tostringtag": hasTostringtag,
   "hasown": hasown,
   "index-of": indexOf,
+  "invariant": invariant,
   "is-array-buffer": isArrayBuffer,
   "is-boolean-object": isBooleanObject,
   "is-builtin-module": isBuiltinModule,
@@ -321,4 +323,4 @@ export const codemods = {
   "typed-array-length": typedArrayLength,
   "typedarray.prototype.slice": typedarrayPrototypeSlice,
   "xtend": xtend,
-};
+};

test/codemod.test.js

@@ -1,6 +1,6 @@
 import { describe, it } from 'node:test';
 import fs from 'node:fs';
-import assert from 'assert';
+import assert from 'node:assert';
 import { codemods } from '../index.js';
 
 describe('codemod', async () => {

test/fixtures/invariant/cjs/after.js

@@ -0,0 +1,7 @@
+const invariant = require("tiny-invariant");
+
+const a = 10;
+
+invariant(a === 10, "a is 10");
+
+invariant(a === 20, "a is 20");

test/fixtures/invariant/cjs/before.js

@@ -0,0 +1,7 @@
+const invariant = require('invariant');
+
+const a = 10;
+
+invariant(a === 10, "a is 10");
+
+invariant(a === 20, "a is 20");

test/fixtures/invariant/cjs/result.js

@@ -0,0 +1,7 @@
+const invariant = require("tiny-invariant");
+
+const a = 10;
+
+invariant(a === 10, "a is 10");
+
+invariant(a === 20, "a is 20");

test/fixtures/invariant/esm-default/after.js

@@ -0,0 +1,7 @@
+import invariant from "tiny-invariant";
+
+const a = 10;
+
+invariant(a === 10, "a is 10");
+
+invariant(a === 20, "a is 20");

test/fixtures/invariant/esm-default/before.js

@@ -0,0 +1,7 @@
+import invariant from "invariant";
+
+const a = 10;
+
+invariant(a === 10, "a is 10");
+
+invariant(a === 20, "a is 20");

test/fixtures/invariant/esm-default/result.js

@@ -0,0 +1,7 @@
+import invariant from "tiny-invariant";
+
+const a = 10;
+
+invariant(a === 10, "a is 10");
+
+invariant(a === 20, "a is 20");

test/fixtures/invariant/esm/after.js

@@ -0,0 +1,7 @@
+import { invariant } from "tiny-invariant";
+
+const a = 10;
+
+invariant(a === 10, "a is 10");
+
+invariant(a === 20, "a is 20");

test/fixtures/invariant/esm/before.js

@@ -0,0 +1,7 @@
+import { invariant } from "invariant";
+
+const a = 10;
+
+invariant(a === 10, "a is 10");
+
+invariant(a === 20, "a is 20");

test/fixtures/invariant/esm/result.js

@@ -0,0 +1,7 @@
+import { invariant } from "tiny-invariant";
+
+const a = 10;
+
+invariant(a === 10, "a is 10");
+
+invariant(a === 20, "a is 20");