yz-yuki-plugin 1.0.1-rc.0

package/lib/apps/bilibili.js

@@ -0,0 +1,402 @@
+import JSON from 'json5';
+import lodash from 'lodash';
+import { Messages, setBotTask, Bot, Redis } from 'yunzai';
+import { BiliQuery } from '../models/bilibili/bilibili.query.js';
+import { BiliTask } from '../models/bilibili/bilibili.task.js';
+import Config from '../utils/config.js';
+import { BiliGetWebData } from '../models/bilibili/bilibili.get.web.data.js';
+import { applyLoginQRCode, pollLoginQRCode, saveLoginCookie, postGateway, exitBiliLogin, checkBiliLogin, readSavedCookieItems, saveLocalBiliCk, readSyncCookie, getNewTempCk, readTempCk } from '../models/bilibili/bilibili.models.js';
+
+const message = new Messages('message');
+let biliConfigData = Config.getConfigData("config", "bilibili", "config");
+let biliPushData = Config.getConfigData("config", "bilibili", "push");
+setBotTask(async (Bot) => {
+    try {
+        newPushTask();
+        if (biliConfigData.pushTaskLog) {
+            Bot.logger.mark("yuki插件---B站动态推送定时任务");
+        }
+    }
+    catch (err) {
+        console.error('B站动态推送定时任务', err);
+    }
+}, biliConfigData.pushStatus ? biliConfigData.pushTime : "");
+async function newPushTask(e) {
+    await new BiliTask(e).runTask();
+}
+message.use(async (e) => {
+    await newPushTask(e);
+}, [/^(#|\/)(yuki|优纪)?执行(b站|B站|bili|bilibili|哔哩|哔哩哔哩)任务$/]);
+message.use(async (e) => {
+    if (!e.isMaster) {
+        e.reply("未取得bot主人身份，无权限添加B站动态订阅");
+    }
+    else {
+        const uid = e.msg.replace(/^(#|\/)(yuki|优纪)?(订阅|添加|add|ADD)(b站|B站|bili|bilibili|哔哩|哔哩哔哩)推送\s*(视频\s*|图文\s*|文章\s*|转发\s*|直播\s*)*/g, "").trim();
+        if (!uid) {
+            e.reply(`请在指令末尾指定订阅的B站up主的UID！`);
+            return true;
+        }
+        let subData = biliPushData || { group: {}, private: {} };
+        let chatType = e.isGroup ? "group" : "private";
+        let chatId = e.isGroup ? e.group_id : e.user_id;
+        if (!subData[chatType][chatId]) {
+            subData[chatType][chatId] = [];
+        }
+        const upData = subData[chatType][chatId].find((item) => item.uid === uid);
+        if (upData) {
+            upData.type = BiliQuery.typeHandle(upData, e.msg, "add");
+            biliPushData = subData;
+            Config.saveConfig("config", "bilibili", "push", subData);
+            e.reply(`修改b站推送动态类型成功~\n${upData.name}：${uid}`);
+            return;
+        }
+        const res = await new BiliGetWebData(e).getBiliDynamicListDataByUid(uid);
+        if (res.statusText !== "OK") {
+            e.reply("出了点网络问题，等会再试试吧~");
+            return false;
+        }
+        const resData = res.data || {};
+        if (resData.code !== 0) {
+            e.reply(`订阅校验失败~\nup主uid：${uid} 可能是无效的，或遭遇了风控，请稍后再试~`);
+            logger.mark(`yuki-plugin addDynamicSub Failed：${JSON.stringify(resData)}`);
+            return true;
+        }
+        const { items } = resData.data || {};
+        let name = items.length > 0 ? (items[0].modules.module_author?.name || uid) : uid;
+        subData[chatType][chatId].push({
+            bot_id: e.self_id,
+            uid,
+            name: name,
+            type: BiliQuery.typeHandle({ uid, name }, e.msg, "add"),
+        });
+        biliPushData = subData;
+        Config.saveConfig("config", "bilibili", "push", subData);
+        e.reply(`添加b站推送成功~\n${name}：${uid}`);
+    }
+}, [/^(#|\/)(yuki|优纪)?(订阅|添加|add|ADD)(b站|B站|bili|bilibili|哔哩|哔哩哔哩)推送\s*(视频\s*|图文\s*|文章\s*|转发\s*|直播\s*)*.*$/]);
+message.use(async (e) => {
+    if (!e.isMaster) {
+        e.reply("未取得bot主人身份，无权限删除B站动态订阅");
+    }
+    else {
+        const uid = e.msg.replace(/^(#|\/)(yuki|优纪)?(取消|删除|del|DEL)(b站|B站|bili|bilibili|哔哩|哔哩哔哩)推送\s*(视频\s*|图文\s*|文章\s*|转发\s*|直播\s*)*/g, "").trim();
+        if (!uid) {
+            e.reply(`请在指令末尾指定订阅的B站up主的UID！`);
+            return;
+        }
+        let subData = Config.getConfigData("config", "bilibili", "push") || { group: {}, private: {} };
+        let chatType = e.isGroup ? "group" : "private";
+        let chatId = e.isGroup ? e.group_id : e.user_id;
+        if (!subData[chatType][chatId]) {
+            subData[chatType][chatId] = [];
+        }
+        const upData = subData[chatType][chatId].find((item) => item.uid == uid);
+        if (!upData) {
+            e.reply(`订阅列表中没有找到该UID~\n${uid}可能是无效的`);
+            return;
+        }
+        const newType = BiliQuery.typeHandle(upData, e.msg, "del");
+        let isDel = false;
+        if (newType.length) {
+            subData[chatType][chatId] = subData[chatType][chatId].map((item) => {
+                if (item.uid == uid) {
+                    item.type = newType;
+                }
+                return item;
+            });
+        }
+        else {
+            isDel = true;
+            subData[chatType][chatId] = subData[chatType][chatId].filter((item) => item.uid !== uid);
+        }
+        Config.saveConfig("config", "bilibili", "push", subData);
+        e.reply(`${isDel ? "删除" : "修改"}b站推送成功~\n${uid}`);
+    }
+}, [/^(#|\/)(yuki|优纪)?(取消|删除|del|DEL)(b站|B站|bili|bilibili|哔哩|哔哩哔哩)推送\s*(视频\s*|图文\s*|文章\s*|转发\s*|直播\s*)*.*$/]);
+message.use(async (e) => {
+    if (e.isMaster) {
+        try {
+            const token = await applyLoginQRCode(e);
+            let biliLoginCk = await pollLoginQRCode(e, token);
+            if (lodash.trim(biliLoginCk).length != 0) {
+                await saveLoginCookie(e, biliLoginCk);
+                e.reply(`get bilibili LoginCk：成功！`);
+                const result = await postGateway(biliLoginCk);
+                const { code, data } = await result.data;
+                switch (code) {
+                    case 0:
+                        (logger ?? Bot.logger)?.mark(`优纪插件：获取biliLoginCK，Gateway校验成功：${JSON.stringify(data)}`);
+                        break;
+                    default:
+                        (logger ?? Bot.logger)?.mark(`优纪插件：获取biliLoginCK，Gateway校验失败：${JSON.stringify(data)}`);
+                        break;
+                }
+            }
+            else {
+                e.reply(`get bilibili LoginCk：失败X﹏X`);
+            }
+        }
+        catch (Error) {
+            (logger ?? Bot.logger)?.info(`yuki-plugin Login bilibili Failed：${Error}`);
+        }
+    }
+    else {
+        e.reply("未取得bot主人身份，无权限配置B站登录ck");
+    }
+}, [/^(#|\/)(yuki|优纪)?(扫码|添加|ADD|add)(b站|B站|bili|bilibili|哔哩|哔哩哔哩)登录$/]);
+message.use(async (e) => {
+    if (e.isMaster) {
+        await exitBiliLogin(e);
+        await Redis.set("Yz:yuki:bili:loginCookie", "", { EX: 3600 * 24 * 180 });
+        e.reply(`登陆的B站ck并已删除~`);
+    }
+    else {
+        e.reply("未取得bot主人身份，无权限删除B站登录ck");
+    }
+}, [/^(#|\/)(yuki|优纪)?(取消|删除|del|DEL)(b站|B站|bili|bilibili|哔哩|哔哩哔哩)登录$/]);
+message.use(async (e) => {
+    if (e.isMaster) {
+        await checkBiliLogin(e);
+    }
+    else {
+        e.reply("未取得bot主人身份，无权限查看B站登录状态");
+    }
+}, [/^(#|\/)(yuki|优纪)?我的(b站|B站|bili|bilibili|哔哩|哔哩哔哩)登录$/]);
+message.use(async (e) => {
+    if (e.isMaster) {
+        if (e.isPrivate) {
+            await e.reply('请注意账号安全，请手动撤回发送的cookie，并私聊进行添加绑定！');
+        }
+        else {
+            let localBiliCookie = e.msg.replace(/^(#|\/)(yuki|优纪)?(绑定|添加|ADD|add)(b站|B站|bili|bilibili|哔哩|哔哩哔哩)(ck|CK|cookie|COOKIE)(:|：)?/g, "").trim();
+            let param = {};
+            localBiliCookie.split(';').forEach((v) => {
+                let tmp = lodash.trim(v).replace('=', '$').split('$');
+                param[tmp[0]] = tmp[1];
+            });
+            if (!param.buvid3 || !param._uuid || !param.buvid4 || !param.DedeUserID) {
+                await e.reply('发送的cookie字段缺失\n请添加完整cookie\n获取方法查看仓库主页。');
+                const missingCookies = [];
+                if (!param.buvid3 || param.buvid3.length === 0) {
+                    missingCookies.push('buvid3');
+                }
+                if (!param.buvid4 || param.buvid4.length === 0) {
+                    missingCookies.push('buvid4');
+                }
+                if (!param._uuid || param._uuid.length === 0) {
+                    missingCookies.push('_uuid');
+                }
+                if (!param.DedeUserID || param.DedeUserID.length === 0) {
+                    missingCookies.push('DedeUserID');
+                }
+                if (missingCookies.length > 0) {
+                    await e.reply(`当前缺失字段：\n${missingCookies.join('\n')}`);
+                }
+                return;
+            }
+            localBiliCookie = await readSavedCookieItems(localBiliCookie, ['buvid3', 'buvid4', '_uuid', 'SESSDATA', 'DedeUserID', 'DedeUserID__ckMd5', 'bili_jct', 'b_nut', 'b_lsid'], false);
+            await saveLocalBiliCk(localBiliCookie);
+            logger.mark(`${e.logFnc} 保存B站cookie成功 [UID:${param.DedeUserID}]`);
+            let uidMsg = [`好耶~绑定B站cookie成功：\n${param.DedeUserID}`];
+            await e.reply(uidMsg);
+            const result = await postGateway(localBiliCookie);
+            const { code, data } = await result.data;
+            switch (code) {
+                case 0:
+                    (logger ?? Bot.logger)?.mark(`优纪插件：绑定localCK，Gateway校验成功：${JSON.stringify(data)}`);
+                    break;
+                default:
+                    (logger ?? Bot.logger)?.mark(`优纪插件：绑定localCK，Gateway校验失败：${JSON.stringify(data)}`);
+                    break;
+            }
+        }
+    }
+    else {
+        e.reply("未取得bot主人身份，无权限配置B站登录ck");
+    }
+}, [/^^(#|\/)(yuki|优纪)?(绑定|添加|ADD|add)(b站|B站|bili|bilibili|哔哩|哔哩哔哩)本地(ck|CK|cookie|COOKIE)(:|：)?.*$/]);
+message.use(async (e) => {
+    if (e.isMaster) {
+        await saveLocalBiliCk("");
+        await e.reply(`手动绑定的B站ck已删除~`);
+    }
+    else {
+        e.reply("未取得bot主人身份，无权限删除B站登录ck");
+    }
+}, [/^(#|\/)(yuki|优纪)?(取消|删除|del|DEL)(b站|B站|bili|bilibili|哔哩|哔哩哔哩)本地(ck|CK|cookie|COOKIE)$/]);
+message.use(async (e) => {
+    if (e.isGroup) {
+        await e.reply('注意账号安全，请私聊查看叭');
+    }
+    else {
+        if (e.isMaster) {
+            let { cookie, mark } = await readSyncCookie();
+            if (mark === "localCk") {
+                e.reply(`当前使用本地获取的B站cookie：`);
+                e.reply(`${cookie}`);
+            }
+            else if (mark === "loginCk") {
+                e.reply(`当前使用扫码登录的B站cookie：`);
+                e.reply(`${cookie}`);
+            }
+            else if (mark === "tempCk") {
+                e.reply(`当前使用自动获取的临时B站cookie：`);
+                e.reply(`${cookie}`);
+            }
+            else if (mark == 'ckIsEmpty') {
+                e.reply(`当前无可使用的B站cookie。`);
+            }
+        }
+        else {
+            e.reply("未取得bot主人身份，无权限查看当前使用的B站cookie");
+        }
+    }
+}, [/^(#|\/)(yuki|优纪)?(取消|删除|del|DEL)(b站|B站|bili|bilibili|哔哩|哔哩哔哩)本地(ck|CK|cookie|COOKIE)$/]);
+message.use(async (e) => {
+    try {
+        await getNewTempCk();
+        let newTempCk = await readTempCk();
+        if ((newTempCk !== null) && (newTempCk !== undefined) && (newTempCk.length !== 0) && (newTempCk !== '')) {
+            e.reply(`~yuki-plugin:\n临时b站ck刷新成功~❤~\n接下来如果获取动态失败，请重启bot(手动或发送指令 #重启)刷新状态~\n如果重启续仍不可用，请考虑 #优纪添加b站登录 吧~`);
+        }
+        else {
+            e.reply(`~yuki-plugin:\n临时b站ck刷新失败X﹏X\n请重启bot(手动或发送指令 #重启)后重试`);
+        }
+    }
+    catch (error) {
+        e.reply(`~yuki-plugin:\n临时b站ck刷新失败X﹏X\n请重启bot(手动或发送指令 #重启)后重试`);
+        (logger ?? Bot.logger)?.mark(`优纪插件：B站临时ck刷新error：${error}`);
+    }
+}, [/^(#|\/)(yuki|优纪)?刷新(b站|B站|bili|bilibili|哔哩|哔哩哔哩)临时(ck|CK|cookie|COOKIE)$/]);
+message.use(async (e) => {
+    if (!e.isMaster) {
+        e.reply("未取得bot主人身份，无权限查看Bot的全部B站订阅列表");
+    }
+    else {
+        let subData = Config.getConfigData("config", "bilibili", "push") || { group: {}, private: {} };
+        const messages = [];
+        const typeMap = {
+            DYNAMIC_TYPE_AV: "视频",
+            DYNAMIC_TYPE_WORD: "图文",
+            DYNAMIC_TYPE_DRAW: "图文",
+            DYNAMIC_TYPE_ARTICLE: "文章",
+            DYNAMIC_TYPE_FORWARD: "转发",
+            DYNAMIC_TYPE_LIVE_RCMD: "直播",
+        };
+        if (subData.group && Object.keys(subData.group).length > 0) {
+            messages.push("------群组B站订阅------");
+            Object.keys(subData.group).forEach((groupId) => {
+                messages.push(`群组ID：${groupId}：`);
+                subData.group[groupId].forEach((item) => {
+                    const types = new Set();
+                    if (item.type && item.type.length) {
+                        item.type.forEach((typeItem) => {
+                            if (typeMap[typeItem]) {
+                                types.add(typeMap[typeItem]);
+                            }
+                        });
+                    }
+                    messages.push(`${item.name}：${item.uid}  ${types.size ? `[${Array.from(types).join("、")}]` : " [全部动态]"}`);
+                });
+            });
+        }
+        if (subData.private && Object.keys(subData.private).length > 0) {
+            messages.push("------私聊B站订阅------");
+            Object.keys(subData.private).forEach((userId) => {
+                messages.push(`用户ID：${userId}：`);
+                subData.private[userId].forEach((item) => {
+                    const types = new Set();
+                    if (item.type && item.type.length) {
+                        item.type.forEach((typeItem) => {
+                            if (typeMap[typeItem]) {
+                                types.add(typeMap[typeItem]);
+                            }
+                        });
+                    }
+                    messages.push(`${item.name}：${item.uid}  ${types.size ? `[${Array.from(types).join("、")}]` : " [全部动态]"}`);
+                });
+            });
+        }
+        e.reply(`推送列表如下：\n${messages.join("\n")}`);
+    }
+}, [/^(#|\/)(yuki|优纪)?(b站|B站|bili|bilibili|哔哩|哔哩哔哩)全部(推送|动态|订阅)列表$/]);
+message.use(async (e) => {
+    let subData = Config.getConfigData("config", "bilibili", "push") || { group: {}, private: {} };
+    const messages = [];
+    const typeMap = {
+        DYNAMIC_TYPE_AV: "视频",
+        DYNAMIC_TYPE_WORD: "图文",
+        DYNAMIC_TYPE_DRAW: "图文",
+        DYNAMIC_TYPE_ARTICLE: "文章",
+        DYNAMIC_TYPE_FORWARD: "转发",
+        DYNAMIC_TYPE_LIVE_RCMD: "直播",
+    };
+    let chatType = e.isGroup ? "group" : "private";
+    let chatId = e.isGroup ? e.group_id : e.user_id;
+    if (!subData[chatType][chatId]) {
+        subData[chatType][chatId] = [];
+    }
+    subData[chatType][chatId].forEach((item) => {
+        const types = new Set();
+        if (item.type && item.type.length) {
+            item.type.forEach((typeItem) => {
+                if (typeMap[typeItem]) {
+                    types.add(typeMap[typeItem]);
+                }
+            });
+        }
+        messages.push(`${item.name}：${item.uid}  ${types.size ? `[${Array.from(types).join("、")}]` : " [全部动态]"}`);
+    });
+    e.reply(`推送列表如下：\n${messages.join("\n")}`);
+}, [/^(#|\/)(yuki|优纪)?(b站|B站|bili|bilibili|哔哩|哔哩哔哩)(推送|动态|订阅)列表$/]);
+message.use(async (e) => {
+    let uid = e.msg.replace(/^(#|\/)(yuki|优纪)?(b站|B站|bili|bilibili|哔哩|哔哩哔哩)(up|UP)主/g, "").trim();
+    const res = await new BiliGetWebData(e).getBilibiUserInfoByUid(uid);
+    if (res.statusText !== 'OK') {
+        e.reply("诶嘿，出了点网络问题，等会再试试吧~");
+        return;
+    }
+    const { code, data } = res.data || {};
+    if (code === -799) {
+        e.reply("遭遇风控：请求过于频繁，请稍后再试。");
+        return;
+    }
+    else if (code === -404) {
+        e.reply("输入的uid无效。");
+        return;
+    }
+    const message = [
+        `昵称：${data?.name}`,
+        `\n性别：${data?.sex}`,
+        `\n等级：${data?.level}`,
+    ];
+    if (data.live_room) {
+        message.push(`\n***********\n---直播信息---`, `\n直播标题：${data?.live_room?.title}`, `\n直播房间：${data?.live_room?.roomid}`, `\n直播状态：${data?.live_room?.liveStatus ? "直播中" : "未开播"}`, `\n直播链接：${data?.live_room?.url}`, `\n观看人数：${data?.live_room?.watched_show?.num}人`);
+    }
+    e.reply(message);
+}, [/^(#|\/)(yuki|优纪)?(b站|B站|bili|bilibili|哔哩|哔哩哔哩)(up|UP)主.*$/]);
+message.use(async (e) => {
+    let keyword = e.msg.replace(/^(#|\/)(yuki|优纪)?搜索(b站|B站|bili|bilibili|哔哩|哔哩哔哩)(up|UP)主/g, "").trim();
+    const res = await new BiliGetWebData(e).searchBiliUserInfoByKeyword(keyword);
+    if (res.statusText !== 'OK') {
+        e.reply("诶嘿，出了点网络问题，等会再试试吧~");
+        return;
+    }
+    const { code, data } = await res.data || {};
+    if (code !== 0 || !data.result) {
+        e.reply("哦豁~没有搜索到该用户捏，请换个关键词试试吧~");
+        return;
+    }
+    const messages = [];
+    data.result.map((item, index) => {
+        if (index < 5) {
+            messages.push(`${item.uname}\nUID：${item.mid}\n粉丝数：${item.fans}${index < 4 ? "\n" : ""}`);
+        }
+        return item;
+    });
+    e.reply(messages.join("\n"));
+}, [/^(#|\/)(yuki|优纪)?搜索(b站|B站|bili|bilibili|哔哩|哔哩哔哩)(up|UP)主.*$/]);
+const YukiBli = message.ok;
+
+export { YukiBli };

package/lib/apps/help.js

@@ -0,0 +1,37 @@
+import { Messages, Segment } from 'yunzai';
+import Image from '../utils/image.js';
+import Help from '../models/help/help.js';
+
+const message = new Messages('message');
+message.use(async (e) => {
+    const helpData = await Help.get();
+    const renderData = {
+        data: helpData.map((item) => ({
+            group: item.group,
+            list: item.list.map((listItem) => ({
+                icon: listItem.icon,
+                title: listItem.title,
+                desc: listItem.desc
+            }))
+        }))
+    };
+    const ScreenshotOptionsData = {
+        isSplit: false,
+        modelName: 'yukiHelp',
+    };
+    const helpImg = await Image.renderPage("help", "Help", renderData, ScreenshotOptionsData);
+    let imgRes;
+    if (helpImg !== false) {
+        const { img } = helpImg;
+        imgRes = { img };
+    }
+    else {
+        return;
+    }
+    let msg = [];
+    msg.push(Segment.image(imgRes.img[0]));
+    await e.reply(msg);
+}, [/^(#|\/)(yuki|优纪)帮助$/]);
+const YukiHelp = message.ok;
+
+export { YukiHelp };

package/lib/apps/index.js

@@ -0,0 +1,4 @@
+export { YukiBli } from './bilibili.js';
+export { YukiWeibo } from './weibo.js';
+export { YukiHelp } from './help.js';
+export { YukiVersion } from './version.js';

package/lib/apps/version.js

@@ -0,0 +1,34 @@
+import { Messages, Segment } from 'yunzai';
+import Image from '../utils/image.js';
+import VersionData from '../models/version/version.js';
+
+const message = new Messages('message');
+message.use(async (e) => {
+    const version = new VersionData;
+    const versionData = await version.getChangelogContent();
+    const renderData = {
+        data: versionData.map((item) => ({
+            version: item.version,
+            data: item.data
+        }))
+    };
+    const ScreenshotOptionsData = {
+        isSplit: false,
+        modelName: 'yukiVersion',
+    };
+    const helpImg = await Image.renderPage("version", "Version", renderData, ScreenshotOptionsData);
+    let imgRes;
+    if (helpImg !== false) {
+        const { img } = helpImg;
+        imgRes = { img };
+    }
+    else {
+        return;
+    }
+    let msg = [];
+    msg.push(Segment.image(imgRes.img[0]));
+    await e.reply(msg);
+}, [/^(#|\/)(yuki|优纪)版本$/]);
+const YukiVersion = message.ok;
+
+export { YukiVersion };