@anthonyalayo/drizzle-kit 1.0.2 → 1.0.4

package/src/cli/commands/mysqlIntrospect.ts

@@ -1,53 +0,0 @@
-import { renderWithTask } from 'hanji';
-import { Minimatch } from 'minimatch';
-import { originUUID } from '../../global';
-import type { MySqlSchema } from '../../serializer/mysqlSchema';
-import { fromDatabase } from '../../serializer/mysqlSerializer';
-import type { DB } from '../../utils';
-import { ProgressView } from '../views';
-
-export const mysqlPushIntrospect = async (
-	db: DB,
-	databaseName: string,
-	filters: string[],
-) => {
-	const matchers = filters.map((it) => {
-		return new Minimatch(it);
-	});
-
-	const filter = (tableName: string) => {
-		if (matchers.length === 0) return true;
-
-		let flags: boolean[] = [];
-
-		for (let matcher of matchers) {
-			if (matcher.negate) {
-				if (!matcher.match(tableName)) {
-					flags.push(false);
-				}
-			}
-
-			if (matcher.match(tableName)) {
-				flags.push(true);
-			}
-		}
-
-		if (flags.length > 0) {
-			return flags.every(Boolean);
-		}
-		return false;
-	};
-
-	const progress = new ProgressView(
-		'Pulling schema from database...',
-		'Pulling schema from database...',
-	);
-	const res = await renderWithTask(
-		progress,
-		fromDatabase(db, databaseName, filter),
-	);
-
-	const schema = { id: originUUID, prevId: '', ...res } as MySqlSchema;
-	const { internal, ...schemaWithoutInternals } = schema;
-	return { schema: schemaWithoutInternals };
-};

package/src/cli/commands/mysqlPushUtils.ts

@@ -1,352 +0,0 @@
-import chalk from 'chalk';
-import { render } from 'hanji';
-import { TypeOf } from 'zod';
-import { JsonAlterColumnTypeStatement, JsonStatement } from '../../jsonStatements';
-import { mysqlSchema, MySqlSquasher } from '../../serializer/mysqlSchema';
-import type { DB } from '../../utils';
-import { Select } from '../selector-ui';
-import { withStyle } from '../validations/outputs';
-
-export const filterStatements = (
-	statements: JsonStatement[],
-	currentSchema: TypeOf<typeof mysqlSchema>,
-	prevSchema: TypeOf<typeof mysqlSchema>,
-) => {
-	return statements.filter((statement) => {
-		if (statement.type === 'alter_table_alter_column_set_type') {
-			// Don't need to handle it on migrations step and introspection
-			// but for both it should be skipped
-			if (
-				statement.oldDataType.startsWith('tinyint')
-				&& statement.newDataType.startsWith('boolean')
-			) {
-				return false;
-			}
-
-			if (
-				statement.oldDataType.startsWith('bigint unsigned')
-				&& statement.newDataType.startsWith('serial')
-			) {
-				return false;
-			}
-
-			if (
-				statement.oldDataType.startsWith('serial')
-				&& statement.newDataType.startsWith('bigint unsigned')
-			) {
-				return false;
-			}
-		} else if (statement.type === 'alter_table_alter_column_set_default') {
-			if (
-				statement.newDefaultValue === false
-				&& statement.oldDefaultValue === 0
-				&& statement.newDataType === 'boolean'
-			) {
-				return false;
-			}
-			if (
-				statement.newDefaultValue === true
-				&& statement.oldDefaultValue === 1
-				&& statement.newDataType === 'boolean'
-			) {
-				return false;
-			}
-		} else if (statement.type === 'delete_unique_constraint') {
-			const unsquashed = MySqlSquasher.unsquashUnique(statement.data);
-			// only if constraint was removed from a serial column, than treat it as removed
-			// const serialStatement = statements.find(
-			//   (it) => it.type === "alter_table_alter_column_set_type"
-			// ) as JsonAlterColumnTypeStatement;
-			// if (
-			//   serialStatement?.oldDataType.startsWith("bigint unsigned") &&
-			//   serialStatement?.newDataType.startsWith("serial") &&
-			//   serialStatement.columnName ===
-			//     MySqlSquasher.unsquashUnique(statement.data).columns[0]
-			// ) {
-			//   return false;
-			// }
-			// Check if uniqueindex was only on this column, that is serial
-
-			// if now serial and was not serial and was unique index
-			if (
-				unsquashed.columns.length === 1
-				&& currentSchema.tables[statement.tableName].columns[unsquashed.columns[0]]
-						.type === 'serial'
-				&& prevSchema.tables[statement.tableName].columns[unsquashed.columns[0]]
-						.type === 'serial'
-				&& currentSchema.tables[statement.tableName].columns[unsquashed.columns[0]]
-						.name === unsquashed.columns[0]
-			) {
-				return false;
-			}
-		} else if (statement.type === 'alter_table_alter_column_drop_notnull') {
-			// only if constraint was removed from a serial column, than treat it as removed
-			const serialStatement = statements.find(
-				(it) => it.type === 'alter_table_alter_column_set_type',
-			) as JsonAlterColumnTypeStatement;
-			if (
-				serialStatement?.oldDataType.startsWith('bigint unsigned')
-				&& serialStatement?.newDataType.startsWith('serial')
-				&& serialStatement.columnName === statement.columnName
-				&& serialStatement.tableName === statement.tableName
-			) {
-				return false;
-			}
-			if (statement.newDataType === 'serial' && !statement.columnNotNull) {
-				return false;
-			}
-			if (statement.columnAutoIncrement) {
-				return false;
-			}
-		}
-
-		return true;
-	});
-};
-
-export const logSuggestionsAndReturn = async (
-	db: DB,
-	statements: JsonStatement[],
-	json2: TypeOf<typeof mysqlSchema>,
-) => {
-	let shouldAskForApprove = false;
-	const statementsToExecute: string[] = [];
-	const infoToPrint: string[] = [];
-
-	const tablesToRemove: string[] = [];
-	const columnsToRemove: string[] = [];
-	const schemasToRemove: string[] = [];
-	const tablesToTruncate: string[] = [];
-
-	for (const statement of statements) {
-		if (statement.type === 'drop_table') {
-			const res = await db.query(
-				`select count(*) as count from \`${statement.tableName}\``,
-			);
-			const count = Number(res[0].count);
-			if (count > 0) {
-				infoToPrint.push(
-					`· You're about to delete ${
-						chalk.underline(
-							statement.tableName,
-						)
-					} table with ${count} items`,
-				);
-				tablesToRemove.push(statement.tableName);
-				shouldAskForApprove = true;
-			}
-		} else if (statement.type === 'alter_table_drop_column') {
-			const res = await db.query(
-				`select count(*) as count from \`${statement.tableName}\``,
-			);
-			const count = Number(res[0].count);
-			if (count > 0) {
-				infoToPrint.push(
-					`· You're about to delete ${
-						chalk.underline(
-							statement.columnName,
-						)
-					} column in ${statement.tableName} table with ${count} items`,
-				);
-				columnsToRemove.push(`${statement.tableName}_${statement.columnName}`);
-				shouldAskForApprove = true;
-			}
-		} else if (statement.type === 'drop_schema') {
-			const res = await db.query(
-				`select count(*) as count from information_schema.tables where table_schema = \`${statement.name}\`;`,
-			);
-			const count = Number(res[0].count);
-			if (count > 0) {
-				infoToPrint.push(
-					`· You're about to delete ${
-						chalk.underline(
-							statement.name,
-						)
-					} schema with ${count} tables`,
-				);
-				schemasToRemove.push(statement.name);
-				shouldAskForApprove = true;
-			}
-		} else if (statement.type === 'alter_table_alter_column_set_type') {
-			const res = await db.query(
-				`select count(*) as count from \`${statement.tableName}\``,
-			);
-			const count = Number(res[0].count);
-			if (count > 0) {
-				infoToPrint.push(
-					`· You're about to change ${
-						chalk.underline(
-							statement.columnName,
-						)
-					} column type from ${
-						chalk.underline(
-							statement.oldDataType,
-						)
-					} to ${chalk.underline(statement.newDataType)} with ${count} items`,
-				);
-				statementsToExecute.push(`truncate table ${statement.tableName};`);
-				tablesToTruncate.push(statement.tableName);
-				shouldAskForApprove = true;
-			}
-		} else if (statement.type === 'alter_table_alter_column_drop_default') {
-			if (statement.columnNotNull) {
-				const res = await db.query(
-					`select count(*) as count from \`${statement.tableName}\``,
-				);
-
-				const count = Number(res[0].count);
-				if (count > 0) {
-					infoToPrint.push(
-						`· You're about to remove default value from ${
-							chalk.underline(
-								statement.columnName,
-							)
-						} not-null column with ${count} items`,
-					);
-
-					tablesToTruncate.push(statement.tableName);
-					statementsToExecute.push(`truncate table ${statement.tableName};`);
-
-					shouldAskForApprove = true;
-				}
-			}
-			// shouldAskForApprove = true;
-		} else if (statement.type === 'alter_table_alter_column_set_notnull') {
-			if (typeof statement.columnDefault === 'undefined') {
-				const res = await db.query(
-					`select count(*) as count from \`${statement.tableName}\``,
-				);
-
-				const count = Number(res[0].count);
-				if (count > 0) {
-					infoToPrint.push(
-						`· You're about to set not-null constraint to ${
-							chalk.underline(
-								statement.columnName,
-							)
-						} column without default, which contains ${count} items`,
-					);
-
-					tablesToTruncate.push(statement.tableName);
-					statementsToExecute.push(`truncate table ${statement.tableName};`);
-
-					shouldAskForApprove = true;
-				}
-			}
-		} else if (statement.type === 'alter_table_alter_column_drop_pk') {
-			const res = await db.query(
-				`select count(*) as count from \`${statement.tableName}\``,
-			);
-
-			// if drop pk and json2 has autoincrement in table -> exit process with error
-			if (
-				Object.values(json2.tables[statement.tableName].columns).filter(
-					(column) => column.autoincrement,
-				).length > 0
-			) {
-				console.log(
-					`${
-						withStyle.errorWarning(
-							`You have removed the primary key from a ${statement.tableName} table without removing the auto-increment property from this table. As the database error states: 'there can be only one auto column, and it must be defined as a key. Make sure to remove autoincrement from ${statement.tableName} table`,
-						)
-					}`,
-				);
-				process.exit(1);
-			}
-
-			const count = Number(res[0].count);
-			if (count > 0) {
-				infoToPrint.push(
-					`· You're about to change ${
-						chalk.underline(
-							statement.tableName,
-						)
-					} primary key. This statements may fail and you table may left without primary key`,
-				);
-
-				tablesToTruncate.push(statement.tableName);
-				shouldAskForApprove = true;
-			}
-		} else if (statement.type === 'delete_composite_pk') {
-			// if drop pk and json2 has autoincrement in table -> exit process with error
-			if (
-				Object.values(json2.tables[statement.tableName].columns).filter(
-					(column) => column.autoincrement,
-				).length > 0
-			) {
-				console.log(
-					`${
-						withStyle.errorWarning(
-							`You have removed the primary key from a ${statement.tableName} table without removing the auto-increment property from this table. As the database error states: 'there can be only one auto column, and it must be defined as a key. Make sure to remove autoincrement from ${statement.tableName} table`,
-						)
-					}`,
-				);
-				process.exit(1);
-			}
-		} else if (statement.type === 'alter_table_add_column') {
-			if (
-				statement.column.notNull
-				&& typeof statement.column.default === 'undefined'
-			) {
-				const res = await db.query(
-					`select count(*) as count from \`${statement.tableName}\``,
-				);
-				const count = Number(res[0].count);
-				if (count > 0) {
-					infoToPrint.push(
-						`· You're about to add not-null ${
-							chalk.underline(
-								statement.column.name,
-							)
-						} column without default value, which contains ${count} items`,
-					);
-
-					tablesToTruncate.push(statement.tableName);
-					statementsToExecute.push(`truncate table ${statement.tableName};`);
-
-					shouldAskForApprove = true;
-				}
-			}
-		} else if (statement.type === 'create_unique_constraint') {
-			const res = await db.query(
-				`select count(*) as count from \`${statement.tableName}\``,
-			);
-			const count = Number(res[0].count);
-			if (count > 0) {
-				const unsquashedUnique = MySqlSquasher.unsquashUnique(statement.data);
-				console.log(
-					`· You're about to add ${
-						chalk.underline(
-							unsquashedUnique.name,
-						)
-					} unique constraint to the table, which contains ${count} items. If this statement fails, you will receive an error from the database. Do you want to truncate ${
-						chalk.underline(
-							statement.tableName,
-						)
-					} table?\n`,
-				);
-				const { status, data } = await render(
-					new Select([
-						'No, add the constraint without truncating the table',
-						`Yes, truncate the table`,
-					]),
-				);
-				if (data?.index === 1) {
-					tablesToTruncate.push(statement.tableName);
-					statementsToExecute.push(`truncate table ${statement.tableName};`);
-					shouldAskForApprove = true;
-				}
-			}
-		}
-	}
-
-	return {
-		statementsToExecute,
-		shouldAskForApprove,
-		infoToPrint,
-		columnsToRemove: [...new Set(columnsToRemove)],
-		schemasToRemove: [...new Set(schemasToRemove)],
-		tablesToTruncate: [...new Set(tablesToTruncate)],
-		tablesToRemove: [...new Set(tablesToRemove)],
-	};
-};

package/src/cli/commands/mysqlUp.ts

@@ -1,102 +0,0 @@
-import chalk from 'chalk';
-import fs, { writeFileSync } from 'fs';
-import path from 'path';
-import { Column, MySqlSchema, MySqlSchemaV4, MySqlSchemaV5, mysqlSchemaV5, Table } from '../../serializer/mysqlSchema';
-import { prepareOutFolder, validateWithReport } from '../../utils';
-
-export const upMysqlHandler = (out: string) => {};
-
-export const upMySqlHandlerV4toV5 = (obj: MySqlSchemaV4): MySqlSchemaV5 => {
-	const mappedTables: Record<string, Table> = {};
-
-	for (const [key, table] of Object.entries(obj.tables)) {
-		const mappedColumns: Record<string, Column> = {};
-		for (const [ckey, column] of Object.entries(table.columns)) {
-			let newDefault: any = column.default;
-			let newType: string = column.type;
-			let newAutoIncrement: boolean | undefined = column.autoincrement;
-
-			if (column.type.toLowerCase().startsWith('datetime')) {
-				if (typeof column.default !== 'undefined') {
-					if (column.default.startsWith("'") && column.default.endsWith("'")) {
-						newDefault = `'${
-							column.default
-								.substring(1, column.default.length - 1)
-								.replace('T', ' ')
-								.slice(0, 23)
-						}'`;
-					} else {
-						newDefault = column.default.replace('T', ' ').slice(0, 23);
-					}
-				}
-
-				newType = column.type.toLowerCase().replace('datetime (', 'datetime(');
-			} else if (column.type.toLowerCase() === 'date') {
-				if (typeof column.default !== 'undefined') {
-					if (column.default.startsWith("'") && column.default.endsWith("'")) {
-						newDefault = `'${
-							column.default
-								.substring(1, column.default.length - 1)
-								.split('T')[0]
-						}'`;
-					} else {
-						newDefault = column.default.split('T')[0];
-					}
-				}
-				newType = column.type.toLowerCase().replace('date (', 'date(');
-			} else if (column.type.toLowerCase().startsWith('timestamp')) {
-				if (typeof column.default !== 'undefined') {
-					if (column.default.startsWith("'") && column.default.endsWith("'")) {
-						newDefault = `'${
-							column.default
-								.substring(1, column.default.length - 1)
-								.replace('T', ' ')
-								.slice(0, 23)
-						}'`;
-					} else {
-						newDefault = column.default.replace('T', ' ').slice(0, 23);
-					}
-				}
-				newType = column.type
-					.toLowerCase()
-					.replace('timestamp (', 'timestamp(');
-			} else if (column.type.toLowerCase().startsWith('time')) {
-				newType = column.type.toLowerCase().replace('time (', 'time(');
-			} else if (column.type.toLowerCase().startsWith('decimal')) {
-				newType = column.type.toLowerCase().replace(', ', ',');
-			} else if (column.type.toLowerCase().startsWith('enum')) {
-				newType = column.type.toLowerCase();
-			} else if (column.type.toLowerCase().startsWith('serial')) {
-				newAutoIncrement = true;
-			}
-			mappedColumns[ckey] = {
-				...column,
-				default: newDefault,
-				type: newType,
-				autoincrement: newAutoIncrement,
-			};
-		}
-
-		mappedTables[key] = {
-			...table,
-			columns: mappedColumns,
-			compositePrimaryKeys: {},
-			uniqueConstraints: {},
-			checkConstraint: {},
-		};
-	}
-
-	return {
-		version: '5',
-		dialect: obj.dialect,
-		id: obj.id,
-		prevId: obj.prevId,
-		tables: mappedTables,
-		schemas: obj.schemas,
-		_meta: {
-			schemas: {} as Record<string, string>,
-			tables: {} as Record<string, string>,
-			columns: {} as Record<string, string>,
-		},
-	};
-};

package/src/cli/commands/pgIntrospect.ts

@@ -1,55 +0,0 @@
-import { renderWithTask } from 'hanji';
-import { Minimatch } from 'minimatch';
-import { originUUID } from '../../global';
-import type { PgSchema, PgSchemaInternal } from '../../serializer/pgSchema';
-import { fromDatabase } from '../../serializer/pgSerializer';
-import type { DB } from '../../utils';
-import { Entities } from '../validations/cli';
-import { ProgressView } from '../views';
-
-export const pgPushIntrospect = async (
-	db: DB,
-	filters: string[],
-	schemaFilters: string[],
-	entities: Entities,
-	tsSchema?: PgSchemaInternal,
-) => {
-	const matchers = filters.map((it) => {
-		return new Minimatch(it);
-	});
-
-	const filter = (tableName: string) => {
-		if (matchers.length === 0) return true;
-
-		let flags: boolean[] = [];
-
-		for (let matcher of matchers) {
-			if (matcher.negate) {
-				if (!matcher.match(tableName)) {
-					flags.push(false);
-				}
-			}
-
-			if (matcher.match(tableName)) {
-				flags.push(true);
-			}
-		}
-
-		if (flags.length > 0) {
-			return flags.every(Boolean);
-		}
-		return false;
-	};
-	const progress = new ProgressView(
-		'Pulling schema from database...',
-		'Pulling schema from database...',
-	);
-	const res = await renderWithTask(
-		progress,
-		fromDatabase(db, filter, schemaFilters, entities, undefined, tsSchema),
-	);
-
-	const schema = { id: originUUID, prevId: '', ...res } as PgSchema;
-	const { internal, ...schemaWithoutInternals } = schema;
-	return { schema: schemaWithoutInternals };
-};