feat: add accounts command and root-aware logout handling

Store root client id in Redis at startup; add an accounts command to
list
connected clients; allow logout with an optional clientId and handle
root
client logout; change logout command role to admin

Author: mrepol742

src/commands/accounts.ts

@@ -0,0 +1,28 @@
+import { Message } from "../types/message";
+import { download } from "../components/utils/download";
+import { getClientIds } from "../components/services/account";
+
+export const info = {
+  command: "accounts",
+  description: "List all accounts currently connected in.",
+  usage: "accounts",
+  example: "accounts",
+  role: "user",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message): Promise<void> {
+  let accounts = "";
+  const accountsIds = await getClientIds();
+  await Promise.all(
+    Array.from(accountsIds.entries()).map(([clientId, isRoot]) => {
+      accounts += `\`${clientId}\` - ${isRoot ? "Root" : "User"}\n`;
+    }),
+  );
+
+  msg.reply(`
+  \`Accounts\`
+
+  ${accounts}
+  `);
+}

src/commands/logout.ts

@@ -2,18 +2,30 @@ import { Message } from "../types/message";
 import { removeAccount } from "../components/client";
 import log from "../components/utils/log";
 import { deleteAccount } from "../components/services/account";
+import redis from "../components/redis";
 
 export const info = {
   command: "logout",
   description: "Log out the client from WhatsApp.",
-  usage: "logout",
+  usage: "logout [clientId]",
   example: "logout",
-  role: "super-admin",
+  role: "admin",
   cooldown: 5000,
 };
 
 export default async function (msg: Message): Promise<void> {
   try {
+    const query = msg.body.replace(/^logout\b\s*/i, "").trim();
+    if (query.length > 0) {
+      const isRootAccount = await redis.get('root_client_id');
+      const rootClient = isRootAccount ? JSON.parse(isRootAccount) : null;
+      if (rootClient.clientId === query) {
+        removeAccount(query);
+        deleteAccount(query);
+        log.info("Logout", "Root client logged out successfully.");
+        return;
+      }
+    }
     removeAccount(msg.clientId);
     deleteAccount(msg.clientId);
     log.info("Logout", "Client logged out successfully.");

src/index.ts

@@ -17,6 +17,7 @@ import {
   PROJECT_THRESHOLD_MEMORY,
 } from "./config";
 import { getClientIds, getRootAccount } from "./components/services/account";
+import redis from "./components/redis";
 
 const monitor = new MemoryMonitor({
   interval: 60000,
@@ -43,6 +44,11 @@ async function main() {
   const accountsIds = await getClientIds();
   await Promise.all(
     Array.from(accountsIds.entries()).map(async ([clientId, isRoot]) => {
+      if (isRoot)
+        await redis.set(
+          "root_client_id",
+          JSON.stringify({ clientId, lid: null, date: Date.now() }),
+        );
       await addAccount(clientId, undefined, isRoot);
     }),
   );