Merge pull request #9 from replit/jt/devprod/add_select_resolver_pg_suggestions

devprod/add select resolver pg suggestions

Author: joe-thom

drizzle-kit/README.md

@@ -13,7 +13,13 @@ Make any changes you require to the `drizzle-kit/api` file ([see here](./drizzle
 ```bash
 pnpm build
 ```
-This will build a `dist` file that you can import into `repl-it-web` using the `file:` protocol in `package.json` like so:
+This will build a `dist` file that you can import into `repl-it-web` using the `file:` by running:
+```bash
+cd pkg/pid2/
+
+pnpm add @drizzle-team/drizzle-kit@file:../../../drizzle-orm/drizzle-kit/dist
+```
+Which should add the following to `pkg/pid2/package.json`:
 ```
 "@drizzle-team/drizzle-kit": "file:../drizzle-orm/drizzle-kit/dist",
 ```

drizzle-kit/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@drizzle-team/drizzle-kit",
-	"version": "0.31.8",
+	"version": "0.31.10",
 	"homepage": "https://orm.drizzle.team",
 	"keywords": [
 		"drizzle",

drizzle-kit/src/api.ts

@@ -166,6 +166,7 @@ export const getPgClientPool = async (
 	return pool;
 };
 
+export type { SelectResolverInput, SelectResolverOutput } from './cli/commands/pgPushUtils';
 export { applyPgSnapshotsDiff } from './snapshotsDiffer';
 export type {
 	ColumnsResolverInput,

drizzle-kit/src/cli/commands/migrate.ts

@@ -248,11 +248,7 @@ export const policyResolver = async (
 export const indPolicyResolver = async (
 	input: PolicyResolverInput<Policy>,
 ): Promise<PolicyResolverOutput<Policy>> => {
-	const result = await promptNamedConflict(
-		input.created,
-		input.deleted,
-		'policy',
-	);
+	const result = await promptNamedConflict(input.created, input.deleted, 'policy');
 	return {
 		created: result.created,
 		deleted: result.deleted,

drizzle-kit/src/cli/commands/pgPushUtils.ts

@@ -56,7 +56,30 @@ function tableNameWithSchemaFrom(
 	return concatSchemaAndTableName(newSchemaName, newTableName);
 }
 
-export const pgSuggestions = async (db: DB, statements: JsonStatement[]) => {
+export type SelectResolverInput = {
+	entity: {
+		type: 'createUniqueConstraint';
+		name: string;
+		count: number;
+		tableName: string;
+	};
+	items: string[];
+};
+
+export type SelectResolverOutput = {
+	data: {
+		index: number;
+		value: string;
+	};
+};
+
+export const pgSuggestions = async (
+	db: DB,
+	statements: JsonStatement[],
+	selectResolver?: (
+		input: SelectResolverInput,
+	) => Promise<SelectResolverOutput>,
+) => {
 	let shouldAskForApprove = false;
 	const statementsToExecute: string[] = [];
 	const infoToPrint: string[] = [];
@@ -236,8 +259,20 @@ export const pgSuggestions = async (db: DB, statements: JsonStatement[]) => {
 						)
 					} table?\n`,
 				);
-				const { status, data } = await render(
-					new Select(['No, add the constraint without truncating the table', `Yes, truncate the table`]),
+
+				const entity = {
+					type: 'createUniqueConstraint' as const,
+					name: unsquashedUnique.name,
+					count,
+					tableName: statement.tableName,
+				};
+				const items = ['no', 'yes'];
+
+				const { data } = selectResolver ? await selectResolver({ entity, items }) : await render(
+					new Select([
+						'No, add the constraint without truncating the table',
+						`Yes, truncate the table`,
+					]),
 				);
 				if (data?.index === 1) {
 					tablesToTruncate.push(statement.tableName);