Refractor

mrepol742 · mrepol742 · commit e10730ea4c1e · 2025-07-18T07:46:13.000+08:00
diff --git a/src/commands/alert.ts b/src/commands/alert.ts
@@ -2,7 +2,6 @@ import { Message, MessageMedia } from "whatsapp-web.js";
 import axios from "axios";
 import log from "npmlog";
 import fs from "fs/promises";
-import { client } from "../index";
 
 export const command = "alert";
 export const role = "user";
diff --git a/src/commands/caution.ts b/src/commands/caution.ts
@@ -2,7 +2,7 @@ import { Message, MessageMedia } from "whatsapp-web.js";
 import axios from "axios";
 import log from "npmlog";
 import fs from "fs/promises";
-import { client } from "../index";
+import { client } from "../components/client";
 
 export const command = "caution";
 export const role = "user";
diff --git a/src/commands/github.ts b/src/commands/github.ts
@@ -2,7 +2,7 @@ import { Message, MessageMedia } from "whatsapp-web.js";
 import axios from "axios";
 import log from "npmlog";
 import fs from "fs/promises";
-import { client } from "../index";
+import { client } from "../components/client";
 
 export const command = "github";
 export const role = "user";
diff --git a/src/commands/god.ts b/src/commands/god.ts
@@ -2,7 +2,7 @@ import { Message, MessageMedia } from "whatsapp-web.js";
 import axios from "axios";
 import log from "npmlog";
 import fs from "fs/promises";
-import { client } from "../index";
+import { client } from "../components/client";
 
 export const command = "god";
 export const role = "user";
diff --git a/src/commands/hwaifu.ts b/src/commands/hwaifu.ts
@@ -2,7 +2,7 @@ import { Message, MessageMedia } from "whatsapp-web.js";
 import axios from "axios";
 import log from "npmlog";
 import fs from "fs/promises";
-import { client } from "../index";
+import { client } from "../components/client";
 
 export const command = "hwaifu";
 export const role = "admin";
diff --git a/src/commands/npm.ts b/src/commands/npm.ts
@@ -2,7 +2,7 @@ import { Message, MessageMedia } from "whatsapp-web.js";
 import axios from "axios";
 import log from "npmlog";
 import fs from "fs/promises";
-import { client } from "../index";
+import { client } from "../components/client";
 
 export const command = "npm";
 export const role = "user";
diff --git a/src/commands/pickupline.ts b/src/commands/pickupline.ts
@@ -2,7 +2,7 @@ import { Message, MessageMedia } from "whatsapp-web.js";
 import axios from "axios";
 import log from "npmlog";
 import fs from "fs/promises";
-import { client } from "../index";
+import { client } from "../components/client";
 
 export const command = "pickupline";
 export const role = "user";
diff --git a/src/commands/poli.ts b/src/commands/poli.ts
@@ -2,7 +2,7 @@ import { Message, MessageMedia } from "whatsapp-web.js";
 import axios from "axios";
 import log from "npmlog";
 import fs from "fs/promises";
-import { client } from "../index";
+import { client } from "../components/client";
 
 export const command = "poli";
 export const role = "user";
diff --git a/src/commands/pooh.ts b/src/commands/pooh.ts
@@ -2,7 +2,7 @@ import { Message, MessageMedia } from "whatsapp-web.js";
 import axios from "axios";
 import log from "npmlog";
 import fs from "fs/promises";
-import { client } from "../index";
+import { client } from "../components/client";
 
 export const command = "pooh";
 export const role = "user";
diff --git a/src/commands/qrcode.ts b/src/commands/qrcode.ts
@@ -2,7 +2,7 @@ import { Message, MessageMedia } from "whatsapp-web.js";
 import axios from "axios";
 import log from "npmlog";
 import fs from "fs/promises";
-import { client } from "../index";
+import { client } from "../components/client";
 
 export const command = "qrcode";
 export const role = "user";
diff --git a/src/commands/randomcolor.ts b/src/commands/randomcolor.ts
@@ -2,7 +2,7 @@ import { Message, MessageMedia } from "whatsapp-web.js";
 import axios from "axios";
 import log from "npmlog";
 import fs from "fs/promises";
-import { client } from "../index";
+import { client } from "../components/client";
 
 export const command = "randomcolor";
 export const role = "user";
diff --git a/src/commands/sadcat.ts b/src/commands/sadcat.ts
@@ -2,7 +2,7 @@ import { Message, MessageMedia } from "whatsapp-web.js";
 import axios from "axios";
 import log from "npmlog";
 import fs from "fs/promises";
-import { client } from "../index";
+import { client } from "../components/client";
 
 export const command = "sadcat";
 export const role = "user";
diff --git a/src/commands/say.ts b/src/commands/say.ts
@@ -3,7 +3,7 @@ import log from "npmlog";
 import * as GoogleTTS from "google-tts-api";
 import fs from "fs";
 import axios from "axios";
-import { client } from "../index";
+import { client } from "../components/client";
 
 export const command = "say";
 export const role = "user";
diff --git a/src/commands/stats.ts b/src/commands/stats.ts
@@ -1,9 +1,10 @@
 import { Message } from "whatsapp-web.js";
 import log from "npmlog";
 import os from "os";
+import { name, version } from "../../package.json";
 
 export const command = "stats";
-export const role = "admin";
+export const role = "user";
 
 export default async function (msg: Message) {
   const stats = {
@@ -24,6 +25,7 @@ export default async function (msg: Message) {
       - Free Memory: ${(stats.freeMemory / 1024 ** 3).toFixed(2)} GB
       - Uptime: ${(stats.uptime / 3600).toFixed(2)} hours
       - Node.js Version: ${process.version}
+      - ${name}: ${version}
     `;
 
   await msg.reply(statsMessage);
diff --git a/src/commands/waifu.ts b/src/commands/waifu.ts
@@ -2,7 +2,7 @@ import { Message, MessageMedia } from "whatsapp-web.js";
 import axios from "axios";
 import log from "npmlog";
 import fs from "fs/promises";
-import { client } from "../index";
+import { client } from "../components/client";
 
 export const command = "waifu";
 export const role = "user";
diff --git a/src/components/client.ts b/src/components/client.ts
@@ -0,0 +1,94 @@
+import log from "npmlog";
+import { Client, LocalAuth, Message } from "whatsapp-web.js";
+import qrcode from "qrcode-terminal";
+import rateLimiter from "./rateLimiter";
+import { commands } from "../index";
+
+const commandPrefix = process.env.COMMAND_PREFIX || "!";
+const commandPrefixLess = process.env.COMMAND_PREFIX_LESS === "true";
+const debug = process.env.DEBUG === "true";
+const superAdmin = process.env.SUPER_ADMIN || "";
+
+const client = new Client({
+  authStrategy: new LocalAuth(),
+});
+
+client.on("qr", (qr) => {
+  // Generate and scan this code with your phone
+  log.info("QR Code", "Scan this QR code with your WhatsApp app:");
+  qrcode.generate(qr, { small: true });
+});
+
+client.on("ready", () => {
+  log.info("Client", "WhatsApp client is ready!");
+});
+
+// client.on("message", (msg) => messageEvent(msg));
+client.on("message_create", (msg) => messageEvent(msg));
+
+client.on("auth_failure", (msg) => {
+  log.error("Auth", "Authentication failed. Please try again.");
+});
+
+client.initialize();
+
+const messageEvent = (msg: Message) => {
+  // ignore message if it is older than 10 seconds
+  if (msg.timestamp < Date.now() / 1000 - 10)
+    return;
+
+  const prefix = !msg.body.startsWith(commandPrefix);
+  const senderId = msg.from.split("@")[0];
+
+  /*
+   * Prefix
+   */
+  if (!commandPrefixLess && prefix) return;
+  if (msg.fromMe && prefix) return;
+
+  /*
+   * Check if the message starts with the command prefix.
+   */
+  const keyWithPrefix = msg.body.split(" ")[0];
+  const key = keyWithPrefix.startsWith(commandPrefix)
+    ? keyWithPrefix.slice(commandPrefix.length)
+    : keyWithPrefix;
+  const handler = commands[key.toLocaleLowerCase()];
+  if (!handler) return;
+
+  /*
+   * Rate limit commands to prevent abuse.
+   */
+  if (senderId !== superAdmin) {
+    const rate = rateLimiter(msg.from);
+    if (rate === null) return;
+    if (!rate) {
+      msg.reply("You are sending commands too fast. Please wait a minute.");
+      return;
+    }
+  }
+
+  /*
+   * Role base restrictions.
+   */
+  if (handler.role === "admin" && !msg.fromMe && senderId !== superAdmin) {
+    return;
+  }
+
+  if (debug) log.info("Message", msg.body.slice(0, 255));
+  msg.body = commandPrefixLess
+    ? msg.body
+    : msg.body.slice(commandPrefix.length).trim();
+
+  /*
+   * Execute the command handler.
+   */
+  try {
+    handler.exec(msg);
+  } catch (error) {
+    log.error("Command", "Error occured while processing the request:", error);
+    msg.reply("An error occurred while processing your request.");
+  }
+};
+
+export { client };
diff --git a/src/components/loader.ts b/src/components/loader.ts
@@ -0,0 +1,29 @@
+import log from "npmlog";
+import fs from "fs";
+import path from "path";
+import { commands } from "../index";
+
+export const command = "load";
+export const role = "admin";
+
+const commandsPath = path.join(__dirname, "..", "commands");
+
+export default function Loader(file: string) {
+  if (/\.js$|\.ts$/.test(file)) {
+    const filePath = path.join(commandsPath, file);
+    delete require.cache[require.resolve(filePath)];
+    const commandModule = require(filePath);
+
+    if (
+      typeof commandModule.command === "string" &&
+      typeof commandModule.default === "function"
+    ) {
+      commands[commandModule.command] = {
+        command: commandModule.command,
+        role: commandModule.role || "user",
+        exec: commandModule.default,
+      };
+      log.info("Loader", `Loaded command: ${commandModule.command}`);
+    }
+  }
+}
diff --git a/src/components/process.ts b/src/components/process.ts
@@ -0,0 +1,35 @@
+import log from "npmlog";
+
+process.on("SIGHUP", function () {
+  process.exit(0);
+});
+
+process.on("SIGTERM", function () {
+  process.exit(0);
+});
+
+process.on("SIGINT", function () {
+  process.kill(process.pid);
+  process.exit(0);
+});
+
+process.on("uncaughtException", (err, origin) => {
+  log.error(
+    "Uncaught Exception",
+    `Exception: ${err.message}\nOrigin: ${origin}`
+  );
+});
+
+process.on("unhandledRejection", (reason, promise) => {
+  log.error("Unhandled Rejection", `Reason: ${reason}\nPromise: ${promise}`);
+});
+
+process.on("beforeExit", (code) => {
+  log.info("Before Exit", `Process is about to exit with code: ${code}`);
+});
+
+process.on("exit", (code) => {
+  console.log("");
+});
+
+log.info("Process", "Event listeners for process signals have been set up.");
diff --git a/src/components/server.ts b/src/components/server.ts
@@ -0,0 +1,13 @@
+import http from "http";
+import log from "npmlog";
+
+const port = process.env.PORT || 3000;
+
+http
+  .createServer((req, res) => {
+    res.writeHead(200, { "Content-Type": "text/plain" });
+    res.end("Bot is running\n");
+  })
+  .listen(port, () => {
+    log.info("Server", `HTTP server started on port ${port}`);
+  });
diff --git a/src/index.ts b/src/index.ts
diff --git a/src/server.ts b/src/server.ts
diff --git a/tsconfig.json b/tsconfig.json
