code-server-2/packages/vscode/src/fill/node-pty.ts

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) => {
			const nodePty = ae.require("node-pty") as typeof import("node-pty");

			ae.preserveEnv(options);

			const ptyProc = nodePty.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;