Elijah/CollabVM-Agent-Server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

log vm connection

Browse source
This commit is contained in:
Elijah 2023-12-10 16:00:24 -05:00
parent 028f0f1f66
commit 49728dc841
2 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/VM.ts
View file

@ -17,12 +17,14 @@ export default class VM {
#noNop : boolean = false; #noNop : boolean = false;
connected : boolean = false; connected : boolean = false;
#events : EventEmitter; #events : EventEmitter;
#id : string;
#currentChunkSize : number | null = null; #currentChunkSize : number | null = null;
#currentChunk : Buffer | null = null; #currentChunk : Buffer | null = null;
#currentChunkOffset : number = 0; #currentChunkOffset : number = 0;
constructor(socketpath : string) { constructor(socketpath : string, id : string) {
this.#socketpath = socketpath; this.#socketpath = socketpath;
this.#id = id;
this.#socket = new Socket(); this.#socket = new Socket();
this.#socket.connect(socketpath); this.#socket.connect(socketpath);
this.#socket.on('connect', () => { this.#socket.on('connect', () => {
@ -100,6 +102,7 @@ export default class VM {
if (this.#nopTimeout) clearInterval(this.#nopTimeout); if (this.#nopTimeout) clearInterval(this.#nopTimeout);
this.#nopTimeout = setInterval(() => this.#nopTimeoutFunc(), 5000); this.#nopTimeout = setInterval(() => this.#nopTimeoutFunc(), 5000);
if (!this.isConnectedToVM) { if (!this.isConnectedToVM) {
log("INFO", `Got connection to VM at ${this.#id}`);
if (this.#messageQueue.size > 0) this.messageQueueLoop(); if (this.#messageQueue.size > 0) this.messageQueueLoop();
this.isConnectedToVM = true; this.isConnectedToVM = true;
this.#noNop = false; this.#noNop = false;
@ -120,6 +123,7 @@ export default class VM {
} else { } else {
this.isConnectedToVM = false; this.isConnectedToVM = false;
clearInterval(this.#nopTimeout!); clearInterval(this.#nopTimeout!);
log("INFO", `Connection to VM at ${this.#id} timed out`);
} }
} }
} }

2
src/index.ts
View file

@ -28,7 +28,7 @@ if (config.DiscordWebhook) {
var VMs = new Map<string, VM>(); var VMs = new Map<string, VM>();
config.VMs.forEach((v) => { config.VMs.forEach((v) => {
VMs.set(v.ID, new VM(v.SocketPath)); VMs.set(v.ID, new VM(v.SocketPath, v.ID));
}); });
var IPs = new Map<string, RateLimit>(); var IPs = new Map<string, RateLimit>();