Add commands (#2)

* Add remote command execution

* Add tests for environment variables and resize

* Fix tab spacing, add newlines

* Remove extra newline

* Add fork

packages/server/src/node/server.ts

@@ -1,13 +1,19 @@
 import { logger, field } from "@coder/logger";
+import { TextDecoder } from "text-encoding";
 import { ClientMessage } from "../proto";
 import { evaluate } from "./evaluate";
 import { ReadWriteConnection } from "../common/connection";
+import { Process, handleNewSession } from "./command";
 
 export class Server {
 
+	private readonly sessions: Map<number, Process>;
+
 	public constructor(
 		private readonly connection: ReadWriteConnection,
 	) {
+		this.sessions = new Map();
+
 		connection.onMessage((data) => {
 			try {
 				this.handleMessage(ClientMessage.deserializeBinary(data));
@@ -20,7 +26,42 @@ export class Server {
 	private handleMessage(message: ClientMessage): void {
 		if (message.hasNewEval()) {
 			evaluate(this.connection, message.getNewEval()!);
+		} else if (message.hasNewSession()) {
+			const session = handleNewSession(this.connection, message.getNewSession()!, () => {
+				this.sessions.delete(message.getNewSession()!.getId());				
+			});
+
+			this.sessions.set(message.getNewSession()!.getId(), session);
+		} else if (message.hasCloseSessionInput()) {
+			const s = this.getSession(message.getCloseSessionInput()!.getId());
+			if (!s || !s.stdin) {
+				return;
+			}
+			s.stdin.end();
+		} else if (message.hasResizeSessionTty()) {
+			const s = this.getSession(message.getResizeSessionTty()!.getId());
+			if (!s || !s.resize) {
+				return;
+			}
+			const tty = message.getResizeSessionTty()!.getTtyDimensions()!;
+			s.resize(tty.getWidth(), tty.getHeight());
+		} else if (message.hasShutdownSession()) {
+			const s = this.getSession(message.getShutdownSession()!.getId());
+			if (!s) {
+				return;
+			}
+			s.kill(message.getShutdownSession()!.getSignal());
+		} else if (message.hasWriteToSession()) {
+			const s = this.getSession(message.getWriteToSession()!.getId());
+			if (!s) {
+				return;
+			}
+			s.write(new TextDecoder().decode(message.getWriteToSession()!.getData_asU8()));
 		}
 	}
 
-}
+	private getSession(id: number): Process | undefined {
+		return this.sessions.get(id);
+	}
+
+}