zod-v3-to-v4 1.6.0 → 1.7.0

package/dist/convert-name-to-top-level-api.js

@@ -267,20 +267,58 @@ function convertNameToTopLevelApiAndWrapInUnion(node, options) {
     const { zodName, oldName, nameToWrap, renames } = options;
     const names = [nameToWrap];
     getCallExpressionsToConvert(node, { oldName: nameToWrap, names }).forEach((callExpression) => {
-        // Remove deprecated names from the chain
         callExpression
             .getDescendantsOfKind(SyntaxKind.PropertyAccessExpression)
             .filter((expression) => names.includes(expression.getName()))
             .forEach((expression) => {
             const parent = expression.getFirstAncestorByKind(SyntaxKind.CallExpression);
-            parent?.replaceWithText(expression.getExpression().getText());
+            if (!parent) {
+                return;
+            }
+            const text = expression.getExpression().getText();
+            // Check if the method call has arguments (e.g., .ip({ version: "v4" }))
+            const args = parent.getArguments();
+            if (args.length > 0) {
+                // Try to extract version from the arguments
+                const firstArg = args[0];
+                if (firstArg?.isKind(SyntaxKind.ObjectLiteralExpression)) {
+                    const versionProperty = firstArg.getProperties().find((prop) => {
+                        if (prop.isKind(SyntaxKind.PropertyAssignment)) {
+                            const name = prop.getName();
+                            return name === "version" || name === '"version"';
+                        }
+                        return false;
+                    });
+                    if (versionProperty?.isKind(SyntaxKind.PropertyAssignment)) {
+                        const initializer = versionProperty.getInitializer();
+                        if (initializer?.isKind(SyntaxKind.StringLiteral)) {
+                            const version = initializer.getLiteralValue();
+                            if (version === "v4" && nameToWrap === "ip") {
+                                parent.replaceWithText(`${text}.ipv4()`);
+                                return;
+                            }
+                            else if (version === "v6" && nameToWrap === "ip") {
+                                parent.replaceWithText(`${text}.ipv6()`);
+                                return;
+                            }
+                            else if (version === "v4" && nameToWrap === "cidr") {
+                                parent.replaceWithText(`${text}.cidrv4()`);
+                                return;
+                            }
+                            else if (version === "v6" && nameToWrap === "cidr") {
+                                parent.replaceWithText(`${text}.cidrv6()`);
+                                return;
+                            }
+                        }
+                    }
+                }
+            }
+            // Default behavior: create union
+            const unionText = renames
+                .map(({ name }) => `${text}.${name}()`)
+                .join(", ");
+            parent.replaceWithText(`${zodName}.union([${unionText}])`);
         });
-        // Nest the whole expression inside a union for ipv4() & ipv6()
-        const text = callExpression.getText();
-        const unionText = renames
-            .map(({ name }) => `${text}.${name}()`)
-            .join(", ");
-        callExpression?.replaceWithText(`${zodName}.union([${unionText}])`);
         convertNameToTopLevelApi(callExpression, {
             zodName,
             oldName,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zod-v3-to-v4",
-  "version": "1.6.0",
+  "version": "1.7.0",
   "description": "Migrate Zod from v3 to v4",
   "keywords": [
     "zod",