From 822149e9309b4ac1227c321f185a88503673830e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Julien=20Valverd=C3=A9?= <julien.valverde@mailo.com>
Date: Tue, 16 Jan 2024 15:08:24 +0100
Subject: [PATCH] Jsonifiable schema helpers

---
 src/jsonifiable/index.ts            |  1 +
 src/jsonifiable/schema/bigint.ts    | 18 ++++++++++++++++++
 src/jsonifiable/schema/date.ts      | 18 ++++++++++++++++++
 src/jsonifiable/schema/decimal.ts   | 19 +++++++++++++++++++
 src/jsonifiable/schema/index.ts     |  4 ++++
 src/jsonifiable/schema/schemable.ts | 25 +++++++++++++++++++++++++
 6 files changed, 85 insertions(+)
 create mode 100644 src/jsonifiable/schema/bigint.ts
 create mode 100644 src/jsonifiable/schema/date.ts
 create mode 100644 src/jsonifiable/schema/decimal.ts
 create mode 100644 src/jsonifiable/schema/index.ts
 create mode 100644 src/jsonifiable/schema/schemable.ts

diff --git a/src/jsonifiable/index.ts b/src/jsonifiable/index.ts
index 26a26be..74e7d23 100644
--- a/src/jsonifiable/index.ts
+++ b/src/jsonifiable/index.ts
@@ -1,3 +1,4 @@
 export * from "./JsonifiableSchemableClass"
 export * from "./dejsonifySchemable"
 export * from "./makeJsonifiableSchemableClass"
+export * from "./schema"
diff --git a/src/jsonifiable/schema/bigint.ts b/src/jsonifiable/schema/bigint.ts
new file mode 100644
index 0000000..b0f0093
--- /dev/null
+++ b/src/jsonifiable/schema/bigint.ts
@@ -0,0 +1,18 @@
+import { z } from "zod"
+
+
+export const jsonifyBigIntSchema = <S extends z.ZodBigInt>(schema: S) =>
+    schema.transform(v => v.toString())
+
+export const dejsonifyBigIntSchema = <S extends z.ZodBigInt>(schema: S) =>
+    z
+        .string()
+        .transform(v => {
+            try {
+                return BigInt(v)
+            }
+            catch (e) {
+                return v
+            }
+        })
+        .pipe(schema)
diff --git a/src/jsonifiable/schema/date.ts b/src/jsonifiable/schema/date.ts
new file mode 100644
index 0000000..b5ac677
--- /dev/null
+++ b/src/jsonifiable/schema/date.ts
@@ -0,0 +1,18 @@
+import { z } from "zod"
+
+
+export const jsonifyDateSchema = <S extends z.ZodDate>(schema: S) =>
+    schema.transform(v => v.toString())
+
+export const dejsonifyDateSchema = <S extends z.ZodDate>(schema: S) =>
+    z
+        .string()
+        .transform(v => {
+            try {
+                return new Date(v)
+            }
+            catch (e) {
+                return v
+            }
+        })
+        .pipe(schema)
diff --git a/src/jsonifiable/schema/decimal.ts b/src/jsonifiable/schema/decimal.ts
new file mode 100644
index 0000000..58c492d
--- /dev/null
+++ b/src/jsonifiable/schema/decimal.ts
@@ -0,0 +1,19 @@
+import { Decimal } from "decimal.js"
+import { z } from "zod"
+
+
+export const jsonifyDecimalSchema = <S extends z.ZodType<Decimal, z.ZodTypeDef, Decimal>>(schema: S) =>
+    schema.transform(v => v.toJSON())
+
+export const dejsonifyDecimalSchema = <S extends z.ZodType<Decimal, z.ZodTypeDef, Decimal>>(schema: S) =>
+    z
+        .string()
+        .transform(v => {
+            try {
+                return new Decimal(v)
+            }
+            catch (e) {
+                return v
+            }
+        })
+        .pipe(schema)
diff --git a/src/jsonifiable/schema/index.ts b/src/jsonifiable/schema/index.ts
new file mode 100644
index 0000000..08ecef8
--- /dev/null
+++ b/src/jsonifiable/schema/index.ts
@@ -0,0 +1,4 @@
+export * from "./bigint"
+export * from "./date"
+export * from "./decimal"
+export * from "./schemable"
diff --git a/src/jsonifiable/schema/schemable.ts b/src/jsonifiable/schema/schemable.ts
new file mode 100644
index 0000000..c6e2a2d
--- /dev/null
+++ b/src/jsonifiable/schema/schemable.ts
@@ -0,0 +1,25 @@
+import { z } from "zod"
+import { JsonifiableSchemableClass, JsonifiableSchemableConfig } from ".."
+
+
+// TODO: try to find a way to get rid of the 'class_' arg
+export const jsonifySchemableSchema = <
+    C       extends JsonifiableSchemableClass<$Config>,
+    $Config extends JsonifiableSchemableConfig,
+    S       extends z.ZodType<InstanceType<C>, z.ZodTypeDef, InstanceType<C>>,
+>(
+    class_:  C | JsonifiableSchemableClass<$Config>,
+    schema:  S,
+) =>
+    schema.pipe(class_.jsonifySchema)
+
+// TODO: try to find a way to get rid of the 'class_' arg
+export const dejsonifySchemableSchema = <
+    C       extends JsonifiableSchemableClass<$Config>,
+    $Config extends JsonifiableSchemableConfig,
+    S       extends z.ZodType<InstanceType<C>, z.ZodTypeDef, InstanceType<C>>,
+>(
+    class_:  C | JsonifiableSchemableClass<$Config>,
+    schema:  S,
+) =>
+    class_.dejsonifySchema.transform(v => new class_(v)).pipe(schema)