@anthonyalayo/drizzle-kit 1.0.2 → 1.0.4

package/src/cli/commands/sqlitePushUtils.ts

@@ -1,322 +0,0 @@
-import chalk from 'chalk';
-
-import { SQLiteSchemaInternal, SQLiteSchemaSquashed, SQLiteSquasher } from '../../serializer/sqliteSchema';
-import {
-	CreateSqliteIndexConvertor,
-	fromJson,
-	SQLiteCreateTableConvertor,
-	SQLiteDropTableConvertor,
-	SqliteRenameTableConvertor,
-} from '../../sqlgenerator';
-
-import type { JsonStatement } from '../../jsonStatements';
-import { findAddedAndRemoved, type SQLiteDB } from '../../utils';
-
-export const _moveDataStatements = (
-	tableName: string,
-	json: SQLiteSchemaSquashed,
-	dataLoss: boolean = false,
-) => {
-	const statements: string[] = [];
-
-	const newTableName = `__new_${tableName}`;
-
-	// create table statement from a new json2 with proper name
-	const tableColumns = Object.values(json.tables[tableName].columns);
-	const referenceData = Object.values(json.tables[tableName].foreignKeys);
-	const compositePKs = Object.values(
-		json.tables[tableName].compositePrimaryKeys,
-	).map((it) => SQLiteSquasher.unsquashPK(it));
-	const checkConstraints = Object.values(json.tables[tableName].checkConstraints);
-
-	const mappedCheckConstraints: string[] = checkConstraints.map((it) =>
-		it.replaceAll(`"${tableName}".`, `"${newTableName}".`)
-			.replaceAll(`\`${tableName}\`.`, `\`${newTableName}\`.`)
-			.replaceAll(`${tableName}.`, `${newTableName}.`)
-			.replaceAll(`'${tableName}'.`, `\`${newTableName}\`.`)
-	);
-
-	const fks = referenceData.map((it) => SQLiteSquasher.unsquashPushFK(it));
-
-	// create new table
-	statements.push(
-		new SQLiteCreateTableConvertor().convert({
-			type: 'sqlite_create_table',
-			tableName: newTableName,
-			columns: tableColumns,
-			referenceData: fks,
-			compositePKs,
-			checkConstraints: mappedCheckConstraints,
-		}),
-	);
-
-	// move data
-	if (!dataLoss) {
-		const columns = Object.keys(json.tables[tableName].columns).map(
-			(c) => `"${c}"`,
-		);
-
-		statements.push(
-			`INSERT INTO \`${newTableName}\`(${
-				columns.join(
-					', ',
-				)
-			}) SELECT ${columns.join(', ')} FROM \`${tableName}\`;`,
-		);
-	}
-
-	statements.push(
-		new SQLiteDropTableConvertor().convert({
-			type: 'drop_table',
-			tableName: tableName,
-			schema: '',
-		}),
-	);
-
-	// rename table
-	statements.push(
-		new SqliteRenameTableConvertor().convert({
-			fromSchema: '',
-			tableNameFrom: newTableName,
-			tableNameTo: tableName,
-			toSchema: '',
-			type: 'rename_table',
-		}),
-	);
-
-	for (const idx of Object.values(json.tables[tableName].indexes)) {
-		statements.push(
-			new CreateSqliteIndexConvertor().convert({
-				type: 'create_index',
-				tableName: tableName,
-				schema: '',
-				data: idx,
-			}),
-		);
-	}
-
-	return statements;
-};
-
-export const getOldTableName = (
-	tableName: string,
-	meta: SQLiteSchemaInternal['_meta'],
-) => {
-	for (const key of Object.keys(meta.tables)) {
-		const value = meta.tables[key];
-		if (`"${tableName}"` === value) {
-			return key.substring(1, key.length - 1);
-		}
-	}
-	return tableName;
-};
-
-export const getNewTableName = (
-	tableName: string,
-	meta: SQLiteSchemaInternal['_meta'],
-) => {
-	if (typeof meta.tables[`"${tableName}"`] !== 'undefined') {
-		return meta.tables[`"${tableName}"`].substring(
-			1,
-			meta.tables[`"${tableName}"`].length - 1,
-		);
-	}
-	return tableName;
-};
-
-export const logSuggestionsAndReturn = async (
-	connection: SQLiteDB,
-	statements: JsonStatement[],
-	json1: SQLiteSchemaSquashed,
-	json2: SQLiteSchemaSquashed,
-	meta: SQLiteSchemaInternal['_meta'],
-) => {
-	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 connection.query<{ count: string }>(
-				`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;
-			}
-
-			const fromJsonStatement = fromJson([statement], 'sqlite', 'push');
-			statementsToExecute.push(
-				...(Array.isArray(fromJsonStatement) ? fromJsonStatement : [fromJsonStatement]),
-			);
-		} else if (statement.type === 'alter_table_drop_column') {
-			const tableName = statement.tableName;
-			const columnName = statement.columnName;
-
-			const res = await connection.query<{ count: string }>(
-				`select count(\`${tableName}\`.\`${columnName}\`) as count from \`${tableName}\``,
-			);
-			const count = Number(res[0].count);
-			if (count > 0) {
-				infoToPrint.push(
-					`· You're about to delete ${
-						chalk.underline(
-							columnName,
-						)
-					} column in ${tableName} table with ${count} items`,
-				);
-				columnsToRemove.push(`${tableName}_${statement.columnName}`);
-				shouldAskForApprove = true;
-			}
-
-			const fromJsonStatement = fromJson([statement], 'sqlite', 'push');
-			statementsToExecute.push(
-				...(Array.isArray(fromJsonStatement) ? fromJsonStatement : [fromJsonStatement]),
-			);
-		} else if (
-			statement.type === 'sqlite_alter_table_add_column'
-			&& (statement.column.notNull && !statement.column.default)
-		) {
-			const tableName = statement.tableName;
-			const columnName = statement.column.name;
-			const res = await connection.query<{ count: string }>(
-				`select count(*) as count from \`${tableName}\``,
-			);
-			const count = Number(res[0].count);
-			if (count > 0) {
-				infoToPrint.push(
-					`· You're about to add not-null ${
-						chalk.underline(
-							columnName,
-						)
-					} column without default value, which contains ${count} items`,
-				);
-
-				tablesToTruncate.push(tableName);
-				statementsToExecute.push(`delete from ${tableName};`);
-
-				shouldAskForApprove = true;
-			}
-
-			const fromJsonStatement = fromJson([statement], 'sqlite', 'push');
-			statementsToExecute.push(
-				...(Array.isArray(fromJsonStatement) ? fromJsonStatement : [fromJsonStatement]),
-			);
-		} else if (statement.type === 'recreate_table') {
-			const tableName = statement.tableName;
-			const oldTableName = getOldTableName(tableName, meta);
-
-			let dataLoss = false;
-
-			const prevColumnNames = Object.keys(json1.tables[oldTableName].columns);
-			const currentColumnNames = Object.keys(json2.tables[tableName].columns);
-			const { removedColumns, addedColumns } = findAddedAndRemoved(
-				prevColumnNames,
-				currentColumnNames,
-			);
-
-			if (removedColumns.length) {
-				for (const removedColumn of removedColumns) {
-					const res = await connection.query<{ count: string }>(
-						`select count(\`${tableName}\`.\`${removedColumn}\`) as count from \`${tableName}\``,
-					);
-
-					const count = Number(res[0].count);
-					if (count > 0) {
-						infoToPrint.push(
-							`· You're about to delete ${
-								chalk.underline(
-									removedColumn,
-								)
-							} column in ${tableName} table with ${count} items`,
-						);
-						columnsToRemove.push(removedColumn);
-						shouldAskForApprove = true;
-					}
-				}
-			}
-
-			if (addedColumns.length) {
-				for (const addedColumn of addedColumns) {
-					const [res] = await connection.query<{ count: string }>(
-						`select count(*) as count from \`${tableName}\``,
-					);
-
-					const columnConf = json2.tables[tableName].columns[addedColumn];
-
-					const count = Number(res.count);
-					if (count > 0 && columnConf.notNull && !columnConf.default) {
-						dataLoss = true;
-						infoToPrint.push(
-							`· You're about to add not-null ${
-								chalk.underline(
-									addedColumn,
-								)
-							} column without default value to table, which contains ${count} items`,
-						);
-						shouldAskForApprove = true;
-						tablesToTruncate.push(tableName);
-
-						statementsToExecute.push(`DELETE FROM \`${tableName}\`;`);
-					}
-				}
-			}
-
-			// check if some tables referencing current for pragma
-			const tablesReferencingCurrent: string[] = [];
-
-			for (const table of Object.values(json2.tables)) {
-				const tablesRefs = Object.values(json2.tables[table.name].foreignKeys)
-					.filter((t) => SQLiteSquasher.unsquashPushFK(t).tableTo === tableName)
-					.map((it) => SQLiteSquasher.unsquashPushFK(it).tableFrom);
-
-				tablesReferencingCurrent.push(...tablesRefs);
-			}
-
-			if (!tablesReferencingCurrent.length) {
-				statementsToExecute.push(..._moveDataStatements(tableName, json2, dataLoss));
-				continue;
-			}
-
-			const [{ foreign_keys: pragmaState }] = await connection.query<{
-				foreign_keys: number;
-			}>(`PRAGMA foreign_keys;`);
-
-			if (pragmaState) {
-				statementsToExecute.push(`PRAGMA foreign_keys=OFF;`);
-			}
-			statementsToExecute.push(..._moveDataStatements(tableName, json2, dataLoss));
-			if (pragmaState) {
-				statementsToExecute.push(`PRAGMA foreign_keys=ON;`);
-			}
-		} else {
-			const fromJsonStatement = fromJson([statement], 'sqlite', 'push');
-			statementsToExecute.push(
-				...(Array.isArray(fromJsonStatement) ? fromJsonStatement : [fromJsonStatement]),
-			);
-		}
-	}
-
-	return {
-		statementsToExecute,
-		shouldAskForApprove,
-		infoToPrint,
-		columnsToRemove: [...new Set(columnsToRemove)],
-		schemasToRemove: [...new Set(schemasToRemove)],
-		tablesToTruncate: [...new Set(tablesToTruncate)],
-		tablesToRemove: [...new Set(tablesToRemove)],
-	};
-};

package/src/cli/commands/sqliteUp.ts

@@ -1,52 +0,0 @@
-import chalk from 'chalk';
-import { writeFileSync } from 'fs';
-import { mapEntries } from 'src/global';
-import { SQLiteSchema, sqliteSchemaV5 } from 'src/serializer/sqliteSchema';
-import { prepareOutFolder, validateWithReport } from 'src/utils';
-
-export const upSqliteHandler = (out: string) => {
-	const { snapshots } = prepareOutFolder(out, 'sqlite');
-	const report = validateWithReport(snapshots, 'sqlite');
-
-	report.nonLatest
-		.map((it) => ({
-			path: it,
-			raw: report.rawMap[it]!! as Record<string, any>,
-		}))
-		.forEach((it) => {
-			const path = it.path;
-			const result = updateUpToV6(it.raw);
-
-			console.log(`[${chalk.green('✓')}] ${path}`);
-
-			writeFileSync(path, JSON.stringify(result, null, 2));
-		});
-
-	console.log("Everything's fine 🐶🔥");
-};
-
-const updateUpToV6 = (json: Record<string, any>): SQLiteSchema => {
-	const schema = sqliteSchemaV5.parse(json);
-
-	const tables = mapEntries(schema.tables, (tableKey, table) => {
-		const columns = mapEntries(table.columns, (key, value) => {
-			if (
-				value.default
-				&& (typeof value.default === 'object' || Array.isArray(value.default))
-			) {
-				value.default = `'${JSON.stringify(value.default)}'`;
-			}
-			return [key, value];
-		});
-		table.columns = columns;
-		return [tableKey, table];
-	});
-
-	return {
-		...schema,
-		version: '6',
-		dialect: 'sqlite',
-		tables: tables,
-		views: {},
-	};
-};