prepareLibSQLCreateReferencesJson() — drizzle-orm Function Reference

Source Code

drizzle-kit/src/jsonStatements.ts lines 2939–2986

export const prepareLibSQLCreateReferencesJson = (
	tableName: string,
	schema: string,
	foreignKeys: Record<string, string>,
	json2: SQLiteSchemaSquashed,
	action?: 'push',
): JsonCreateReferenceStatement[] => {
	return Object.values(foreignKeys).map((fkData) => {
		const { columnsFrom, tableFrom, columnsTo } = action === 'push'
			? SQLiteSquasher.unsquashPushFK(fkData)
			: SQLiteSquasher.unsquashFK(fkData);

		// When trying to alter table in lib sql it is necessary to pass all config for column like "NOT NULL", "DEFAULT", etc.
		// If it is multicolumn reference it is not possible to pass this data for all columns
		// Pass multicolumn flag for sql statements to not generate migration
		let isMulticolumn = false;

		if (columnsFrom.length > 1 || columnsTo.length > 1) {
			isMulticolumn = true;

			return {
				type: 'create_reference',
				tableName,
				data: fkData,
				schema,
				isMulticolumn,
			};
		}

		const columnFrom = columnsFrom[0];

		const {
			notNull: columnNotNull,
			default: columnDefault,
			type: columnType,
		} = json2.tables[tableFrom].columns[columnFrom];

		return {
			type: 'create_reference',
			tableName,
			data: fkData,
			schema,
			columnNotNull,
			columnDefault,
			columnType,
		};
	});
};