import Decimal from "decimal.js"
import { z } from "zod"


type JsonifiedDecimalBrand = "@thilawyn/zod-schema-class/JsonifiedDecimal"


export function decimal() {
    return z.custom<Decimal>(v => Decimal.isDecimal(v))
}

export module decimal {
    export function jsonifyDecimalSchema<S extends z.ZodType<Decimal>>(schema: S) {
        return schema
            .transform(v => v.toJSON())
            .brand<JsonifiedDecimalBrand>()
    }

    export function dejsonifyDecimalSchema<S extends z.ZodType<Decimal>>(schema: S) {
        return z
            .custom<string & z.BRAND<JsonifiedDecimalBrand>>()
            .pipe(
                z
                    .string()
                    .transform(v => {
                        try {
                            return new Decimal(v)
                        }
                        catch (e) {
                            return v
                        }
                    })
            )
            .pipe(schema)
    }
}