squashPgScheme() — drizzle-orm Function Reference

Source Code

drizzle-kit/src/serializer/pgSchema.ts lines 758–868

export const squashPgScheme = (
	json: PgSchema,
	action?: 'push' | undefined,
): PgSchemaSquashed => {
	const mappedTables = Object.fromEntries(
		Object.entries(json.tables).map((it) => {
			const squashedIndexes = mapValues(it[1].indexes, (index) => {
				return action === 'push'
					? PgSquasher.squashIdxPush(index)
					: PgSquasher.squashIdx(index);
			});

			const squashedFKs = mapValues(it[1].foreignKeys, (fk) => {
				return PgSquasher.squashFK(fk);
			});

			const squashedPKs = mapValues(it[1].compositePrimaryKeys, (pk) => {
				return PgSquasher.squashPK(pk);
			});

			const mappedColumns = Object.fromEntries(
				Object.entries(it[1].columns).map((it) => {
					const mappedIdentity = it[1].identity
						? PgSquasher.squashIdentity(it[1].identity)
						: undefined;
					return [
						it[0],
						{
							...it[1],
							identity: mappedIdentity,
						},
					];
				}),
			);

			const squashedUniqueConstraints = mapValues(
				it[1].uniqueConstraints,
				(unq) => {
					return PgSquasher.squashUnique(unq);
				},
			);

			const squashedPolicies = mapValues(it[1].policies, (policy) => {
				return action === 'push'
					? PgSquasher.squashPolicyPush(policy)
					: PgSquasher.squashPolicy(policy);
			});
			const squashedChecksContraints = mapValues(
				it[1].checkConstraints,
				(check) => {
					return PgSquasher.squashCheck(check);
				},
			);

			return [
				it[0],
				{
					name: it[1].name,
					schema: it[1].schema,
					columns: mappedColumns,
					indexes: squashedIndexes,
					foreignKeys: squashedFKs,
					compositePrimaryKeys: squashedPKs,
					uniqueConstraints: squashedUniqueConstraints,
					policies: squashedPolicies,
					checkConstraints: squashedChecksContraints,
					isRLSEnabled: it[1].isRLSEnabled ?? false,
				},
			];
		}),
	);

	const mappedSequences = Object.fromEntries(
		Object.entries(json.sequences).map((it) => {
			return [
				it[0],
				{
					name: it[1].name,
					schema: it[1].schema,
					values: PgSquasher.squashSequence(it[1]),
				},