2019-06-29 05:37:23 +07:00
|
|
|
import * as path from "path";
|
|
|
|
|
2019-07-02 02:00:11 +07:00
|
|
|
import { VSBuffer } from "vs/base/common/buffer";
|
2019-06-29 05:37:23 +07:00
|
|
|
import { Emitter, Event } from "vs/base/common/event";
|
2019-07-02 02:00:11 +07:00
|
|
|
import { IDisposable } from "vs/base/common/lifecycle";
|
|
|
|
import { Schemas } from "vs/base/common/network";
|
2019-06-29 05:37:23 +07:00
|
|
|
import { OS } from "vs/base/common/platform";
|
2019-07-02 02:00:11 +07:00
|
|
|
import { URI, UriComponents } from "vs/base/common/uri";
|
2019-06-29 05:37:23 +07:00
|
|
|
import { IServerChannel } from "vs/base/parts/ipc/common/ipc";
|
|
|
|
import { IDiagnosticInfo } from "vs/platform/diagnostics/common/diagnosticsService";
|
|
|
|
import { IEnvironmentService } from "vs/platform/environment/common/environment";
|
|
|
|
import { FileDeleteOptions, FileOverwriteOptions, FileType, IStat, IWatchOptions, FileOpenOptions } from "vs/platform/files/common/files";
|
2019-07-02 02:00:11 +07:00
|
|
|
import { ILogService } from "vs/platform/log/common/log";
|
2019-06-29 05:37:23 +07:00
|
|
|
import { IRemoteAgentEnvironment } from "vs/platform/remote/common/remoteAgentEnvironment";
|
2019-07-02 02:00:11 +07:00
|
|
|
import { DiskFileSystemProvider } from "vs/workbench/services/files/node/diskFileSystemProvider";
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Extend the file provider to allow unwatching.
|
|
|
|
*/
|
|
|
|
class Watcher extends DiskFileSystemProvider {
|
|
|
|
public readonly watches = new Map<number, IDisposable>();
|
|
|
|
|
|
|
|
public dispose(): void {
|
|
|
|
this.watches.forEach((w) => w.dispose());
|
|
|
|
this.watches.clear();
|
|
|
|
super.dispose();
|
|
|
|
}
|
|
|
|
|
|
|
|
public _watch(req: number, resource: URI, opts: IWatchOptions): void {
|
|
|
|
this.watches.set(req, this.watch(resource, opts));
|
|
|
|
}
|
|
|
|
|
|
|
|
public unwatch(req: number): void {
|
|
|
|
this.watches.get(req)!.dispose();
|
|
|
|
this.watches.delete(req);
|
|
|
|
}
|
|
|
|
}
|
2019-06-29 05:37:23 +07:00
|
|
|
|
|
|
|
/**
|
|
|
|
* See: src/vs/platform/remote/common/remoteAgentFileSystemChannel.ts.
|
|
|
|
*/
|
|
|
|
export class FileProviderChannel implements IServerChannel {
|
2019-07-02 02:00:11 +07:00
|
|
|
private readonly provider: DiskFileSystemProvider;
|
|
|
|
private readonly watchers = new Map<string, Watcher>();
|
|
|
|
|
|
|
|
public constructor(private readonly logService: ILogService) {
|
|
|
|
this.provider = new DiskFileSystemProvider(this.logService);
|
|
|
|
}
|
|
|
|
|
|
|
|
public listen(_: unknown, event: string, args?: any): Event<any> {
|
2019-06-29 05:37:23 +07:00
|
|
|
switch (event) {
|
2019-07-02 02:00:11 +07:00
|
|
|
// This is where the actual file changes are sent. The watch method just
|
|
|
|
// adds things that will fire here. That means we have to split up
|
|
|
|
// watchers based on the session otherwise sessions would get events for
|
|
|
|
// other sessions. There is also no point in having the watcher unless
|
|
|
|
// something is listening. I'm not sure there is a different way to
|
|
|
|
// dispose, anyway.
|
2019-06-29 05:37:23 +07:00
|
|
|
case "filechange":
|
2019-07-02 02:00:11 +07:00
|
|
|
const session = args[0];
|
|
|
|
const emitter = new Emitter({
|
|
|
|
onFirstListenerAdd: () => {
|
|
|
|
const provider = new Watcher(this.logService);
|
|
|
|
this.watchers.set(session, provider);
|
|
|
|
provider.onDidChangeFile((events) => {
|
|
|
|
emitter.fire(events.map((event) => ({
|
|
|
|
...event,
|
|
|
|
resource: event.resource.with({ scheme: Schemas.vscodeRemote }),
|
|
|
|
})));
|
|
|
|
});
|
|
|
|
provider.onDidErrorOccur((event) => emitter.fire(event));
|
|
|
|
},
|
|
|
|
onLastListenerRemove: () => {
|
|
|
|
this.watchers.get(session)!.dispose();
|
|
|
|
this.watchers.delete(session);
|
|
|
|
},
|
|
|
|
});
|
|
|
|
|
|
|
|
return emitter.event;
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
|
|
|
|
throw new Error(`Invalid listen "${event}"`);
|
|
|
|
}
|
|
|
|
|
|
|
|
public call(_: unknown, command: string, args?: any): Promise<any> {
|
|
|
|
switch (command) {
|
|
|
|
case "stat": return this.stat(args[0]);
|
|
|
|
case "open": return this.open(args[0], args[1]);
|
|
|
|
case "close": return this.close(args[0]);
|
2019-07-02 02:00:11 +07:00
|
|
|
case "read": return this.read(args[0], args[1], args[2]);
|
2019-06-29 05:37:23 +07:00
|
|
|
case "write": return this.write(args[0], args[1], args[2], args[3], args[4]);
|
|
|
|
case "delete": return this.delete(args[0], args[1]);
|
|
|
|
case "mkdir": return this.mkdir(args[0]);
|
|
|
|
case "readdir": return this.readdir(args[0]);
|
|
|
|
case "rename": return this.rename(args[0], args[1], args[2]);
|
|
|
|
case "copy": return this.copy(args[0], args[1], args[2]);
|
2019-07-02 02:00:11 +07:00
|
|
|
case "watch": return this.watch(args[0], args[1], args[2], args[3]);
|
|
|
|
case "unwatch": return this.unwatch(args[0], args[1]);
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
|
|
|
|
throw new Error(`Invalid call "${command}"`);
|
|
|
|
}
|
|
|
|
|
2019-07-02 02:00:11 +07:00
|
|
|
private async stat(resource: UriComponents): Promise<IStat> {
|
|
|
|
return this.provider.stat(URI.from(resource));
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
|
2019-07-02 02:00:11 +07:00
|
|
|
private async open(resource: UriComponents, opts: FileOpenOptions): Promise<number> {
|
|
|
|
return this.provider.open(URI.from(resource), opts);
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
|
|
|
|
private async close(fd: number): Promise<void> {
|
2019-07-02 02:00:11 +07:00
|
|
|
return this.provider.close(fd);
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
|
2019-07-02 02:00:11 +07:00
|
|
|
private async read(fd: number, pos: number, length: number): Promise<[VSBuffer, number]> {
|
|
|
|
const buffer = VSBuffer.alloc(length);
|
|
|
|
const bytesRead = await this.provider.read(fd, pos, buffer.buffer, 0, length);
|
|
|
|
return [buffer, bytesRead];
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
|
2019-07-02 02:00:11 +07:00
|
|
|
private write(fd: number, pos: number, buffer: VSBuffer, offset: number, length: number): Promise<number> {
|
|
|
|
return this.provider.write(fd, pos, buffer.buffer, offset, length);
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
|
2019-07-02 02:00:11 +07:00
|
|
|
private async delete(resource: UriComponents, opts: FileDeleteOptions): Promise<void> {
|
|
|
|
return this.provider.delete(URI.from(resource), opts);
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
|
2019-07-02 02:00:11 +07:00
|
|
|
private async mkdir(resource: UriComponents): Promise<void> {
|
|
|
|
return this.provider.mkdir(URI.from(resource));
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
|
2019-07-02 02:00:11 +07:00
|
|
|
private async readdir(resource: UriComponents): Promise<[string, FileType][]> {
|
|
|
|
return this.provider.readdir(URI.from(resource));
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
|
2019-07-02 02:00:11 +07:00
|
|
|
private async rename(resource: UriComponents, target: UriComponents, opts: FileOverwriteOptions): Promise<void> {
|
|
|
|
return this.provider.rename(URI.from(resource), URI.from(target), opts);
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
|
2019-07-02 02:00:11 +07:00
|
|
|
private copy(resource: UriComponents, target: UriComponents, opts: FileOverwriteOptions): Promise<void> {
|
|
|
|
return this.provider.copy(URI.from(resource), URI.from(target), opts);
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
|
2019-07-02 02:00:11 +07:00
|
|
|
private async watch(session: string, req: number, resource: UriComponents, opts: IWatchOptions): Promise<void> {
|
|
|
|
this.watchers.get(session)!._watch(req, URI.from(resource), opts);
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
|
2019-07-02 02:00:11 +07:00
|
|
|
private async unwatch(session: string, req: number): Promise<void> {
|
|
|
|
this.watchers.get(session)!.unwatch(req);
|
2019-06-29 05:37:23 +07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* See: src/vs/workbench/services/remote/common/remoteAgentEnvironmentChannel.ts.
|
|
|
|
*/
|
|
|
|
export class ExtensionEnvironmentChannel implements IServerChannel {
|
|
|
|
public constructor(private readonly environment: IEnvironmentService) {}
|
|
|
|
|
|
|
|
public listen(_context: any, event: string): Event<any> {
|
|
|
|
throw new Error(`Invalid listen "${event}"`);
|
|
|
|
}
|
|
|
|
|
2019-07-02 02:18:30 +07:00
|
|
|
public call(_: unknown, command: string, _args?: any): Promise<any> {
|
2019-06-29 05:37:23 +07:00
|
|
|
switch (command) {
|
|
|
|
case "getEnvironmentData": return this.getEnvironmentData();
|
|
|
|
case "getDiagnosticInfo": return this.getDiagnosticInfo();
|
|
|
|
case "disableTelemetry": return this.disableTelemetry();
|
|
|
|
}
|
|
|
|
throw new Error(`Invalid call "${command}"`);
|
|
|
|
}
|
|
|
|
|
|
|
|
private async getEnvironmentData(): Promise<IRemoteAgentEnvironment> {
|
2019-07-02 02:00:11 +07:00
|
|
|
// TODO: this `with` stuff feels a bit jank.
|
|
|
|
// Maybe it should already come in like this instead.
|
2019-06-29 05:37:23 +07:00
|
|
|
return {
|
|
|
|
pid: process.pid,
|
2019-07-02 02:00:11 +07:00
|
|
|
appRoot: URI.file(this.environment.appRoot).with({ scheme: Schemas.vscodeRemote }),
|
|
|
|
appSettingsHome: this.environment.appSettingsHome.with({ scheme: Schemas.vscodeRemote }),
|
|
|
|
settingsPath: this.environment.machineSettingsHome.with({ scheme: Schemas.vscodeRemote }),
|
|
|
|
logsPath: URI.file(this.environment.logsPath).with({ scheme: Schemas.vscodeRemote }),
|
|
|
|
extensionsPath: URI.file(this.environment.extensionsPath).with({ scheme: Schemas.vscodeRemote }),
|
|
|
|
extensionHostLogsPath: URI.file(path.join(this.environment.logsPath, "extension-host")).with({ scheme: Schemas.vscodeRemote }), // TODO
|
|
|
|
globalStorageHome: URI.file(this.environment.globalStorageHome).with({ scheme: Schemas.vscodeRemote }),
|
|
|
|
userHome: URI.file(this.environment.userHome).with({ scheme: Schemas.vscodeRemote }),
|
2019-06-29 05:37:23 +07:00
|
|
|
extensions: [], // TODO
|
|
|
|
os: OS,
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
private getDiagnosticInfo(): Promise<IDiagnosticInfo> {
|
|
|
|
throw new Error("not implemented");
|
|
|
|
}
|
|
|
|
|
|
|
|
private disableTelemetry(): Promise<void> {
|
|
|
|
throw new Error("not implemented");
|
|
|
|
}
|
|
|
|
}
|