createTableColumns() — drizzle-orm Function Reference

Source Code

drizzle-kit/src/introspect-pg.ts lines 1112–1199

const createTableColumns = (
	tableName: string,
	columns: Column[],
	fks: ForeignKey[],
	enumTypes: Set<string>,
	schemas: Record<string, string>,
	casing: Casing,
	internals: PgKitInternals,
): string => {
	let statement = '';

	// no self refs and no cyclic
	const oneColumnsFKs = Object.values(fks)
		.filter((it) => {
			return !isSelf(it);
		})
		.filter((it) => it.columnsFrom.length === 1);

	const fkByColumnName = oneColumnsFKs.reduce((res, it) => {
		const arr = res[it.columnsFrom[0]] || [];
		arr.push(it);
		res[it.columnsFrom[0]] = arr;
		return res;
	}, {} as Record<string, ForeignKey[]>);

	columns.forEach((it) => {
		const columnStatement = column(
			tableName,
			it.type,
			it.name,
			enumTypes,
			it.typeSchema ?? 'public',
			casing,
			it.default,
			internals,
		);
		statement += '\t';
		statement += columnStatement;
		// Provide just this in column function
		if (internals?.tables[tableName]?.columns[it.name]?.isArray) {
			statement += dimensionsInArray(internals?.tables[tableName]?.columns[it.name]?.dimensions);
		}
		statement += mapDefault(tableName, it.type, it.name, enumTypes, it.typeSchema ?? 'public', it.default, internals);
		statement += it.primaryKey ? '.primaryKey()' : '';
		statement += it.notNull && !it.identity ? '.notNull()' : '';

		statement += it.identity ? generateIdentityParams(it.identity) : '';

		statement += it.generated ? `.generatedAlwaysAs(sql\`${it.generated.as}\`)` : '';

		// const fks = fkByColumnName[it.name];
		// Andrii: I switched it off until we will get a custom naem setting in references
		// if (fks) {
		// 	const fksStatement = fks
		// 		.map((it) => {
		// 			const onDelete = it.onDelete && it.onDelete !== 'no action' ? it.onDelete : null;
		// 			const onUpdate = it.onUpdate && it.onUpdate !== 'no action' ? it.onUpdate : null;
		// 			const params = { onDelete, onUpdate };

		// 			const typeSuffix = isCyclic(it) ? ': AnyPgColumn' : '';

		// 			const paramsStr = objToStatement2(params);
		// 			const tableSchema = schemas[it.schemaTo || ''];
		// 			const paramName = paramNameFor(it.tableTo, tableSchema);
		// 			if (paramsStr) {
		// 				return `.references(()${typeSuffix} => ${
		// 					withCasing(
		// 						paramName,
		// 						casing,
		// 					)
		// 				}.${withCasing(it.columnsTo[0], casing)}, ${paramsStr} )`;
		// 			}
		// 			return `.references(()${typeSuffix} => ${
		// 				withCasing(
		// 					paramName,
		// 					casing,
		// 				)
		// 			}.${withCasing(it.columnsTo[0], casing)})`;
		// 		})
		// 		.join('');
		// 	statement += fksStatement;