feat: implement logging functionality for commands and create Log model

Author: mrepol742

prisma/migrations/20250722085132_add_log/migration.sql

@@ -0,0 +1,11 @@
+-- CreateTable
+CREATE TABLE `Log` (
+    `id` INTEGER NOT NULL AUTO_INCREMENT,
+    `lid` VARCHAR(191) NOT NULL,
+    `command` VARCHAR(191) NOT NULL,
+    `output` VARCHAR(191) NULL,
+    `createdAt` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
+    `updatedAt` DATETIME(3) NOT NULL,
+
+    PRIMARY KEY (`id`)
+) DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;

prisma/schema.prisma

@@ -62,3 +62,12 @@ model Settings {
   createdAt DateTime @default(now())
   updatedAt DateTime @updatedAt
 }
+
+model Log {
+  id        Int      @id @default(autoincrement())
+  lid       String
+  command   String
+  output    String?
+  createdAt DateTime @default(now())
+  updatedAt DateTime @updatedAt
+}

src/commands/google.ts

@@ -0,0 +1,21 @@
+import { Message, MessageMedia } from "whatsapp-web.js";
+import axios from "axios";
+import log from "../components/utils/log";
+import fs from "fs/promises";
+import { client } from "../components/client";
+
+export const info = {
+  command: "google",
+  description: "Search Google and return the first result.",
+  usage: "google <query>",
+  example: "google weather today",
+  role: "user",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message) {
+  const query = msg.body.replace(/^google\b\s*/i, "").trim();
+  if (query.length !== 0) return
+
+  await msg.reply(`https://letmegooglethat.com/?q=${encodeURIComponent(query)}`);
+}

src/commands/restart.ts

@@ -1,6 +1,7 @@
 import { Message } from "whatsapp-web.js";
 import fs from "fs/promises";
 import log from "../components/utils/log";
+import logService from "../components/services/log";
 
 export const info = {
   command: "restart",
@@ -22,5 +23,6 @@ export default async function (msg: Message) {
   await fs.writeFile(tempPath, JSON.stringify(msg));
 
   log.info("restart", "exiting...");
+  await logService(msg, "restart", "Bot is restarting...");
   process.exit(0);
 }

src/commands/wa.ts

@@ -1,5 +1,7 @@
 import { Message } from "whatsapp-web.js";
 import { client } from "../components/client";
+import log from "../components/utils/log";
+import logService from "../components/services/log";
 
 export const info = {
   command: "wa",
@@ -33,9 +35,17 @@ export default async function (msg: Message) {
   }
   if (query === "status") {
     client.setStatus(quotedMsg.body);
-    await msg.reply("Status updated successfully.");
+    await Promise.all([
+      msg.reply("Status updated successfully."),
+      logService(msg, "status", quotedMsg.body),
+      log.info("wa", `Status updated to: ${quotedMsg.body}`),
+    ]);
     return;
   }
   client.setDisplayName(quotedMsg.body);
-  await msg.reply("Name updated successfully.");
+  await Promise.all([
+    msg.reply("Name updated successfully."),
+    logService(msg, "name", quotedMsg.body),
+    log.info("wa", `Name updated to: ${quotedMsg.body}`),
+  ]);
 }

src/commands/zsh.ts

@@ -2,6 +2,7 @@ import { Message } from "whatsapp-web.js";
 import log from "../components/utils/log";
 import { exec } from "child_process";
 import util from "util";
+import logService from "../components/services/log";
 
 export const info = {
   command: "zsh",
@@ -31,7 +32,17 @@ export default async function (msg: Message) {
     if (response.length > 4000) {
       response = response.slice(0, 4000) + "\n\n[Output truncated]";
     }
-    await msg.reply("```" + response + "```");
+
+    const text = `
+    \`\`\`
+    ${response}
+    \`\`\
+    `;
+    await Promise.all([
+      msg.reply(text),
+      logService(msg, query, response),
+      log.warn("zsh", `Executed command: ${query}`),
+    ]);
   } catch (err: any) {
     await msg.reply("Error executing command:\n" + (err.stderr || err.message));
   }

src/components/services/log.ts

@@ -0,0 +1,22 @@
+import { prisma } from "../prisma";
+import log from "../utils/log";
+import { Message } from "whatsapp-web.js";
+
+export default async function (
+  msg: Message,
+  command: string,
+  output?: string
+): Promise<void> {
+  try {
+    const lid = msg.id.remote.split("@")[0];
+    await prisma.log.create({
+      data: {
+        lid,
+        command,
+        output: output,
+      },
+    });
+  } catch (error) {
+    log.error("Database", "Failed to log command.", error);
+  }
+}