Add createServer (#18)

packages/protocol/src/node/command.ts

@@ -3,7 +3,7 @@ import * as net from "net";
 import * as nodePty from "node-pty";
 import * as stream from "stream";
 import { TextEncoder } from "text-encoding";
-import { NewSessionMessage, ServerMessage, SessionDoneMessage, SessionOutputMessage, IdentifySessionMessage, NewConnectionMessage, ConnectionEstablishedMessage, NewConnectionFailureMessage, ConnectionCloseMessage, ConnectionOutputMessage } from "../proto";
+import { NewSessionMessage, ServerMessage, SessionDoneMessage, SessionOutputMessage, IdentifySessionMessage, NewConnectionMessage, ConnectionEstablishedMessage, NewConnectionFailureMessage, ConnectionCloseMessage, ConnectionOutputMessage, NewServerMessage, ServerEstablishedMessage, NewServerFailureMessage, ServerCloseMessage, ServerConnectionEstablishedMessage } from "../proto";
 import { SendableConnection } from "../common/connection";
 import { ServerOptions } from "./server";
 
@@ -180,3 +180,48 @@ export const handleNewConnection = (connection: SendableConnection, newConnectio
 
 	return socket;
 };
+
+export const handleNewServer = (connection: SendableConnection, newServer: NewServerMessage, addSocket: (socket: net.Socket) => number, onExit: () => void): net.Server => {
+	const s = net.createServer();
+
+	try {
+		s.listen(newServer.getPath() ? newServer.getPath() : newServer.getPort(), () => {
+			const se = new ServerEstablishedMessage();
+			se.setId(newServer.getId());
+			const sm = new ServerMessage();
+			sm.setServerEstablished(se);
+			connection.send(sm.serializeBinary());
+		});
+	} catch (ex) {
+		const sf = new NewServerFailureMessage();
+		sf.setId(newServer.getId());
+		const sm = new ServerMessage();
+		sm.setServerFailure(sf);
+		connection.send(sm.serializeBinary());
+	
+		onExit();
+	}
+
+	s.on("close", () => {
+		const sc = new ServerCloseMessage();
+		sc.setId(newServer.getId());
+		const sm = new ServerMessage();
+		sm.setServerClose(sc);
+		connection.send(sm.serializeBinary());
+
+		onExit();
+	});
+
+	s.on("connection", (socket) => {
+		const socketId = addSocket(socket);
+
+		const sock = new ServerConnectionEstablishedMessage();
+		sock.setServerId(newServer.getId());
+		sock.setConnectionId(socketId);
+		const sm = new ServerMessage();
+		sm.setServerConnectionEstablished(sock);
+		connection.send(sm.serializeBinary());
+	});
+
+	return s;
+};

packages/protocol/src/node/server.ts

@@ -8,7 +8,7 @@ import { logger, field } from "@coder/logger";
 import { ClientMessage, WorkingInitMessage, ServerMessage, NewSessionMessage, WriteToSessionMessage } from "../proto";
 import { evaluate } from "./evaluate";
 import { ReadWriteConnection } from "../common/connection";
-import { Process, handleNewSession, handleNewConnection } from "./command";
+import { Process, handleNewSession, handleNewConnection, handleNewServer } from "./command";
 import * as net from "net";
 
 export interface ServerOptions {
@@ -22,6 +22,9 @@ export class Server {
 
 	private readonly sessions: Map<number, Process> = new Map();
 	private readonly connections: Map<number, net.Socket> = new Map();
+	private readonly servers: Map<number, net.Server> = new Map();
+
+	private connectionId: number = Number.MAX_SAFE_INTEGER;
 
 	public constructor(
 		private readonly connection: ReadWriteConnection,
@@ -147,9 +150,28 @@ export class Server {
 				return;
 			}
 			c.end();
+		} else if (message.hasNewServer()) {
+			const s = handleNewServer(this.connection, message.getNewServer()!, (socket) => {
+				const id = this.connectionId--;
+				this.connections.set(id, socket);
+				return id;
+			}, () => {
+				this.connections.delete(message.getNewServer()!.getId());
+			});
+			this.servers.set(message.getNewServer()!.getId(), s);
+		} else if (message.hasServerClose()) {
+			const s = this.getServer(message.getServerClose()!.getId());
+			if (!s) {
+				return;
+			}
+			s.close();
 		}
 	}
 
+	private getServer(id: number): net.Server | undefined {
+		return this.servers.get(id);
+	}
+
 	private getConnection(id: number): net.Socket | undefined {
 		return this.connections.get(id);
 	}