testEnumsResolver() — drizzle-orm Function Reference

Source Code

drizzle-kit/tests/schemaDiffer.ts lines 248–322

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

		let createdEnums = [...input.created];
		let deletedEnums = [...input.deleted];

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

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

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

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

				const tableFrom = deletedEnums[idxFrom];
				const tableTo = createdEnums[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: deletedEnums[idxFrom],
						to: createdEnums[idxTo],
					});
				}

				delete createdEnums[idxTo];
				delete deletedEnums[idxFrom];

				createdEnums = createdEnums.filter(Boolean);
				deletedEnums = deletedEnums.filter(Boolean);
			}
		}

		result.created = createdEnums;
		result.deleted = deletedEnums;

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