Support creating image in workspace when pasting image data into markdown (#160554)

For #157043

extensions/markdown-language-features/src/languageFeatures/copyPaste.ts

@@ -3,27 +3,99 @@
  *  Licensed under the MIT License. See License.txt in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
 
+import * as path from 'path';
 import * as vscode from 'vscode';
-import { tryGetUriListSnippet } from './dropIntoEditor';
+import { Utils } from 'vscode-uri';
+import { createUriListSnippet, tryGetUriListSnippet } from './dropIntoEditor';
 
-export function registerPasteSupport(selector: vscode.DocumentSelector) {
-	return vscode.languages.registerDocumentPasteEditProvider(selector, new class implements vscode.DocumentPasteEditProvider {
+const supportedImageMimes = new Set([
+	'image/png',
+	'image/jpg',
+]);
 
-		async provideDocumentPasteEdits(
-			document: vscode.TextDocument,
-			_ranges: readonly vscode.Range[],
-			dataTransfer: vscode.DataTransfer,
-			token: vscode.CancellationToken,
-		): Promise<vscode.DocumentPasteEdit | undefined> {
-			const enabled = vscode.workspace.getConfiguration('markdown', document).get('experimental.editor.pasteLinks.enabled', true);
-			if (!enabled) {
-				return;
-			}
+class PasteEditProvider implements vscode.DocumentPasteEditProvider {
 
-			const snippet = await tryGetUriListSnippet(document, dataTransfer, token);
-			return snippet ? new vscode.DocumentPasteEdit(snippet) : undefined;
+	async provideDocumentPasteEdits(
+		document: vscode.TextDocument,
+		_ranges: readonly vscode.Range[],
+		dataTransfer: vscode.DataTransfer,
+		token: vscode.CancellationToken,
+	): Promise<vscode.DocumentPasteEdit | undefined> {
+		const enabled = vscode.workspace.getConfiguration('markdown', document).get('experimental.editor.pasteLinks.enabled', true);
+		if (!enabled) {
+			return;
 		}
-	}, {
-		pasteMimeTypes: ['text/uri-list']
+
+		for (const imageMime of supportedImageMimes) {
+			const file = dataTransfer.get(imageMime)?.asFile();
+			if (file) {
+				const edit = await this.makeCreateImagePasteEdit(document, file, token);
+				if (token.isCancellationRequested) {
+					return;
+				}
+
+				if (edit) {
+					return edit;
+				}
+			}
+		}
+
+		const snippet = await tryGetUriListSnippet(document, dataTransfer, token);
+		return snippet ? new vscode.DocumentPasteEdit(snippet) : undefined;
+	}
+
+	private async makeCreateImagePasteEdit(document: vscode.TextDocument, file: vscode.DataTransferFile, token: vscode.CancellationToken): Promise<vscode.DocumentPasteEdit | undefined> {
+		if (file.uri) {
+			// If file is already in workspace, we don't want to create a copy of it
+			const workspaceFolder = vscode.workspace.getWorkspaceFolder(file.uri);
+			if (workspaceFolder) {
+				const snippet = createUriListSnippet(document, [file.uri]);
+				return snippet ? new vscode.DocumentPasteEdit(snippet) : undefined;
+			}
+		}
+
+		const uri = await this.getNewFileName(document, file);
+		if (token.isCancellationRequested) {
+			return;
+		}
+
+		const snippet = createUriListSnippet(document, [uri]);
+		if (!snippet) {
+			return;
+		}
+
+		// Note that there is currently no way to undo the file creation :/
+		const workspaceEdit = new vscode.WorkspaceEdit();
+		workspaceEdit.createFile(uri, { contents: await file.data() });
+
+		const pasteEdit = new vscode.DocumentPasteEdit(snippet);
+		pasteEdit.additionalEdit = workspaceEdit;
+		return pasteEdit;
+	}
+
+	private async getNewFileName(document: vscode.TextDocument, file: vscode.DataTransferFile): Promise<vscode.Uri> {
+		const root = Utils.dirname(document.uri);
+
+		const ext = path.extname(file.name);
+		const baseName = path.basename(file.name, ext);
+		for (let i = 0; ; ++i) {
+			const name = i === 0 ? baseName : `${baseName}-${i}`;
+			const uri = vscode.Uri.joinPath(root, `${name}.${ext}`);
+			try {
+				await vscode.workspace.fs.stat(uri);
+			} catch {
+				// Does not exist
+				return uri;
+			}
+		}
+	}
+}
+
+export function registerPasteSupport(selector: vscode.DocumentSelector,) {
+	return vscode.languages.registerDocumentPasteEditProvider(selector, new PasteEditProvider(), {
+		pasteMimeTypes: [
+			'text/uri-list',
+			...supportedImageMimes,
+		]
 	});
 }

extensions/markdown-language-features/src/languageFeatures/dropIntoEditor.ts

@@ -53,8 +53,12 @@ export async function tryGetUriListSnippet(document: vscode.TextDocument, dataTr
 		}
 	}
 
+	return createUriListSnippet(document, uris);
+}
+
+export function createUriListSnippet(document: vscode.TextDocument, uris: readonly vscode.Uri[]): vscode.SnippetString | undefined {
 	if (!uris.length) {
-		return;
+		return undefined;
 	}
 
 	const dir = getDocumentDir(document);
@@ -74,7 +78,6 @@ export async function tryGetUriListSnippet(document: vscode.TextDocument, dataTr
 			snippet.appendText(' ');
 		}
 	});
-
 	return snippet;
 }