feat: implement user blocking and unblocking commands, enhance user management, and add new database models

Author: mrepol742

.env.example

@@ -22,9 +22,6 @@ AUTO_RELOAD=false
 # this should be a valid phone number in the format
 SUPER_ADMIN=09123456789
 
-# the database provider (e.g., postgresql, mysql, sqlite)
-DATABASE_PROVIDER=mysql
-
 # the database connection URL
 DATABASE_URL=mysql://root@localhost:3306/project_canis
 

package-lock.json

@@ -33,7 +33,9 @@
     "npmlog": "^7.0.1",
     "qrcode-terminal": "^0.12.0",
     "systeminformation": "^5.27.7",
-    "whatsapp-web.js": "^1.31.0"
+    "whatsapp-web.js": "^1.31.0",
+    "youtubei.js": "^15.0.0",
+    "ytdl-core": "^4.11.5"
   },
   "devDependencies": {
     "@types/dotenv": "^6.1.1",

package.json

@@ -0,0 +1,42 @@
+-- CreateTable
+CREATE TABLE `User` (
+    `id` INTEGER NOT NULL AUTO_INCREMENT,
+    `lid` VARCHAR(191) NOT NULL,
+    `name` VARCHAR(191) NOT NULL,
+    `number` VARCHAR(191) NOT NULL,
+    `countryCode` VARCHAR(191) NOT NULL,
+    `type` VARCHAR(191) NOT NULL,
+    `mode` VARCHAR(191) NOT NULL,
+    `about` VARCHAR(191) NULL,
+    `createdAt` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
+    `updatedAt` DATETIME(3) NOT NULL,
+
+    UNIQUE INDEX `User_lid_key`(`lid`),
+    PRIMARY KEY (`id`)
+) DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;
+
+-- CreateTable
+CREATE TABLE `Block` (
+    `id` INTEGER NOT NULL AUTO_INCREMENT,
+    `lid` VARCHAR(191) NOT NULL,
+    `mode` VARCHAR(191) NOT NULL,
+    `reason` VARCHAR(191) NULL,
+    `createdAt` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
+    `updatedAt` DATETIME(3) NOT NULL,
+
+    UNIQUE INDEX `Block_lid_key`(`lid`),
+    PRIMARY KEY (`id`)
+) DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;
+
+-- CreateTable
+CREATE TABLE `Group` (
+    `id` INTEGER NOT NULL AUTO_INCREMENT,
+    `gid` VARCHAR(191) NOT NULL,
+    `name` VARCHAR(191) NOT NULL,
+    `description` VARCHAR(191) NULL,
+    `createdAt` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
+    `updatedAt` DATETIME(3) NOT NULL,
+
+    UNIQUE INDEX `Group_gid_key`(`gid`),
+    PRIMARY KEY (`id`)
+) DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;

prisma/migrations/20250719173602_init/migration.sql

@@ -0,0 +1,3 @@
+# Please do not edit this file manually
+# It should be added in your version-control system (e.g., Git)
+provider = "mysql"

prisma/migrations/migration_lock.toml

@@ -10,12 +10,37 @@ generator client {
 }
 
 datasource db {
-  provider = env("DATABASE_PROVIDER")
+  provider = "mysql"
   url      = env("DATABASE_URL")
 }
 
 model User {
-  id    Int     @id @default(autoincrement())
-  name  String
-  number String  @unique
-}
+  id          Int      @id @default(autoincrement())
+  lid         String   @unique
+  name        String
+  number      String
+  countryCode String
+  type        String
+  mode        String
+  about       String?
+  createdAt   DateTime @default(now())
+  updatedAt   DateTime @updatedAt
+}
+
+model Block {
+  id        Int      @id @default(autoincrement())
+  lid       String   @unique
+  mode      String
+  reason    String?
+  createdAt DateTime @default(now())
+  updatedAt DateTime @updatedAt
+}
+
+model Group {
+  id          Int      @id @default(autoincrement())
+  gid         String   @unique
+  name        String
+  description String?
+  createdAt   DateTime @default(now())
+  updatedAt   DateTime @updatedAt
+}

prisma/schema.prisma

@@ -0,0 +1,27 @@
+import { Message } from "whatsapp-web.js";
+import log from "../components/utils/log";
+import { exec } from "child_process";
+import util from "util";
+import { prisma } from "../components/prisma";
+
+export const command = "block";
+export const role = "admin";
+
+export default async function block(msg: Message) {
+  if (msg.mentionedIds.length === 0) {
+    await msg.reply("Please mention a user to block.");
+    return;
+  }
+
+  for (const userId of msg.mentionedIds) {
+    const lid = userId.split("@")[0];
+
+    await prisma.block.upsert({
+      where: { lid },
+      update: {},
+      create: { lid, mode: msg.author ? "group" : "private" },
+    });
+  }
+
+  await msg.react("✅");
+}

src/commands/block.ts

@@ -0,0 +1,68 @@
+import { Message, MessageMedia } from "whatsapp-web.js";
+import fs from "fs";
+import path from "path";
+import innertube from "../components/innertube";
+import ytdl from "ytdl-core";
+import { client } from "../components/client";
+
+export const command = "play";
+export const role = "admin";
+
+// Type guard
+function isVideoNode(node: any): node is {
+  type: string;
+  video_id: string;
+  title: { toString(): string };
+  length_text?: { toString(): string };
+} {
+  return node?.type === "Video" && typeof node.video_id === "string";
+}
+
+export default async function play(msg: Message) {
+  const query = msg.body.replace(/^play\b\s*/i, "").trim();
+  if (!query) {
+    await msg.reply("Please provide a search query.");
+    return;
+  }
+
+  const yt = await innertube();
+  const search = await yt.search(query, { type: "video" });
+
+  const video = search.results.find((node) => isVideoNode(node));
+  if (!video) {
+    await msg.reply("Unable to find resources for the given query.");
+    return;
+  }
+
+  const videoId = video.video_id;
+  const videoUrl = `https://www.youtube.com/watch?v=${videoId}`;
+  const title = video.title.toString();
+  const duration = video.length_text?.toString() || "unknown";
+
+  await msg.reply(`🎵 Downloading audio for *${title}* (${duration})`);
+
+  const filePath = path.resolve(__dirname, `../../.temp/audio-${videoId}.webm`);
+  const audioStream = ytdl(videoUrl, {
+    filter: "audioonly",
+    quality: "highestaudio",
+  });
+  const writeStream = fs.createWriteStream(filePath);
+  audioStream.pipe(writeStream);
+
+  await new Promise<void>((resolve, reject) => {
+    writeStream.on("finish", resolve);
+    writeStream.on("error", reject);
+  });
+
+  const audioBuffer = fs.readFileSync(filePath);
+  const media = new MessageMedia(
+    "audio/webm",
+    audioBuffer.toString("base64"),
+    `audio-${videoId}.webm`
+  );
+  await client.sendMessage(msg.from, media, {
+    caption: `🎶 *${title}*`,
+  });
+
+  fs.unlinkSync(filePath);
+}

src/commands/play.ts

@@ -0,0 +1,25 @@
+import { Message } from "whatsapp-web.js";
+import log from "../components/utils/log";
+import { exec } from "child_process";
+import util from "util";
+import { prisma } from "../components/prisma";
+
+export const command = "unblock";
+export const role = "admin";
+
+export default async function unblock(msg: Message) {
+  if (msg.mentionedIds.length === 0) {
+    await msg.reply("Please mention a user to ublock.");
+    return;
+  }
+
+  for (const userId of msg.mentionedIds) {
+    const lid = userId.split("@")[0];
+
+    await prisma.block.delete({
+      where: { lid },
+    });
+  }
+
+  await msg.react("✅");
+}