import { CamelCase } from "type-fest";
import { ZodRawShape } from "zod/lib/types";
import { ZodDiscriminatedUnion, ZodLiteral, ZodObject } from "zod";
import { EntityCore, EntityCoreShape, EntityCoreShapeType, Extender, UnsafeZentity, Zentity, ZodShapeType } from "./zentity";
type ExtractType<T, D extends string> = T extends {
    [K in D]: infer V;
} ? {
    [K in keyof T]: K extends D ? V : T[K];
} : never;
type ShapeRecord<S extends ZodRawShape> = Record<string, S>;
type KeyedExtender<B extends ZodRawShape, KS extends ShapeRecord<any>, R extends object> = {
    [K in keyof KS]?: Extender<KS[K] & B, R>;
};
type Flatten<O extends ShapeRecord<ZodRawShape>, D extends string> = ValueUnion<{
    [K in keyof O]: ZodShapeType<O[K]> & {
        [Q in D]: K;
    };
}>;
type CC<P extends string, K extends string> = CamelCase<`${P}-${K}`>;
type Clean<T> = T extends undefined ? {} : T;
type OShape<B extends ZodRawShape, O extends ShapeRecord<ZodRawShape>, K extends keyof O> = O[K] & Clean<B>;
type OType<B extends ZodRawShape, O extends ShapeRecord<ZodRawShape>, K extends keyof O> = ZodShapeType<OShape<B, O, K>>;
type ReturnType<E> = E extends (arg: any) => infer R ? R : {};
type ToZodD<BS extends ZodRawShape, KS extends ShapeRecord<any>, D extends string> = ZodObject<ValueUnion<{
    [key in keyof KS]: KS[key] & {
        [K in D]: ZodLiteral<any>;
    };
}> & BS & EntityCoreShapeType>;
type ZentityNewFn<BS extends ZodRawShape, KS extends ShapeRecord<ZodRawShape>, D extends string, KSE extends KeyedExtender<BS, KS, any>, TD extends TransitionDef<KS>> = {
    [key in (keyof KS & string) as CC<`new`, key>]: (params: OType<BS, KS, key> & {
        id: string;
    }) => Full<BS, KS, D, KSE, TD, key>;
};
type ZentityCreateFn<BS extends ZodRawShape, KS extends ShapeRecord<ZodRawShape>, D extends string, KSE extends KeyedExtender<BS, KS, any>, TD extends TransitionDef<KS>> = {
    [key in (keyof KS & string) as CC<`create`, key>]: (params: OType<BS, KS, key> & EntityCore) => Full<BS, KS, D, KSE, TD, key>;
};
type ZentityUnsafeFn<BS extends ZodRawShape, KS extends ShapeRecord<ZodRawShape>, D extends string, KSE extends KeyedExtender<BS, KS, any>, TD extends TransitionDef<KS>> = {
    [key in (keyof KS & string) as CC<`unsafe`, key>]: (params: OType<BS, KS, key> & EntityCore) => UnsafeFull<BS, KS, D, KSE, TD, key>;
};
type ZentityGuardFn<BS extends ZodRawShape, KS extends ShapeRecord<ZodRawShape>, D extends string, KSE extends KeyedExtender<BS, KS, any>, TD extends TransitionDef<KS>> = {
    [key in (keyof KS & string) as CC<'is', key>]: (data: any) => data is Full<BS, KS, D, KSE, TD, key>;
};
type ZentityShape<BS extends ZodRawShape, KS extends ShapeRecord<ZodRawShape>, D extends string> = {
    [key in (keyof KS & string) as CC<key, 'shape'>]: OShape<BS, KS, key> & {
        [K in D]: ZodLiteral<key>;
    } & typeof EntityCoreShape;
};
type ZentitySchema<BS extends ZodRawShape, KS extends ShapeRecord<ZodRawShape>, D extends string> = {
    [key in (keyof KS & string) as CC<key, 'schema'>]: ZodObject<OShape<BS, KS, key> & {
        [K in D]: ZodLiteral<key>;
    } & typeof EntityCoreShape>;
};
type Full<BS extends ZodRawShape, KS extends ShapeRecord<ZodRawShape>, D extends string, KE extends KeyedExtender<BS, KS, any>, TD extends TransitionDef<KS>, key extends keyof KS> = Zentity<{
    [Q in D]: key;
} & OType<BS, KS, key>, ReturnType<KE[key]> & Transition<BS, KS, KE, TD, key>>;
type UnsafeFull<BS extends ZodRawShape, KS extends ShapeRecord<ZodRawShape>, D extends string, KE extends KeyedExtender<BS, KS, any>, TD extends TransitionDef<KS>, key extends keyof KS> = UnsafeZentity<{
    [Q in D]: key;
} & OType<BS, KS, key>, ReturnType<KE[key]> & Transition<BS, KS, KE, TD, key>>;
type DiscriminatedZentityClass<BS extends ZodRawShape, KS extends ShapeRecord<ZodRawShape>, D extends string, KE extends KeyedExtender<BS, KS, any>, TD extends TransitionDef<KS>> = {
    create: <T extends Flatten<KS, D>>(params: T & EntityCore & ZodShapeType<BS>) => Zentity<ZodShapeType<BS> & ExtractType<T, D>, ReturnType<KE[T[D]]> & Transition<BS, KS, KE, TD, T[D]>>;
    new: <T extends Flatten<KS, D>>(params: T & ZodShapeType<BS> & {
        id: string;
    }) => Zentity<ExtractType<T, D>, ReturnType<KE[T[D]]> & Transition<BS, KS, KE, TD, T[D]>>;
    unsafe: <T extends Flatten<KS, D>>(params: T & ZodShapeType<BS>) => UnsafeZentity<ExtractType<T, D>, ReturnType<KE[T[D]]> & Transition<BS, KS, KE, TD, T[D]>>;
    baseShape: BS & typeof EntityCoreShape;
    schema: ZodDiscriminatedUnion<D, [ToZodD<BS, KS, D>, ...ToZodD<BS, KS, D>[]]>;
} & ZentityNewFn<BS, KS, D, KE, TD> & ZentityCreateFn<BS, KS, D, KE, TD> & ZentityUnsafeFn<BS, KS, D, KE, TD> & ZentityGuardFn<BS, KS, D, KE, TD> & ZentitySchema<BS, KS, D> & ZentityShape<BS, KS, D>;
type ValueUnion<T> = Required<T[keyof T]>;
type TransitionDef<KS extends ShapeRecord<ZodRawShape>> = {
    [key in keyof KS]: (keyof Omit<KS, key>)[];
};
type ExclusiveToB<B, A> = {
    [K in keyof B as K extends keyof A ? never : K]: B[K];
};
type Transition<B extends ZodRawShape, KS extends ShapeRecord<ZodRawShape>, KE extends KeyedExtender<B, KS, any>, TD extends TransitionDef<KS>, K extends keyof KS> = {
    [key in (TD[K][number] & string) as CC<`to`, key>]: (param: ExclusiveToB<ZodShapeType<KS[key]>, ZodShapeType<KS[K]>>) => Zentity<OType<B, KS, key>, ReturnType<KE[key]> & Transition<B, KS, KE, TD, key>>;
};
export declare function statefulEntity<BS extends ZodRawShape, KS extends ShapeRecord<ZodRawShape>, D extends string, KSE extends KeyedExtender<BS, KS, any>, T extends TransitionDef<KS>>(options: {
    discriminator: D;
    base: BS;
    states: KS;
    extend: KSE;
    transitions: T;
}): DiscriminatedZentityClass<BS, KS, D, KSE, T>;
type UnionOfStateShapeTypesWithDiscriminator<KS extends ShapeRecord<ZodRawShape>, D extends string> = ValueUnion<{
    [key in keyof KS]: ZodShapeType<KS[key]> & {
        [Q in D]: key;
    };
}>;
export type inferStatefulZentityTypeFromClass<T> = T extends DiscriminatedZentityClass<infer BS, infer KS, infer D, infer KE, infer TD> ? Zentity<UnionOfStateShapeTypesWithDiscriminator<KS, D> & Clean<BS>, ReturnType<KE[keyof KS]> & Transition<BS, KS, KE, TD, keyof KS>> : never;
export {};
//# sourceMappingURL=stateful-zentity.d.ts.map