koishi-plugin-vr-fever 0.0.1

package/README.md

@@ -0,0 +1,21 @@
+我就跟你实话实说了
+
+
+# koishi-plugin-vr-fever
+
+[![npm](https://img.shields.io/npm/v/koishi-plugin-vr-fever?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-vr-fever)
+
+自用改型回旋插件
+
+## weibo相关
+
+期间内同博主多条微博&点赞（按发布时间预估点赞）
+
+照搬了以下仓库，进行魔改。如果你追求更好的使用体验，请使用它们：
+
+- https://github.com/moehuhu/weibo-monitor#readme
+- https://github.com/MingxiaGuo/koishi-plugin-weibo-notify#readme
+
+## 其他
+
+tbd

package/lib/commands/weibo.d.ts

@@ -0,0 +1,7 @@
+import { Context } from "koishi";
+import type Puppeteer from "koishi-plugin-puppeteer";
+import type { Config } from "../index";
+export interface WeiboContext extends Context {
+    puppeteer: Puppeteer;
+}
+export declare const registerWeiboCommand: (ctx: WeiboContext, config: Config, pollWeibo: () => Promise<void>) => void;

package/lib/commands/weibo.js

@@ -0,0 +1,177 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.registerWeiboCommand = void 0;
+const login_1 = require("../service/login");
+// import { formatCommentsMessages } from "../service/comment/formatter";
+// import { getWeiboCommentsByWeiboID } from "../service/weibo-fetch";
+const constants_1 = require("../util/constants");
+const puppeteer_cookie_1 = require("../util/puppeteer-cookie");
+const send_msg_1 = require("../util/send-msg");
+const registerWeiboCommand = (ctx, config, pollWeibo) => {
+    ctx.command("weibo <message>").action(async (argv, message) => {
+        if (!ctx?.puppeteer) {
+            return "please install puppeteer plugin";
+        }
+        if (message === "list") {
+            const groupID = (argv.session?.channel)
+                .id;
+            const subscribes = await ctx.database
+                .select("weibo_subscribes")
+                .where({ groupID, isActive: true })
+                .execute();
+            if (!subscribes) {
+                return argv.session.sendQueued("未找到订阅");
+            }
+            const msg = ["当前订阅的UID:"];
+            Array.from(subscribes).forEach((subscribe) => {
+                msg.push(`- ${subscribe.isActive ? "💚" : "🩶"} ${subscribe.weiboUID} ${subscribe.weiboName} `);
+            });
+            await argv.session.sendQueued(msg.join("\n"));
+            return;
+        }
+        if (message === "add") {
+            let [, weiboUID] = argv.args;
+            weiboUID = weiboUID?.trim();
+            if (!weiboUID || !constants_1.REGEX.IS_WEIBO_UID.test(weiboUID)) {
+                return argv.session.sendQueued("请输入正确格式的UID");
+            }
+            const groupID = (argv.session?.channel)
+                .id;
+            if (!groupID) {
+                return argv.session.sendQueued("未找到群ID");
+            }
+            const beforeSubscribe = await ctx.database
+                .select("weibo_subscribes")
+                .where({ id: `${weiboUID}-${groupID}` })
+                .execute();
+            if (beforeSubscribe.length > 0) {
+                return argv.session.sendQueued("已订阅，无需重复订阅");
+            }
+            await ctx.database.create("weibo_subscribes", {
+                id: `${weiboUID}-${groupID}`,
+                weiboUID,
+                groupID,
+                isActive: true,
+                createdAt: new Date(),
+            });
+            // const bot = ctx.bots[`onebot:${config.adminAccount.trim()}`];
+            // return bot.sendMessage(groupID, `订阅成功: ${weiboUID}`);
+            argv.session.sendQueued(`订阅成功: ${weiboUID}`);
+            return;
+        }
+        if (message === "remove") {
+            let [, weiboUID] = argv.args;
+            weiboUID = weiboUID?.trim();
+            if (!weiboUID || !constants_1.REGEX.IS_WEIBO_UID.test(weiboUID)) {
+                return argv.session.sendQueued("请输入正确格式的UID");
+            }
+            const groupID = (argv.session?.channel)
+                .id;
+            if (!groupID) {
+                return argv.session.sendQueued("未找到群ID");
+            }
+            const beforeSubscribe = await ctx.database
+                .select("weibo_subscribes")
+                .where({ id: `${weiboUID}-${groupID}` })
+                .execute();
+            if (beforeSubscribe.length === 0) {
+                return argv.session.sendQueued("未找到订阅，无需取消订阅");
+            }
+            argv.session.sendQueued(`正在取消订阅.../nUID: ${weiboUID} ${beforeSubscribe[0].weiboName}`);
+            await ctx.database.set("weibo_subscribes", {
+                weiboUID,
+                groupID,
+            }, {
+                isActive: false,
+            });
+            // const bot = ctx.bots[`onebot:${config.adminAccount.trim()}`];
+            // return bot.sendMessage(groupID, `移除订阅成功: ${weiboUID}`);
+            argv.session.sendQueued(`移除订阅成功: ${weiboUID}`);
+            return;
+        }
+        if (message === "help") {
+            (0, send_msg_1.sendMsg)([
+                "weibo 命令帮助：",
+                "💚",
+                "help - 显示本帮助",
+                "list - 查看本群订阅列表",
+                "add <UID> - 订阅博主",
+                "remove <UID> - 取消订阅",
+                "login - 扫码登录微博",
+                "pull - 立即拉取一次微博卡片",
+                "💚",
+                "weibo list",
+                "weibo add 8376019184",
+                "weibo remove 1234567890",
+                "weibo login",
+                "weibo pull",
+            ].join("\n"), argv.session);
+        }
+        console.log(message);
+        if (message === "login") {
+            try {
+                await (0, puppeteer_cookie_1.ensurePuppeteerBrowser)(ctx);
+                await (0, login_1.getQRcode)(ctx, argv.session);
+            }
+            catch (error) {
+                (0, send_msg_1.sendMsg)((0, puppeteer_cookie_1.formatPuppeteerError)(error), argv.session);
+            }
+        }
+        // if (message === "cookie") {
+        //   const cookies = await loadCookiesFromDatabase(ctx);
+        //   if (!cookies || cookies.length === 0) {
+        //     return argv.session.sendQueued("no cookies found");
+        //   }
+        //   return argv.session.sendQueued(JSON.stringify(cookies));
+        // }
+        // if (message === "comments") {
+        //   try {
+        //     let [, weiboUID, weiboID] = argv.args;
+        //     weiboUID = weiboUID?.trim();
+        //     weiboID = weiboID?.trim();
+        //     if (
+        //       !weiboUID ||
+        //       !REGEX.IS_WEIBO_UID.test(weiboUID) ||
+        //       !weiboID ||
+        //       !/^\d+$/.test(weiboID)
+        //     ) {
+        //       return argv.session.sendQueued("请输入正确格式的UID和微博ID");
+        //     }
+        //     const result = await getWeiboCommentsByWeiboID(weiboID, weiboUID, ctx);
+        //     if (!result?.comments.length) {
+        //       return argv.session.sendQueued("未找到评论数据");
+        //     }
+        //     for (const msg of formatCommentsMessages(result)) {
+        //       await argv.session.sendQueued(msg);
+        //     }
+        //     return;
+        //   } catch (error: any) {
+        //     return argv.session.sendQueued(error.message);
+        //   }
+        // }
+        if (message === "立刻获取") {
+            try {
+                await pollWeibo();
+            }
+            catch (error) {
+                return argv.session.sendQueued((0, puppeteer_cookie_1.formatPuppeteerError)(error));
+            }
+        }
+        if (message === "check") {
+            try {
+                const loginStatus = await (0, login_1.checkLoginStatus)(ctx);
+                if (loginStatus) {
+                    return argv.session.sendQueued("微博登录状态正常");
+                }
+                else {
+                    return argv.session.sendQueued("微博登录状态异常");
+                }
+            }
+            catch (error) {
+                return argv.session.sendQueued((0, puppeteer_cookie_1.formatPuppeteerError)(error));
+            }
+        }
+        return;
+    });
+};
+exports.registerWeiboCommand = registerWeiboCommand;

package/lib/index.d.ts

@@ -0,0 +1,37 @@
+import { Context, Schema } from "koishi";
+export declare const name = "vr-fever";
+export interface WeiboCookie {
+    name: string;
+    value: string;
+    domain: string;
+    updatedAt: Date;
+}
+interface WeiboSubscribe {
+    id: string;
+    weiboUID: string;
+    groupID: string;
+    createdAt: Date;
+    isActive: boolean;
+    weiboName?: string;
+    remark?: string;
+}
+declare module "koishi" {
+    interface Tables {
+        weibo_cookies: WeiboCookie;
+        weibo_subscribes: WeiboSubscribe;
+    }
+}
+export declare const using: string[];
+export declare const inject: {
+    required: string[];
+    optional: string[];
+};
+export interface Config {
+    adminAccount: string;
+    adminGroupID: string;
+    waitMinutes: number;
+    isTextMode: boolean;
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, config: Config): Promise<void>;
+export {};

package/lib/index.js

@@ -0,0 +1,110 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Config = exports.inject = exports.using = exports.name = void 0;
+exports.apply = apply;
+const koishi_1 = require("koishi");
+const weibo_1 = require("./commands/weibo");
+const poll_1 = require("./service/poll");
+// import { ensurePuppeteerBrowser } from "./util/puppeteer-cookie";
+const timer_1 = require("./util/timer");
+const login_1 = require("./service/login");
+exports.name = "vr-fever";
+exports.using = ["puppeteer", "database", "http"];
+exports.inject = {
+    required: [...exports.using],
+    optional: ["console", "server"],
+};
+exports.Config = koishi_1.Schema.object({
+    adminAccount: koishi_1.Schema.string().description("账号(qq号)"),
+    adminGroupID: koishi_1.Schema.string().description("管理员群ID，用于微博是否掉登录"),
+    waitMinutes: koishi_1.Schema.number()
+        .default(3)
+        .min(3)
+        .description("隔多久拉取一次最新微博 (分钟)，最少3分钟"),
+    isTextMode: koishi_1.Schema.boolean()
+        .default(false)
+        .description("开启后以文本推送微博，关闭则以截图图片推送"),
+});
+async function apply(ctx, config) {
+    ctx.model.extend("weibo_cookies", {
+        name: "string",
+        value: "string",
+        domain: "string",
+        updatedAt: "timestamp",
+    }, {
+        primary: ["name", "domain"],
+    });
+    ctx.model.extend("weibo_subscribes", {
+        id: "string",
+        weiboUID: "string",
+        groupID: "string",
+        createdAt: "timestamp",
+        isActive: "boolean",
+        weiboName: "string",
+        remark: "string",
+    }, {
+        primary: ["id"],
+    });
+    /**
+     * 创建一个用于发送消息到OneBot的会话对象
+     * @param groupId - 目标群组的ID
+     * @returns 返回一个Session对象，包含发送消息的方法
+     */
+    const sendMsgOnebot = (groupId) => {
+        // 获取管理员账号并去除首尾空格
+        const account = config.adminAccount.trim();
+        // 从上下文中获取指定账号的OneBot实例
+        const bot = ctx.bots[`onebot:${account}`];
+        // 返回一个Session对象
+        return {
+            // 发送消息方法，支持队列
+            sendQueued: (content) => {
+                // 检查机器人实例是否存在
+                if (!bot) {
+                    // 如果不存在，记录警告日志并显示可用的机器人实例
+                    ctx.logger.warn(`未找到机器人实例: onebot:${account}，当前可用: ${Object.keys(ctx.bots).join(", ") || "无"}`);
+                    return Promise.resolve([]);
+                }
+                return bot.sendMessage(groupId, content);
+            },
+        };
+    };
+    let checkingLoginStatus = false;
+    const checkLoginStatusProcess = async () => {
+        if (checkingLoginStatus) {
+            ctx.logger.debug("登录状态检查进行中，跳过本次定时任务");
+            return;
+        }
+        checkingLoginStatus = true;
+        try {
+            // ctx.logger.info("开始检测登录状态...");
+            if (!config.adminGroupID) {
+                ctx.logger.error("管理员群ID未设置，无法发送消息");
+                return;
+            }
+            const loginStatus = await (0, login_1.checkLoginStatus)(ctx);
+            const formatter = (status) => {
+                return `微博登录状态${status ? "正常" : "异常"}，当前时间戳: ${new Date().toLocaleString()}`;
+            };
+            if (!loginStatus) {
+                await sendMsgOnebot(config.adminGroupID).sendQueued(formatter(false));
+                ctx.logger.error(formatter(false));
+                return;
+            }
+            // await sendMsgOnebot(config.adminGroupID).sendQueued(formatter(true));
+            // ctx.logger.info(formatter(true));
+        }
+        finally {
+            checkingLoginStatus = false;
+        }
+    };
+    const pollWeibo = (0, poll_1.createPollWeibo)(ctx, config, sendMsgOnebot);
+    // 定时任务 - 抓微博
+    ctx.setInterval(checkLoginStatusProcess, (0, timer_1.getWaitMs)(30));
+    // 定时任务 - 登录状态检测
+    ctx.setInterval(pollWeibo, (0, timer_1.getWaitMs)(config.waitMinutes));
+    if (!ctx.puppeteer.browser?.connected) {
+        await ctx.puppeteer.start();
+    }
+    (0, weibo_1.registerWeiboCommand)(ctx, config, pollWeibo);
+}

package/lib/service/comment/fetch-for-timeline.d.ts

@@ -0,0 +1,4 @@
+import type { Context } from "koishi";
+import type { TimelineEntry } from "../drawer/types";
+/** 为时间线中的原创/转发微博批量拉取评论并写回 entry */
+export declare const attachCommentsToEntries: (ctx: Context, entryByUID: Map<string, TimelineEntry | null>, minutes: number) => Promise<void>;

package/lib/service/comment/fetch-for-timeline.js

@@ -0,0 +1,41 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.attachCommentsToEntries = void 0;
+const timeline_1 = require("../timeline");
+const constants_1 = require("../../util/constants");
+const weibo_fetch_1 = require("../weibo-fetch");
+const filter_1 = require("./filter");
+/** 为时间线中的原创/转发微博批量拉取评论并写回 entry */
+const attachCommentsToEntries = async (ctx, entryByUID, minutes) => {
+    const tasks = [];
+    for (const [weiboUID, entry] of entryByUID) {
+        if (!entry?.timeline)
+            continue;
+        for (const item of entry.timeline) {
+            if (item.activityType !== timeline_1.EActivityType.like) {
+                tasks.push({ weiboUID, weiboID: String(item.id) });
+            }
+        }
+    }
+    if (!tasks.length)
+        return;
+    const results = await Promise.all(tasks.map(async ({ weiboUID, weiboID }) => {
+        const result = await (0, weibo_fetch_1.getWeiboCommentsByWeiboID)(weiboID, weiboUID, ctx, {
+            count: constants_1.CONSTANTS.MAX_RENDER_COMMENTS,
+        });
+        const comments = (0, filter_1.filterCommentsWithinMinutes)(result?.comments ?? [], minutes);
+        return { weiboID, comments };
+    }));
+    const commentsByPostId = new Map(results
+        .filter((item) => item.comments.length > 0)
+        .map((item) => [item.weiboID, item.comments]));
+    for (const entry of entryByUID.values()) {
+        if (!entry?.timeline)
+            continue;
+        entry.timeline = entry.timeline.map((post) => {
+            const comments = commentsByPostId.get(String(post.id));
+            return comments?.length ? { ...post, comments } : post;
+        });
+    }
+};
+exports.attachCommentsToEntries = attachCommentsToEntries;

package/lib/service/comment/filter.d.ts

@@ -0,0 +1,3 @@
+import type { NormalizedComment } from "./types";
+/** 保留指定分钟数内的评论（含楼中楼） */
+export declare const filterCommentsWithinMinutes: (comments: NormalizedComment[], minutes: number) => NormalizedComment[];

package/lib/service/comment/filter.js

@@ -0,0 +1,25 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.filterCommentsWithinMinutes = void 0;
+const isWithinWindow = (time, cutoff) => time != null && time >= cutoff;
+const filterReplyTree = (comment, cutoff) => {
+    const replies = (comment.replies ?? [])
+        .map((reply) => filterReplyTree(reply, cutoff))
+        .filter((reply) => reply != null);
+    const inWindow = isWithinWindow(comment.createdAtTime, cutoff);
+    if (!inWindow && !replies.length)
+        return null;
+    return {
+        ...comment,
+        replies: replies.length ? replies : undefined,
+    };
+};
+/** 保留指定分钟数内的评论（含楼中楼） */
+const filterCommentsWithinMinutes = (comments, minutes) => {
+    const windowMs = minutes > 0 ? minutes * 60 * 1000 : 60000;
+    const cutoff = Date.now() - windowMs;
+    return comments
+        .map((comment) => filterReplyTree(comment, cutoff))
+        .filter((comment) => comment != null);
+};
+exports.filterCommentsWithinMinutes = filterCommentsWithinMinutes;

package/lib/service/comment/formatter.d.ts

@@ -0,0 +1,3 @@
+import type { WeiboCommentsResult } from "./types";
+/** 将评论格式化为可读文本，并按 QQ 消息长度分段 */
+export declare const formatCommentsMessages: (result: WeiboCommentsResult) => string[];

package/lib/service/comment/formatter.js

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.formatCommentsMessages = void 0;
+const MAX_CHUNK_LENGTH = 1500;
+const formatCommentLines = (comment, prefix) => {
+    const name = comment.user?.screen_name ?? "匿名";
+    const author = comment.isAuthor ? "[博主] " : "";
+    const likes = comment.likesCount && comment.likesCount > 0
+        ? ` (👍${comment.likesCount})`
+        : "";
+    const lines = [
+        `${prefix}[${comment.createdAtText}] ${author}${name}: ${comment.text}${likes}`,
+    ];
+    comment.replies?.forEach((reply) => {
+        lines.push(...formatCommentLines(reply, "  ↳ "));
+    });
+    return lines;
+};
+/** 将评论格式化为可读文本，并按 QQ 消息长度分段 */
+const formatCommentsMessages = (result) => {
+    const header = `共 ${result.total} 条评论，本次 ${result.comments.length} 条：`;
+    const lines = [
+        header,
+        ...result.comments.flatMap((comment, index) => formatCommentLines(comment, `${index + 1}. `)),
+    ];
+    const chunks = [];
+    let current = "";
+    for (const line of lines) {
+        const next = current ? `${current}\n${line}` : line;
+        if (next.length > MAX_CHUNK_LENGTH && current) {
+            chunks.push(current);
+            current = line;
+            continue;
+        }
+        current = next;
+    }
+    if (current) {
+        chunks.push(current);
+    }
+    return chunks;
+};
+exports.formatCommentsMessages = formatCommentsMessages;

package/lib/service/comment/normalizer.d.ts

@@ -0,0 +1,2 @@
+import type { NormalizedComment } from "./types";
+export declare const normalizeComments: (payload: any) => NormalizedComment[];

package/lib/service/comment/normalizer.js

@@ -0,0 +1,53 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.normalizeComments = void 0;
+const constants_1 = require("../../util/constants");
+const weibo_date_1 = require("../../util/weibo-date");
+/** 优先 text_raw；否则从 HTML text 提取纯文本与表情 alt */
+const extractCommentText = (comment) => {
+    if (comment?.text_raw)
+        return comment.text_raw;
+    const text = String(comment?.text || "");
+    return text
+        .replace(/<img[^>]*alt="([^"]*)"[^>]*>/gi, "$1")
+        .replace(/<br\s*\/?>/gi, "\n")
+        .replace(/<[^>]+>/g, "")
+        .trim();
+};
+const normalizeComment = (comment, isReply = false) => {
+    if (!comment?.id)
+        return null;
+    const createdAt = comment?.created_at || "";
+    const createdAtDate = (0, weibo_date_1.parseWeiboDateString)(createdAt);
+    const replies = !isReply && Array.isArray(comment.comments) && comment.comments.length > 0
+        ? comment.comments
+            .slice(0, constants_1.CONSTANTS.MAX_RENDER_COMMENT_REPLIES)
+            .map((item) => normalizeComment(item, true))
+            .filter((item) => item != null)
+        : undefined;
+    return {
+        id: String(comment.id),
+        rootId: String(comment.rootidstr ?? comment.rootid ?? comment.id),
+        text: extractCommentText(comment),
+        createdAt,
+        createdAtTime: createdAtDate?.getTime() ?? null,
+        createdAtText: createdAtDate ? (0, weibo_date_1.formatWeiboDate)(createdAtDate) : createdAt,
+        user: comment?.user,
+        likesCount: comment?.like_counts ?? comment?.like_count ?? 0,
+        source: comment?.source,
+        replyCount: Number(comment?.total_number) || replies?.length || 0,
+        isAuthor: Boolean(comment?.is_mblog_author),
+        replies: replies?.length ? replies : undefined,
+    };
+};
+const normalizeComments = (payload) => {
+    const list = Array.isArray(payload)
+        ? payload
+        : Array.isArray(payload?.data)
+            ? payload.data
+            : [];
+    return list
+        .map((item) => normalizeComment(item))
+        .filter((item) => item != null);
+};
+exports.normalizeComments = normalizeComments;

package/lib/service/comment/types.d.ts

@@ -0,0 +1,23 @@
+export interface NormalizedComment {
+    id: string;
+    /** 所属一级评论 ID；一级评论通常等于自身 id */
+    rootId: string;
+    text: string;
+    createdAt: string;
+    createdAtText: string;
+    createdAtTime: number | null;
+    user?: Record<string, any>;
+    likesCount?: number;
+    source?: string;
+    /** 楼中楼总数（来自 total_number） */
+    replyCount?: number;
+    /** 是否为博主回复 */
+    isAuthor?: boolean;
+    /** 楼中楼（fetch_level=0 响应内嵌的 comments 字段） */
+    replies?: NormalizedComment[];
+}
+export interface WeiboCommentsResult {
+    comments: NormalizedComment[];
+    total: number;
+    maxId: string | null;
+}

package/lib/service/comment/types.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/lib/service/drawer/html-builder.d.ts

@@ -0,0 +1,6 @@
+import type { ProfileData, ResolvedMediaPost } from "./types";
+export declare const buildTimelineHtml: (profile: ProfileData, normalizedTimeline: ResolvedMediaPost[]) => string;
+export declare const buildMultiTimelineHtml: (entries: {
+    profile: ProfileData;
+    timeline: ResolvedMediaPost[];
+}[]) => string;