feat: add strip-ansi codemod

codemods/strip-ansi/index.js

@@ -0,0 +1,55 @@
+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: 'strip-ansi',
+		transform: ({ file }) => {
+			const j = jscodeshift;
+			const root = j(file.source);
+			let dirtyFlag = false;
+
+			const importDeclarations = root.find(j.ImportDeclaration, {
+				source: {
+					value: 'strip-ansi',
+				},
+			});
+
+			let importName = 'stripAnsi';
+
+			for (const declaration of importDeclarations.nodes()) {
+				const name = declaration.specifiers?.[0].local?.name;
+				if (name) {
+					importName = name;
+				}
+				declaration.source.value = 'node:util';
+				declaration.specifiers = [j.importSpecifier(j.identifier('stripVTControlCharacters'))];
+				dirtyFlag = true;
+			}
+
+			const callExpression = root.find(j.CallExpression, {
+				callee: {
+					type: 'Identifier',
+					name: importName,
+				},
+			});
+
+			for (const expression of callExpression.nodes()) {
+				if (expression.callee.type === 'Identifier') {
+					expression.callee.name = 'stripVTControlCharacters';
+					dirtyFlag = true;
+				}
+			}
+
+			return dirtyFlag ? root.toSource() : file.source;
+		},
+	};
+}

index.js

@@ -61,11 +61,11 @@ import functionsHaveNames from './codemods/functions-have-names/index.js';
 import getSymbolDescription from './codemods/get-symbol-description/index.js';
 import global from './codemods/global/index.js';
 import gopd from './codemods/gopd/index.js';
-import has from './codemods/has/index.js';
 import hasOwnProp from './codemods/has-own-prop/index.js';
 import hasProto from './codemods/has-proto/index.js';
 import hasSymbols from './codemods/has-symbols/index.js';
 import hasTostringtag from './codemods/has-tostringtag/index.js';
+import has from './codemods/has/index.js';
 import hasown from './codemods/hasown/index.js';
 import indexOf from './codemods/index-of/index.js';
 import isArrayBuffer from './codemods/is-array-buffer/index.js';
@@ -76,8 +76,8 @@ import isEven from './codemods/is-even/index.js';
 import isNan from './codemods/is-nan/index.js';
 import isNegativeZero from './codemods/is-negative-zero/index.js';
 import isNpm from './codemods/is-npm/index.js';
-import isNumber from './codemods/is-number/index.js';
 import isNumberObject from './codemods/is-number-object/index.js';
+import isNumber from './codemods/is-number/index.js';
 import isOdd from './codemods/is-odd/index.js';
 import isPlainObject from './codemods/is-plain-object/index.js';
 import isPrimitive from './codemods/is-primitive/index.js';
@@ -96,8 +96,8 @@ import mathClz32 from './codemods/math.clz32/index.js';
 import mathF16round from './codemods/math.f16round/index.js';
 import mathFround from './codemods/math.fround/index.js';
 import mathImul from './codemods/math.imul/index.js';
-import mathLog10 from './codemods/math.log10/index.js';
 import mathLog1p from './codemods/math.log1p/index.js';
+import mathLog10 from './codemods/math.log10/index.js';
 import mathSign from './codemods/math.sign/index.js';
 import md5 from './codemods/md5/index.js';
 import numberIsfinite from './codemods/number.isfinite/index.js';
@@ -146,6 +146,7 @@ import stringPrototypeTrimleft from './codemods/string.prototype.trimleft/index.
 import stringPrototypeTrimright from './codemods/string.prototype.trimright/index.js';
 import stringPrototypeTrimstart from './codemods/string.prototype.trimstart/index.js';
 import stringRaw from './codemods/string.raw/index.js';
+import stripAnsi from './codemods/strip-ansi/index.js';
 import symbolPrototypeDescription from './codemods/symbol.prototype.description/index.js';
 import traverse from './codemods/traverse/index.js';
 import typedArrayBuffer from './codemods/typed-array-buffer/index.js';
@@ -313,6 +314,7 @@ export const codemods = {
   "string.prototype.trimright": stringPrototypeTrimright,
   "string.prototype.trimstart": stringPrototypeTrimstart,
   "string.raw": stringRaw,
+  "strip-ansi": stripAnsi,
   "symbol.prototype.description": symbolPrototypeDescription,
   "traverse": traverse,
   "typed-array-buffer": typedArrayBuffer,
@@ -321,4 +323,4 @@ export const codemods = {
   "typed-array-length": typedArrayLength,
   "typedarray.prototype.slice": typedarrayPrototypeSlice,
   "xtend": xtend,
-};
+};

test/fixtures/strip-ansi/case-1/after.js

@@ -0,0 +1,3 @@
+import { stripVTControlCharacters } from "node:util";
+
+stripVTControlCharacters('\u001B[4mUnicorn\u001B[0m');

test/fixtures/strip-ansi/case-1/before.js

@@ -0,0 +1,3 @@
+import stripAnsi from 'strip-ansi';
+
+stripAnsi('\u001B[4mUnicorn\u001B[0m');

test/fixtures/strip-ansi/case-1/result.js

@@ -0,0 +1,3 @@
+import { stripVTControlCharacters } from "node:util";
+
+stripVTControlCharacters('\u001B[4mUnicorn\u001B[0m');

test/fixtures/strip-ansi/case-2/after.js

@@ -0,0 +1,3 @@
+import { stripVTControlCharacters } from "node:util";
+
+stripVTControlCharacters('\u001B[4mUnicorn\u001B[0m');

test/fixtures/strip-ansi/case-2/before.js

@@ -0,0 +1,3 @@
+import strip from 'strip-ansi';
+
+strip('\u001B[4mUnicorn\u001B[0m');

test/fixtures/strip-ansi/case-2/result.js

@@ -0,0 +1,3 @@
+import { stripVTControlCharacters } from "node:util";
+
+stripVTControlCharacters('\u001B[4mUnicorn\u001B[0m');