rake-db 2.33.9 → 2.33.11

package/dist/index.mjs

@@ -521,6 +521,7 @@ const addOrDropChanges = [];
 function add(item, options) {
 	consumeColumnName();
 	setName(this, item);
+	if (isNoForeignKeyChangeInput(item)) throw new Error("t.noForeignKey() is only supported in t.change(...)");
 	if (item instanceof Column) {
 		const result = addOrDrop("add", item, options);
 		if (result.type === "change") return result;
@@ -544,6 +545,7 @@ function add(item, options) {
 const drop = function(item, options) {
 	consumeColumnName();
 	setName(this, item);
+	if (isNoForeignKeyChangeInput(item)) throw new Error("t.noForeignKey() is only supported in t.change(...)");
 	if (item instanceof Column) {
 		const result = addOrDrop("drop", item, options);
 		if (result.type === "change") return result;
@@ -587,6 +589,28 @@ const addOrDrop = (type, item, options) => {
 		dropMode: options?.dropMode
 	};
 };
+const isColumnForeignKeyChangeInput = (item) => {
+	return "columnForeignKey" in item;
+};
+const isNoForeignKeyChangeInput = (item) => {
+	return "noForeignKey" in item;
+};
+const isCheckConstraintItem = (item) => {
+	return !!item.constraint?.check;
+};
+const changeInputToColumnChange = (item, opposite) => {
+	if (item instanceof Column || "type" in item) return columnTypeToColumnChange(item);
+	if (isColumnForeignKeyChangeInput(item)) return { foreignKeys: [item.columnForeignKey] };
+	if (isNoForeignKeyChangeInput(item)) {
+		if (!isColumnForeignKeyChangeInput(opposite)) throw new Error("t.noForeignKey() in t.change(...) must be paired with t.foreignKey(...)");
+		return { foreignKeys: [] };
+	}
+	if (isCheckConstraintItem(item)) return { checks: [{
+		sql: item.constraint.check,
+		name: item.constraint.name
+	}] };
+	throw new Error("t.change(...) supports t.check(...) only for check constraints in this form");
+};
 const columnTypeToColumnChange = (item, name) => {
 	if (item instanceof Column) {
 		let column = item;
@@ -618,6 +642,18 @@ const setName = (self, item) => {
 	else if (item instanceof Column) item.data.name ??= name;
 	else item.name ??= name;
 };
+function foreignKey(...args) {
+	if (Array.isArray(args[0])) {
+		const [columns, fnOrTable, foreignColumns, options] = args;
+		return tableDataMethods.foreignKey(columns, fnOrTable, foreignColumns, options);
+	}
+	const [fnOrTable, foreignColumn, options] = args;
+	return { columnForeignKey: {
+		fnOrTable,
+		foreignColumns: [foreignColumn],
+		options
+	} };
+}
 const tableChangeMethods = {
 	...tableMethods,
 	...tableDataMethods,
@@ -629,10 +665,14 @@ const tableChangeMethods = {
 	},
 	add,
 	drop,
+	foreignKey,
+	noForeignKey() {
+		return { noForeignKey: true };
+	},
 	change(from, to, using) {
 		consumeColumnName();
-		const f = columnTypeToColumnChange(from);
-		const t = columnTypeToColumnChange(to);
+		const f = changeInputToColumnChange(from, to);
+		const t = changeInputToColumnChange(to, from);
 		setName(this, f);
 		setName(this, t);
 		return {