Refactoring

package.json

@@ -54,6 +54,7 @@
     "rollup": "^4.9.4",
     "rollup-plugin-cleanup": "^3.2.1",
     "rollup-plugin-ts": "^3.4.5",
+    "ts-functional-pipe": "^3.1.2",
     "tsx": "^4.7.0",
     "typescript": "^5.3.3"
   }

src/extendSchemableClass.ts

@@ -34,15 +34,25 @@ export function extendSchemableClass<
         ExtendDefaultValues
     >,
 
-    schemaApplier: (schema: C["schema"]) => z.ZodObject<
+    props: {
+        schema: (
+            schema: z.ZodObject<
+                ExtendSchemaT,
+                ExtendSchemaUnknownKeys,
+                ExtendSchemaCatchall,
+                ExtendSchemaValues,
+                ExtendSchemaValues
+            >
+        ) => z.ZodObject<
             SchemaT,
             SchemaUnknownKeys,
             SchemaCatchall,
             SchemaValues,
             SchemaValues
-    >,
+        >
 
-    defaultValuesApplier: (defaultValues: ExtendDefaultValues) => DefaultValues,
+        defaultValues: (defaultValues: ExtendDefaultValues) => DefaultValues
+    },
 ) {
     type Class<T, Arguments extends unknown[]> = (
         C extends ConcreteClass<any>
@@ -50,8 +60,8 @@ export function extendSchemableClass<
             : AbstractClass<T, Arguments>
     )
 
-    const schema        = schemaApplier(extend.schema)
+    const schema        = props.schema(extend.schema)
-    const defaultValues = defaultValuesApplier(extend.defaultValues)
+    const defaultValues = props.defaultValues(extend.defaultValues)
 
     return class extends extend {
         static readonly schema = schema

src/makeSchemableClass.ts

@@ -21,15 +21,17 @@ export function makeSchemableClassFrom<
 >(
     extend: C,
 
+    { schema, defaultValues }: {
         schema: z.ZodObject<
             SchemaT,
             SchemaUnknownKeys,
             SchemaCatchall,
             Values,
             Values
-    >,
+        >
 
-    defaultValues: DefaultValues,
+        defaultValues: DefaultValues
+    },
 ) {
     type Class<T, Arguments extends unknown[]> = (
         C extends ConcreteClass<any>
@@ -77,15 +79,17 @@ export function makeSchemableClass<
     Values            extends {},
     DefaultValues     extends Partial<Values>,
 >(
+    props: {
         schema: z.ZodObject<
             SchemaT,
             SchemaUnknownKeys,
             SchemaCatchall,
             Values,
             Values
-    >,
+        >
 
-    defaultValues: DefaultValues,
+        defaultValues: DefaultValues
-) {
+    }
-    return makeSchemableClassFrom(Object, schema, defaultValues)
+) {
+    return makeSchemableClassFrom(Object, props)
 }

src/tests.ts

@@ -1,3 +1,4 @@
+import { pipeInto } from "ts-functional-pipe"
 import { z } from "zod"
 import { makeSchemableClass, newSchemable } from "."
 import { makeJsonifiableSchemableClass } from "./jsonifiable"
@@ -7,18 +8,35 @@ import { dejsonifyBigIntSchema, jsonifyBigIntSchema } from "./legacy/jsonifiable
 const UserLevel = z.enum(["User", "Admin"])
 
 
-const UserProto = makeSchemableClass(z.object({
+const UserProto = makeSchemableClass({
+    schema: z.object({
         id: z.bigint(),
         name: z.string(),
         level: UserLevel,
-}), {
+    }),
+
+    defaultValues: {
         level: "User"
-} as const)
+    } as const,
+})
 
 UserProto.defaultValues
 
 
-const JsonifiableUserProto = makeJsonifiableSchemableClass(UserProto, {
+class User extends pipeInto(
+    makeSchemableClass({
+        schema: z.object({
+            id:     z.bigint(),
+            name:   z.string(),
+            level:  UserLevel,
+        }),
+
+        defaultValues: {
+            level: "User"
+        } as const,
+    }),
+
+    v => makeJsonifiableSchemableClass(v, {
         jsonifySchema: ({ schema, shape }) => schema.extend({
             id: jsonifyBigIntSchema(shape.id)
         }),
@@ -27,9 +45,7 @@ const JsonifiableUserProto = makeJsonifiableSchemableClass(UserProto, {
             id: dejsonifyBigIntSchema(shape.id)
         }),
     })
-
+) {}
-
-class User extends JsonifiableUserProto {}
 
 
 const user1 = newSchemable(User, { id: 1n, name: "User" })