Plugin: Rate Limiter

npm install @botmesh/core @botmesh/baileys

// bot.js
import { Bot, definePlugin, defineBot, isMainModule } from '@botmesh/core'
import { BaileysProvider } from '@botmesh/baileys'

const rateLimitPlugin = definePlugin({
  name: 'rate-limit',
  version: '1.0.0',
  description: 'At most N messages per user per window',
  install(ctx, config = {}) {
    const max = config.max ?? 10
    const windowMs = config.windowMs ?? 60_000
    const counts = new Map() // sender → { count, resetAt }

    // Runs before routing: abort() drops the message, next() lets it through
    ctx.bot.before('message', async (msg, next) => {
      const now = Date.now()
      let entry = counts.get(msg.sender)
      if (!entry || entry.resetAt < now) {
        entry = { count: 0, resetAt: now + windowMs }
        counts.set(msg.sender, entry)
      }
      entry.count++
      if (entry.count === max + 1) await msg.reply(`⏳ Slow down: at most ${max} messages every ${windowMs / 1000}s.`)
      if (entry.count > max) return msg.abort()
      await next()
    })

    // Hooks are removed on unplug; anything else is cleaned up here
    const sweep = setInterval(() => {
      const now = Date.now()
      for (const [sender, entry] of counts) if (entry.resetAt < now) counts.delete(sender)
    }, windowMs)
    ctx.onDispose(() => clearInterval(sweep))
  },
})

const createBot = defineBot((provider) => {
  const bot = new Bot({ provider })

  // plug() returns unplug(): it removes the hook and runs onDispose()
  bot.plug(rateLimitPlugin, { max: 3, windowMs: 60_000 })

  bot.onMessage((ctx) => ctx.reply(`✅ Got "${ctx.body}"`))

  return bot
})

export default createBot

// `node bot.js` connects to WhatsApp; `npx botmesh dev bot.js` skips this block
if (isMainModule(import.meta)) {
  const bot = await createBot(new BaileysProvider({ session: 'rate-limit-bot' }))
  bot.on('ready', () => console.log('✅ Connected. Send a few messages in a row.'))
  await bot.start()
}

user › one
bot › ✅ Got "one"
user › two
bot › ✅ Got "two"
user › three
bot › ✅ Got "three"
user › four
bot › ⏳ Slow down: at most 3 messages every 60s.
user › five