mirror of https://git.tuxpa.in/a/code-server.git
95 lines
2.5 KiB
TypeScript
95 lines
2.5 KiB
TypeScript
import { client } from "@coder/ide/src/fill/client";
|
|
import { EventEmitter } from "events";
|
|
import * as nodePty from "node-pty";
|
|
import { ActiveEvalHelper } from "@coder/protocol";
|
|
import { logger } from "@coder/logger";
|
|
|
|
/**
|
|
* Implementation of nodePty for the browser.
|
|
*/
|
|
class Pty implements nodePty.IPty {
|
|
private readonly emitter = new EventEmitter();
|
|
private readonly ae: ActiveEvalHelper;
|
|
private _pid = -1;
|
|
private _process = "";
|
|
|
|
public constructor(file: string, args: string[] | string, options: nodePty.IPtyForkOptions) {
|
|
this.ae = client.run((ae, file, args, options) => {
|
|
ae.preserveEnv(options);
|
|
|
|
const ptyProc = ae.modules.pty.spawn(file, args, options);
|
|
|
|
let process = ptyProc.process;
|
|
ae.emit("process", process);
|
|
ae.emit("pid", ptyProc.pid);
|
|
|
|
const timer = setInterval(() => {
|
|
if (ptyProc.process !== process) {
|
|
process = ptyProc.process;
|
|
ae.emit("process", process);
|
|
}
|
|
}, 200);
|
|
|
|
ptyProc.on("exit", (code, signal) => {
|
|
clearTimeout(timer);
|
|
ae.emit("exit", code, signal);
|
|
});
|
|
|
|
ptyProc.on("data", (data) => ae.emit("data", data));
|
|
|
|
ae.on("resize", (cols: number, rows: number) => ptyProc.resize(cols, rows));
|
|
ae.on("write", (data: string) => ptyProc.write(data));
|
|
ae.on("kill", (signal: string) => ptyProc.kill(signal));
|
|
|
|
return {
|
|
onDidDispose: (cb): void => ptyProc.on("exit", cb),
|
|
dispose: (): void => {
|
|
ptyProc.kill();
|
|
setTimeout(() => ptyProc.kill("SIGKILL"), 5000); // Double tap.
|
|
},
|
|
};
|
|
}, file, args, options);
|
|
|
|
this.ae.on("error", (error) => logger.error(error.message));
|
|
|
|
this.ae.on("pid", (pid) => this._pid = pid);
|
|
this.ae.on("process", (process) => this._process = process);
|
|
|
|
this.ae.on("exit", (code, signal) => this.emitter.emit("exit", code, signal));
|
|
this.ae.on("data", (data) => this.emitter.emit("data", data));
|
|
}
|
|
|
|
public get pid(): number {
|
|
return this._pid;
|
|
}
|
|
|
|
public get process(): string {
|
|
return this._process;
|
|
}
|
|
|
|
// tslint:disable-next-line no-any
|
|
public on(event: string, listener: (...args: any[]) => void): void {
|
|
this.emitter.on(event, listener);
|
|
}
|
|
|
|
public resize(columns: number, rows: number): void {
|
|
this.ae.emit("resize", columns, rows);
|
|
}
|
|
|
|
public write(data: string): void {
|
|
this.ae.emit("write", data);
|
|
}
|
|
|
|
public kill(signal?: string): void {
|
|
this.ae.emit("kill", signal);
|
|
}
|
|
}
|
|
|
|
const ptyType: typeof nodePty = {
|
|
spawn: (file: string, args: string[] | string, options: nodePty.IPtyForkOptions): nodePty.IPty => {
|
|
return new Pty(file, args, options);
|
|
},
|
|
};
|
|
|
|
module.exports = ptyType;
|