cogsbox-shape 0.5.4 → 0.5.5

package/dist/schema.d.ts

@@ -1243,6 +1243,12 @@ type Prettify<T> = {
 } & {};
 type InferSchema<T> = {
     [K in keyof T as K extends "_tableName" | "__schemaId" ? never : K]: T[K] extends {
+        type: "reference";
+        to: () => infer RefField;
+    } ? RefField extends {
+        zodClientSchema: infer ClientType extends z.ZodTypeAny;
+        toClient?: (dbValue: any) => infer TransformedType;
+    } ? RefField["toClient"] extends Function ? z.ZodType<TransformedType> : ClientType extends z.ZodNever ? z.ZodOptional<z.ZodDate> : ClientType : never : T[K] extends {
         zodClientSchema: infer ClientType extends z.ZodTypeAny;
         toClient?: (dbValue: any) => infer TransformedType;
     } ? T[K]["toClient"] extends Function ? z.ZodType<TransformedType> : ClientType extends z.ZodNever ? z.ZodOptional<z.ZodDate> : ClientType : T[K] extends () => {
@@ -1250,6 +1256,12 @@ type InferSchema<T> = {
         schema: infer S;
     } ? z.ZodArray<z.ZodObject<{
         [P in keyof S as P extends "_tableName" | "__schemaId" ? never : P]: S[P] extends {
+            type: "reference";
+            to: () => infer RefField;
+        } ? RefField extends {
+            zodClientSchema: z.ZodTypeAny;
+            toClient?: (dbValue: any) => any;
+        } ? RefField["toClient"] extends Function ? z.ZodType<ReturnType<RefField["toClient"]>> : RefField["zodClientSchema"] extends z.ZodNever ? z.ZodOptional<z.ZodDate> : RefField["zodClientSchema"] : never : S[P] extends {
             zodClientSchema: z.ZodTypeAny;
             toClient?: (dbValue: any) => any;
         } ? S[P]["toClient"] extends Function ? z.ZodType<ReturnType<S[P]["toClient"]>> : S[P]["zodClientSchema"] extends z.ZodNever ? z.ZodOptional<z.ZodDate> : S[P]["zodClientSchema"] : never;
@@ -1258,6 +1270,12 @@ type InferSchema<T> = {
         schema: infer S;
     } ? z.ZodObject<{
         [P in keyof S as P extends "_tableName" | "__schemaId" ? never : P]: S[P] extends {
+            type: "reference";
+            to: () => infer RefField;
+        } ? RefField extends {
+            zodClientSchema: z.ZodTypeAny;
+            toClient?: (dbValue: any) => any;
+        } ? RefField["toClient"] extends Function ? z.ZodType<ReturnType<RefField["toClient"]>> : RefField["zodClientSchema"] extends z.ZodNever ? z.ZodOptional<z.ZodDate> : RefField["zodClientSchema"] : never : S[P] extends {
             zodClientSchema: z.ZodTypeAny;
             toClient?: (dbValue: any) => any;
         } ? S[P]["toClient"] extends Function ? z.ZodType<ReturnType<S[P]["toClient"]>> : S[P]["zodClientSchema"] extends z.ZodNever ? z.ZodOptional<z.ZodDate> : S[P]["zodClientSchema"] : never;
@@ -1265,6 +1283,11 @@ type InferSchema<T> = {
 };
 export type InferDBSchema<T> = {
     [K in keyof T as K extends "_tableName" | "__schemaId" ? never : K]: T[K] extends {
+        type: "reference";
+        to: () => infer RefField;
+    } ? RefField extends {
+        zodDbSchema: infer DbType extends z.ZodTypeAny;
+    } ? DbType : never : T[K] extends {
         zodDbSchema: infer DbType extends z.ZodTypeAny;
     } ? DbType : T[K] extends {
         dbType: infer DbType extends z.ZodTypeAny;
@@ -1273,6 +1296,11 @@ export type InferDBSchema<T> = {
         schema: infer S;
     } ? z.ZodArray<z.ZodObject<{
         [P in keyof S as P extends "_tableName" | "__schemaId" ? never : P]: S[P] extends {
+            type: "reference";
+            to: () => infer RefField;
+        } ? RefField extends {
+            zodDbSchema: infer DbType extends z.ZodTypeAny;
+        } ? DbType : never : S[P] extends {
             zodDbSchema: infer DbType extends z.ZodTypeAny;
         } ? DbType : never;
     }>> : T[K] extends () => {
@@ -1280,6 +1308,11 @@ export type InferDBSchema<T> = {
         schema: infer S;
     } ? z.ZodObject<{
         [P in keyof S as P extends "_tableName" | "__schemaId" ? never : P]: S[P] extends {
+            type: "reference";
+            to: () => infer RefField;
+        } ? RefField extends {
+            zodDbSchema: infer DbType extends z.ZodTypeAny;
+        } ? DbType : never : S[P] extends {
             zodDbSchema: infer DbType extends z.ZodTypeAny;
         } ? DbType : never;
     }> : never;
@@ -1408,9 +1441,9 @@ type InferSerializedSchema<T> = {
         };
     } : never;
 };
-export declare function reference<T extends () => BaseSchemaField>(config: {
-    to: T;
-}): SchemaField;
+export declare function reference<T extends BaseSchemaField>(config: {
+    to: () => T;
+}): ReferenceField<T>;
 export declare function createSchema<T extends Schema<any>>(schema: T): {
     dbSchema: z.ZodObject<Prettify<InferDBSchema<T>>>;
     clientSchema: z.ZodObject<Prettify<OmitNever<InferSchema<T>>>>;

package/dist/schema.js

@@ -388,13 +388,14 @@ function createSerializableSchema(schema) {
     }
     return serializableSchema;
 }
+// Update the createSchema function to handle references in the main loop
 export function createSchema(schema) {
     const serialized = createSerializableSchema(schema);
     const dbFields = {};
     const clientFields = {};
     const defaultValues = {};
     for (const [key, value] of Object.entries(schema)) {
-        if (key === "_tableName")
+        if (key === "_tableName" || key === "__schemaId")
             continue;
         if (typeof value === "function") {
             const relation = value();
@@ -436,10 +437,22 @@ export function createSchema(schema) {
             }
             continue;
         }
-        dbFields[key] = value.zodDbSchema;
-        clientFields[key] = value.zodClientSchema;
-        defaultValues[key] =
-            value.defaultValue ?? inferDefaultFromZod(value.zodClientSchema);
+        // Handle references - this is the key addition
+        if (value && typeof value === "object" && value.type === "reference") {
+            const referencedField = value.to();
+            dbFields[key] = referencedField.zodDbSchema;
+            clientFields[key] = referencedField.zodClientSchema;
+            defaultValues[key] =
+                referencedField.defaultValue ??
+                    inferDefaultFromZod(referencedField.zodClientSchema);
+        }
+        else {
+            // Handle regular fields
+            dbFields[key] = value.zodDbSchema;
+            clientFields[key] = value.zodClientSchema;
+            defaultValues[key] =
+                value.defaultValue ?? inferDefaultFromZod(value.zodClientSchema);
+        }
     }
     return {
         dbSchema: z.object(dbFields),

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "cogsbox-shape",
-    "version": "0.5.4",
+    "version": "0.5.5",
     "description": "A TypeScript library for creating type-safe database schemas with Zod validation, SQL type definitions, and automatic client/server transformations. Unifies client, server, and database types through a single schema definition, with built-in support for relationships and serialization.",
     "main": "dist/index.js",
     "types": "dist/index.d.ts",