zod-v3-to-v4 1.11.0 → 1.13.0

package/README.md

@@ -133,6 +133,9 @@ pnpm playground:interactive
       <td align="center" valign="top" width="14.28%"><a href="http://adam.doussan.info"><img src="https://avatars.githubusercontent.com/u/26745150?v=4?s=100" width="100px;" alt="acdoussan"/><br /><sub><b>acdoussan</b></sub></a><br /><a href="https://github.com/nicoespeon/zod-v3-to-v4/issues?q=author%3Aacdoussan" title="Bug reports">🐛</a></td>
       <td align="center" valign="top" width="14.28%"><a href="http://matsjfunke.com"><img src="https://avatars.githubusercontent.com/u/125814808?v=4?s=100" width="100px;" alt="Mats Julius Funke"/><br /><sub><b>Mats Julius Funke</b></sub></a><br /><a href="#ideas-matsjfunke" title="Ideas, Planning, & Feedback">🤔</a></td>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/popfendi"><img src="https://avatars.githubusercontent.com/u/84185235?v=4?s=100" width="100px;" alt="popfendi"/><br /><sub><b>popfendi</b></sub></a><br /><a href="https://github.com/nicoespeon/zod-v3-to-v4/issues?q=author%3Apopfendi" title="Bug reports">🐛</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="https://github.com/luis-azevedo-visma"><img src="https://avatars.githubusercontent.com/u/149402066?v=4?s=100" width="100px;" alt="Luís Azevedo"/><br /><sub><b>Luís Azevedo</b></sub></a><br /><a href="#ideas-luis-azevedo-visma" title="Ideas, Planning, & Feedback">🤔</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="https://florian-lefebvre.dev"><img src="https://avatars.githubusercontent.com/u/69633530?v=4?s=100" width="100px;" alt="Florian Lefebvre"/><br /><sub><b>Florian Lefebvre</b></sub></a><br /><a href="https://github.com/nicoespeon/zod-v3-to-v4/issues?q=author%3Aflorian-lefebvre" title="Bug reports">🐛</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="http://www.robertcooper.me"><img src="https://avatars.githubusercontent.com/u/16786990?v=4?s=100" width="100px;" alt="Robert Cooper"/><br /><sub><b>Robert Cooper</b></sub></a><br /><a href="https://github.com/nicoespeon/zod-v3-to-v4/issues?q=author%3Arobertcoopercode" title="Bug reports">🐛</a></td>
     </tr>
   </tbody>
   <tfoot>

package/dist/collect-imports.js

@@ -25,16 +25,38 @@ export function getZodImport(importDeclaration) {
         ?.getNamedImports()
         .find((namedImport) => namedImport.getName() === "z");
 }
-export function collectZodReferences(importDeclarations) {
-    return importDeclarations.flatMap((importDeclaration) => importDeclaration.getNamedImports().flatMap((namedImport) => {
-        const namedNode = namedImport.getAliasNode() ?? namedImport.getNameNode();
-        if (!namedNode.isKind(SyntaxKind.Identifier)) {
-            return [];
+export function collectZodReferences(sourceFile, importDeclarations) {
+    const identifierMap = buildIdentifierMap(sourceFile);
+    return {
+        zodReferences: importDeclarations.flatMap((importDeclaration) => importDeclaration.getNamedImports().flatMap((namedImport) => {
+            const namedNode = namedImport.getAliasNode() ?? namedImport.getNameNode();
+            if (!namedNode.isKind(SyntaxKind.Identifier)) {
+                return [];
+            }
+            const name = namedNode.getText();
+            const allReferences = identifierMap.get(name) ?? [];
+            return allReferences.filter((node) => node !== namedNode);
+        })),
+        identifierMap,
+    };
+}
+/**
+ * Build a map of identifier name -> all nodes with that name.
+ * This allows O(1) lookups instead of expensive findReferencesAsNodes() calls.
+ */
+function buildIdentifierMap(sourceFile) {
+    const identifierMap = new Map();
+    for (const identifier of sourceFile.getDescendantsOfKind(SyntaxKind.Identifier)) {
+        const name = identifier.getText();
+        const existing = identifierMap.get(name);
+        if (existing) {
+            existing.push(identifier);
+        }
+        else {
+            identifierMap.set(name, [identifier]);
         }
-        return namedNode
-            .findReferencesAsNodes()
-            .filter((node) => node !== namedNode);
-    }));
+    }
+    return identifierMap;
 }
 /**
  * Collect references to variables that are derived from Zod schemas.
@@ -47,7 +69,7 @@ export function collectZodReferences(importDeclarations) {
  * This function will return references to `baseSchema` (like the one used
  * in `constrainedSchema`), so we can also migrate those.
  */
-export function collectDerivedZodSchemaReferences(zodReferences) {
+export function collectDerivedZodSchemaReferences(zodReferences, identifierMap) {
     const derivedReferences = [];
     const visited = new Set();
     function collectFromReferences(refs) {
@@ -67,11 +89,9 @@ export function collectDerivedZodSchemaReferences(zodReferences) {
             if (!nameNode.isKind(SyntaxKind.Identifier)) {
                 continue;
             }
-            // Find all references to this variable (excluding the declaration itself)
-            const variableReferences = nameNode
-                .findReferencesAsNodes()
-                .filter((node) => node !== nameNode)
-                .filter((node) => node.isKind(SyntaxKind.Identifier));
+            // Find all references to this variable
+            const allReferences = identifierMap.get(variableName) ?? [];
+            const variableReferences = allReferences.filter((node) => node !== nameNode);
             derivedReferences.push(...variableReferences);
             // Recursively collect references from these derived schemas
             collectFromReferences(variableReferences);

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

@@ -85,6 +85,8 @@ export function convertZCoercePatternsToTopLevelApi(node, zodName) {
     });
 }
 export function convertZObjectPatternsToTopLevelApi(node, zodName) {
+    // First, convert z.object() methods like .strict() or .passthrough()
+    convertObjectMethodOnSchemaReference(node, zodName);
     convertNameToTopLevelApi(node, {
         zodName,
         oldName: "object",
@@ -98,6 +100,56 @@ export function convertZObjectPatternsToTopLevelApi(node, zodName) {
     });
     convertZObjectMergeToExtend(node);
 }
+/**
+ * @example
+ * UserSchema.pick({ id: true }).strict()
+ * // becomes
+ * z.strictObject(UserSchema.pick({ id: true }).shape)
+ */
+function convertObjectMethodOnSchemaReference(node, zodName) {
+    const methodMappings = {
+        strict: "strictObject",
+        passthrough: "looseObject",
+        strip: "object",
+        nonstrict: "object",
+    };
+    const methodNames = Object.keys(methodMappings);
+    node
+        .getDescendantsOfKind(SyntaxKind.PropertyAccessExpression)
+        // Start from the deepest to handle nested cases
+        .reverse()
+        .filter((e) => methodNames.includes(e.getName()))
+        .forEach((propertyAccess) => {
+        const methodName = propertyAccess.getName();
+        const parent = propertyAccess.getFirstAncestorByKind(SyntaxKind.CallExpression);
+        if (!parent || parent.getExpression() !== propertyAccess) {
+            return;
+        }
+        const baseExpression = propertyAccess.getExpression();
+        const baseText = baseExpression.getText();
+        const hasZObjectInChain = baseExpression
+            .getDescendantsOfKind(SyntaxKind.PropertyAccessExpression)
+            .some((e) => e.getName() === "object" && e.getExpression().getText() === zodName);
+        if (hasZObjectInChain) {
+            return;
+        }
+        if (baseExpression.isKind(SyntaxKind.CallExpression) &&
+            baseExpression
+                .getExpression()
+                .isKind(SyntaxKind.PropertyAccessExpression)) {
+            const propAccess = baseExpression.getExpression();
+            if (propAccess.isKind(SyntaxKind.PropertyAccessExpression) &&
+                propAccess.getName() === "object" &&
+                propAccess.getExpression().getText() === zodName) {
+                return;
+            }
+        }
+        // This is a schema reference with .strict(), etc.
+        // Transform it to `z.strictObject(schemaRef.shape)`
+        const newName = methodMappings[methodName];
+        parent.replaceWithText(`${zodName}.${newName}(${baseText}.shape)`);
+    });
+}
 function convertZObjectMergeToExtend(node) {
     node
         .getDescendantsOfKind(SyntaxKind.PropertyAccessExpression)

package/dist/convert-zod-errors.js

@@ -1,5 +1,41 @@
 import { SyntaxKind, } from "ts-morph";
 import { getDirectDescendantsOfKind, isZodReference, } from "./zod-node.js";
+export function convertSetErrorMapToConfig(sourceFile, zodName) {
+    sourceFile
+        .getDescendantsOfKind(SyntaxKind.CallExpression)
+        .filter((expression) => {
+        const callee = expression.getExpression();
+        if (!callee.isKind(SyntaxKind.PropertyAccessExpression)) {
+            return false;
+        }
+        const objectName = callee.getExpression().getText();
+        const methodName = callee.getName();
+        return objectName === zodName && methodName === "setErrorMap";
+    })
+        .forEach((expression) => {
+        const argument = expression.getArguments()[0];
+        if (!argument) {
+            return;
+        }
+        expression.replaceWithText(`${zodName}.config({ customError: ${argument.getText()} })`);
+    });
+}
+export function convertZodErrorMapType(sourceFile, zodName) {
+    sourceFile
+        .getDescendantsOfKind(SyntaxKind.TypeReference)
+        .filter((typeRef) => {
+        const typeName = typeRef.getTypeName();
+        if (!typeName.isKind(SyntaxKind.QualifiedName)) {
+            return false;
+        }
+        const left = typeName.getLeft().getText();
+        const right = typeName.getRight().getText();
+        return left === zodName && right === "ZodErrorMap";
+    })
+        .forEach((typeRef) => {
+        typeRef.replaceWithText(`${zodName}.core.$ZodErrorMap`);
+    });
+}
 export function convertErrorMapToErrorFunction(node) {
     // Find all errorMap properties
     getDirectDescendantsOfKind(node, SyntaxKind.Identifier)

package/dist/index.js

@@ -5,7 +5,7 @@ import * as fs from "node:fs";
 import * as path from "node:path";
 import { promisify } from "node:util";
 import { Project } from "ts-morph";
-import z from "zod";
+import { z } from "zod";
 import { migrateZodV3ToV4 } from "./migrate.js";
 const execAsync = promisify(exec);
 intro(`🏗️  Let's migrate Zod from v3 to v4`);

package/dist/migrate.js

@@ -3,7 +3,7 @@ import { findRootExpression } from "./ast.js";
 import { AstroZodModuleSpecifiers, collectDerivedZodSchemaReferences, collectZodImportDeclarations, collectZodReferences, getZodName, } from "./collect-imports.js";
 import { convertAstroDeprecatedZodImports } from "./convert-astro-imports.js";
 import { convertZArrayPatternsToTopLevelApi, convertZCoercePatternsToTopLevelApi, convertZFunctionPatternsToTopLevelApi, convertZNumberPatternsToZInt, convertZObjectPatternsToTopLevelApi, convertZRecordPatternsToTopLevelApi, convertZStringPatternsToTopLevelApi, } from "./convert-name-to-top-level-api.js";
-import { convertDeprecatedErrorKeysToErrorFunction, convertErrorMapToErrorFunction, convertMessageKeyToError, convertZodErrorAddIssueToDirectPushes, convertZodErrorToTreeifyError, } from "./convert-zod-errors.js";
+import { convertDeprecatedErrorKeysToErrorFunction, convertErrorMapToErrorFunction, convertMessageKeyToError, convertSetErrorMapToConfig, convertZodErrorAddIssueToDirectPushes, convertZodErrorMapType, convertZodErrorToTreeifyError, } from "./convert-zod-errors.js";
 import { replaceDeletedTypes } from "./replace-deleted-types.js";
 import { isZodNode } from "./zod-node.js";
 export function migrateZodV3ToV4(sourceFile, options = {}) {
@@ -25,8 +25,8 @@ export function migrateZodV3ToV4(sourceFile, options = {}) {
         }
     });
     // Collect references before modifying imports
-    const zodReferences = collectZodReferences(importDeclarations);
-    const derivedReferences = collectDerivedZodSchemaReferences(zodReferences);
+    const { zodReferences, identifierMap } = collectZodReferences(sourceFile, importDeclarations);
+    const derivedReferences = collectDerivedZodSchemaReferences(zodReferences, identifierMap);
     replaceDeletedTypes(importDeclarations, zodReferences);
     [...zodReferences, ...derivedReferences].forEach((node) => {
         if (node.wasForgotten()) {
@@ -56,6 +56,8 @@ export function migrateZodV3ToV4(sourceFile, options = {}) {
     });
     convertZodErrorToTreeifyError(sourceFile, zodName);
     convertZodErrorAddIssueToDirectPushes(sourceFile, zodName);
+    convertSetErrorMapToConfig(sourceFile, zodName);
+    convertZodErrorMapType(sourceFile, zodName);
     renameZSchemaEnumToLowercase(sourceFile, zodName);
     convertAstroDeprecatedZodImports(importDeclarations);
     return sourceFile.getFullText();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zod-v3-to-v4",
-  "version": "1.11.0",
+  "version": "1.13.0",
   "description": "Migrate Zod from v3 to v4",
   "keywords": [
     "zod",
@@ -44,22 +44,22 @@
     "*": "prettier --ignore-unknown --write"
   },
   "dependencies": {
-    "@clack/prompts": "1.0.0-alpha.1",
-    "ts-morph": "26.0.0"
+    "@clack/prompts": "1.0.0-alpha.9",
+    "ts-morph": "26.0.0",
+    "zod": "3.25.12"
   },
   "devDependencies": {
-    "@types/node": "20.9.0",
+    "@types/node": "25.0.3",
     "husky": "9.1.7",
-    "lint-staged": "16.0.0",
+    "lint-staged": "16.2.7",
     "prettier": "3.5.3",
     "prettier-plugin-curly": "0.3.1",
     "prettier-plugin-organize-imports": "4.3.0",
     "prettier-plugin-packagejson": "2.5.10",
-    "prettier-plugin-sh": "0.15.0",
+    "prettier-plugin-sh": "0.17.4",
     "rimraf": "6.0.1",
-    "typescript": "5.8.3",
-    "vitest": "3.1.4",
-    "zod": "3.25.12"
+    "typescript": "5.9.3",
+    "vitest": "3.1.4"
   },
   "packageManager": "pnpm@9.15.4+sha512.b2dc20e2fc72b3e18848459b37359a32064663e5627a51e4c74b2c29dd8e8e0491483c3abb40789cfd578bf362fb6ba8261b05f0387d76792ed6e23ea3b1b6a0"
 }