koishi-plugin-bilirice 0.0.4 → 0.0.6

package/lib/index.d.ts

@@ -1,39 +1,9 @@
 import { Context, Schema } from 'koishi';
-export declare const name = "bilirice";
-interface BiliriceConfig {
-    anchors: [string, string][];
-    pollInterval?: number;
-    templates?: {
-        liveStart?: string;
-        liveEnd?: string;
-    };
-    timeout?: number;
-    bilibiliCookie?: string;
+export declare const name = "koishi-plugin-bilirice";
+export interface Config {
+    roomId: number;
+    channelId: string;
+    pollInterval: number;
 }
-export declare const Config: Schema<Schemastery.ObjectS<{
-    anchors: Schema<[string?, string?, ...any[]][], [string?, string?, ...any[]][]>;
-    pollInterval: Schema<number, number>;
-    timeout: Schema<number, number>;
-    bilibiliCookie: Schema<string, string>;
-    templates: Schema<Schemastery.ObjectS<{
-        liveStart: Schema<string, string>;
-        liveEnd: Schema<string, string>;
-    }>, Schemastery.ObjectT<{
-        liveStart: Schema<string, string>;
-        liveEnd: Schema<string, string>;
-    }>>;
-}>, Schemastery.ObjectT<{
-    anchors: Schema<[string?, string?, ...any[]][], [string?, string?, ...any[]][]>;
-    pollInterval: Schema<number, number>;
-    timeout: Schema<number, number>;
-    bilibiliCookie: Schema<string, string>;
-    templates: Schema<Schemastery.ObjectS<{
-        liveStart: Schema<string, string>;
-        liveEnd: Schema<string, string>;
-    }>, Schemastery.ObjectT<{
-        liveStart: Schema<string, string>;
-        liveEnd: Schema<string, string>;
-    }>>;
-}>>;
-export declare function apply(ctx: Context, config: BiliriceConfig): void;
-export {};
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -36,267 +36,181 @@ __export(src_exports, {
 });
 module.exports = __toCommonJS(src_exports);
 var import_koishi = require("koishi");
-var import_axios = __toESM(require("axios"));
-var import_https = __toESM(require("https"));
-var name = "bilirice";
-var Config = import_koishi.Schema.object({
-  anchors: import_koishi.Schema.array(import_koishi.Schema.tuple([
-    import_koishi.Schema.string().description("B站主播UID（数字，如老番茄14663353）"),
-    import_koishi.Schema.string().description("通知群号（OneBot格式加group_前缀，如group_123456789）")
-  ])).required().description("监听的主播列表 → 点击「添加项」可配置多个主播").role("table"),
-  pollInterval: import_koishi.Schema.number().default(10).description("轮询间隔（秒），建议≥5秒（避免触发B站API风控）").role("slider", { min: 5, max: 30, step: 1 }),
-  // 控制台显示滑块，更易用
-  timeout: import_koishi.Schema.number().default(1e4).description("API请求超时时间（毫秒），建议5000-15000").role("number", { min: 5e3, max: 3e4 }),
-  // 核心：Cookie字段适配控制台（密码类型，输入隐藏 + 详细说明）
-  bilibiliCookie: import_koishi.Schema.string().description(`【必填】B站登录Cookie（解决-352错误）
-    获取方式：
-    1. 登录B站直播页 https://live.bilibili.com/
-    2. F12 → 网络 → 搜索 getInfoByRoom
-    3. 复制请求头中的「Cookie」完整值
-    （包含SESSDATA/bili_jct/DedeUserID等核心字段）`).required().role("password"),
-  // 控制台显示为密码输入框，输入内容隐藏
-  // 通知模板：控制台支持多行文本编辑
-  templates: import_koishi.Schema.object({
-    liveStart: import_koishi.Schema.string().default("【{uname} 开播啦】\n标题：{title}\n分区：{area}\n链接：{url}\n开播时间：{startTime}\n封面：{cover}").description("开播通知模板，支持占位符：{uname}/{title}/{area}/{url}/{startTime}/{cover}").role("textarea"),
-    // 控制台显示多行文本框
-    liveEnd: import_koishi.Schema.string().default("【{uname} 下播啦】\n直播时长：{liveTime}\n最高在线：{peakOnline}\n观看人数：{watchCount}\n弹幕数：{dmCount}\n人均弹幕：{avgDmPerUser}\n下播时间：{endTime}").description("下播通知模板，额外支持：{liveTime}/{peakOnline}/{watchCount}/{dmCount}/{avgDmPerUser}/{endTime}").role("textarea")
-  }).description("自定义通知模板（保留占位符即可自动替换数据）")
-});
-function apply(ctx, config) {
-  const finalConfig = {
-    pollInterval: config.pollInterval || 10,
-    timeout: config.timeout || 1e4,
-    templates: {
-      liveStart: config.templates?.liveStart || "【{uname} 开播啦】\n标题：{title}\n分区：{area}\n链接：{url}\n开播时间：{startTime}\n封面：{cover}",
-      liveEnd: config.templates?.liveEnd || "【{uname} 下播啦】\n直播时长：{liveTime}\n最高在线：{peakOnline}\n观看人数：{watchCount}\n弹幕数：{dmCount}\n人均弹幕：{avgDmPerUser}\n下播时间：{endTime}"
-    },
-    anchors: config.anchors || [],
-    bilibiliCookie: config.bilibiliCookie || ""
-  };
-  const anchorStateCache = /* @__PURE__ */ new Map();
-  const api = import_axios.default.create({
-    baseURL: "https://api.live.bilibili.com",
-    timeout: finalConfig.timeout,
-    httpsAgent: new import_https.default.Agent({ rejectUnauthorized: false }),
-    headers: {
-      "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36 Edg/130.0.0.0",
-      "Referer": "https://live.bilibili.com/",
-      "Origin": "https://live.bilibili.com",
-      "Accept": "application/json, text/plain, */*",
-      "Accept-Language": "zh-CN,zh;q=0.9,en;q=0.8,en-US;q=0.7",
-      "Cache-Control": "no-cache",
-      "Pragma": "no-cache",
-      "Sec-Fetch-Dest": "empty",
-      "Sec-Fetch-Mode": "cors",
-      "Sec-Fetch-Site": "same-site",
-      "Cookie": finalConfig.bilibiliCookie
-      // 读取控制台配置的Cookie
-    },
-    transformResponse: [(data) => data]
-  });
-  api.interceptors.response.use(
-    (response) => {
-      try {
-        response.data = JSON.parse(response.data);
-        return response;
-      } catch (err) {
-        throw new Error(`API响应解析失败: ${err.message}`);
-      }
-    },
-    (error) => {
-      throw new Error(`API请求失败: ${error.message || error.code}`);
-    }
-  );
-  async function getRoomIdByMid(mid) {
-    const res = await api.get("/room/v1/Room/getRoomInfoOld", {
-      params: { mid: Number(mid) }
-    });
-    if (res.data.code !== 0) throw new Error(`获取直播间ID失败: ${res.data.code} ${res.data.message || ""}`);
-    return res.data.data.roomid.toString();
+
+// src/session.ts
+var LiveSession = class {
+  static {
+    __name(this, "LiveSession");
   }
-  __name(getRoomIdByMid, "getRoomIdByMid");
-  async function getRoomBaseInfo(roomId) {
-    const res = await api.get("/room/v1/Room/get_info", {
-      params: { room_id: Number(roomId) }
-    });
-    if (res.data.code !== 0) throw new Error(`获取直播间基础信息失败: ${res.data.code} ${res.data.message || ""}`);
-    return res.data.data;
+  startTime = 0;
+  danmuCount = 0;
+  maxOnline = 0;
+  guardNew = 0;
+  guardRenew = 0;
+  start() {
+    this.startTime = Date.now();
   }
-  __name(getRoomBaseInfo, "getRoomBaseInfo");
-  async function getRoomStat(roomId) {
-    const res = await api.get("/xlive/web-room/v1/index/getInfoByRoom", {
-      params: {
-        room_id: Number(roomId),
-        platform: "web",
-        version: "1.0.0",
-        ts: Date.now()
-      }
-    });
-    if (res.data.code !== 0) throw new Error(`获取直播间统计失败: ${res.data.code} ${res.data.message || ""}`);
-    const data = res.data.data;
-    if (!data || !data.room_info) throw new Error("获取直播间统计失败: 返回数据结构异常");
+  end() {
     return {
-      online: data.room_info.online || 0,
-      watch_num: data.room_info.watch_num || 0,
-      interact_num: data.room_info.interact_num || 0,
-      dm_count: data.room_info.dm_count || 0,
-      medal_count: data.room_info.medal_count || 0,
-      fans_count: data.anchor_info?.base_info?.fans_count || 0,
-      live_time: data.room_info.live_time || 0
+      durationMs: Date.now() - this.startTime,
+      danmuCount: this.danmuCount,
+      maxOnline: this.maxOnline,
+      guardNew: this.guardNew,
+      guardRenew: this.guardRenew
     };
   }
-  __name(getRoomStat, "getRoomStat");
-  async function getAnchorInfo(roomId) {
-    const res = await api.get("/live_user/v1/UserInfo/get_anchor_in_room", {
-      params: { roomid: Number(roomId) }
-    });
-    if (res.data.code !== 0) throw new Error(`获取主播信息失败: ${res.data.code} ${res.data.message || ""}`);
-    return res.data.data;
+  onDanmu() {
+    this.danmuCount++;
   }
-  __name(getAnchorInfo, "getAnchorInfo");
-  function formatTime(timestamp) {
-    return new Date(timestamp * 1e3).toLocaleString("zh-CN", {
-      year: "numeric",
-      month: "2-digit",
-      day: "2-digit",
-      hour: "2-digit",
-      minute: "2-digit",
-      second: "2-digit"
-    });
+  onOnline(count) {
+    this.maxOnline = Math.max(this.maxOnline, count);
   }
-  __name(formatTime, "formatTime");
-  function formatSeconds(seconds) {
-    const h = Math.floor(seconds / 3600);
-    const m = Math.floor(seconds % 3600 / 60);
-    const s = seconds % 60;
-    return [h, m, s].map((v) => v.toString().padStart(2, "0")).join(":");
+  onGuard(num) {
+    num > 1 ? this.guardRenew++ : this.guardNew++;
   }
-  __name(formatSeconds, "formatSeconds");
-  function calcLiveStat(startTime, endTime, lastStat, currentStat) {
-    let fansChange = 0;
-    if (lastStat && currentStat) fansChange = currentStat.fans_count - lastStat.fans_count;
-    return {
-      liveTime: endTime - startTime,
-      peakOnline: currentStat.online,
-      watchCount: currentStat.watch_num,
-      interactCount: currentStat.interact_num,
-      dmCount: currentStat.dm_count,
-      avgDmPerUser: Number((currentStat.dm_count / Math.max(currentStat.watch_num, 1)).toFixed(2)),
-      medalCount: currentStat.medal_count,
-      fansChange,
-      endTime
-    };
+  get avgDanmu() {
+    return this.maxOnline ? (this.danmuCount / this.maxOnline).toFixed(2) : "0";
+  }
+  formatDuration(ms) {
+    const h2 = Math.floor(ms / 36e5);
+    const m = Math.floor(ms % 36e5 / 6e4);
+    return `${h2}小时${m}分钟`;
+  }
+};
+
+// src/index.ts
+var import_ws = __toESM(require("ws"));
+var name = "koishi-plugin-bilirice";
+var logger = new import_koishi.Logger("bilirice");
+var Config = import_koishi.Schema.object({
+  roomId: import_koishi.Schema.number().description("Bilibili 直播间 room_id").required(),
+  channelId: import_koishi.Schema.string().description("推送到的群 / 频道 ID").required(),
+  pollInterval: import_koishi.Schema.number().default(30).description("轮询间隔（秒）")
+});
+function apply(ctx, config) {
+  const session = new LiveSession();
+  let isLiving = false;
+  let ws = null;
+  let liveStartTime = 0;
+  async function fetchJSON(url) {
+    const res = await ctx.http.get(url);
+    if (res?.code !== 0) throw new Error(res?.message || "API Error");
+    return res.data;
   }
-  __name(calcLiveStat, "calcLiveStat");
-  function renderTemplate(template, data) {
-    return template.replace(/\{(\w+)\}/g, (_, key) => data[key] ?? `{${key}}`);
+  __name(fetchJSON, "fetchJSON");
+  async function send(text, image) {
+    const content = image ? import_koishi.h.image(image) + "\n" + text : text;
+    await ctx.broadcast([config.channelId], content);
   }
-  __name(renderTemplate, "renderTemplate");
-  async function sendGroupMessage(groupId, message) {
-    const bot = ctx.bots.values().next().value;
-    if (!bot) throw new Error("未找到可用的Bot实例");
-    await bot.sendMessage(groupId, message);
+  __name(send, "send");
+  function formatTime(ts = Date.now()) {
+    return new Date(ts).toLocaleString();
   }
-  __name(sendGroupMessage, "sendGroupMessage");
-  async function initAnchorState(mid) {
-    if (anchorStateCache.has(mid)) return;
-    const roomId = await getRoomIdByMid(mid);
-    const roomInfo = await getRoomBaseInfo(roomId);
-    const isOnline = roomInfo.live_status === 1;
-    const lastStat = isOnline ? await getRoomStat(roomId) : null;
-    anchorStateCache.set(mid, {
-      mid,
-      roomId,
-      isOnline,
-      liveStartTime: isOnline ? roomInfo.live_time : 0,
-      lastStat
+  __name(formatTime, "formatTime");
+  const getRoomInfo = /* @__PURE__ */ __name(() => fetchJSON(`https://api.live.bilibili.com/room/v1/Room/get_info?room_id=${config.roomId}`), "getRoomInfo");
+  const getAnchorInfo = /* @__PURE__ */ __name(() => fetchJSON(`https://api.live.bilibili.com/live_user/v1/UserInfo/get_anchor_in_room?roomid=${config.roomId}`), "getAnchorInfo");
+  async function onLiveStart(room, anchor) {
+    session.start();
+    liveStartTime = Date.now();
+    const text = `【开播提醒】${anchor.info.uname} 开始直播啦！
+
+📝 直播间标题：${room.title}
+📂 直播分区：${room.parent_area_name} / ${room.area_name}
+🏷️ 标签：${room.tags || "无"}
+⏰ 开播时间：${formatTime(room.live_time * 1e3)}
+
+快来直播间一起玩吧~
+👉 https://live.bilibili.com/${config.roomId}`;
+    await send(text, room.user_cover);
+    connectWS();
+  }
+  __name(onLiveStart, "onLiveStart");
+  async function onLiveEnd(anchor) {
+    disconnectWS();
+    const result = session.end();
+    const text = `【下播提醒】${anchor.info.uname} 结束直播啦！
+
+📊 直播时长：${session.formatDuration(Date.now() - liveStartTime)}
+👥 最高在线：${result.maxOnline}
+💬 弹幕统计：${result.danmuCount}
+📈 人均弹幕：${session.avgDanmu}
+⚓ 舰队统计：新增 ${result.guardNew} 人，续费 ${result.guardRenew} 人
+⏰ 下播时间：${formatTime()}
+
+感谢大家的陪伴 💖`;
+    await send(text);
+  }
+  __name(onLiveEnd, "onLiveEnd");
+  function connectWS() {
+    if (ws) return;
+    ws = new import_ws.default("wss://broadcastlv.chat.bilibili.com/sub");
+    ws.on("open", () => {
+      logger.info("WebSocket connected");
+    });
+    ws.on("message", (data) => {
+      try {
+        const text = data.toString();
+        if (!text.includes("{")) return;
+        const msg = JSON.parse(text);
+        handleWSMessage(msg);
+      } catch {
+      }
+    });
+    ws.on("close", () => {
+      ws = null;
+      logger.warn("WebSocket closed");
     });
   }
-  __name(initAnchorState, "initAnchorState");
-  async function checkAnchorState(mid, groupIds) {
-    const state = anchorStateCache.get(mid);
-    if (!state) {
-      await initAnchorState(mid);
-      return;
-    }
-    const roomInfo = await getRoomBaseInfo(state.roomId);
-    const currentOnline = roomInfo.live_status === 1;
-    const anchorInfo = await getAnchorInfo(state.roomId);
-    const now = Math.floor(Date.now() / 1e3);
-    if (!state.isOnline && currentOnline) {
-      const currentStat = await getRoomStat(state.roomId);
-      state.isOnline = true;
-      state.liveStartTime = roomInfo.live_time || now;
-      state.lastStat = currentStat;
-      const templateData = {
-        uname: anchorInfo.info.uname,
-        title: roomInfo.title,
-        area: `${roomInfo.parent_area_name || ""} - ${roomInfo.area_name || ""}`,
-        url: `https://live.bilibili.com/${state.roomId}`,
-        startTime: formatTime(state.liveStartTime),
-        cover: import_koishi.segment.image(roomInfo.user_cover || "")
-      };
-      const message = renderTemplate(finalConfig.templates.liveStart, templateData);
-      for (const gid of groupIds) await sendGroupMessage(gid, message);
-      ctx.logger.info(`主播${mid}开播，已发送通知到指定群聊`);
-    } else if (state.isOnline && !currentOnline && state.liveStartTime > 0) {
-      const currentStat = await getRoomStat(state.roomId);
-      const liveStat = calcLiveStat(state.liveStartTime, now, state.lastStat, currentStat);
-      state.isOnline = false;
-      state.liveStartTime = 0;
-      const templateData = {
-        uname: anchorInfo.info.uname,
-        liveTime: formatSeconds(liveStat.liveTime),
-        peakOnline: liveStat.peakOnline.toLocaleString(),
-        watchCount: liveStat.watchCount.toLocaleString(),
-        dmCount: liveStat.dmCount.toLocaleString(),
-        avgDmPerUser: liveStat.avgDmPerUser,
-        medalCount: liveStat.medalCount.toLocaleString(),
-        fansChange: liveStat.fansChange > 0 ? `+${liveStat.fansChange}` : liveStat.fansChange.toString(),
-        endTime: formatTime(liveStat.endTime),
-        url: `https://live.bilibili.com/${state.roomId}`
-      };
-      const message = renderTemplate(finalConfig.templates.liveEnd, templateData);
-      for (const gid of groupIds) await sendGroupMessage(gid, message);
-      ctx.logger.info(`主播${mid}下播，已发送统计通知到指定群聊`);
-    } else if (state.isOnline && currentOnline) {
-      const currentStat = await getRoomStat(state.roomId);
-      state.lastStat = currentStat;
-    }
+  __name(connectWS, "connectWS");
+  function disconnectWS() {
+    ws?.close();
+    ws = null;
   }
-  __name(checkAnchorState, "checkAnchorState");
-  ctx.on("ready", async () => {
-    if (!finalConfig.anchors || finalConfig.anchors.length === 0) {
-      ctx.logger.error("bilirice插件未配置任何主播，无法启动！");
-      return;
+  __name(disconnectWS, "disconnectWS");
+  function handleWSMessage(msg) {
+    const cmd = msg?.cmd;
+    if (cmd === "DANMU_MSG") {
+      session.onDanmu();
     }
-    if (!finalConfig.bilibiliCookie) {
-      ctx.logger.error("bilirice插件未配置B站Cookie，无法解决-352错误，请补充配置！");
-      return;
+    if (cmd === "SUPER_CHAT_MESSAGE") {
+      const sc = msg.data;
+      const price = sc.price * 10;
+      const text = `【醒目留言 ${price}电池】
+${sc.user_info.uname}（${sc.uid}）：
+${sc.message}`;
+      send(text);
     }
-    for (const [mid, _] of finalConfig.anchors) {
-      await initAnchorState(mid).catch((err) => ctx.logger.error(`初始化主播${mid}失败:`, err.message));
+    if (cmd === "GUARD_BUY") {
+      const g = msg.data;
+      session.onGuard(g.num);
+      const price = g.price * g.num * 10;
+      const text = `【舰队事件】
+${g.username}（${g.uid}）
+开通 ${g.gift_name}（${price}电池）`;
+      send(text);
     }
-    ctx.setInterval(async () => {
-      for (const [mid, groupStr] of finalConfig.anchors) {
-        const groupIds = groupStr.split(",").filter((g) => g.trim());
-        await checkAnchorState(mid, groupIds).catch((err) => ctx.logger.error(`检查主播${mid}失败:`, err.message));
-      }
-    }, finalConfig.pollInterval * 1e3);
-    ctx.logger.info(`bilirice插件已启动，监听主播数：${finalConfig.anchors.length}，轮询间隔：${finalConfig.pollInterval}秒`);
-  });
-  ctx.command("bilirice <mid>", "手动检查指定主播的直播间状态").action(async ({ session }, mid) => {
-    if (!mid) return "请输入主播UID";
-    if (!finalConfig.bilibiliCookie) return "插件未配置B站Cookie，无法执行操作！";
+    if (cmd === "ROOM_REAL_TIME_MESSAGE_UPDATE") {
+      session.onOnline(msg.data.online);
+    }
+  }
+  __name(handleWSMessage, "handleWSMessage");
+  ctx.setInterval(async () => {
     try {
-      const groupIds = session.guildId ? [session.guildId] : [];
-      await initAnchorState(mid);
-      await checkAnchorState(mid, groupIds);
-      return `已检查主播${mid}的直播间状态（含完整统计数据）`;
-    } catch (err) {
-      return `检查失败：${err.message}`;
+      const room = await getRoomInfo();
+      const anchor = await getAnchorInfo();
+      const living = room.live_status === 1;
+      if (living && !isLiving) {
+        isLiving = true;
+        await onLiveStart(room, anchor);
+      }
+      if (!living && isLiving) {
+        isLiving = false;
+        await onLiveEnd(anchor);
+      }
+    } catch (e) {
+      logger.warn(String(e));
     }
-  });
+  }, config.pollInterval * 1e3);
 }
 __name(apply, "apply");
 // Annotate the CommonJS export names for ESM import in node:

package/lib/session.d.ts

@@ -0,0 +1,20 @@
+export declare class LiveSession {
+    private startTime;
+    danmuCount: number;
+    maxOnline: number;
+    guardNew: number;
+    guardRenew: number;
+    start(): void;
+    end(): {
+        durationMs: number;
+        danmuCount: number;
+        maxOnline: number;
+        guardNew: number;
+        guardRenew: number;
+    };
+    onDanmu(): void;
+    onOnline(count: number): void;
+    onGuard(num: number): void;
+    get avgDanmu(): string;
+    formatDuration(ms: number): string;
+}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-bilirice",
   "description": "这是一个bilibili直播间事件监控插件",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [
@@ -19,6 +19,10 @@
   },
   "dependencies": {
     "axios": "^1.13.4",
-    "koishi-plugin-adapter-onebot": "^6.8.0"
+    "koishi-plugin-adapter-onebot": "^6.8.0",
+    "ws": "^8.19.0"
+  },
+  "devDependencies": {
+    "@types/ws": "^8"
   }
 }