orchid-orm-test-factory 0.7.65 → 1.0.1

package/dist/index.d.ts

@@ -1,23 +1,29 @@
-import { AnyZodObject, ZodTypeAny, ZodObject } from 'zod';
-import { CreateData, Query } from 'pqb';
-import { MaybePromise, EmptyObject, RecordUnknown, ColumnShapeOutput } from 'orchid-core';
+import { CreateData, CreateSelf } from 'pqb';
+import { PickQueryShape, MaybePromise, EmptyObject, RecordUnknown, ColumnTypeBase, ColumnShapeOutput } from 'orchid-core';
 
-type UniqueField = {
-    key: string;
-    type: 'text';
-    kind?: 'email' | 'url';
-    max?: number;
-    length?: number;
-} | {
-    key: string;
-    type: 'number';
-    gt?: number;
-    gte?: number;
-};
-interface FactoryOptions {
+type FakeDataFn = (sequence: number) => unknown;
+interface FakeDataDefineFns {
+    [K: string]: (column: ColumnTypeBase) => FakeDataFn;
+}
+interface FakeDataFns {
+    [K: string]: FakeDataFn;
+}
+interface FactoryConfig {
     sequence?: number;
     sequenceDistance?: number;
     maxTextLength?: number;
+    fakeDataForTypes?: FakeDataDefineFns;
+}
+type FactoryExtend<T extends PickQueryShape> = {
+    [K in keyof T['shape']]?: (sequence: number) => T['shape'][K]['outputType'];
+};
+interface TableFactoryConfig<T extends PickQueryShape> extends FactoryConfig {
+    extend?: FactoryExtend<T>;
+}
+interface OrmFactoryConfig<T> extends FactoryConfig {
+    extend?: {
+        [K in keyof T]?: T[K] extends PickQueryShape ? FactoryExtend<T[K]> : never;
+    };
 }
 type metaKey = typeof metaKey;
 declare const metaKey: unique symbol;
@@ -36,20 +42,19 @@ type CreateArg<T extends TestFactory> = CreateData<{
     } : T['table'][K];
 }>;
 type CreateResult<T extends TestFactory> = Result<T, ColumnShapeOutput<T['table']['shape']>>;
-declare class TestFactory<Q extends Query = Query, Schema extends AnyZodObject = AnyZodObject, Type extends EmptyObject = EmptyObject> {
+declare class TestFactory<Q extends CreateSelf = CreateSelf, Type extends EmptyObject = EmptyObject> {
     table: Q;
-    schema: Schema;
-    private uniqueFields;
-    private readonly data;
+    fns: FakeDataFns;
     sequence: number;
     private readonly omitValues;
     private readonly pickValues;
+    private readonly data;
     [metaKey]: {
         type: Type;
         omit: EmptyObject;
         pick: EmptyObject;
     };
-    constructor(table: Q, schema: Schema, uniqueFields: UniqueField[], data?: RecordUnknown, options?: FactoryOptions);
+    constructor(table: Q, fns: FakeDataFns, options?: TableFactoryConfig<PickQueryShape>);
     set<T extends this, Meta extends {
         type: EmptyObject;
     }, Data extends {
@@ -85,19 +90,13 @@ declare class TestFactory<Q extends Query = Query, Schema extends AnyZodObject =
     createMany<T extends this, Args extends CreateArg<T>[]>(this: T, ...arr: Args): Promise<{
         [K in keyof Args]: CreateResult<T>;
     }>;
-    extend<T extends this>(this: T): new () => TestFactory<Q, Schema, Type>;
+    extend<T extends this>(this: T): new () => TestFactory<Q, Type>;
 }
-type TableFactory<T extends Query> = T['shape'] extends Record<string, {
-    inputSchema: ZodTypeAny;
-}> ? TestFactory<T, ZodObject<{
-    [K in keyof T['shape']]: T['shape'][K]['inputSchema'];
-}, 'strip'>, ColumnShapeOutput<T['shape']>> : never;
-declare const tableFactory: <T extends Query>(table: T, options?: FactoryOptions) => TableFactory<T>;
+type TableFactory<T extends CreateSelf> = TestFactory<T, ColumnShapeOutput<T['shape']>>;
+declare const tableFactory: <T extends CreateSelf>(table: T, config?: TableFactoryConfig<T>) => TableFactory<T>;
 type ORMFactory<T> = {
-    [K in keyof T]: T[K] extends Query & {
-        definedAs: string;
-    } ? TableFactory<T[K]> : never;
+    [K in keyof T]: T[K] extends CreateSelf ? TableFactory<T[K]> : never;
 };
-declare const ormFactory: <T>(orm: T, options?: FactoryOptions) => ORMFactory<T>;
+declare const ormFactory: <T>(orm: T, options?: OrmFactoryConfig<T>) => ORMFactory<T>;
 
-export { CreateArg, TestFactory, ormFactory, tableFactory };
+export { type CreateArg, type FactoryConfig, type OrmFactoryConfig, type TableFactoryConfig, TestFactory, ormFactory, tableFactory };