zod-v3-to-v4 1.6.1 → 1.7.0

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

@@ -272,11 +272,52 @@ function convertNameToTopLevelApiAndWrapInUnion(node, options) {
             .filter((expression) => names.includes(expression.getName()))
             .forEach((expression) => {
             const parent = expression.getFirstAncestorByKind(SyntaxKind.CallExpression);
+            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}])`);
+            parent.replaceWithText(`${zodName}.union([${unionText}])`);
         });
         convertNameToTopLevelApi(callExpression, {
             zodName,

package/package.json

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