testPolicyResolver() — drizzle-orm Function Reference

Source Code

drizzle-kit/tests/schemaDiffer.ts lines 464–526

export const testPolicyResolver = (renames: Set<string>) =>
async (
	input: TablePolicyResolverInput<Policy>,
): Promise<TablePolicyResolverOutput<Policy>> => {
	try {
		if (
			input.created.length === 0
			|| input.deleted.length === 0
			|| renames.size === 0
		) {
			return {
				tableName: input.tableName,
				schema: input.schema,
				created: input.created,
				renamed: [],
				deleted: input.deleted,
			};
		}

		let createdPolicies = [...input.created];
		let deletedPolicies = [...input.deleted];

		const renamed: { from: Policy; to: Policy }[] = [];

		const schema = input.schema || 'public';

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

			const idxFrom = deletedPolicies.findIndex((it) => {
				return `${schema}.${input.tableName}.${it.name}` === from;
			});

			if (idxFrom >= 0) {
				const idxTo = createdPolicies.findIndex((it) => {
					return `${schema}.${input.tableName}.${it.name}` === to;
				});

				renamed.push({
					from: deletedPolicies[idxFrom],
					to: createdPolicies[idxTo],
				});

				delete createdPolicies[idxTo];
				delete deletedPolicies[idxFrom];

				createdPolicies = createdPolicies.filter(Boolean);
				deletedPolicies = deletedPolicies.filter(Boolean);
			}
		}

		return {
			tableName: input.tableName,
			schema: input.schema,
			created: createdPolicies,
			deleted: deletedPolicies,
			renamed,
		};
	} catch (e) {
		console.error(e);
		throw e;
	}
};