testSequencesResolver() — drizzle-orm Function Reference

Source Code

drizzle-kit/tests/schemaDiffer.ts lines 172–246

export const testSequencesResolver = (renames: Set<string>) =>
async (
	input: ResolverInput<Sequence>,
): Promise<ResolverOutputWithMoved<Sequence>> => {
	try {
		if (
			input.created.length === 0
			|| input.deleted.length === 0
			|| renames.size === 0
		) {
			return {
				created: input.created,
				moved: [],
				renamed: [],
				deleted: input.deleted,
			};
		}

		let createdSequences = [...input.created];
		let deletedSequences = [...input.deleted];

		const result: {
			created: Sequence[];
			moved: { name: string; schemaFrom: string; schemaTo: string }[];
			renamed: { from: Sequence; to: Sequence }[];
			deleted: Sequence[];
		} = { created: [], renamed: [], deleted: [], moved: [] };

		for (let rename of renames) {
			const [from, to] = rename.split('->');

			const idxFrom = deletedSequences.findIndex((it) => {
				return `${it.schema || 'public'}.${it.name}` === from;
			});

			if (idxFrom >= 0) {
				const idxTo = createdSequences.findIndex((it) => {
					return `${it.schema || 'public'}.${it.name}` === to;
				});

				const tableFrom = deletedSequences[idxFrom];
				const tableTo = createdSequences[idxFrom];

				if (tableFrom.schema !== tableTo.schema) {
					result.moved.push({
						name: tableFrom.name,
						schemaFrom: tableFrom.schema,
						schemaTo: tableTo.schema,
					});
				}

				if (tableFrom.name !== tableTo.name) {
					result.renamed.push({
						from: deletedSequences[idxFrom],
						to: createdSequences[idxTo],
					});
				}

				delete createdSequences[idxTo];
				delete deletedSequences[idxFrom];

				createdSequences = createdSequences.filter(Boolean);
				deletedSequences = deletedSequences.filter(Boolean);
			}
		}

		result.created = createdSequences;
		result.deleted = deletedSequences;

		return result;
	} catch (e) {
		console.error(e);
		throw e;
	}
};