Add Create Project from Database quickpick (#16572)

* Add Create Project from Database quickpick

* fix test

* pr comments

* Add comment

extensions/sql-database-projects/src/dialogs/createProjectFromDatabaseDialog.ts

@@ -353,36 +353,23 @@ export class CreateProjectFromDatabaseDialog {
 	}
 
 	public async handleCreateButtonClick(): Promise<void> {
+		const azdataApi = getAzdataApi()!;
+		const connectionUri = await azdataApi!.connection.getUriForConnection(this.connectionId!);
 		const model: ImportDataModel = {
-			serverId: this.connectionId!,
+			connectionUri: connectionUri,
 			database: <string>this.sourceDatabaseDropDown!.value,
 			projName: this.projectNameTextBox!.value!,
 			filePath: this.projectLocationTextBox!.value!,
 			version: '1.0.0.0',
-			extractTarget: this.mapExtractTargetEnum(<string>this.folderStructureDropDown!.value)
+			extractTarget: mapExtractTargetEnum(<string>this.folderStructureDropDown!.value)
 		};
 
-		getAzdataApi()!.window.closeDialog(this.dialog);
+		azdataApi!.window.closeDialog(this.dialog);
 		await this.createProjectFromDatabaseCallback!(model);
 
 		this.dispose();
 	}
 
-	private mapExtractTargetEnum(inputTarget: any): mssql.ExtractTarget {
-		if (inputTarget) {
-			switch (inputTarget) {
-				case constants.file: return mssql.ExtractTarget['file'];
-				case constants.flat: return mssql.ExtractTarget['flat'];
-				case constants.objectType: return mssql.ExtractTarget['objectType'];
-				case constants.schema: return mssql.ExtractTarget['schema'];
-				case constants.schemaObjectType: return mssql.ExtractTarget['schemaObjectType'];
-				default: throw new Error(constants.invalidInput(inputTarget));
-			}
-		} else {
-			throw new Error(constants.extractTargetRequired);
-		}
-	}
-
 	async validate(): Promise<boolean> {
 		try {
 			if (await getDataWorkspaceExtensionApi().validateWorkspace() === false) {
@@ -415,3 +402,18 @@ export class CreateProjectFromDatabaseDialog {
 		};
 	}
 }
+
+export function mapExtractTargetEnum(inputTarget: string): mssql.ExtractTarget {
+	if (inputTarget) {
+		switch (inputTarget) {
+			case constants.file: return mssql.ExtractTarget.file;
+			case constants.flat: return mssql.ExtractTarget.flat;
+			case constants.objectType: return mssql.ExtractTarget.objectType;
+			case constants.schema: return mssql.ExtractTarget.schema;
+			case constants.schemaObjectType: return mssql.ExtractTarget.schemaObjectType;
+			default: throw new Error(constants.invalidInput(inputTarget));
+		}
+	} else {
+		throw new Error(constants.extractTargetRequired);
+	}
+}

extensions/sql-database-projects/src/dialogs/createProjectFromDatabaseQuickpick.ts

@@ -0,0 +1,121 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as vscode from 'vscode';
+import * as path from 'path';
+import * as constants from '../common/constants';
+import { exists, getVscodeMssqlApi } from '../common/utils';
+import { IConnectionInfo } from 'vscode-mssql';
+import { defaultProjectNameFromDb, defaultProjectSaveLocation } from '../tools/newProjectTool';
+import { ImportDataModel } from '../models/api/import';
+import { mapExtractTargetEnum } from './createProjectFromDatabaseDialog';
+
+/**
+ * Create flow for a New Project using only VS Code-native APIs such as QuickPick
+ */
+export async function createNewProjectFromDatabaseWithQuickpick(): Promise<ImportDataModel | undefined> {
+
+	// 1. Select connection
+	const vscodeMssqlApi = await getVscodeMssqlApi();
+	let connectionProfile: IConnectionInfo | undefined = undefined;
+	let connectionUri: string = '';
+	let dbs: string[] | undefined = undefined;
+	while (!dbs) {
+		connectionProfile = await vscodeMssqlApi.promptForConnection(true);
+		if (!connectionProfile) {
+			// User cancelled
+			return undefined;
+		}
+		// Get the list of databases now to validate that the connection is valid and re-prompt them if it isn't
+		try {
+			connectionUri = await vscodeMssqlApi.connect(connectionProfile);
+			dbs = (await vscodeMssqlApi.listDatabases(connectionUri))
+				.filter(db => !constants.systemDbs.includes(db)); // Filter out system dbs
+		} catch (err) {
+			// no-op, the mssql extension handles showing the error to the user. We'll just go
+			// back and prompt the user for a connection again
+		}
+	}
+
+	// 2. Select database
+	const selectedDatabase = await vscode.window.showQuickPick(
+		dbs,
+		{ title: constants.selectDatabase, ignoreFocusOut: true });
+	if (!selectedDatabase) {
+		// User cancelled
+		return undefined;
+	}
+
+	// 3. Prompt for project name
+	const projectName = await vscode.window.showInputBox(
+		{
+			title: constants.projectNamePlaceholderText,
+			value: defaultProjectNameFromDb(selectedDatabase),
+			validateInput: (value) => {
+				return value ? undefined : constants.nameMustNotBeEmpty;
+			},
+			ignoreFocusOut: true
+		});
+	if (!projectName) {
+		return undefined;
+	}
+
+	// 4. Prompt for Project location
+	// Show quick pick with just browse option to give user context about what the file dialog is for (since that doesn't always have a title)
+	const browseProjectLocation = await vscode.window.showQuickPick(
+		[constants.browseEllipsis],
+		{ title: constants.projectLocationPlaceholderText, ignoreFocusOut: true });
+	if (!browseProjectLocation) {
+		return undefined;
+	}
+	// We validate that the folder doesn't already exist, and if it does keep prompting them to pick a new one
+	let valid = false;
+	let projectLocation = '';
+	while (!valid) {
+		const locations = await vscode.window.showOpenDialog({
+			canSelectFiles: false,
+			canSelectFolders: true,
+			canSelectMany: false,
+			openLabel: constants.selectString,
+			title: constants.selectProjectLocation,
+			defaultUri: defaultProjectSaveLocation()
+		});
+		if (!locations) {
+			// User cancelled
+			return undefined;
+		}
+		projectLocation = locations[0].fsPath;
+		const locationExists = await exists(path.join(projectLocation, projectName));
+		if (locationExists) {
+			// Show the browse quick pick again with the title updated with the error
+			const browseProjectLocation = await vscode.window.showQuickPick(
+				[constants.browseEllipsis],
+				{ title: constants.folderAlreadyExistsChooseNewLocation(projectName), ignoreFocusOut: true });
+			if (!browseProjectLocation) {
+				return undefined;
+			}
+		} else {
+			valid = true;
+		}
+	}
+
+	// 5: Prompt for folder structure
+	const folderStructure = await vscode.window.showQuickPick(
+		[constants.file, constants.flat, constants.objectType, constants.schema, constants.schemaObjectType],
+		{ title: constants.selectFolderStructure, ignoreFocusOut: true, });
+	if (!folderStructure) {
+		// User cancelled
+		return undefined;
+	}
+
+	return {
+		connectionUri: connectionUri,
+		database: selectedDatabase,
+		projName: projectName,
+		filePath: projectLocation,
+		version: '1.0.0.0',
+		extractTarget: mapExtractTargetEnum(folderStructure)
+	};
+}