prepareAndMigratePg() — drizzle-orm Function Reference

Source Code

drizzle-kit/src/cli/commands/migrate.ts lines 302–369

export const prepareAndMigratePg = async (config: GenerateConfig) => {
	const outFolder = config.out;
	const schemaPath = config.schema;
	const casing = config.casing;

	try {
		assertV1OutFolder(outFolder);

		const { snapshots, journal } = prepareMigrationFolder(
			outFolder,
			'postgresql',
		);

		const { prev, cur, custom } = await preparePgMigrationSnapshot(
			snapshots,
			schemaPath,
			casing,
		);

		const validatedPrev = pgSchema.parse(prev);
		const validatedCur = pgSchema.parse(cur);

		if (config.custom) {
			writeResult({
				cur: custom,
				sqlStatements: [],
				journal,
				outFolder,
				name: config.name,
				breakpoints: config.breakpoints,
				type: 'custom',
				prefixMode: config.prefix,
			});
			return;
		}

		const squashedPrev = squashPgScheme(validatedPrev);
		const squashedCur = squashPgScheme(validatedCur);

		const { sqlStatements, _meta } = await applyPgSnapshotsDiff(
			squashedPrev,
			squashedCur,
			schemasResolver,
			enumsResolver,
			sequencesResolver,
			policyResolver,
			indPolicyResolver,
			roleResolver,
			tablesResolver,
			columnsResolver,
			viewsResolver,
			validatedPrev,
			validatedCur,
		);

		writeResult({
			cur,
			sqlStatements,
			journal,
			outFolder,
			name: config.name,
			breakpoints: config.breakpoints,
			prefixMode: config.prefix,
		});
	} catch (e) {
		console.error(e);
	}
};