Microsoft/vscode
1
0
Fork 0
mirror of https://github.com/microsoft/vscode.git synced 2026-04-26 11:38:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

commands: allow transferring arraybuffers/uint8arrays over commands

Browse Source
This commit is contained in:
Connor Peet
2021-09-15 16:39:03 -07:00
parent 8845812f6a
commit 35bba41595
3 changed files with 19 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/vs/workbench/api/common/extHost.protocol.ts
View File

@@ -140,7 +140,7 @@ export interface MainThreadClipboardShape extends IDisposable {
export interface MainThreadCommandsShape extends IDisposable {
$registerCommand(id: string): void;
$unregisterCommand(id: string): void;
$executeCommand<T>(id: string, args: any[], retry: boolean): Promise<T | undefined>;
$executeCommand<T>(id: string, args: any[] | SerializableObjectWithBuffers<any[]>, retry: boolean): Promise<T | undefined>;
$getCommands(): Promise<string[]>;
}

14
src/vs/workbench/api/common/extHostCommands.ts
View File

@@ -22,6 +22,8 @@ import { createDecorator } from 'vs/platform/instantiation/common/instantiation'
import { IExtHostRpcService } from 'vs/workbench/api/common/extHostRpcService';
import { ISelection } from 'vs/editor/common/core/selection';
import { TestItemImpl } from 'vs/workbench/api/common/extHostTestingPrivateApi';
import { VSBuffer } from 'vs/base/common/buffer';
import { SerializableObjectWithBuffers } from 'vs/workbench/services/extensions/common/proxyIdentifier';
interface CommandHandler {
callback: Function;
@@ -84,6 +86,9 @@ export class ExtHostCommands implements ExtHostCommandsShape {
if (Range.isIRange((obj as modes.Location).range) && URI.isUri((obj as modes.Location).uri)) {
return extHostTypeConverter.location.to(obj);
}
if (obj instanceof VSBuffer) {
return obj.buffer.buffer;
}
if (!Array.isArray(obj)) {
return obj;
}
@@ -164,6 +169,7 @@ export class ExtHostCommands implements ExtHostCommandsShape {
} else {
// automagically convert some argument types
let hasBuffers = false;
const toArgs = cloneAndChange(args, function (value) {
if (value instanceof extHostTypes.Position) {
return extHostTypeConverter.Position.from(value);
@@ -173,6 +179,12 @@ export class ExtHostCommands implements ExtHostCommandsShape {
return extHostTypeConverter.location.from(value);
} else if (extHostTypes.NotebookRange.isNotebookRange(value)) {
return extHostTypeConverter.NotebookRange.from(value);
} else if (value instanceof ArrayBuffer) {
hasBuffers = true;
return VSBuffer.wrap(new Uint8Array(value));
} else if (value instanceof Uint8Array) {
hasBuffers = true;
return VSBuffer.wrap(value);
}
if (!Array.isArray(value)) {
return value;
@@ -180,7 +192,7 @@ export class ExtHostCommands implements ExtHostCommandsShape {
});
try {
const result = await this._proxy.$executeCommand<T>(id, toArgs, retry);
const result = await this._proxy.$executeCommand<T>(id, hasBuffers ? new SerializableObjectWithBuffers(toArgs) : toArgs, retry);
return revive<any>(result);
} catch (e) {
// Rerun the command when it wasn't known, had arguments, and when retry