SeedPromise Class — drizzle-orm Architecture

Source Code

drizzle-seed/src/index.ts lines 137–206

class SeedPromise<
	DB extends
		| PgDatabase<any, any>
		| MySqlDatabase<any, any>
		| BaseSQLiteDatabase<any, any>,
	SCHEMA extends {
		[key: string]: PgTable | PgSchema | MySqlTable | MySqlSchema | SQLiteTable | Relations;
	},
	VERSION extends string | undefined,
> implements Promise<void> {
	static readonly entityKind: string = 'SeedPromise';

	[Symbol.toStringTag] = 'SeedPromise';

	constructor(
		private db: DB,
		private schema: SCHEMA,
		private options?: { count?: number; seed?: number; version?: VERSION },
	) {}

	then<TResult1 = void, TResult2 = never>(
		onfulfilled?:
			| ((value: void) => TResult1 | PromiseLike<TResult1>)
			| null
			| undefined,
		onrejected?:
			| ((reason: any) => TResult2 | PromiseLike<TResult2>)
			| null
			| undefined,
	): Promise<TResult1 | TResult2> {
		return seedFunc(this.db, this.schema, this.options).then(
			onfulfilled,
			onrejected,
		);
	}

	catch<TResult = never>(
		onrejected?:
			| ((reason: any) => TResult | PromiseLike<TResult>)
			| null
			| undefined,
	): Promise<void | TResult> {
		return this.then(undefined, onrejected);
	}

	finally(onfinally?: (() => void) | null | undefined): Promise<void> {
		return this.then(
			(value) => {
				onfinally?.();
				return value;
			},
			(reason) => {
				onfinally?.();
				throw reason;
			},
		);
	}

	async refine(
		callback: (
			funcs: FunctionsVersioning<VERSION>,
		) => InferCallbackType<DB, SCHEMA>,
	): Promise<void> {
		const refinements = this.options?.version === undefined || this.options.version === '2'
			? callback(generatorsFuncsV2 as FunctionsVersioning<VERSION>) as RefinementsType
			: callback(generatorsFuncs as FunctionsVersioning<VERSION>) as RefinementsType;

		await seedFunc(this.db, this.schema, this.options, refinements);
	}
}