import { Opaque } from "type-fest"
import { z } from "zod"
import { identity } from "../../util"


export type JsonifiedBigInt = Opaque<string, "@thilawyn/schemable-class/JsonifiedBigInt">


export function jsonifyBigIntSchema<S extends z.ZodBigInt>(schema: S) {
    return schema.transform(v => v.toString() as JsonifiedBigInt)
}

export function dejsonifyBigIntSchema<S extends z.ZodBigInt>(schema: S) {
    return z
        .custom<JsonifiedBigInt>(identity)
        .pipe(z
            .string()
            .transform(v => {
                try {
                    return BigInt(v)
                }
                catch (e) {
                    return v
                }
            })
            .pipe(schema)
        )
}