koishi-plugin-bot-ddos-protect/lib/index.mjs

Provider your koishi bot with ddos protect

var __defProp = Object.defineProperty;
var __name = (target, value) => __defProp(target, "name", { value, configurable: true });

// src/index.ts
import { Schema } from "koishi";
function apply(ctx, config) {
  const states = {};
  function getState(key) {
    if (!states[key]) {
      states[key] = { count: 0 };
    }
    return states[key];
  }
  __name(getState, "getState");
  function addState(key) {
    const state = getState(key);
    state.count++;
    state.interval?.();
    state.interval = ctx.setTimeout(() => {
      delete state[key];
    }, 6e4);
    return state.count >= config.limit;
  }
  __name(addState, "addState");
  ctx.before("command/execute", ({ command, session }) => {
    if (command.name !== "echo") {
      return;
    }
    if (!config.echoProtect) {
      return;
    }
    const key = session.guildId || session.userId;
    if (addState(key)) {
      ctx.logger.debug("你的 bot 可能在遭受 echo 攻击");
      return "";
    }
  });
  ctx.before("send", (session) => {
    if (!config.sendMessageProtect) {
      return;
    }
    const key = session.guildId || session.userId;
    if (addState(key)) {
      return false;
    }
  });
  ctx.on("message", async (session) => {
    if (!config.receiveMessageProtect) {
      return;
    }
    const key = session.guildId || session.userId;
    if (addState(key)) {
      const bot = session.bot;
      const adapter = bot.adapter;
      await bot.stop();
      if (adapter.stop) {
        await adapter.stop();
      }
      bot.offline(new Error("你的 bot 正在被攻击，暂时下线"));
      ctx.logger.error("你的 bot 可能遭受攻击，已暂时下线");
      ctx.setTimeout(
        async () => {
          ctx.logger.info("正在重新上线 bot");
          ctx.loader.fullReload(52);
        },
        60 * config.restartTime * 1e3
      );
      return false;
    }
  });
}
__name(apply, "apply");
var Config = Schema.object({
  limit: Schema.number().description("限制的最小消息触发次数（一分钟内）").default(50),
  restartTime: Schema.number().description(
    "当适配器被断开后多少时间内重启 Koishi 以上线 bot（按分钟计算）"
  ).min(1).default(3),
  echoProtect: Schema.boolean().description("是否启用 echo 保护。开启后将不允许大量的 echo 消息调用。").default(true),
  receiveMessageProtect: Schema.boolean().description(
    "是否启用接收消息保护。开启后将不允许大量的接收消息（不推荐开启）。"
  ).default(false),
  sendMessageProtect: Schema.boolean().description("是否启用发送消息保护。开启后将不允许大量的发送消息。").default(true)
});
var name = "bot-ddos-protect";
export {
  Config,
  apply,
  name
};
//# sourceMappingURL=index.mjs.map