Fixing copy headers (#23280)

src/sql/workbench/services/query/common/gridDataProvider.ts

@@ -25,6 +25,12 @@ export interface IGridDataProvider {
 	 */
 	copyResults(selection: Slick.Range[], includeHeaders?: boolean, tableView?: IDisposableDataProvider<Slick.SlickData>): Promise<void>;
 
+	/**
+	 * Sends a copy request to copy table headers to the clipboard
+	 * @param selection The selection range to copy
+	 */
+	copyHeaders(selection: Slick.Range[]): Promise<void>;
+
 	/**
 	 * Gets the EOL terminator to use for this data type.
 	 */
@@ -100,11 +106,8 @@ export async function getResultsString(provider: IGridDataProvider, selection: S
 	// Make sure all these tasks have executed
 	await Promise.all(actionedTasks);
 
-	const sortResults = (e1: [number, any], e2: [number, any]) => {
-		return e1[0] - e2[0];
-	};
-	headers = new Map([...headers].sort(sortResults));
-	rows = new Map([...rows].sort(sortResults));
+	headers = sortMapEntriesByColumnOrder(headers);
+	rows = sortMapEntriesByColumnOrder(rows);
 
 	let copyString = '';
 	if (includeHeaders) {
@@ -132,6 +135,42 @@ export async function getResultsString(provider: IGridDataProvider, selection: S
 	return copyString;
 }
 
+export function getTableHeaderString(provider: IGridDataProvider, selection: Slick.Range[]): string {
+	let headers: Map<number, string> = new Map(); // Maps a column index -> header
+
+	selection.forEach((range) => {
+		let startCol = range.fromCell;
+		let columnHeaders = provider.getColumnHeaders(range);
+		if (columnHeaders !== undefined) {
+			let idx = 0;
+			for (let header of columnHeaders) {
+				headers.set(startCol + idx, header);
+				idx++;
+			}
+		}
+	});
+
+	headers = sortMapEntriesByColumnOrder(headers)
+
+	const copyString = Array.from(headers.values())
+		.map(colHeader => colHeader ? colHeader : '')
+		.join('\t');
+
+	return copyString;
+}
+
+/**
+ * Ensures that table entries in the map appear in column order instead of the order that they were selected.
+ * @param map Contains the entries selected in a table
+ * @returns Sorted map with entries appearing in column order.
+ */
+function sortMapEntriesByColumnOrder(map: Map<number, any>): Map<number, any> {
+	const leftToRight = (e1: [number, any], e2: [number, any]) => {
+		return e1[0] - e2[0];
+	};
+
+	return new Map([...map].sort(leftToRight));
+}
 
 function removeNewLines(inputString: string): string {
 	// This regex removes all newlines in all OS types

src/sql/workbench/services/query/common/queryRunner.ts

@@ -21,7 +21,7 @@ import { IConfigurationService } from 'vs/platform/configuration/common/configur
 import { URI } from 'vs/base/common/uri';
 import * as perf from 'vs/base/common/performance';
 import { mssqlProviderName } from 'sql/platform/connection/common/constants';
-import { IGridDataProvider, getResultsString } from 'sql/workbench/services/query/common/gridDataProvider';
+import { IGridDataProvider, getResultsString, getTableHeaderString } from 'sql/workbench/services/query/common/gridDataProvider';
 import { getErrorMessage } from 'vs/base/common/errors';
 import { ILogService } from 'vs/platform/log/common/log';
 import { IRange, Range } from 'vs/editor/common/core/range';
@@ -564,7 +564,16 @@ export class QueryGridDataProvider implements IGridDataProvider {
 			const results = await getResultsString(this, selection, includeHeaders, tableView);
 			await this._clipboardService.writeText(results);
 		} catch (error) {
-			this._notificationService.error(nls.localize('copyFailed', "Copy failed with error {0}", getErrorMessage(error)));
+			this._notificationService.error(nls.localize('copyFailed', "Copy failed with error: {0}", getErrorMessage(error)));
+		}
+	}
+
+	async copyHeaders(selection: Slick.Range[]): Promise<void> {
+		try {
+			const results = getTableHeaderString(this, selection);
+			await this._clipboardService.writeText(results);
+		} catch (error) {
+			this._notificationService.error(nls.localize('copyFailed', "Copy failed with error: {0}", getErrorMessage(error)));
 		}
 	}