applyPgDiffs() — drizzle-orm Function Reference

Source Code

drizzle-kit/tests/schemaDiffer.ts lines 1130–1212

export const applyPgDiffs = async (
	sn: PostgresSchema,
	casing: CasingType | undefined,
) => {
	const dryRun = {
		version: '7',
		dialect: 'postgresql',
		id: '0',
		prevId: '0',
		tables: {},
		enums: {},
		views: {},
		schemas: {},
		sequences: {},
		policies: {},
		roles: {},
		_meta: {
			schemas: {},
			tables: {},
			columns: {},
		},
	} as const;

	const tables = Object.values(sn).filter((it) => is(it, PgTable)) as PgTable[];

	const schemas = Object.values(sn).filter((it) => is(it, PgSchema)) as PgSchema[];

	const enums = Object.values(sn).filter((it) => isPgEnum(it)) as PgEnum<any>[];

	const sequences = Object.values(sn).filter((it) => isPgSequence(it)) as PgSequence[];

	const roles = Object.values(sn).filter((it) => is(it, PgRole)) as PgRole[];

	const views = Object.values(sn).filter((it) => isPgView(it)) as PgView[];

	const policies = Object.values(sn).filter((it) => is(it, PgPolicy)) as PgPolicy[];

	const materializedViews = Object.values(sn).filter((it) => isPgMaterializedView(it)) as PgMaterializedView[];

	const serialized1 = generatePgSnapshot(
		tables,
		enums,
		schemas,
		sequences,
		roles,
		policies,
		views,
		materializedViews,
		casing,
	);

	const { version: v1, dialect: d1, ...rest1 } = serialized1;

	const sch1 = {
		version: '7',
		dialect: 'postgresql',
		id: '0',
		prevId: '0',
		...rest1,
	} as const;

	const sn1 = squashPgScheme(sch1);

	const validatedPrev = pgSchema.parse(dryRun);
	const validatedCur = pgSchema.parse(sch1);

	const { sqlStatements, statements } = await applyPgSnapshotsDiff(
		dryRun,
		sn1,
		testSchemasResolver(new Set()),
		testEnumsResolver(new Set()),
		testSequencesResolver(new Set()),
		testPolicyResolver(new Set()),
		testIndPolicyResolver(new Set()),
		testRolesResolver(new Set()),
		testTablesResolver(new Set()),
		testColumnsResolver(new Set()),
		testViewsResolver(new Set()),
		validatedPrev,
		validatedCur,
	);