Merge pull request #1261 from paoloricciuti/fix-or-migration

feat: add `or` migration

Author: fabian-hiller

codemod/zod-to-valibot/__testfixtures__/or/input.ts

@@ -0,0 +1,4 @@
+import { z } from "zod";
+
+const Schema1 = z.string().or(z.number());
+const Schema2 = z.string().or(z.number()).or(z.boolean());

codemod/zod-to-valibot/__testfixtures__/or/output.ts

@@ -0,0 +1,4 @@
+import * as v from "valibot";
+
+const Schema1 = v.union([v.string(), v.number()]);
+const Schema2 = v.union([v.union([v.string(), v.number()]), v.boolean()]);

codemod/zod-to-valibot/src/test-setup.test.ts

@@ -51,6 +51,7 @@ defineTests(transform, [
   'object-strict',
   'object-strip',
   'optional-schema',
+  'or',
   'parsing',
   'readonly',
   'record-schema',

codemod/zod-to-valibot/src/transform/schemas-and-links/constants.ts

@@ -144,6 +144,7 @@ export const ZOD_METHODS = [
   'keyof',
   'omit',
   'optional',
+  'or',
   'merge',
   'nullable',
   'nullish',

codemod/zod-to-valibot/src/transform/schemas-and-links/methods/index.ts

@@ -11,6 +11,7 @@ export * from './nullable';
 export * from './nullish';
 export * from './omit';
 export * from './optional';
+export * from './or';
 export * from './parse';
 export * from './parseAsync';
 export * from './partial';

codemod/zod-to-valibot/src/transform/schemas-and-links/methods/or/index.ts

@@ -0,0 +1 @@
+export * from './or';

codemod/zod-to-valibot/src/transform/schemas-and-links/methods/or/or.ts

@@ -0,0 +1,12 @@
+import j from 'jscodeshift';
+
+export function transformOr(
+  valibotIdentifier: string,
+  schemaExp: j.CallExpression | j.MemberExpression | j.Identifier,
+  inputArgs: j.CallExpression['arguments']
+) {
+  return j.callExpression(
+    j.memberExpression(j.identifier(valibotIdentifier), j.identifier('union')),
+    [j.arrayExpression([schemaExp, ...inputArgs])]
+  );
+}

codemod/zod-to-valibot/src/transform/schemas-and-links/schemas-and-links.ts

@@ -23,6 +23,7 @@ import {
   transformNullish,
   transformOmit,
   transformOptional as transformOptionalMethod,
+  transformOr,
   transformParse,
   transformParseAsync,
   transformPartial,
@@ -359,6 +360,8 @@ function toValibotMethodExp(
       return transformKeyof(...args);
     case 'omit':
       return transformOmit(...args);
+    case 'or':
+      return transformOr(...args);
     case 'optional':
       return transformOptionalMethod(...args);
     case 'merge':