feat(util/update-binding): add new utility to update binding in the file

Author: brunocroh

utils/src/ast-grep/remove-binding.ts

@@ -1,12 +1,5 @@
 import type { SgNode, Edit, Range, Kinds, TypesMap } from "@codemod.com/jssg-types/main";
-
-const requireKinds = ["lexical_declaration", "variable_declarator"];
-const importKinds = ["import_statement", "import_clause"];
-
-type RemoveBindingReturnType = {
-	edit?: Edit;
-	lineToRemove?: Range;
-};
+import { updateBinding } from "./update-binding.ts";
 
 /**
  * Removes a specific binding from an import or require statement.
@@ -40,165 +33,8 @@ type RemoveBindingReturnType = {
  * // Returns: undefined (no destructured binding found)
  * ```
  */
-export function removeBinding(
-	node: SgNode<TypesMap, Kinds<TypesMap>>,
-	binding: string,
-): RemoveBindingReturnType {
-	const nodeKind = node.kind().toString();
-
-	const identifier = node.find({
-		rule: {
-			any: [
-				{
-					kind: "identifier",
-					inside: {
-						kind: "variable_declarator",
-					},
-				},
-				{
-					kind: "identifier",
-					inside: {
-						kind: "import_clause",
-					},
-				},
-			],
-		},
-	});
-
-	if (identifier && identifier.text() === binding) {
-		return {
-			lineToRemove: node.range(),
-		};
-	}
-
-	if (requireKinds.includes(nodeKind)) {
-		return handleNamedRequireBindings(node, binding);
-	}
-
-	if (importKinds.includes(nodeKind)) {
-		return handleNamedImportBindings(node, binding);
-	}
-}
-
-function handleNamedImportBindings(
-	node: SgNode<TypesMap, Kinds<TypesMap>>,
-	binding: string,
-): RemoveBindingReturnType {
-	const namespaceImport = node.find({
-		rule: {
-			kind: "identifier",
-			inside: {
-				kind: "namespace_import",
-			},
-		},
-	});
-
-	if (Boolean(namespaceImport) && namespaceImport.text() === binding) {
-		return {
-			lineToRemove: node.range(),
-		};
-	}
-
-	const namedImports = node.findAll({
-		rule: {
-			kind: "import_specifier",
-			// ignore imports with alias (renamed imports)
-			not: {
-				has: {
-					field: "alias",
-					kind: "identifier",
-				},
-			},
-		},
+export function removeBinding(node: SgNode<TypesMap, Kinds<TypesMap>>, binding: string) {
+	return updateBinding(node, binding, {
+		newBinding: undefined,
 	});
-
-	for (const namedImport of namedImports) {
-		const text = namedImport.text();
-		if (text === binding) {
-			if (namedImports.length === 1) {
-				return {
-					lineToRemove: node.range(),
-				};
-			}
-
-			const namedImportsNode = node.find({
-				rule: {
-					kind: "named_imports",
-				},
-			});
-			const restNamedImports = namedImports.map((d) => d.text()).filter((d) => d !== binding);
-
-			return {
-				edit: namedImportsNode.replace(`{ ${restNamedImports.join(", ")} }`),
-			};
-		}
-	}
-
-	const renamedImports = node.findAll({
-		rule: {
-			has: {
-				field: "alias",
-				kind: "identifier",
-			},
-		},
-	});
-
-	for (const renamedImport of renamedImports) {
-		if (renamedImport.text() === binding) {
-			if (renamedImports.length === 1 && namedImports.length === 0) {
-				return {
-					lineToRemove: node.range(),
-				};
-			}
-
-			const namedImportsNode = node.find({
-				rule: {
-					kind: "named_imports",
-				},
-			});
-
-			const aliasStatement = renamedImports.map((alias) => alias.parent());
-
-			const restNamedImports = [...namedImports, ...aliasStatement]
-				.map((d) => d.text())
-				.filter((d) => d !== renamedImport.parent().text());
-
-			return {
-				edit: namedImportsNode.replace(`{ ${restNamedImports.join(", ")} }`),
-			};
-		}
-	}
-}
-
-function handleNamedRequireBindings(
-	node: SgNode<TypesMap, Kinds<TypesMap>>,
-	binding: string,
-): RemoveBindingReturnType {
-	const objectPattern = node.find({
-		rule: {
-			kind: "object_pattern",
-		},
-	});
-
-	if (!objectPattern) return;
-
-	const declarations = node.findAll({
-		rule: {
-			kind: "shorthand_property_identifier_pattern",
-		},
-	});
-
-	if (declarations.length === 1) {
-		return {
-			lineToRemove: node.range(),
-		};
-	}
-
-	if (declarations.length > 1) {
-		const restDeclarations = declarations.map((d) => d.text()).filter((d) => d !== binding);
-
-		return {
-			edit: objectPattern.replace(`{ ${restDeclarations.join(", ")} }`),
-		};
-	}
 }