arkormx 1.1.0 → 1.2.1

package/dist/cli.mjs

@@ -129,6 +129,95 @@ var ForeignKeyBuilder = class {
 
 //#endregion
 //#region src/database/TableBuilder.ts
+const PRISMA_ENUM_MEMBER_REGEX$1 = /^[A-Za-z][A-Za-z0-9_]*$/;
+const normalizeEnumMember = (columnName, value) => {
+	const normalized = value.trim();
+	if (!normalized) throw new Error(`Enum column [${columnName}] must define only non-empty values.`);
+	if (!PRISMA_ENUM_MEMBER_REGEX$1.test(normalized)) throw new Error(`Enum column [${columnName}] contains invalid Prisma enum value [${normalized}].`);
+	return normalized;
+};
+const normalizeEnumMembers = (columnName, values) => {
+	const normalizedValues = values.map((value) => normalizeEnumMember(columnName, value));
+	const seen = /* @__PURE__ */ new Set();
+	for (const value of normalizedValues) {
+		if (seen.has(value)) throw new Error(`Enum column [${columnName}] contains duplicate enum value [${value}].`);
+		seen.add(value);
+	}
+	return normalizedValues;
+};
+/**
+* The EnumBuilder class provides a fluent interface for configuring enum columns
+* after they are defined on a table.
+*
+* @author Legacy (3m1n3nc3)
+* @since 0.2.3
+*/
+var EnumBuilder = class {
+	tableBuilder;
+	columnName;
+	constructor(tableBuilder, columnName) {
+		this.tableBuilder = tableBuilder;
+		this.columnName = columnName;
+	}
+	/**
+	* Defines the Prisma enum name for this column.
+	*
+	* @param name
+	* @returns
+	*/
+	enumName(name) {
+		this.tableBuilder.enumName(name, this.columnName);
+		return this;
+	}
+	/**
+	* Marks the enum column as nullable.
+	*
+	* @returns
+	*/
+	nullable() {
+		this.tableBuilder.nullable(this.columnName);
+		return this;
+	}
+	/**
+	* Marks the enum column as unique.
+	*
+	* @returns
+	*/
+	unique() {
+		this.tableBuilder.unique(this.columnName);
+		return this;
+	}
+	/**
+	* Sets a default value for the enum column.
+	*
+	* @param value
+	* @returns
+	*/
+	default(value) {
+		this.tableBuilder.default(value, this.columnName);
+		return this;
+	}
+	/**
+	* Positions the enum column after another column when supported.
+	*
+	* @param referenceColumn
+	* @returns
+	*/
+	after(referenceColumn) {
+		this.tableBuilder.after(referenceColumn, this.columnName);
+		return this;
+	}
+	/**
+	* Maps the enum column to a custom database column name.
+	*
+	* @param name
+	* @returns
+	*/
+	map(name) {
+		this.tableBuilder.map(name, this.columnName);
+		return this;
+	}
+};
 /**
 * The TableBuilder class provides a fluent interface for defining 
 * the structure of a database table in a migration, including columns to add or drop.
@@ -181,6 +270,27 @@ var TableBuilder = class {
 		return this.column(name, "uuid", options);
 	}
 	/**
+	* Defines an enum column in the table.
+	*
+	* @param name      The name of the enum column.
+	* @param values    Either an array of string values for the enum or the name of an existing enum to reuse.
+	* @param options   Additional options for the enum column.
+	* @returns
+	*/
+	enum(name, valuesOrEnumName, options = {}) {
+		const isEnumReuse = typeof valuesOrEnumName === "string";
+		if (!isEnumReuse && valuesOrEnumName.length === 0) throw new Error(`Enum column [${name}] must define at least one value.`);
+		const normalizedEnumValues = isEnumReuse ? void 0 : normalizeEnumMembers(name, valuesOrEnumName);
+		const enumName = isEnumReuse ? valuesOrEnumName.trim() : options.enumName?.trim();
+		if (isEnumReuse && !enumName) throw new Error(`Enum column [${name}] must define an enum name.`);
+		this.column(name, "enum", {
+			...options,
+			enumName,
+			enumValues: normalizedEnumValues
+		});
+		return new EnumBuilder(this, name);
+	}
+	/**
 	* Defines a string column in the table.
 	* 
 	* @param name      The name of the string column.
@@ -350,6 +460,21 @@ var TableBuilder = class {
 		return this;
 	}
 	/**
+	* Sets the Prisma enum name for an enum column.
+	*
+	* @param name       The enum name to assign.
+	* @param columnName Optional explicit target column name. When omitted, applies to the latest defined column.
+	* @returns          The current TableBuilder instance for chaining.
+	*/
+	enumName(name, columnName) {
+		const column = this.resolveColumn(columnName);
+		if (column.type !== "enum") throw new Error(`Column [${column.name}] is not an enum column.`);
+		const enumName = name.trim();
+		if (!enumName) throw new Error(`Enum column [${column.name}] must define an enum name.`);
+		column.enumName = enumName;
+		return this;
+	}
+	/**
 	* Sets a default value for a column.
 	*
 	* @param value      The default scalar value or Prisma expression (e.g. 'now()').
@@ -424,7 +549,7 @@ var TableBuilder = class {
 	*/
 	foreign(column) {
 		const columnName = this.resolveColumn(column).name;
-		return this.foreignKey(columnName + (column ? "" : "Id"));
+		return this.foreignKey(column ?? columnName);
 	}
 	/**
 	* Returns a deep copy of the defined columns for the table.
@@ -432,7 +557,10 @@ var TableBuilder = class {
 	* @returns 
 	*/
 	getColumns() {
-		return this.columns.map((column) => ({ ...column }));
+		return this.columns.map((column) => ({
+			...column,
+			enumValues: column.enumValues ? [...column.enumValues] : void 0
+		}));
 	}
 	/**
 	* Returns a copy of the defined column names to be dropped from the table.
@@ -473,6 +601,8 @@ var TableBuilder = class {
 		this.columns.push({
 			name,
 			type,
+			enumName: options.enumName,
+			enumValues: options.enumValues ? [...options.enumValues] : void 0,
 			map: options.map,
 			nullable: options.nullable,
 			unique: options.unique,
@@ -572,7 +702,10 @@ var SchemaBuilder = class {
 		return this.operations.map((operation) => {
 			if (operation.type === "createTable") return {
 				...operation,
-				columns: operation.columns.map((column) => ({ ...column })),
+				columns: operation.columns.map((column) => ({
+					...column,
+					enumValues: column.enumValues ? [...column.enumValues] : void 0
+				})),
 				indexes: operation.indexes.map((index) => ({
 					...index,
 					columns: [...index.columns]
@@ -581,7 +714,10 @@ var SchemaBuilder = class {
 			};
 			if (operation.type === "alterTable") return {
 				...operation,
-				addColumns: operation.addColumns.map((column) => ({ ...column })),
+				addColumns: operation.addColumns.map((column) => ({
+					...column,
+					enumValues: column.enumValues ? [...column.enumValues] : void 0
+				})),
 				dropColumns: [...operation.dropColumns],
 				addIndexes: operation.addIndexes.map((index) => ({
 					...index,
@@ -597,6 +733,8 @@ var SchemaBuilder = class {
 //#endregion
 //#region src/helpers/migrations.ts
 const PRISMA_MODEL_REGEX = /model\s+(\w+)\s*\{[\s\S]*?\n\}/g;
+const PRISMA_ENUM_REGEX = /enum\s+(\w+)\s*\{[\s\S]*?\n\}/g;
+const PRISMA_ENUM_MEMBER_REGEX = /^[A-Za-z][A-Za-z0-9_]*$/;
 /**
 * Convert a table name to a PascalCase model name, with basic singularization.
 * 
@@ -624,6 +762,7 @@ const escapeRegex = (value) => value.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
 */
 const resolvePrismaType = (column) => {
 	if (column.type === "id") return "Int";
+	if (column.type === "enum") return resolveEnumName(column);
 	if (column.type === "uuid") return "String";
 	if (column.type === "string" || column.type === "text") return "String";
 	if (column.type === "integer") return "Int";
@@ -633,6 +772,11 @@ const resolvePrismaType = (column) => {
 	if (column.type === "json") return "Json";
 	return "DateTime";
 };
+const resolveEnumName = (column) => {
+	if (column.type !== "enum") throw new ArkormException(`Column [${column.name}] is not an enum column.`);
+	if (column.enumName && column.enumName.trim().length > 0) return column.enumName.trim();
+	throw new ArkormException(`Enum column [${column.name}] must define an enum name.`);
+};
 /** 
 * Format a default value for inclusion in a Prisma schema field definition, based on its type.
 * 
@@ -647,6 +791,61 @@ const formatDefaultValue = (value) => {
 	if (typeof value === "boolean") return `@default(${value ? "true" : "false"})`;
 };
 /**
+* Format a default value for an enum column as a Prisma @default attribute, validating that it is a non-empty string.
+* 
+* @param value 
+* @returns 
+*/
+const formatEnumDefaultValue = (value) => {
+	if (value == null) return void 0;
+	if (typeof value !== "string" || value.trim().length === 0) throw new ArkormException("Enum default values must be provided as non-empty strings.");
+	return `@default(${value.trim()})`;
+};
+/**
+* Normalize an enum value by ensuring it is a non-empty string and trimming whitespace.
+* 
+* @param value 
+* @returns 
+*/
+const normalizeEnumValue = (value) => {
+	if (typeof value !== "string" || value.trim().length === 0) throw new ArkormException("Enum values must be provided as non-empty strings.");
+	const normalized = value.trim();
+	if (!PRISMA_ENUM_MEMBER_REGEX.test(normalized)) throw new ArkormException(`Enum value [${normalized}] is not a valid Prisma enum member name.`);
+	return normalized;
+};
+/**
+* Extract the enum values from a Prisma enum block string.
+* 
+* @param block 
+* @returns 
+*/
+const extractEnumBlockValues = (block) => {
+	return block.split("\n").slice(1, -1).map((line) => line.trim()).filter(Boolean);
+};
+const validateEnumValues = (column, enumName, enumValues) => {
+	const normalizedValues = enumValues.map(normalizeEnumValue);
+	const seen = /* @__PURE__ */ new Set();
+	for (const value of normalizedValues) {
+		if (seen.has(value)) throw new ArkormException(`Prisma enum [${enumName}] for column [${column.name}] contains duplicate value [${value}].`);
+		seen.add(value);
+	}
+	return normalizedValues;
+};
+/**
+* Validate that a default value for an enum column is included in the defined enum values.
+* 
+* @param column 
+* @param enumName 
+* @param enumValues 
+* @returns 
+*/
+const validateEnumDefaultValue = (column, enumName, enumValues) => {
+	if (column.default == null) return;
+	const normalizedDefault = normalizeEnumValue(column.default);
+	if (enumValues.includes(normalizedDefault)) return;
+	throw new ArkormException(`Enum default value [${normalizedDefault}] is not defined in Prisma enum [${enumName}] for column [${column.name}].`);
+};
+/**
 * Build a single line of a Prisma model field definition based on a SchemaColumn, including type and modifiers.
 * 
 * @param column 
@@ -667,11 +866,82 @@ const buildFieldLine = (column) => {
 	const primary = column.primary ? " @id" : "";
 	const mapped = typeof column.map === "string" && column.map.trim().length > 0 ? ` @map("${column.map.replace(/"/g, "\\\"")}")` : "";
 	const updatedAt = column.updatedAt ? " @updatedAt" : "";
-	const defaultValue = formatDefaultValue(column.default) ?? (column.type === "uuid" && column.primary ? "@default(uuid())" : void 0);
+	const defaultValue = column.type === "enum" ? formatEnumDefaultValue(column.default) : formatDefaultValue(column.default) ?? (column.type === "uuid" && column.primary ? "@default(uuid())" : void 0);
 	const defaultSuffix = defaultValue ? ` ${defaultValue}` : "";
 	return `  ${column.name} ${scalar}${nullable}${primary}${unique}${defaultSuffix}${updatedAt}${mapped}`;
 };
 /**
+* Build a Prisma enum block string based on an enum name and its values, validating that 
+* at least one value is provided.
+* 
+* @param enumName      The name of the enum to create.
+* @param values        The array of values for the enum.
+* @returns             The Prisma enum block string.
+*/
+const buildEnumBlock = (enumName, values) => {
+	if (values.length === 0) throw new ArkormException(`Enum [${enumName}] must define at least one value.`);
+	return `enum ${enumName} {\n${values.map((value) => `  ${value}`).join("\n")}\n}`;
+};
+/**
+* Find the Prisma enum block in a schema string that corresponds to a given enum 
+* name, returning its details if found.
+* 
+* @param schema        The Prisma schema string to search for the enum block.
+* @param enumName      The name of the enum to find in the schema.
+* @returns 
+*/
+const findEnumBlock = (schema, enumName) => {
+	const candidates = [...schema.matchAll(PRISMA_ENUM_REGEX)];
+	for (const match of candidates) {
+		const block = match[0];
+		const matchedEnumName = match[1];
+		const start = match.index ?? 0;
+		const end = start + block.length;
+		if (matchedEnumName === enumName) return {
+			enumName: matchedEnumName,
+			block,
+			start,
+			end
+		};
+	}
+	return null;
+};
+/**
+* Ensure that Prisma enum blocks exist in the schema for any enum columns defined in a 
+* create or alter table operation, adding them if necessary and validating against 
+* existing blocks.
+* 
+* @param schema    The current Prisma schema string to check and modify.
+* @param columns   The array of schema column definitions to check for enum types and ensure corresponding blocks exist for.
+* @returns 
+*/
+const ensureEnumBlocks = (schema, columns) => {
+	let nextSchema = schema;
+	for (const column of columns) {
+		if (column.type !== "enum") continue;
+		const enumName = resolveEnumName(column);
+		const enumValues = column.enumValues ?? [];
+		const existing = findEnumBlock(nextSchema, enumName);
+		if (existing) {
+			const existingValues = validateEnumValues(column, enumName, extractEnumBlockValues(existing.block));
+			if (enumValues.length === 0) {
+				validateEnumDefaultValue(column, enumName, existingValues);
+				continue;
+			}
+			const normalizedEnumValues = validateEnumValues(column, enumName, enumValues);
+			if (existingValues.join("|") !== normalizedEnumValues.join("|")) throw new ArkormException(`Prisma enum [${enumName}] already exists with different values.`);
+			validateEnumDefaultValue(column, enumName, existingValues);
+			continue;
+		}
+		if (enumValues.length === 0) throw new ArkormException(`Prisma enum [${enumName}] was not found for column [${column.name}].`);
+		const normalizedEnumValues = validateEnumValues(column, enumName, enumValues);
+		validateEnumDefaultValue(column, enumName, normalizedEnumValues);
+		const block = buildEnumBlock(enumName, normalizedEnumValues);
+		nextSchema = `${nextSchema.trimEnd()}\n\n${block}\n`;
+	}
+	return nextSchema;
+};
+/**
 * Build a Prisma model-level @@index definition line.
 * 
 * @param index     The schema index definition to convert to a Prisma \@\@index line.
@@ -697,24 +967,23 @@ const deriveRelationFieldName = (columnName) => {
 	if (trimmed.endsWith("_id") && trimmed.length > 3) return trimmed.slice(0, -3).replace(/_([a-zA-Z0-9])/g, (_, letter) => letter.toUpperCase());
 	return `${trimmed.charAt(0).toLowerCase()}${trimmed.slice(1)}`;
 };
-const pascalWords = (value) => {
-	return value.match(/[A-Z][a-z0-9]*/g) ?? [value];
-};
 /**
-* Derive a relation name for the inverse side of a relation based on the 
+* Derive a relation name for both sides of a relation based on the 
 * source and target model names, using an explicit alias if provided or a 
-* convention of combining the target model name with the last segment of 
-* the source model name.
+* convention of combining the full source model name with the target model name.
 * 
 * @param sourceModelName    The name of the source model in the relation.
 * @param targetModelName    The name of the target model in the relation.
-* @param explicitAlias      An optional explicit alias for the inverse relation.
-* @returns                  The derived or explicit inverse relation alias.
+* @param explicitAlias      An optional explicit alias for the relation.
+* @returns                  The derived or explicit relation alias.
 */
-const deriveInverseRelationAlias = (sourceModelName, targetModelName, explicitAlias) => {
+const deriveRelationAlias = (sourceModelName, targetModelName, explicitAlias) => {
 	if (explicitAlias && explicitAlias.trim().length > 0) return explicitAlias.trim();
-	const sourceWords = pascalWords(sourceModelName);
-	return `${sourceWords[sourceWords.length - 1] ?? sourceModelName}${targetModelName}`;
+	return [sourceModelName, targetModelName].sort((left, right) => left.localeCompare(right)).join("");
+};
+const deriveSingularFieldName = (modelName) => {
+	if (!modelName) return "item";
+	return `${modelName.charAt(0).toLowerCase()}${modelName.slice(1)}`;
 };
 const deriveCollectionFieldName = (modelName) => {
 	if (!modelName) return "items";
@@ -722,6 +991,12 @@ const deriveCollectionFieldName = (modelName) => {
 	if (camel.endsWith("s")) return `${camel}es`;
 	return `${camel}s`;
 };
+const resolveForeignKeyColumn = (columns, foreignKey) => {
+	return columns.find((column) => column.name === foreignKey.column);
+};
+const isOneToOneForeignKey = (column) => {
+	return Boolean(column?.unique || column?.primary);
+};
 /** 
 * Format a SchemaForeignKeyAction value as a Prisma onDelete action string.
 * 
@@ -742,15 +1017,16 @@ const formatRelationAction = (action) => {
 * @param foreignKey    The foreign key definition to convert to a relation line.
 * @returns             The corresponding Prisma schema line for the relation field.
 */
-const buildRelationLine = (foreignKey) => {
+const buildRelationLine = (sourceModelName, foreignKey, columns = []) => {
 	if (!foreignKey.referencesTable.trim()) throw new ArkormException(`Foreign key [${foreignKey.column}] must define a referenced table.`);
 	if (!foreignKey.referencesColumn.trim()) throw new ArkormException(`Foreign key [${foreignKey.column}] must define a referenced column.`);
+	const sourceColumn = resolveForeignKeyColumn(columns, foreignKey);
 	const fieldName = foreignKey.fieldAlias?.trim() || deriveRelationFieldName(foreignKey.column);
 	const targetModel = toModelName(foreignKey.referencesTable);
-	const relationName = foreignKey.relationAlias?.trim();
-	const relationPrefix = relationName ? `@relation("${relationName.replace(/"/g, "\\\"")}", ` : "@relation(";
+	const relationName = deriveRelationAlias(sourceModelName, targetModel, foreignKey.relationAlias?.trim());
+	const optional = sourceColumn?.nullable ? "?" : "";
 	const onDelete = foreignKey.onDelete ? `, onDelete: ${formatRelationAction(foreignKey.onDelete)}` : "";
-	return `  ${fieldName} ${targetModel} ${relationPrefix}fields: [${foreignKey.column}], references: [${foreignKey.referencesColumn}]${onDelete})`;
+	return `  ${fieldName} ${targetModel}${optional} @relation("${relationName.replace(/"/g, "\\\"")}", fields: [${foreignKey.column}], references: [${foreignKey.referencesColumn}]${onDelete})`;
 };
 /**
 * Build a Prisma relation field line for the inverse side of a relation, based 
@@ -762,8 +1038,11 @@ const buildRelationLine = (foreignKey) => {
 * @param foreignKey        The foreign key definition for the relation.
 * @returns                 The Prisma schema line for the inverse relation field.
 */
-const buildInverseRelationLine = (sourceModelName, targetModelName, foreignKey) => {
-	return `  ${deriveCollectionFieldName(sourceModelName)} ${sourceModelName}[] @relation("${deriveInverseRelationAlias(sourceModelName, targetModelName, foreignKey.inverseRelationAlias).replace(/"/g, "\\\"")}")`;
+const buildInverseRelationLine = (sourceModelName, targetModelName, foreignKey, columns = []) => {
+	const sourceColumn = resolveForeignKeyColumn(columns, foreignKey);
+	const fieldName = isOneToOneForeignKey(sourceColumn) ? deriveSingularFieldName(sourceModelName) : deriveCollectionFieldName(sourceModelName);
+	const relationName = deriveRelationAlias(sourceModelName, targetModelName, foreignKey.relationAlias?.trim());
+	return `  ${fieldName} ${isOneToOneForeignKey(sourceColumn) ? `${sourceModelName}?` : `${sourceModelName}[]`} @relation("${relationName.replace(/"/g, "\\\"")}")`;
 };
 /**
 * Inject a line into the body of a Prisma model block if it does not already 
@@ -791,14 +1070,15 @@ const injectLineIntoModelBody = (bodyLines, line, exists) => {
 * @param foreignKeys An array of foreign key definitions to process.
 * @returns The updated Prisma schema string with inverse relations applied.
 */
-const applyInverseRelations = (schema, sourceModelName, foreignKeys) => {
+const applyInverseRelations = (schema, sourceModelName, foreignKeys, columns = []) => {
 	let nextSchema = schema;
 	for (const foreignKey of foreignKeys) {
 		const targetModel = findModelBlock(nextSchema, foreignKey.referencesTable);
 		if (!targetModel) continue;
-		const inverseLine = buildInverseRelationLine(sourceModelName, targetModel.modelName, foreignKey);
+		const sourceColumn = resolveForeignKeyColumn(columns, foreignKey);
+		const inverseLine = buildInverseRelationLine(sourceModelName, targetModel.modelName, foreignKey, columns);
 		const targetBodyLines = targetModel.block.split("\n");
-		const fieldName = deriveCollectionFieldName(sourceModelName);
+		const fieldName = isOneToOneForeignKey(sourceColumn) ? deriveSingularFieldName(sourceModelName) : deriveCollectionFieldName(sourceModelName);
 		const fieldRegex = new RegExp(`^\\s*${escapeRegex(fieldName)}\\s+`);
 		injectLineIntoModelBody(targetBodyLines, inverseLine, (line) => fieldRegex.test(line));
 		const updatedTarget = targetBodyLines.join("\n");
@@ -817,7 +1097,7 @@ const buildModelBlock = (operation) => {
 	const modelName = toModelName(operation.table);
 	const mapped = operation.table !== modelName.toLowerCase();
 	const fields = operation.columns.map(buildFieldLine);
-	const relations = (operation.foreignKeys ?? []).map(buildRelationLine);
+	const relations = (operation.foreignKeys ?? []).map((foreignKey) => buildRelationLine(modelName, foreignKey, operation.columns));
 	const metadata = [...(operation.indexes ?? []).map(buildIndexLine), ...mapped ? [`  @@map("${str(operation.table).snake()}")`] : []];
 	return `model ${modelName} {\n${(metadata.length > 0 ? [
 		...fields,
@@ -873,8 +1153,9 @@ const findModelBlock = (schema, table) => {
 */
 const applyCreateTableOperation = (schema, operation) => {
 	if (findModelBlock(schema, operation.table)) throw new ArkormException(`Prisma model for table [${operation.table}] already exists.`);
+	const schemaWithEnums = ensureEnumBlocks(schema, operation.columns);
 	const block = buildModelBlock(operation);
-	return applyInverseRelations(`${schema.trimEnd()}\n\n${block}\n`, toModelName(operation.table), operation.foreignKeys ?? []);
+	return applyInverseRelations(`${schemaWithEnums.trimEnd()}\n\n${block}\n`, toModelName(operation.table), operation.foreignKeys ?? [], operation.columns);
 };
 /**
 * Apply an alter table operation to a Prisma schema string, modifying the model 
@@ -887,7 +1168,10 @@ const applyCreateTableOperation = (schema, operation) => {
 const applyAlterTableOperation = (schema, operation) => {
 	const model = findModelBlock(schema, operation.table);
 	if (!model) throw new ArkormException(`Prisma model for table [${operation.table}] was not found.`);
-	let block = model.block;
+	const schemaWithEnums = ensureEnumBlocks(schema, operation.addColumns);
+	const refreshedModel = findModelBlock(schemaWithEnums, operation.table);
+	if (!refreshedModel) throw new ArkormException(`Prisma model for table [${operation.table}] was not found.`);
+	let block = refreshedModel.block;
 	const bodyLines = block.split("\n");
 	operation.dropColumns.forEach((column) => {
 		const columnRegex = new RegExp(`^\\s*${escapeRegex(column)}\\s+`);
@@ -912,12 +1196,12 @@ const applyAlterTableOperation = (schema, operation) => {
 		bodyLines.splice(insertIndex, 0, indexLine);
 	});
 	for (const foreignKey of operation.addForeignKeys ?? []) {
-		const relationLine = buildRelationLine(foreignKey);
+		const relationLine = buildRelationLine(model.modelName, foreignKey, operation.addColumns);
 		const relationRegex = new RegExp(`^\\s*${escapeRegex(foreignKey.fieldAlias?.trim() || deriveRelationFieldName(foreignKey.column))}\\s+`);
 		injectLineIntoModelBody(bodyLines, relationLine, (line) => relationRegex.test(line));
 	}
 	block = bodyLines.join("\n");
-	return applyInverseRelations(`${schema.slice(0, model.start)}${block}${schema.slice(model.end)}`, model.modelName, operation.addForeignKeys ?? []);
+	return applyInverseRelations(`${schemaWithEnums.slice(0, refreshedModel.start)}${block}${schemaWithEnums.slice(refreshedModel.end)}`, model.modelName, operation.addForeignKeys ?? [], operation.addColumns);
 };
 /**
 * Apply a drop table operation to a Prisma schema string, removing the model block