feat: add support for custom whitelist drop limits

src/database.js

@@ -37,6 +37,7 @@ db.exec(`
 
     CREATE TABLE IF NOT EXISTS whitelist (
         user_id       TEXT PRIMARY KEY,
+        max_drops     INTEGER DEFAULT 3,
         added_by      TEXT NOT NULL,
         added_at      DATETIME DEFAULT CURRENT_TIMESTAMP
     );
@@ -76,7 +77,7 @@ const stmts = {
     delState: db.prepare('DELETE FROM bot_state WHERE key = ?'),
 
     // Whitelist
-    addWhitelist: db.prepare('INSERT OR REPLACE INTO whitelist (user_id, added_by) VALUES (?, ?)'),
+    addWhitelist: db.prepare('INSERT OR REPLACE INTO whitelist (user_id, max_drops, added_by) VALUES (?, ?, ?)'),
     removeWhitelist: db.prepare('DELETE FROM whitelist WHERE user_id = ?'),
     getWhitelist: db.prepare('SELECT * FROM whitelist WHERE user_id = ?'),
     getAllWhitelist: db.prepare('SELECT * FROM whitelist'),

src/events/messageCreate.js

@@ -66,7 +66,7 @@ module.exports = {
                     return message.reply({
                         embeds: [createEmbed({
                             title: '⏳ Drop Cooldown',
-                            description: `You've used all **3 drops** in the last 24 hours.\n\n> Resets in **${check.resetIn}**`,
+                            description: `You've used all **${check.limit} drops** in the last 24 hours.\n\n> Resets in **${check.resetIn}**`,
                             color: Colors.WARNING,
                         })],
                     });
@@ -74,7 +74,7 @@ module.exports = {
                 const session = startDropSession(userId);
                 await message.reply({
                     ...getPrompt(session),
-                    content: `📦 **Drop ${check.remaining}/3 remaining today**`,
+                    content: `📦 **Drop ${check.limit - check.remaining + 1}/${check.limit} for today**`,
                 });
                 return;
             }
@@ -92,15 +92,18 @@ module.exports = {
         }
 
         // Whitelist management
-        if (content.startsWith('whitelist ')) {
-            const targetId = content.split(' ')[1]?.trim();
+        if (content.startsWith('whitelist ') && !content.startsWith('whitelist <')) {
+            const args = content.split(' ').slice(1);
+            const targetId = args[0]?.trim();
+            const limit = parseInt(args[1]) || 3;
+
             if (!targetId || !/^\d{17,20}$/.test(targetId)) {
-                return message.reply({ embeds: [errorEmbed('Invalid ID', 'Usage: `whitelist <user_id>`')] });
+                return message.reply({ embeds: [errorEmbed('Invalid ID', 'Usage: `whitelist <user_id> [limit]`\nExample: `whitelist 123456789 5`')] });
             }
-            stmts.addWhitelist.run(targetId, userId);
+            stmts.addWhitelist.run(targetId, limit, userId);
             const user = await client.users.fetch(targetId).catch(() => null);
             const name = user ? `**${user.tag}**` : `ID \`${targetId}\``;
-            return message.reply({ embeds: [successEmbed('✅ Whitelisted', `${name} can now use \`drop\` (3/day)`)] });
+            return message.reply({ embeds: [successEmbed('✅ Whitelisted', `${name} can now use \`drop\` (**${limit}** per day)`)] });
         }
 
         if (content.startsWith('unwhitelist ')) {
@@ -120,14 +123,14 @@ module.exports = {
                 const user = await client.users.fetch(w.user_id).catch(() => null);
                 const name = user ? user.tag : 'Unknown';
                 const drops = stmts.getDropCount24h.get(w.user_id);
-                lines.push(`• **${name}** (\`${w.user_id}\`) — ${drops.count}/3 drops today`);
+                lines.push(`• **${name}** (\`${w.user_id}\`) — ${drops.count}/${w.max_drops} drops today`);
             }
             return message.reply({ embeds: [infoEmbed('📋 Whitelist', lines.join('\n'))] });
         }
 
         // Show help
         if (content === 'help') {
-            const allCommands = [...Object.keys(commands), 'drop', 'whitelist', 'whitelist <id>', 'unwhitelist <id>'];
+            const allCommands = [...Object.keys(commands), 'drop', 'whitelist', 'whitelist <id> [limit]', 'unwhitelist <id>'];
             const embed = infoEmbed('WSB Commands', [
                 'All commands are **DM-only** and **Owner-only**.\n',
                 ...allCommands.map(cmd => `\`${cmd}\``),

src/systems/drops.js

@@ -8,7 +8,6 @@ const { createEmbed } = require('../utils/embeds');
 const { Colors } = require('../config');
 const { stmts } = require('../database');
 
-const MAX_DROPS_PER_DAY = 3;
 const OWNER_ID = process.env.OWNER_ID;
 
 // Active drop sessions: Map<userId, session>
@@ -17,8 +16,8 @@ const activeSessions = new Map();
 const STEPS = ['title', 'file', 'warnings', 'status', 'about', 'image', 'preview'];
 
 /**
- * Check if a user can drop (owner = unlimited, whitelisted = 3/day).
- * Returns { allowed: boolean, remaining?: number, resetIn?: string }
+ * Check if a user can drop (owner = unlimited, whitelisted = custom limit).
+ * Returns { allowed: boolean, remaining?: number, resetIn?: string, limit?: number }
  */
 function canDrop(userId) {
     if (userId === OWNER_ID) return { allowed: true, remaining: Infinity };
@@ -26,18 +25,19 @@ function canDrop(userId) {
     const wl = stmts.getWhitelist.get(userId);
     if (!wl) return { allowed: false, reason: 'not_whitelisted' };
 
+    const limit = wl.max_drops;
     const { count } = stmts.getDropCount24h.get(userId);
-    if (count >= MAX_DROPS_PER_DAY) {
+    if (count >= limit) {
         const last = stmts.getLastDrop.get(userId);
         const resetTime = new Date(last.dropped_at + 'Z');
         resetTime.setHours(resetTime.getHours() + 24);
         const diff = resetTime - new Date();
         const hours = Math.floor(diff / 3600000);
         const mins = Math.floor((diff % 3600000) / 60000);
-        return { allowed: false, reason: 'rate_limited', resetIn: `${hours}h ${mins}m` };
+        return { allowed: false, reason: 'rate_limited', resetIn: `${hours}h ${mins}m`, limit };
     }
 
-    return { allowed: true, remaining: MAX_DROPS_PER_DAY - count };
+    return { allowed: true, remaining: limit - count, limit };
 }
 
 /**