From 74ac3b41ad88a96fb996ba41f50b513b52e9a2e0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Julien=20Valverd=C3=A9?= <julien.valverde@mailo.com>
Date: Fri, 22 Mar 2024 01:11:13 +0100
Subject: [PATCH] Schema work

---
 src/schema/effect/option.ts | 86 ++++++++++++++++++++++++++++++-------
 1 file changed, 70 insertions(+), 16 deletions(-)

diff --git a/src/schema/effect/option.ts b/src/schema/effect/option.ts
index ba6aed1..a3f91aa 100644
--- a/src/schema/effect/option.ts
+++ b/src/schema/effect/option.ts
@@ -3,23 +3,83 @@ import { identity } from "lodash-es"
 import { z } from "zod"
 
 
+type JsonifiedEffectOptionSomeBrand = "@thilawyn/zod-schema-class/JsonifiedEffectOptionSome"
+
+
 export const option = <S extends z.ZodTypeAny>(schema: S) =>
     z.union([option.some(schema), option.none(schema)])
 
-option.some = <S extends z.ZodTypeAny>(schema: S) => z
-    .custom<Option.Some<z.output<S>>>(v => Option.isOption(v) && Option.isSome(v), "Not an Option")
+
+const some = <
+    InnerS extends z.ZodTypeAny
+>(schema: InnerS) => z
+    .custom<Option.Some<z.output<InnerS>>>(v => Option.isOption(v) && Option.isSome(v), "Not an Option.Some")
     .pipe(z.object({ value: schema }).passthrough())
-    .transform<Option.Some<z.output<S>>>(identity)
+    .transform<Option.Some<z.output<InnerS>>>(identity)
 
-option.none = <S extends z.ZodTypeAny | unknown = unknown>(_schema?: S) =>
-    z.custom<Option.None<
-        S extends z.ZodTypeAny
-            ? z.output<S>
-            : unknown
-    >>(v => Option.isOption(v) && Option.isNone(v), "Not an Option")
+some.jsonify = <
+    InnerS          extends z.ZodTypeAny,
+    JsonifiedInnerS extends z.ZodTypeAny = InnerS,
+>(
+    schema:         ZodEffectOptionSome<InnerS>,
+    jsonifySchema:  (schema: InnerS) => JsonifiedInnerS = identity,
+) => schema
+    .transform(v => Option.getOrThrow(v))
+    .pipe(jsonifySchema(effectOptionSomeInnerSchema(schema)))
+    .brand<JsonifiedEffectOptionSomeBrand>()
+
+some.dejsonify = <
+    InnerS            extends z.ZodTypeAny,
+    DejsonifiedInnerS extends z.ZodTypeAny = InnerS,
+>(
+    schema:           ZodEffectOptionSome<InnerS>,
+    dejsonifySchema:  (schema: InnerS) => DejsonifiedInnerS = identity,
+) => z
+    .custom<z.input<DejsonifiedInnerS> & z.BRAND<JsonifiedEffectOptionSomeBrand>>()
+    .pipe(dejsonifySchema(effectOptionSomeInnerSchema(schema)))
+    .transform(v => Option.some<z.output<DejsonifiedInnerS>>(v))
+    .pipe(schema)
+
+option.some = some
 
 
-export type ZodEffectOption<S extends z.ZodTypeAny> = ReturnType<typeof option<S>>
+const none = <
+    InnerS extends z.ZodTypeAny | unknown = unknown
+>(_schema?: InnerS) =>
+    z.custom<
+        Option.None<
+            InnerS extends z.ZodTypeAny
+                ? z.output<InnerS>
+                : unknown
+        >
+    >(v => Option.isOption(v) && Option.isNone(v), "Not an Option.None")
+
+none.jsonify = <
+    InnerS          extends z.ZodTypeAny,
+    JsonifiedInnerS extends z.ZodTypeAny = InnerS,
+>(
+    schema:           ZodEffectOptionNone<InnerS>,
+    _jsonifySchema?:  (schema: InnerS) => JsonifiedInnerS,
+) =>
+    schema.transform(() => null),
+
+none.dejsonify = <
+    InnerS            extends z.ZodTypeAny,
+    DejsonifiedInnerS extends z.ZodTypeAny = InnerS,
+>(
+    schema:             ZodEffectOptionNone<InnerS>,
+    _dejsonifySchema?:  (schema: InnerS) => DejsonifiedInnerS,
+) => z
+    .null()
+    .transform(() => Option.none<z.output<DejsonifiedInnerS>>())
+    .pipe(schema),
+
+option.none = none
+
+
+export type ZodEffectOption<S extends z.ZodTypeAny>     = ReturnType<typeof option<S>>
+export type ZodEffectOptionSome<S extends z.ZodTypeAny> = ReturnType<typeof option.some<S>>
+export type ZodEffectOptionNone<S extends z.ZodTypeAny> = ReturnType<typeof option.none<S>>
 
 export const effectOptionSomeSchema = <S extends z.ZodTypeAny>(schema: ZodEffectOption<S>): ZodEffectOptionSome<S> =>
     schema.options[0]
@@ -27,11 +87,5 @@ export const effectOptionSomeSchema = <S extends z.ZodTypeAny>(schema: ZodEffect
 export const effectOptionNoneSchema = <S extends z.ZodTypeAny>(schema: ZodEffectOption<S>): ZodEffectOptionNone<S> =>
     schema.options[1]
 
-
-export type ZodEffectOptionSome<S extends z.ZodTypeAny> = ReturnType<typeof option.some<S>>
-
 export const effectOptionSomeInnerSchema = <S extends z.ZodTypeAny>(schema: ZodEffectOptionSome<S>): S =>
     schema._def.schema._def.out.shape.value
-
-
-export type ZodEffectOptionNone<S extends z.ZodTypeAny> = ReturnType<typeof option.none<S>>