import { expression } from "@thilawyn/traitify-ts"
import { AbstractClass } from "type-fest"
import { z } from "zod"
import { ExtendableZodSchemaObject } from "./traits/ExtendableZodSchemaObject"
import { InstantiableZodSchemaObject } from "./traits/InstantiableZodSchemaObject"
import { Extend, StaticMembers } from "./util"


export function ZodSchemaClassOf<
    Super             extends AbstractClass<object, []>,

    SchemaT           extends z.ZodRawShape,
    SchemaUnknownKeys extends z.UnknownKeysParam,
    SchemaCatchall    extends z.ZodTypeAny,

    Values            extends {},
    DefaultValues     extends Partial<Values>,
>(
    of: Super,

    { schema, defaultValues }: {
        schema:         z.ZodObject<SchemaT, SchemaUnknownKeys, SchemaCatchall, Values, Values>
        defaultValues:  DefaultValues
    },
) {
    const exp = expression
        .extends(class extends (of as AbstractClass<{}, []>) {
            static readonly schema        = schema
            static readonly defaultValues = defaultValues

            constructor(values: Values) {
                super()
                Object.assign(this, values)
            }
        })
        .expresses(
            InstantiableZodSchemaObject,
            ExtendableZodSchemaObject,
        )
        .build()

    return exp.extends as AbstractClass<
        Extend<[
            InstanceType<Super>,
            InstanceType<typeof exp.extends>,
            Values,
        ]>,

        ConstructorParameters<typeof exp.extends>
    > &
    Extend<[
        StaticMembers<Super>,
        StaticMembers<typeof exp.extends>,
    ]>
}


class DefaultRoot {}

export function ZodSchemaClass<
    SchemaT           extends z.ZodRawShape,
    SchemaUnknownKeys extends z.UnknownKeysParam,
    SchemaCatchall    extends z.ZodTypeAny,

    Values            extends {},
    DefaultValues     extends Partial<Values>,
>(
    props: {
        schema:         z.ZodObject<SchemaT, SchemaUnknownKeys, SchemaCatchall, Values, Values>
        defaultValues:  DefaultValues
    },
) {
    return ZodSchemaClassOf(DefaultRoot, props)
}