@supernova-studio/model 0.13.0 → 0.14.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@supernova-studio/model",
-  "version": "0.13.0",
+  "version": "0.14.0",
   "description": "Supernova Data Models",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/helpers/db.ts

@@ -1,3 +1,17 @@
+import {
+  UnknownKeysParam,
+  ZodNullable,
+  ZodObject,
+  ZodObjectDef,
+  ZodOptional,
+  ZodRawShape,
+  ZodSchema,
+  ZodTypeAny,
+  objectInputType,
+  objectOutputType,
+  z,
+} from "zod";
+
 export type DbCreateInputOmit<T> = Omit<T, "id" | "createdAt" | "updatedAt">;
 
 export type DbUpdateInputOmit<T> = Omit<T, "createdAt" | "updatedAt" | "persistentId">;
@@ -35,3 +49,48 @@ export function zodUpdateInputOmit() {
     persistentId: true,
   } as const;
 }
+
+function zodCreate<
+  T extends ZodRawShape,
+  UnknownKeys extends UnknownKeysParam = UnknownKeysParam,
+  Catchall extends ZodTypeAny = ZodTypeAny,
+  O = objectOutputType<T, Catchall, UnknownKeys>,
+  I = objectInputType<T, Catchall, UnknownKeys>
+>(schema: ZodObject<T, UnknownKeysParam, ZodTypeAny, I, O>) {
+  return schema.omit(zodCreateInputOmit());
+}
+
+/**
+ * Don't use it yet, produces too complex type
+ */
+function zodUpdate<
+  T extends ZodRawShape,
+  UnknownKeys extends UnknownKeysParam = UnknownKeysParam,
+  Catchall extends ZodTypeAny = ZodTypeAny,
+  O = objectOutputType<T, Catchall, UnknownKeys>,
+  I = objectInputType<T, Catchall, UnknownKeys>
+>(schema: ZodObject<T, UnknownKeysParam, ZodTypeAny, I, O>) {
+  const hue = schema.omit(zodUpdateInputOmit());
+  return makeOptionalPropsNullable(hue).partial();
+}
+
+function makeOptionalPropsNullable<Schema extends z.AnyZodObject>(schema: Schema) {
+  type SchemaType = typeof schema;
+
+  // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
+  const entries = Object.entries(schema.shape) as [keyof SchemaType["shape"], z.ZodTypeAny][];
+  const newProps = entries.reduce(
+    (acc, [key, value]) => {
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-unsafe-call, @typescript-eslint/no-unsafe-member-access
+      acc[key] = value instanceof z.ZodOptional ? value.unwrap().nullable() : value;
+      return acc;
+    },
+    {} as {
+      [key in keyof SchemaType["shape"]]: SchemaType["shape"][key] extends z.ZodOptional<infer T>
+        ? z.ZodNullable<T>
+        : SchemaType["shape"][key];
+    }
+  );
+
+  return z.object(newProps);
+}