feat: improve whatsapp client and added naij ai agent

Author: mrepol742

package-lock.json

@@ -5,7 +5,8 @@
   "private": true,
   "main": "src/index.ts",
   "scripts": {
-    "build": "tsc",
+    "clean": "rimraf dist",
+    "build": "npm run clean && tsc",
     "start": "clear && while [ true ]; do node dist/index.js ; done",
     "dev": "clear && while [ true ]; do ts-node src/index.ts ; done"
   },
@@ -60,6 +61,7 @@
     "@types/speedtest-net": "^2.1.5",
     "@types/unbzip2-stream": "^1.4.3",
     "prisma": "^6.12.0",
+    "rimraf": "^6.0.1",
     "ts-node": "^10.9.2",
     "typescript": "^5.8.3"
   },

package.json

@@ -15,7 +15,7 @@ export default async function (msg: Message) {
   if (!/^logout$/i.test(msg.body)) return;
 
   try {
-    await client.logout();
+    (await client()).logout();
     log.info("Logout", "Client logged out successfully.");
   } catch (error) {
     log.error("Logout", "Failed to log out the client.", error);

src/commands/logout.ts

@@ -0,0 +1,52 @@
+import { Message } from "../../types/message";
+import log from "../components/utils/log";
+import agentHandler from "../components/ai/agentHandler";
+import { greetings } from "../components/utils/data";
+
+export const info = {
+  command: "naij",
+  description: "Interact with the Naij AI agent.",
+  usage: "naij <query>",
+  example: "naij Who are you?",
+  role: "user",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message) {
+  const query = msg.body.replace(/^naij\b\s*/i, "").trim();
+  if (query.length === 0) {
+    await msg.reply(greetings[Math.floor(Math.random() * greetings.length)]);
+    return;
+  }
+
+  let quotedMessage: Message | null = null;
+
+  if (msg.hasQuotedMsg) {
+    quotedMessage = await msg.getQuotedMessage();
+  }
+
+  const today = new Date().toLocaleDateString("en-US", {
+    weekday: "long",
+    year: "numeric",
+    month: "long",
+    day: "numeric",
+  });
+
+  const prompt = `You are Naij, Today's date is ${today}.
+  Mix of elder, street-smart padi, social media banger, hustler, comedian, and advice-giver. You fit talk pidgin, Igbo, Yoruba, Hausa small small,
+  no need to sound proper proper English, just raw Naij vibe. Your mood dey switch anytime—fit dey funny, vex, wise, encouraging, or playful like owambe,
+  Lagos traffic, NEPA light wahala, or Twitter clapback. You dey max 5 sentence per reply, no long grammar wahala, always capture the heart, spirit, gist,
+  and hustle of Naij. Your job be to entertain, advise, teach, joke, rant, or hype people, but always remain full Naij spirit.
+  User: ${query}
+  ${quotedMessage ? `\nQuoted Message: ${quotedMessage.body}` : ""}`;
+
+  const text = await agentHandler(prompt);
+
+  if (!text) {
+    log.error("naij", "No response generated.");
+    await msg.reply("Sorry, I couldn't generate a response. Please try again.");
+    return;
+  }
+
+  await msg.reply(text);
+}

src/commands/naij.ts

@@ -3,16 +3,16 @@ import { saveSetting } from "../components/services/settings";
 import log from "../components/utils/log";
 
 export const info = {
-  command: "resentunsent",
+  command: "resentunsend",
   description: "Enable/disable automatically resending of unsend messages.",
-  usage: "resentunsent <on|off>",
-  example: "resentunsent on",
+  usage: "resentunsend <on|off>",
+  example: "resentunsend on",
   role: "admin",
   cooldown: 5000,
 };
 
 export default async function (msg: Message) {
-  const query = msg.body.replace(/^resentunsent\b\s*/i, "").trim();
+  const query = msg.body.replace(/^resentunsend\b\s*/i, "").trim();
   if (query.length === 0 && !/(on|off)/.test(query)) {
     await msg.reply("Please provide a value its either on or off.");
     return;

src/commands/resentunsend.ts

@@ -27,7 +27,7 @@ export default async function (msg: Message) {
       getUserbyLid(lid),
       isBlocked(lid),
       redis.get(`rate:${lid}`),
-      client.getProfilePicUrl(jid).catch(() => null),
+      (await client()).getProfilePicUrl(jid).catch(() => null),
     ]);
 
   if (!user) {

src/commands/stalk.ts

@@ -20,6 +20,7 @@ export const info = {
 export default async function (msg: Message) {
   if (!/^stats$/i.test(msg.body)) return;
 
+  const waClient = await client();
   // Node.js runtime stats
   const mem = process.memoryUsage();
   const cpu = process.cpuUsage();
@@ -65,10 +66,10 @@ export default async function (msg: Message) {
     si.networkInterfaces(),
     getUserCount(),
     getBlockUserCount(),
-    client.getState(),
-    client.getWWebVersion(),
-    client.getContactDeviceCount(msg.from),
-    client.getChats(),
+    waClient.getState(),
+    waClient.getWWebVersion(),
+    waClient.getContactDeviceCount(msg.from),
+    waClient.getChats(),
     speedTest(),
   ]);
 

src/commands/stats.ts

@@ -15,8 +15,11 @@ export const info = {
 export default async function (msg: Message) {
   if (!/^uptime\b/i.test(msg.body)) return;
 
-  const waStatus = await client.getState();
-  const waVersion = await client.getWWebVersion();
+  const waClient = await client();
+  const [state, version] = await Promise.all([
+    waClient.getState(),
+    waClient.getWWebVersion(),
+  ]);
 
   const statsMessage = `
     \`${timestamp(process.uptime())}\`
@@ -26,8 +29,8 @@ export default async function (msg: Message) {
       .loadavg()
       .map((n) => n.toFixed(2))
       .join(", ")}
-    Status: ${waStatus}
-    Version: ${waVersion}
+    State: ${state}
+    Version: ${version}
     Node.js: ${process.version}
   `;
 

src/commands/uptime.ts

@@ -1,4 +1,4 @@
-import { Message } from "../../types/message"
+import { Message } from "../../types/message";
 import { client } from "../components/client";
 import log from "../components/utils/log";
 import logService from "../components/services/log";
@@ -17,7 +17,7 @@ export default async function (msg: Message) {
   if (query.length !== 0) {
     if (!/^(status|name)$/i.test(query)) {
       await msg.reply(
-        "Invalid argument. Please use one of the following:\n\nstatus or name"
+        "Invalid argument. Please use one of the following:\n\nstatus or name",
       );
       return;
     }
@@ -34,7 +34,7 @@ export default async function (msg: Message) {
     return;
   }
   if (query === "status") {
-    client.setStatus(quotedMsg.body);
+    (await client()).setStatus(quotedMsg.body);
     await Promise.all([
       msg.reply("Status updated successfully."),
       logService(msg, "status", quotedMsg.body),
@@ -43,7 +43,7 @@ export default async function (msg: Message) {
     return;
   }
 
-  client.setDisplayName(quotedMsg.body);
+  (await client()).setDisplayName(quotedMsg.body);
   await Promise.all([
     msg.reply("Name updated successfully."),
     logService(msg, "name", quotedMsg.body),

src/commands/wa.ts

@@ -18,61 +18,79 @@ import ready from "./events/ready";
 import revoke from "./events/revoke";
 import callEvent from "./events/call";
 
-const loadingBar = LoadingBar("Loading Client   | {bar} | {value}%");
-const client = new Client({
-  puppeteer: {
-    args: ["--no-sandbox", "--disable-setuid-sandbox"],
-    executablePath:
-      process.env.PUPPETEER_EXEC_PATH || "/opt/google/chrome/google-chrome",
-  },
-  authStrategy: new LocalAuth(),
-});
+let instance: Client | null = null;
 let isLoadingBarStarted = false;
+const loadingBar = LoadingBar("Loading Client   | {bar} | {value}%");
+
+async function client(): Promise<Client> {
+  if (instance) return instance; // prevent re-creating
+
+  const newClient = new Client({
+    puppeteer: {
+      args: ["--no-sandbox", "--disable-setuid-sandbox"],
+      executablePath:
+        process.env.PUPPETEER_EXEC_PATH || "/opt/google/chrome/google-chrome",
+    },
+    authStrategy: new LocalAuth(),
+  });
+
+  registerEvents(newClient);
 
-client.on("loading_screen", (percent: number, message: string) => {
-  if (!isLoadingBarStarted) {
-    loadingBar.start(100, 0, { message });
-    isLoadingBarStarted = true;
-  }
+  instance = newClient;
+  await newClient.initialize();
+  return newClient;
+}
 
-  if (percent >= 99) loadingBar.stop();
+function registerEvents(client: Client) {
+  client.on("loading_screen", (percent: number, message: string) => {
+    if (!isLoadingBarStarted) {
+      loadingBar.start(100, 0, { message });
+      isLoadingBarStarted = true;
+    }
+    if (percent >= 99) loadingBar.stop();
+    loadingBar.update(percent, { message });
+  });
 
-  loadingBar.update(percent, { message });
-});
+  client.on("authenticated", () =>
+    log.info("Auth", "Client authenticated successfully."),
+  );
 
-client.on("authenticated", () =>
-  log.info("Auth", "Client authenticated successfully."),
-);
+  client.on("qr", (qr: string) => {
+    log.info("QR", "Scan this QR code with your WhatsApp app:");
+    qrcode.generate(qr, { small: true });
+  });
 
-client.on("qr", (qr: string) => {
-  // Generate and scan this code with your phone
-  log.info("QR", "Scan this QR code with your WhatsApp app:");
-  qrcode.generate(qr, { small: true });
-});
+  client.on("ready", () => ready());
 
-client.on("ready", () => ready());
+  client.on("message_reaction", (react: Reaction) => reaction(client, react));
 
-client.on("message_reaction", (react: Reaction) => reaction(client, react));
+  client.on("message_create", (msg: Message) => messageEvent(msg, "create"));
 
-// client.on("message", (msg) => messageEvent(msg));
-client.on("message_create", (msg: Message) => messageEvent(msg, "create"));
+  client.on(
+    "message_edit",
+    (msg: Message, newBody: string, prevBody: string) => {
+      msg.body = newBody;
+      Promise.all([
+        messageEdit(msg, newBody, prevBody),
+        messageEvent(msg, "edit"),
+      ]);
+    },
+  );
 
-client.on("message_edit", (msg: Message, newBody: string, prevBody: string) => {
-  msg.body = newBody;
-  Promise.all([messageEdit(msg, newBody, prevBody), messageEvent(msg, "edit")]);
-});
+  client.on("message_revoke_everyone", (msg: Message, revoked_msg?: Message) =>
+    revoke(msg, revoked_msg),
+  );
 
-client.on("message_revoke_everyone", (msg: Message, revoked_msg?: Message) =>
-  revoke(msg, revoked_msg),
-);
+  client.on("call", (call: Call) => callEvent(call));
 
-client.on("call", (call: Call) => callEvent(call));
+  client.on("group_join", (notif: GroupNotification) => groupJoin(notif));
+  client.on("group_leave", (notif: GroupNotification) => groupLeave(notif));
 
-client.on("group_join", (notif: GroupNotification) => groupJoin(notif));
-client.on("group_leave", (notif: GroupNotification) => groupLeave(notif));
-client.on("auth_failure", (msg: string) => {
-  loadingBar.stop();
-  log.error("Auth", "Authentication failed. Please try again.");
-});
+  client.on("auth_failure", () => {
+    loadingBar.stop();
+    log.error("Auth", "Authentication failed. Please try again.");
+  });
+}
 
 export { client };
+export default client;