drizzle-seed 0.3.2-f8a2f3c → 0.4.0-08e4e66

package/{services → generators}/Generators.d.ts

@@ -1,20 +1,25 @@
 import prand from 'pure-rand';
+import type { GeneratedValueType } from '../types/seedService.js';
+import type { Column } from '../types/tables.js';
 export declare abstract class AbstractGenerator<T = {}> {
     static readonly entityKind: string;
     static readonly version: number;
+    isGeneratorUnique: boolean;
     isUnique: boolean;
     notNull: boolean;
     uniqueVersionOfGen?: new (params: T) => AbstractGenerator<T>;
+    maxUniqueCount: number;
     dataType?: string;
     timeSpent?: number;
     arraySize?: number;
     baseColumnDataType?: string;
-    stringLength?: number;
     weightedCountSeed?: number | undefined;
     maxRepeatedValuesCount?: number | {
         weight: number;
         count: number | number[];
     }[] | undefined;
+    typeParams: Column['typeParams'];
+    uniqueKey?: string;
     params: T;
     constructor(params?: T);
     init(params: {
@@ -27,8 +32,10 @@ export declare abstract class AbstractGenerator<T = {}> {
     updateParams(): void;
     abstract generate(params: {
         i: number;
+        columnName?: string;
     }): number | string | boolean | unknown | undefined | void;
     getEntityKind(): string;
+    getMaxUniqueCount(): number;
     replaceIfUnique(): AbstractGenerator<T> | undefined;
     replaceIfArray(): GenerateArray | undefined;
 }
@@ -68,17 +75,22 @@ export declare class GenerateDefault extends AbstractGenerator<{
     static readonly entityKind: string;
     generate(): unknown;
 }
-export declare class GenerateValuesFromArray extends AbstractGenerator<{
-    values: (number | string | boolean | undefined)[] | {
+export type GenerateValuesFromArrayT = {
+    values: GeneratedValueType[] | {
         weight: number;
-        values: (number | string | boolean | undefined)[];
+        values: GeneratedValueType[];
     }[];
     isUnique?: boolean;
     arraySize?: number;
-}> {
+};
+export declare class GenerateValuesFromArray extends AbstractGenerator<GenerateValuesFromArrayT> {
     static readonly entityKind: string;
     private state;
     timeSpent: number;
+    maxUniqueCount: number;
+    private allValuesCount;
+    constructor(params?: GenerateValuesFromArrayT);
+    getMaxUniqueCount(): number;
     checks({ count }: {
         count: number;
     }): void;
@@ -89,7 +101,7 @@ export declare class GenerateValuesFromArray extends AbstractGenerator<{
     generate(): string | number | boolean | undefined;
 }
 export declare class GenerateSelfRelationsValuesFromArray extends AbstractGenerator<{
-    values: (number | string | boolean)[];
+    values: (number | string | bigint)[];
 }> {
     static readonly entityKind: string;
     private state;
@@ -99,11 +111,13 @@ export declare class GenerateSelfRelationsValuesFromArray extends AbstractGenera
     }): void;
     generate({ i }: {
         i: number;
-    }): string | number | boolean | undefined;
+    }): string | number | bigint | undefined;
 }
 export declare class GenerateIntPrimaryKey extends AbstractGenerator<{}> {
     static readonly entityKind: string;
     maxValue?: number | bigint;
+    maxUniqueCount: number;
+    isGeneratorUnique: boolean;
     init({ count }: {
         count: number;
         seed: number;
@@ -128,15 +142,20 @@ export declare class GenerateNumber extends AbstractGenerator<{
     }): void;
     generate(): number;
 }
-export declare class GenerateUniqueNumber extends AbstractGenerator<{
+export type GenerateUniqueNumberT = {
     minValue?: number;
     maxValue?: number;
     precision?: number;
     isUnique?: boolean;
-}> {
+};
+export declare class GenerateUniqueNumber extends AbstractGenerator<GenerateUniqueNumberT> {
     static readonly entityKind: string;
     private state;
-    isUnique: boolean;
+    precision: number;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
+    constructor(params?: GenerateUniqueNumberT);
+    getMaxUniqueCount(): number;
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -158,11 +177,12 @@ export declare class GenerateInt extends AbstractGenerator<{
     }): void;
     generate(): string | number | bigint;
 }
-export declare class GenerateUniqueInt extends AbstractGenerator<{
+export type GenerateUniqueIntT = {
     minValue?: number | bigint;
     maxValue?: number | bigint;
     isUnique?: boolean;
-}> {
+};
+export declare class GenerateUniqueInt extends AbstractGenerator<GenerateUniqueIntT> {
     static readonly entityKind: string;
     genMaxRepeatedValuesCount: GenerateDefault | GenerateWeightedCount | undefined;
     skipCheck?: boolean;
@@ -173,8 +193,11 @@ export declare class GenerateUniqueInt extends AbstractGenerator<{
         intervals: (number | bigint)[][];
         integersCount: Map<number | bigint, number>;
     } | undefined;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
     timeSpent: number;
+    maxUniqueCount: number;
+    constructor(params?: GenerateUniqueIntT);
+    getMaxUniqueCount<T extends number | bigint>(): number;
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -208,8 +231,6 @@ export declare class GenerateDate extends AbstractGenerator<{
     generate(): string | Date;
 }
 export declare class GenerateTime extends AbstractGenerator<{
-    minTime?: string | Date;
-    maxTime?: string | Date;
     arraySize?: number;
 }> {
     static readonly entityKind: string;
@@ -232,8 +253,6 @@ export declare class GenerateTimestampInt extends AbstractGenerator<{
     generate(): number;
 }
 export declare class GenerateTimestamp extends AbstractGenerator<{
-    minTimestamp?: string | Date;
-    maxTimestamp?: string | Date;
     arraySize?: number;
 }> {
     static readonly entityKind: string;
@@ -245,8 +264,6 @@ export declare class GenerateTimestamp extends AbstractGenerator<{
     generate(): string | Date;
 }
 export declare class GenerateDatetime extends AbstractGenerator<{
-    minDatetime?: string | Date;
-    maxDatetime?: string | Date;
     arraySize?: number;
 }> {
     static readonly entityKind: string;
@@ -323,14 +340,18 @@ export declare class GenerateInterval extends AbstractGenerator<{
     }): void;
     generate(): string;
 }
-export declare class GenerateUniqueInterval extends AbstractGenerator<{
+export type GenerateUniqueIntervalT = {
     fields?: 'year' | 'month' | 'day' | 'hour' | 'minute' | 'second' | 'year to month' | 'day to hour' | 'day to minute' | 'day to second' | 'hour to minute' | 'hour to second' | 'minute to second';
     isUnique?: boolean;
-}> {
+};
+export declare class GenerateUniqueInterval extends AbstractGenerator<GenerateUniqueIntervalT> {
     static readonly 'entityKind': string;
     private state;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
     private config;
+    maxUniqueCount: number;
+    constructor(params?: GenerateUniqueIntervalT);
+    getMaxUniqueCount(): number;
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -348,27 +369,31 @@ export declare class GenerateString extends AbstractGenerator<{
         count: number;
         seed: number;
     }): void;
-    generate(): string;
+    generate(): string | Buffer<ArrayBuffer>;
 }
 export declare class GenerateUniqueString extends AbstractGenerator<{
     isUnique?: boolean;
 }> {
     static readonly entityKind: string;
     private state;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
+    getMaxUniqueCount(): number;
     init({ seed }: {
         seed: number;
     }): void;
     generate({ i }: {
         i: number;
-    }): string;
+    }): string | Buffer<ArrayBuffer>;
 }
 export declare class GenerateUUID extends AbstractGenerator<{
     arraySize?: number;
 }> {
     static readonly entityKind: string;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
     private state;
+    getMaxUniqueCount(): number;
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -394,7 +419,9 @@ export declare class GenerateUniqueFirstName extends AbstractGenerator<{
 }> {
     static readonly entityKind: string;
     private state;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
+    getMaxUniqueCount(): number;
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -419,7 +446,9 @@ export declare class GenerateUniqueLastName extends AbstractGenerator<{
 }> {
     static readonly entityKind: string;
     private state;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
+    getMaxUniqueCount(): number;
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -444,8 +473,10 @@ export declare class GenerateUniqueFullName extends AbstractGenerator<{
 }> {
     static readonly entityKind: string;
     private state;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
     timeSpent: number;
+    maxUniqueCount: number;
+    getMaxUniqueCount(): number;
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -458,22 +489,35 @@ export declare class GenerateEmail extends AbstractGenerator<{
     static readonly entityKind: string;
     private state;
     timeSpent: number;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
+    getMaxUniqueCount(): number;
     init({ count, seed }: {
         count: number;
         seed: number;
     }): void;
     generate(): string;
 }
-export declare class GeneratePhoneNumber extends AbstractGenerator<{
+export type GeneratePhoneNumberT = {
     template?: string;
     prefixes?: string[];
     generatedDigitsNumbers?: number | number[];
     arraySize?: number;
-}> {
+};
+export declare class GeneratePhoneNumber extends AbstractGenerator<GeneratePhoneNumberT> {
     static readonly entityKind: string;
     private state;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
+    constructor(params?: GeneratePhoneNumberT);
+    prepareWithTemplate(): {
+        placeholdersCount: number;
+    };
+    prepareWithoutTemplate(): {
+        generatedDigitsNumbers: number[];
+        prefixes: string[];
+    };
+    getMaxUniqueCount(): number;
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -498,7 +542,9 @@ export declare class GenerateUniqueCountry extends AbstractGenerator<{
 }> {
     static readonly entityKind: string;
     private state;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
+    getMaxUniqueCount(): number;
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -534,7 +580,10 @@ export declare class GenerateUniqueStreetAddress extends AbstractGenerator<{
 }> {
     static readonly entityKind: string;
     private state;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
+    streetNumbersCount: number;
+    maxUniqueCount: number;
+    getMaxUniqueCount(): number;
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -559,7 +608,8 @@ export declare class GenerateUniqueCity extends AbstractGenerator<{
 }> {
     static readonly entityKind: string;
     private state;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -584,7 +634,8 @@ export declare class GenerateUniquePostcode extends AbstractGenerator<{
 }> {
     static readonly entityKind: string;
     private state;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -620,7 +671,8 @@ export declare class GenerateUniqueCompanyName extends AbstractGenerator<{
 }> {
     static readonly entityKind: string;
     private state;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -673,16 +725,21 @@ export declare class GeneratePoint extends AbstractGenerator<{
         y: number;
     };
 }
-export declare class GenerateUniquePoint extends AbstractGenerator<{
+export type GenerateUniquePointT = {
     minXValue?: number;
     maxXValue?: number;
     minYValue?: number;
     maxYValue?: number;
     isUnique?: boolean;
-}> {
+};
+export declare class GenerateUniquePoint extends AbstractGenerator<GenerateUniquePointT> {
     static readonly entityKind: string;
     private state;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
+    xCoordinateGen: GenerateUniqueNumber;
+    yCoordinateGen: GenerateUniqueNumber;
+    maxUniqueCount: number;
+    constructor(params?: GenerateUniquePointT);
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -715,7 +772,7 @@ export declare class GenerateLine extends AbstractGenerator<{
         c: number;
     };
 }
-export declare class GenerateUniqueLine extends AbstractGenerator<{
+export type GenerateUniqueLineT = {
     minAValue?: number;
     maxAValue?: number;
     minBValue?: number;
@@ -723,10 +780,16 @@ export declare class GenerateUniqueLine extends AbstractGenerator<{
     minCValue?: number;
     maxCValue?: number;
     isUnique?: boolean;
-}> {
+};
+export declare class GenerateUniqueLine extends AbstractGenerator<GenerateUniqueLineT> {
     static readonly entityKind: string;
     private state;
-    isUnique: boolean;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
+    aCoefficientGen: GenerateUniqueNumber;
+    bCoefficientGen: GenerateUniqueNumber;
+    cCoefficientGen: GenerateUniqueNumber;
+    constructor(params?: GenerateUniqueLineT);
     init({ count, seed }: {
         count: number;
         seed: number;
@@ -737,24 +800,184 @@ export declare class GenerateUniqueLine extends AbstractGenerator<{
         c: number;
     };
 }
-export declare class CustomGenerator<StateT extends Record<string, any> = {}> extends AbstractGenerator<{
-    generate: (params: {
-        this_: CustomGenerator;
-        i: number;
-    }) => any;
-    init?: (params: {
-        this_: CustomGenerator;
+export declare class GenerateBitString extends AbstractGenerator<{
+    dimensions?: number;
+    isUnique?: boolean;
+    arraySize?: number;
+}> {
+    static readonly entityKind: string;
+    dimensions: number;
+    private state;
+    uniqueVersionOfGen: typeof GenerateUniqueBitString;
+    init({ count, seed }: {
         count: number;
         seed: number;
-    }) => void;
+    }): void;
+    generate(): string;
+}
+export declare class GenerateUniqueBitString extends AbstractGenerator<{
+    dimensions?: number;
+    isUnique?: boolean;
+    arraySize?: number;
 }> {
-    state: StateT;
     static readonly entityKind: string;
+    dimensions: number;
+    private state;
+    isGeneratorUnique: boolean;
+    getMaxUniqueCount(): number;
     init({ count, seed }: {
         count: number;
         seed: number;
     }): void;
-    generate({ i }: {
+    generate(): string;
+}
+export declare class GenerateInet extends AbstractGenerator<{
+    ipAddress?: 'ipv4' | 'ipv6';
+    includeCidr?: boolean;
+    isUnique?: boolean;
+    arraySize?: number;
+}> {
+    static readonly entityKind: string;
+    ipAddress: 'ipv4' | 'ipv6';
+    includeCidr: boolean;
+    private state;
+    uniqueVersionOfGen: typeof GenerateUniqueInet;
+    init({ count, seed }: {
+        count: number;
+        seed: number;
+    }): void;
+    generate(): string;
+}
+export type GenerateUniqueInetT = {
+    ipAddress?: 'ipv4' | 'ipv6';
+    includeCidr?: boolean;
+    isUnique?: boolean;
+    arraySize?: number;
+};
+export declare class GenerateUniqueInet extends AbstractGenerator<GenerateUniqueInetT> {
+    static readonly entityKind: string;
+    ipAddress: 'ipv4' | 'ipv6';
+    includeCidr: boolean;
+    delimiter: '.' | ':';
+    private state;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
+    constructor(params?: GenerateUniqueInetT);
+    init({ count, seed }: {
+        count: number;
+        seed: number;
+    }): void;
+    generate(): string;
+}
+export declare class GenerateGeometry extends AbstractGenerator<{
+    type?: 'point';
+    srid?: 4326 | 3857;
+    decimalPlaces?: 1 | 2 | 3 | 4 | 5 | 6 | 7;
+    isUnique?: boolean;
+    arraySize?: number;
+}> {
+    static readonly entityKind: string;
+    type: "point";
+    srid: 4326 | 3857;
+    decimalPlaces: 1 | 2 | 3 | 4 | 5 | 6 | 7;
+    private state;
+    uniqueVersionOfGen: typeof GenerateUniqueGeometry;
+    init({ count, seed }: {
+        count: number;
+        seed: number;
+    }): void;
+    generate(): number[] | {
+        x: number;
+        y: number;
+    };
+}
+export type GenerateUniqueGeometryT = {
+    type?: 'point';
+    srid?: 4326 | 3857;
+    decimalPlaces?: 1 | 2 | 3 | 4 | 5 | 6 | 7;
+    isUnique?: boolean;
+    arraySize?: number;
+};
+export declare class GenerateUniqueGeometry extends AbstractGenerator<GenerateUniqueGeometryT> {
+    static readonly entityKind: string;
+    type: "point";
+    srid: 4326 | 3857;
+    decimalPlaces: 1 | 2 | 3 | 4 | 5 | 6 | 7;
+    private state;
+    isGeneratorUnique: boolean;
+    maxUniqueCount: number;
+    constructor(params?: GenerateUniqueGeometryT);
+    init({ count, seed }: {
+        count: number;
+        seed: number;
+    }): void;
+    generate(): number[] | {
+        x: number;
+        y: number;
+    };
+}
+export declare class GenerateVector extends AbstractGenerator<{
+    dimensions?: number;
+    minValue?: number;
+    maxValue?: number;
+    decimalPlaces?: number;
+    isUnique?: boolean;
+    arraySize?: number;
+}> {
+    static readonly entityKind: string;
+    dimensions: number;
+    minValue: number;
+    maxValue: number;
+    decimalPlaces: number;
+    private state;
+    uniqueVersionOfGen: typeof GenerateUniqueVector;
+    init({ count, seed }: {
+        count: number;
+        seed: number;
+    }): void;
+    generate(): unknown[];
+}
+export type GenerateUniqueVectorT = {
+    dimensions?: number;
+    minValue?: number;
+    maxValue?: number;
+    decimalPlaces?: number;
+    isUnique?: boolean;
+    arraySize?: number;
+};
+export declare class GenerateUniqueVector extends AbstractGenerator<GenerateUniqueVectorT> {
+    static readonly entityKind: string;
+    dimensions: number;
+    minValue: number;
+    maxValue: number;
+    decimalPlaces: number;
+    private state;
+    isGeneratorUnique: boolean;
+    constructor(params?: GenerateUniqueVectorT);
+    getMaxUniqueCount(): number;
+    init({ count, seed }: {
+        count: number;
+        seed: number;
+    }): void;
+    generate(): number[];
+}
+export declare class GenerateCompositeUniqueKey extends AbstractGenerator {
+    static readonly entityKind: string;
+    columnGenerators: {
+        columnName: string;
+        generator: AbstractGenerator<any>;
+        maxUniqueCount?: number;
+        count?: number;
+    }[];
+    private isInitialized;
+    private state;
+    addGenerator(columnName: string, generator: AbstractGenerator<any>): void;
+    init({ count, seed }: {
+        count: number;
+        seed: number;
+    }): void;
+    generate({ i, columnName }: {
         i: number;
+        columnName: string;
     }): any;
 }

package/generators/utils.d.cts

@@ -0,0 +1,38 @@
+export declare const fastCartesianProduct: <SetsT extends ((number | string | boolean | object)[] | OrderedNumberRange | OrderedBigintRange)[]>(sets: SetsT, index: number) => SetsT[number][number][];
+export declare const fastCartesianProductForBigint: <SetsT extends ((number | string | boolean | object)[] | OrderedNumberRange | OrderedBigintRange)[]>(sets: SetsT, index: bigint) => SetsT[number][number][];
+export declare class OrderedNumberRange<T extends number = number> {
+    private readonly min;
+    private readonly max;
+    private readonly step;
+    [index: number]: T;
+    readonly length: number;
+    constructor(min: T, max: T, step: T);
+}
+export declare class OrderedBigintRange<T extends bigint = bigint> {
+    private readonly min;
+    private readonly max;
+    private readonly step;
+    [index: number]: T;
+    readonly length: bigint;
+    constructor(min: T, max: T, step: T);
+}
+export declare const abs: (n: number | bigint) => number | bigint;
+/**
+ * @param weights positive number in range [0, 1], that represents probabilities to choose index of array. Example: weights = [0.2, 0.8]
+ * @param [accuracy=100] approximate number of elements in returning array
+ * @returns Example: with weights = [0.2, 0.8] and accuracy = 10 returning array of indices gonna equal this: [0, 0, 1, 1, 1, 1, 1, 1, 1, 1]
+ */
+export declare const getWeightedIndices: (weights: number[], accuracy?: number) => number[];
+/**
+ * @param param0.template example: "#####" or "#####-####"
+ * @param param0.values example: ["3", "2", "h"]
+ * @param param0.defaultValue example: "0"
+ * @returns
+ */
+export declare const fillTemplate: ({ template, placeholdersCount, values, defaultValue }: {
+    template: string;
+    placeholdersCount?: number;
+    values: string[];
+    defaultValue?: string;
+}) => string;
+export declare const isObject: (value: any) => boolean;

package/generators/utils.d.mts

@@ -0,0 +1,38 @@
+export declare const fastCartesianProduct: <SetsT extends ((number | string | boolean | object)[] | OrderedNumberRange | OrderedBigintRange)[]>(sets: SetsT, index: number) => SetsT[number][number][];
+export declare const fastCartesianProductForBigint: <SetsT extends ((number | string | boolean | object)[] | OrderedNumberRange | OrderedBigintRange)[]>(sets: SetsT, index: bigint) => SetsT[number][number][];
+export declare class OrderedNumberRange<T extends number = number> {
+    private readonly min;
+    private readonly max;
+    private readonly step;
+    [index: number]: T;
+    readonly length: number;
+    constructor(min: T, max: T, step: T);
+}
+export declare class OrderedBigintRange<T extends bigint = bigint> {
+    private readonly min;
+    private readonly max;
+    private readonly step;
+    [index: number]: T;
+    readonly length: bigint;
+    constructor(min: T, max: T, step: T);
+}
+export declare const abs: (n: number | bigint) => number | bigint;
+/**
+ * @param weights positive number in range [0, 1], that represents probabilities to choose index of array. Example: weights = [0.2, 0.8]
+ * @param [accuracy=100] approximate number of elements in returning array
+ * @returns Example: with weights = [0.2, 0.8] and accuracy = 10 returning array of indices gonna equal this: [0, 0, 1, 1, 1, 1, 1, 1, 1, 1]
+ */
+export declare const getWeightedIndices: (weights: number[], accuracy?: number) => number[];
+/**
+ * @param param0.template example: "#####" or "#####-####"
+ * @param param0.values example: ["3", "2", "h"]
+ * @param param0.defaultValue example: "0"
+ * @returns
+ */
+export declare const fillTemplate: ({ template, placeholdersCount, values, defaultValue }: {
+    template: string;
+    placeholdersCount?: number;
+    values: string[];
+    defaultValue?: string;
+}) => string;
+export declare const isObject: (value: any) => boolean;