applyPgSnapshotsDiff() — drizzle-orm Function Reference

Source Code

drizzle-kit/src/snapshotsDiffer.ts lines 559–2131

export const applyPgSnapshotsDiff = async (
	json1: PgSchemaSquashed,
	json2: PgSchemaSquashed,
	schemasResolver: (
		input: ResolverInput<Named>,
	) => Promise<ResolverOutput<Named>>,
	enumsResolver: (
		input: ResolverInput<Enum>,
	) => Promise<ResolverOutputWithMoved<Enum>>,
	sequencesResolver: (
		input: ResolverInput<Sequence>,
	) => Promise<ResolverOutputWithMoved<Sequence>>,
	policyResolver: (
		input: TablePolicyResolverInput<Policy>,
	) => Promise<TablePolicyResolverOutput<Policy>>,
	indPolicyResolver: (
		input: PolicyResolverInput<Policy>,
	) => Promise<PolicyResolverOutput<Policy>>,
	roleResolver: (
		input: RolesResolverInput<Role>,
	) => Promise<RolesResolverOutput<Role>>,
	tablesResolver: (
		input: ResolverInput<Table>,
	) => Promise<ResolverOutputWithMoved<Table>>,
	columnsResolver: (
		input: ColumnsResolverInput<Column>,
	) => Promise<ColumnsResolverOutput<Column>>,
	viewsResolver: (
		input: ResolverInput<View>,
	) => Promise<ResolverOutputWithMoved<View>>,
	prevFull: PgSchema,
	curFull: PgSchema,
	action?: 'push' | undefined,
): Promise<{
	statements: JsonStatement[];
	sqlStatements: string[];
	_meta:
		| {
			schemas: {};
			tables: {};
			columns: {};
		}
		| undefined;
}> => {
	const schemasDiff = diffSchemasOrTables(json1.schemas, json2.schemas);

	const {
		created: createdSchemas,
		deleted: deletedSchemas,
		renamed: renamedSchemas,
	} = await schemasResolver({
		created: schemasDiff.added.map((it) => ({ name: it })),
		deleted: schemasDiff.deleted.map((it) => ({ name: it })),
	});

	const schemasPatchedSnap1 = copy(json1);
	schemasPatchedSnap1.tables = mapEntries(
		schemasPatchedSnap1.tables,
		(_, it) => {
			const { key, schema } = schemaChangeFor(it, renamedSchemas);
			it.schema = schema;
			return [key, it];
		},
	);

	schemasPatchedSnap1.enums = mapEntries(schemasPatchedSnap1.enums, (_, it) => {
		const { key, schema } = schemaChangeFor(it, renamedSchemas);
		it.schema = schema;
		return [key, it];
	});

	const enumsDiff = diffSchemasOrTables(schemasPatchedSnap1.enums, json2.enums);

	const {
		created: createdEnums,
		deleted: deletedEnums,
		renamed: renamedEnums,
		moved: movedEnums,
	} = await enumsResolver({
		created: enumsDiff.added,
		deleted: enumsDiff.deleted,