49 lines
1.2 KiB
TypeScript
49 lines
1.2 KiB
TypeScript
|
import { WebSocket } from "ws";
|
||
|
|
import { SocketVM } from "./SocketVM";
|
||
|
|
|
||
|
|
import * as Shared from "@socketcomputer/shared";
|
||
|
|
|
||
|
|
export class VMUser {
|
||
|
|
public connection: WebSocket;
|
||
|
|
public address: string;
|
||
|
|
public username: string = "";
|
||
|
|
private vm: SocketVM;
|
||
|
|
|
||
|
|
|
||
|
|
constructor(connection: WebSocket, slot: SocketVM, address: string) {
|
||
|
|
this.connection = connection;
|
||
|
|
this.address = address;
|
||
|
|
this.vm = slot;
|
||
|
|
|
||
|
|
this.vm.AddUser(this);
|
||
|
|
|
||
|
|
this.connection.on('message', async (data, isBinary) => {
|
||
|
|
if (!isBinary) this.connection.close(1000);
|
||
|
|
await this.vm.OnWSMessage(this, data as Buffer);
|
||
|
|
});
|
||
|
|
|
||
|
|
this.connection.on('close', async () => {
|
||
|
|
console.log('closed');
|
||
|
|
await this.vm.RemUser(this);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
async SendMessage(messageGenerator: (encoder: Shared.MessageEncoder) => ArrayBuffer) {
|
||
|
|
await this.SendBuffer(messageGenerator(new Shared.MessageEncoder()));
|
||
|
|
}
|
||
|
|
|
||
|
|
async SendBuffer(buffer: ArrayBuffer): Promise<void> {
|
||
|
|
return new Promise((res, rej) => {
|
||
|
|
if (this.connection.readyState !== WebSocket.CLOSED) {
|
||
|
|
this.connection.send(buffer);
|
||
|
|
res();
|
||
|
|
}
|
||
|
|
rej(new Error('connection haves closed'));
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
static GenerateName() {
|
||
|
|
return `guest${Math.floor(Math.random() * (99999 - 10000) + 10000)}`;
|
||
|
|
}
|
||
|
|
}
|