feat: add Message and Settings models

Author: mrepol742

prisma/migrations/20250722011411_add_message/migration.sql

@@ -0,0 +1,20 @@
+-- CreateTable
+CREATE TABLE `Message` (
+    `id` INTEGER NOT NULL AUTO_INCREMENT,
+    `mid` VARCHAR(191) NOT NULL,
+    `gid` INTEGER NULL,
+    `lid` INTEGER NULL,
+    `content` VARCHAR(191) NOT NULL,
+    `type` VARCHAR(191) NOT NULL DEFAULT 'send',
+    `createdAt` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
+    `updatedAt` DATETIME(3) NOT NULL,
+
+    UNIQUE INDEX `Message_mid_key`(`mid`),
+    PRIMARY KEY (`id`)
+) DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;
+
+-- AddForeignKey
+ALTER TABLE `Message` ADD CONSTRAINT `Message_gid_fkey` FOREIGN KEY (`gid`) REFERENCES `Group`(`id`) ON DELETE SET NULL ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE `Message` ADD CONSTRAINT `Message_lid_fkey` FOREIGN KEY (`lid`) REFERENCES `User`(`id`) ON DELETE SET NULL ON UPDATE CASCADE;

prisma/migrations/20250722012831_remove_mid_and_gid/migration.sql

@@ -0,0 +1,27 @@
+/*
+  Warnings:
+
+  - You are about to drop the column `gid` on the `Message` table. All the data in the column will be lost.
+  - You are about to drop the column `mid` on the `Message` table. All the data in the column will be lost.
+  - Made the column `lid` on table `Message` required. This step will fail if there are existing NULL values in that column.
+
+*/
+-- DropForeignKey
+ALTER TABLE `Message` DROP FOREIGN KEY `Message_gid_fkey`;
+
+-- DropForeignKey
+ALTER TABLE `Message` DROP FOREIGN KEY `Message_lid_fkey`;
+
+-- DropIndex
+DROP INDEX `Message_gid_fkey` ON `Message`;
+
+-- DropIndex
+DROP INDEX `Message_lid_fkey` ON `Message`;
+
+-- DropIndex
+DROP INDEX `Message_mid_key` ON `Message`;
+
+-- AlterTable
+ALTER TABLE `Message` DROP COLUMN `gid`,
+    DROP COLUMN `mid`,
+    MODIFY `lid` VARCHAR(191) NOT NULL;

prisma/migrations/20250722013542_add_settings/migration.sql

@@ -0,0 +1,11 @@
+-- CreateTable
+CREATE TABLE `Settings` (
+    `id` INTEGER NOT NULL AUTO_INCREMENT,
+    `name` VARCHAR(191) NOT NULL,
+    `value` VARCHAR(191) NOT NULL,
+    `createdAt` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
+    `updatedAt` DATETIME(3) NOT NULL,
+
+    UNIQUE INDEX `Settings_name_key`(`name`),
+    PRIMARY KEY (`id`)
+) DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;

prisma/schema.prisma

@@ -45,3 +45,20 @@ model Group {
   createdAt   DateTime @default(now())
   updatedAt   DateTime @updatedAt
 }
+
+model Message {
+  id        Int      @id @default(autoincrement())
+  lid       String
+  content   String
+  type      String   @default("send") // "send", "edit", "delete"
+  createdAt DateTime @default(now())
+  updatedAt DateTime @updatedAt
+}
+
+model Settings {
+  id        Int      @id @default(autoincrement())
+  name      String   @unique
+  value     String
+  createdAt DateTime @default(now())
+  updatedAt DateTime @updatedAt
+}

src/components/client.ts

@@ -41,7 +41,7 @@ client.on("message_reaction", async (react: Reaction) =>
 client.on("message_create", async (msg: Message) => messageEvent(msg));
 client.on(
   "message_edit",
-  async (msg: Message, newBody: String, prevBody: String) =>
+  async (msg: Message, newBody: string, prevBody: string) =>
     messageEdit(msg, newBody, prevBody)
 );
 client.on(
@@ -50,7 +50,7 @@ client.on(
 );
 client.on("group_join", async (notif: GroupNotification) => groupJoin(notif));
 client.on("group_leave", async (notif: GroupNotification) => groupLeave(notif));
-client.on("auth_failure", (msg: String) =>
+client.on("auth_failure", (msg: string) =>
   log.error("Auth", "Authentication failed. Please try again.")
 );
 

src/components/events/edit.ts

@@ -1,14 +1,16 @@
 import { Message } from "whatsapp-web.js";
 import { getUserbyLid } from "../services/user";
+import { addMessage } from "../services/message";
 
 export default async function (
   msg: Message,
-  _newBody: String,
-  prevBody: String
+  _newBody: string,
+  prevBody: string
 ) {
   if (msg.fromMe) return;
   const isGroup = !!msg.author;
   const user = await getUserbyLid(msg.from) || "Your";
+  await addMessage(msg, prevBody, "edit");
   await msg.reply(
     `${isGroup ? user : "Your"} message was edited from "${prevBody}".`
   );

src/components/events/revoke.ts

@@ -1,10 +1,12 @@
 import { Message } from "whatsapp-web.js";
 import { getUserbyLid } from "../services/user";
+import { addMessage } from "../services/message";
 
 export default async function (msg: Message, revoked_msg?: Message) {
   if (msg.fromMe || !revoked_msg) return;
   const isGroup = !!msg.author;
-  const user = await getUserbyLid(msg.from) || "Your";
+  const user = (await getUserbyLid(msg.from)) || "Your";
+  await addMessage(msg, revoked_msg?.body, "revoke");
   await msg.reply(
     `${isGroup ? user : "Your"} message "${revoked_msg?.body}" was deleted.`
   );

src/components/services/message.ts

@@ -0,0 +1,23 @@
+import { prisma } from "../prisma";
+import log from "../../components/utils/log";
+import { Message } from "whatsapp-web.js";
+
+export async function addMessage(
+  msg: Message,
+  body: string,
+  type: string
+): Promise<void> {
+  try {
+
+    const lid = msg.id.remote.split("@")[0];
+    await prisma.message.create({
+      data: {
+        lid,
+        content: body,
+        type: type,
+      },
+    });
+  } catch (error) {
+    log.error("Database", "Failed to add message.", error);
+  }
+}

src/components/services/settings.ts

@@ -0,0 +1,45 @@
+import { prisma } from "../prisma";
+import log from "../../components/utils/log";
+import { Message } from "whatsapp-web.js";
+
+export async function saveSetting(name: string, value: string): Promise<void> {
+  try {
+    await prisma.settings.upsert({
+      where: { name },
+      update: {
+        value,
+      },
+      create: {
+        name,
+        value,
+      },
+    });
+  } catch (error) {
+    log.error("Database", "Failed to add message.", error);
+  }
+}
+
+export async function getSetting(name: string): Promise<string | null> {
+  try {
+    const setting = await prisma.settings.findUnique({
+      where: { name },
+    });
+    return setting ? setting.value : null;
+  } catch (error) {
+    log.error("Database", `Failed to get setting: ${name}`, error);
+    return null;
+  }
+}
+
+export async function getAllSettings(): Promise<Record<string, string>> {
+  try {
+    const settings = await prisma.settings.findMany();
+    return settings.reduce((acc, setting) => {
+      acc[setting.name] = setting.value;
+      return acc;
+    }, {} as Record<string, string>);
+  } catch (error) {
+    log.error("Database", "Failed to get all settings.", error);
+    return {};
+  }
+}

src/components/services/user.ts

@@ -1,7 +1,6 @@
 import { prisma } from "../prisma";
 import log from "../../components/utils/log";
 import { Message } from "whatsapp-web.js";
-import { info } from "console";
 
 export async function findOrCreateUser(msg: Message): Promise<boolean> {
   try {