Commit evil (set process names)

5 files changed, 21 insertions, 1 deletions

diff --git a/src/api/start.ts b/src/api/start.ts

index 96c0c348..44e028a0 100644
--- a/src/api/start.ts
+++ b/src/api/start.ts
@@ -26,6 +26,7 @@ config({ quiet: true });
 import { SpacebarServer } from "./Server";
 import cluster from "cluster";
 import os from "os";
+import fs from "fs";
 let cores = 1;
 try {
     cores = Number(process.env.THREADS) || os.cpus().length;
@@ -48,6 +49,9 @@ if (cluster.isPrimary && process.env.NODE_ENV == "production") {
 } else {
     const port = Number(process.env.PORT) || 3001;
 
+    if (fs.existsSync("/proc/self/comm")) fs.writeFileSync("/proc/self/comm", `spacebar-api-${cluster.worker ? cluster.worker.id : port}`);
+    process.title = `sb-api-${cluster.worker ? cluster.worker.id : port}`;
+
     const server = new SpacebarServer({ port });
     server.start().catch(console.error);
 
diff --git a/src/cdn/start.ts b/src/cdn/start.ts

index 4455806a..3d98d5f2 100644
--- a/src/cdn/start.ts
+++ b/src/cdn/start.ts
@@ -22,10 +22,15 @@ import { config } from "dotenv";
 config({ quiet: true });
 
 import { CDNServer } from "./Server";
+import fs from "fs";
+import cluster from "cluster";
 const server = new CDNServer({ port: Number(process.env.PORT) || 3003 });
 server
     .start()
     .then(() => {
+        if (fs.existsSync("/proc/self/comm")) fs.writeFileSync("/proc/self/comm", `spacebar-cdn-${cluster.worker ? cluster.worker.id : server.options.port}`);
+        process.title = `sb-cdn-${cluster.worker ? cluster.worker.id : server.options.port}`;
+
         console.log("[Server] started on :" + server.options.port);
     })
     .catch((e) => console.error("[Server] Error starting: ", e));
diff --git a/src/gateway/start.ts b/src/gateway/start.ts

index 10dfac92..50d69a61 100644
--- a/src/gateway/start.ts
+++ b/src/gateway/start.ts
@@ -23,6 +23,8 @@ process.on("unhandledRejection", console.error);
 
 import { Server } from "./Server";
 import { config } from "dotenv";
+import fs from "fs";
+import cluster from "cluster";
 config({ quiet: true });
 
 let port = Number(process.env.PORT);
@@ -31,4 +33,8 @@ if (isNaN(port)) port = 3002;
 const server = new Server({
     port,
 });
+
+if (fs.existsSync("/proc/self/comm")) fs.writeFileSync("/proc/self/comm", `spacebar-gw-${cluster.worker ? cluster.worker.id : port}`);
+process.title = `sb-gw-${cluster.worker ? cluster.worker.id : port}`;
+
 server.start();
diff --git a/src/util/util/lambert-server/Server.ts b/src/util/util/lambert-server/Server.ts

index 3b7d37a7..47cc457b 100644
--- a/src/util/util/lambert-server/Server.ts
+++ b/src/util/util/lambert-server/Server.ts
@@ -6,7 +6,6 @@ import { HTTPError } from "./HTTPError";
 import bodyParser from "body-parser";
 // import helmet from "helmet";
 import http from "http";
-import chalk from "chalk";
 
 declare global {
     // eslint-disable-next-line @typescript-eslint/no-namespace
diff --git a/src/webrtc/start.ts b/src/webrtc/start.ts

index 5ad9761a..14b4ce3a 100644
--- a/src/webrtc/start.ts
+++ b/src/webrtc/start.ts
@@ -22,6 +22,8 @@ process.on("unhandledRejection", console.error);
 
 import { config } from "dotenv";
 import { Server } from "./Server";
+import fs from "fs";
+import cluster from "cluster";
 config({ quiet: true });
 
 const port = Number(process.env.PORT) || 3004;
@@ -29,4 +31,8 @@ const port = Number(process.env.PORT) || 3004;
 const server = new Server({
     port,
 });
+
+if (fs.existsSync("/proc/self/comm")) fs.writeFileSync("/proc/self/comm", `spacebar-wrtc-${cluster.worker ? cluster.worker.id : port}`);
+process.title = `sb-wrtc-${cluster.worker ? cluster.worker.id : port}`;
+
 server.start();