Jsonify option work

src/schema/jsonified/option.ts

@@ -3,7 +3,7 @@ import { identity } from "lodash-es"
 import { Jsonifiable, Opaque } from "type-fest"
 import { z } from "zod"
 import { effect } from "../effect"
-import { ZodEffectOption, ZodEffectOptionSome, effectOptionSomeInnerType } from "../effect/option"
+import { ZodEffectOption, ZodEffectOptionNone, ZodEffectOptionSome, effectOptionSomeInnerType } from "../effect/option"
 
 
 export type JsonifiedEffectOption<T extends Jsonifiable> = Opaque<T | null, "@thilawyn/zod-schema-class/JsonifiedEffectOption">
@@ -11,13 +11,15 @@ export type JsonifiedEffectOption<T extends Jsonifiable> = Opaque<T | null, "@th
 
 export const jsonifyOption = {
     option: <
-        S extends ZodEffectOption<T>,
-        T extends z.ZodTypeAny,
-        // Jsonify extends (schema: z.ZodType<>),
+        InnerS          extends z.ZodTypeAny,
+        JsonifiedInnerS extends z.ZodTypeAny = InnerS,
     >(
-        schema: S | ZodEffectOption<T>,
-    ) =>
-        (schema as S).transform(v => Option.getOrNull(v)),
+        schema:         ZodEffectOption<InnerS>,
+        jsonifySchema:  (schema: InnerS) => JsonifiedInnerS = identity,
+    ) => z.union([
+        jsonifyOption.some(schema._def.options[0], jsonifySchema),
+        jsonifyOption.none(schema._def.options[1], jsonifySchema),
+    ]),
 
     some: <
         InnerS          extends z.ZodTypeAny,
@@ -30,12 +32,13 @@ export const jsonifyOption = {
         .pipe(jsonifySchema(effectOptionSomeInnerType(schema))),
 
     none: <
-        S extends ReturnType<typeof effect.option.none<T>>,
-        T extends z.ZodSchema | unknown = unknown,
+        InnerS          extends z.ZodTypeAny,
+        JsonifiedInnerS extends z.ZodTypeAny = InnerS,
     >(
-        schema: S | ReturnType<typeof effect.option.none<T>>
+        schema:           ZodEffectOptionNone<InnerS>,
+        _jsonifySchema?:  (schema: InnerS) => JsonifiedInnerS,
     ) =>
-        (schema as S).transform(() => null),
+        schema.transform(() => null),
 } as const
 
 export const dejsonifyOption = {