mirror of
https://github.com/microsoft/vscode.git
synced 2026-04-22 17:48:56 +01:00
Fix range when copying empty selection (#182227)
This fixes the range extensions get when copying an empty selection. As part of this, I've also: - Added tests for this change - Made the paste parts of the api optional. This is useful when a test provider only wants to add data on copy
This commit is contained in:
@@ -0,0 +1,190 @@
|
||||
/*---------------------------------------------------------------------------------------------
|
||||
* Copyright (c) Microsoft Corporation. All rights reserved.
|
||||
* Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
*--------------------------------------------------------------------------------------------*/
|
||||
|
||||
import * as assert from 'assert';
|
||||
import * as vscode from 'vscode';
|
||||
import { assertNoRpc, createRandomFile, usingDisposables } from '../utils';
|
||||
|
||||
const textPlain = 'text/plain';
|
||||
|
||||
(vscode.env.uiKind === vscode.UIKind.Web ? suite.skip : suite)('vscode API - Copy Paste', () => {
|
||||
|
||||
teardown(assertNoRpc);
|
||||
|
||||
test('Copy should be able to overwrite text/plain', usingDisposables(async (disposables) => {
|
||||
const file = await createRandomFile('$abcde@');
|
||||
const doc = await vscode.workspace.openTextDocument(file);
|
||||
|
||||
const editor = await vscode.window.showTextDocument(doc);
|
||||
editor.selections = [new vscode.Selection(0, 1, 0, 6)];
|
||||
|
||||
disposables.push(vscode.languages.registerDocumentPasteEditProvider({ language: 'plaintext' }, new class implements vscode.DocumentPasteEditProvider {
|
||||
async prepareDocumentPaste(_document: vscode.TextDocument, _ranges: readonly vscode.Range[], dataTransfer: vscode.DataTransfer, _token: vscode.CancellationToken): Promise<void> {
|
||||
const existing = dataTransfer.get(textPlain);
|
||||
if (existing) {
|
||||
const str = await existing.asString();
|
||||
const reversed = reverseString(str);
|
||||
dataTransfer.set(textPlain, new vscode.DataTransferItem(reversed));
|
||||
}
|
||||
}
|
||||
}, { copyMimeTypes: [textPlain] }));
|
||||
|
||||
await vscode.commands.executeCommand('editor.action.clipboardCopyAction');
|
||||
const newDocContent = getNextDocumentText(disposables, doc);
|
||||
await vscode.commands.executeCommand('editor.action.clipboardPasteAction');
|
||||
assert.strictEqual(await newDocContent, '$edcba@');
|
||||
}));
|
||||
|
||||
test('Copy with empty selection should copy entire line', usingDisposables(async (disposables) => {
|
||||
const file = await createRandomFile('abc\ndef');
|
||||
const doc = await vscode.workspace.openTextDocument(file);
|
||||
await vscode.window.showTextDocument(doc);
|
||||
|
||||
disposables.push(vscode.languages.registerDocumentPasteEditProvider({ language: 'plaintext' }, new class implements vscode.DocumentPasteEditProvider {
|
||||
async prepareDocumentPaste(_document: vscode.TextDocument, _ranges: readonly vscode.Range[], dataTransfer: vscode.DataTransfer, _token: vscode.CancellationToken): Promise<void> {
|
||||
const existing = dataTransfer.get(textPlain);
|
||||
if (existing) {
|
||||
const str = await existing.asString();
|
||||
// text/plain includes the trailing new line in this case
|
||||
// On windows, this will always be `\r\n` even if the document uses `\n`
|
||||
const eol = str.match(/\r?\n$/);
|
||||
const reversed = reverseString(str.slice(0, -eol![0].length));
|
||||
dataTransfer.set(textPlain, new vscode.DataTransferItem(reversed + '\n'));
|
||||
}
|
||||
}
|
||||
}, { copyMimeTypes: [textPlain] }));
|
||||
|
||||
await vscode.commands.executeCommand('editor.action.clipboardCopyAction');
|
||||
const newDocContent = getNextDocumentText(disposables, doc);
|
||||
await vscode.commands.executeCommand('editor.action.clipboardPasteAction');
|
||||
assert.strictEqual(await newDocContent, `cba\nabc\ndef`);
|
||||
}));
|
||||
|
||||
test('Copy with multiple selections should get all selections', usingDisposables(async (disposables) => {
|
||||
const file = await createRandomFile('111\n222\n333');
|
||||
const doc = await vscode.workspace.openTextDocument(file);
|
||||
const editor = await vscode.window.showTextDocument(doc);
|
||||
|
||||
editor.selections = [
|
||||
new vscode.Selection(0, 0, 0, 3),
|
||||
new vscode.Selection(2, 0, 2, 3),
|
||||
];
|
||||
|
||||
disposables.push(vscode.languages.registerDocumentPasteEditProvider({ language: 'plaintext' }, new class implements vscode.DocumentPasteEditProvider {
|
||||
async prepareDocumentPaste(document: vscode.TextDocument, ranges: readonly vscode.Range[], dataTransfer: vscode.DataTransfer, _token: vscode.CancellationToken): Promise<void> {
|
||||
const existing = dataTransfer.get(textPlain);
|
||||
if (existing) {
|
||||
const selections = ranges.map(range => document.getText(range));
|
||||
dataTransfer.set(textPlain, new vscode.DataTransferItem(`(${ranges.length})${selections.join(' ')}`));
|
||||
}
|
||||
}
|
||||
}, { copyMimeTypes: [textPlain] }));
|
||||
|
||||
await vscode.commands.executeCommand('editor.action.clipboardCopyAction');
|
||||
editor.selections = [new vscode.Selection(0, 0, 0, 0)];
|
||||
const newDocContent = getNextDocumentText(disposables, doc);
|
||||
await vscode.commands.executeCommand('editor.action.clipboardPasteAction');
|
||||
|
||||
assert.strictEqual(await newDocContent, `(2)111 333111\n222\n333`);
|
||||
}));
|
||||
|
||||
test('Earlier invoked copy providers should win when writing values', usingDisposables(async (disposables) => {
|
||||
const file = await createRandomFile('abc\ndef');
|
||||
const doc = await vscode.workspace.openTextDocument(file);
|
||||
|
||||
const editor = await vscode.window.showTextDocument(doc);
|
||||
editor.selections = [new vscode.Selection(0, 0, 0, 3)];
|
||||
|
||||
const callOrder: string[] = [];
|
||||
const a_id = 'a';
|
||||
const b_id = 'b';
|
||||
|
||||
let providerAResolve: () => void;
|
||||
const providerAFinished = new Promise<void>(resolve => providerAResolve = resolve);
|
||||
|
||||
disposables.push(vscode.languages.registerDocumentPasteEditProvider({ language: 'plaintext' }, new class implements vscode.DocumentPasteEditProvider {
|
||||
async prepareDocumentPaste(_document: vscode.TextDocument, _ranges: readonly vscode.Range[], dataTransfer: vscode.DataTransfer, _token: vscode.CancellationToken): Promise<void> {
|
||||
callOrder.push(a_id);
|
||||
dataTransfer.set(textPlain, new vscode.DataTransferItem('a'));
|
||||
providerAResolve();
|
||||
}
|
||||
}, { copyMimeTypes: [textPlain] }));
|
||||
|
||||
// Later registered providers will be called first
|
||||
disposables.push(vscode.languages.registerDocumentPasteEditProvider({ language: 'plaintext' }, new class implements vscode.DocumentPasteEditProvider {
|
||||
async prepareDocumentPaste(_document: vscode.TextDocument, _ranges: readonly vscode.Range[], dataTransfer: vscode.DataTransfer, _token: vscode.CancellationToken): Promise<void> {
|
||||
callOrder.push(b_id);
|
||||
|
||||
// Wait for the first provider to finish even though we were called first.
|
||||
// This tests that resulting order does not depend on the order the providers
|
||||
// return in.
|
||||
await providerAFinished;
|
||||
|
||||
dataTransfer.set(textPlain, new vscode.DataTransferItem('b'));
|
||||
}
|
||||
}, { copyMimeTypes: [textPlain] }));
|
||||
|
||||
await vscode.commands.executeCommand('editor.action.clipboardCopyAction');
|
||||
const newDocContent = getNextDocumentText(disposables, doc);
|
||||
await vscode.commands.executeCommand('editor.action.clipboardPasteAction');
|
||||
assert.strictEqual(await newDocContent, 'b\ndef');
|
||||
|
||||
// Confirm provider call order is what we expected
|
||||
assert.deepStrictEqual(callOrder, [b_id, a_id]);
|
||||
}));
|
||||
|
||||
test('Copy providers should not be able to effect the data transfer of another', usingDisposables(async (disposables) => {
|
||||
const file = await createRandomFile('abc\ndef');
|
||||
const doc = await vscode.workspace.openTextDocument(file);
|
||||
|
||||
const editor = await vscode.window.showTextDocument(doc);
|
||||
editor.selections = [new vscode.Selection(0, 0, 0, 3)];
|
||||
|
||||
|
||||
let providerAResolve: () => void;
|
||||
const providerAFinished = new Promise<void>(resolve => providerAResolve = resolve);
|
||||
|
||||
disposables.push(vscode.languages.registerDocumentPasteEditProvider({ language: 'plaintext' }, new class implements vscode.DocumentPasteEditProvider {
|
||||
async prepareDocumentPaste(_document: vscode.TextDocument, _ranges: readonly vscode.Range[], dataTransfer: vscode.DataTransfer, _token: vscode.CancellationToken): Promise<void> {
|
||||
dataTransfer.set(textPlain, new vscode.DataTransferItem('xyz'));
|
||||
providerAResolve();
|
||||
}
|
||||
}, { copyMimeTypes: [textPlain] }));
|
||||
|
||||
disposables.push(vscode.languages.registerDocumentPasteEditProvider({ language: 'plaintext' }, new class implements vscode.DocumentPasteEditProvider {
|
||||
async prepareDocumentPaste(_document: vscode.TextDocument, _ranges: readonly vscode.Range[], dataTransfer: vscode.DataTransfer, _token: vscode.CancellationToken): Promise<void> {
|
||||
|
||||
// Wait for the first provider to finish
|
||||
await providerAFinished;
|
||||
|
||||
// We we access the data transfer here, we should not see changes made by the first provider
|
||||
const entry = dataTransfer.get(textPlain);
|
||||
const str = await entry!.asString();
|
||||
dataTransfer.set(textPlain, new vscode.DataTransferItem(reverseString(str)));
|
||||
}
|
||||
}, { copyMimeTypes: [textPlain] }));
|
||||
|
||||
await vscode.commands.executeCommand('editor.action.clipboardCopyAction');
|
||||
const newDocContent = getNextDocumentText(disposables, doc);
|
||||
await vscode.commands.executeCommand('editor.action.clipboardPasteAction');
|
||||
assert.strictEqual(await newDocContent, 'cba\ndef');
|
||||
|
||||
}));
|
||||
});
|
||||
|
||||
function reverseString(str: string) {
|
||||
return str.split("").reverse().join("");
|
||||
}
|
||||
|
||||
function getNextDocumentText(disposables: vscode.Disposable[], doc: vscode.TextDocument): Promise<string> {
|
||||
return new Promise<string>(resolve => {
|
||||
disposables.push(vscode.workspace.onDidChangeTextDocument(e => {
|
||||
if (e.document === doc) {
|
||||
resolve(doc.getText());
|
||||
}
|
||||
}));
|
||||
});
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user