koishi-plugin-oni-wiki-qq 0.0.2

package/lib/index.d.ts

@@ -0,0 +1,12 @@
+import { Context, Schema } from "koishi";
+export declare const name = "oni-wiki-qq";
+export declare const inject: string[];
+export declare const usage = "\n  - 0.0.2 \u57FA\u672C\u529F\u80FD\u5B8C\u6210\n  - 0.0.1 \u521D\u59CB\u5316\uFF0C\u660E\u5929\u518D\u6162\u6162\u5199\u4E86\uFF08\n";
+export interface Config {
+    mdId: string;
+    api: string;
+    imgPath: string;
+    publicUrl: string;
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -0,0 +1,249 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.apply = exports.Config = exports.usage = exports.inject = exports.name = void 0;
+const koishi_1 = require("koishi");
+const fs_1 = __importDefault(require("fs"));
+const os_1 = __importDefault(require("os"));
+const lib = __importStar(require("./lib"));
+exports.name = "oni-wiki-qq";
+exports.inject = ["puppeteer"];
+exports.usage = `
+  - 0.0.2 基本功能完成
+  - 0.0.1 初始化，明天再慢慢写了（
+`;
+exports.Config = koishi_1.Schema.object({
+    mdId: koishi_1.Schema.string().description("模板ID"),
+    api: koishi_1.Schema.string()
+        .description("api地址")
+        .default("https://oxygennotincluded.fandom.com/zh/api.php"),
+    imgPath: koishi_1.Schema.string()
+        .description("图片保存路径")
+        .default(os_1.default.homedir() + "\\wikiImg\\"),
+    publicUrl: koishi_1.Schema.string().description("图片公网访问路径"),
+});
+function apply(ctx, config) {
+    const logger = ctx.logger(exports.name);
+    // 注册指令
+    ctx
+        .command("c <itemName>", "查询缺氧中文wiki")
+        .alias("/查wiki")
+        .option("update", "-u 更新本地缓存", { authority: 2 })
+        .option("delete", "-d 删除本地缓存", { authority: 2 })
+        .option("rename", "-r <newName> 重命名本地缓存", { authority: 2 })
+        .action(async ({ session, options }, itemName) => {
+        let filePath = config.imgPath +
+            itemName.replace(/\//g, "-").replace(/:/g, "-").replace(/'/g, "-") +
+            ".jpeg";
+        if (options.update) {
+            let url = "https://oxygennotincluded.fandom.com/zh/wiki/" + encodeURI(itemName);
+            await lib.screenShot(url, ctx, itemName, config);
+            return `已尝试为您更新${itemName}的缓存...}`;
+        }
+        if (options.delete) {
+            let filePath = config.imgPath + itemName + ".jpeg";
+            if (lib.checkFileExists(filePath)) {
+                fs_1.default.unlinkSync(filePath);
+                return `已尝试删除${itemName}的缓存...`;
+            }
+            else {
+                return `文件不存在...`;
+            }
+        }
+        if (options.rename) {
+            if (lib.checkFileExists(filePath)) {
+                fs_1.default.renameSync(filePath, config.imgPath + options.rename + ".jpeg");
+                return `已尝试重命名文件...`;
+            }
+            else {
+                return `文件不存在...`;
+            }
+        }
+        // 主流程
+        session.send("正在查询中，请稍后...");
+        await lib.delay(1000);
+        // 判断文件是否在本地
+        if (lib.checkFileExists(filePath)) {
+            return `文件缓存已命中，缓存时间为：${lib.getFileModifyTime(filePath)} 请前往以下网址查看: \n ${config.publicUrl +
+                encodeURI(itemName.replace(/\//g, "-").replace(/:/g, "-").replace(/'/g, "-"))}.jpeg`;
+        }
+        else {
+            // 没有缓存
+            let res = await lib.getFromFandom(config.api, ctx, itemName);
+            if (res.length === 0) {
+                // session.send()
+                return `在Wiki里没找到或API查询超时或....`;
+            }
+            else {
+                const title = [...res[0]];
+                let res_url = [...res[1]];
+                logger.info(`API返回的数据为: ${title}`);
+                if (title[0] === itemName) {
+                    return lib.screenShot(res_url[0], ctx, itemName, config);
+                }
+                else {
+                    let [one = "曼德拉草汤", two = "火龙果派", three = "大肉汤", four = "饺子", five = "萌新骨头汤",] = title;
+                    // 发送md消息
+                    await session.bot.internal.sendMessage(session.guildId, {
+                        content: "111",
+                        msg_type: 2,
+                        markdown: {
+                            custom_template_id: config.mdId,
+                            params: [
+                                {
+                                    key: `one`,
+                                    values: [one],
+                                },
+                                {
+                                    key: `two`,
+                                    values: [two],
+                                },
+                                {
+                                    key: `three`,
+                                    values: [three],
+                                },
+                                {
+                                    key: `four`,
+                                    values: [four],
+                                },
+                                {
+                                    key: `five`,
+                                    values: [five],
+                                },
+                            ],
+                        },
+                        keyboard: {
+                            content: {
+                                rows: [
+                                    {
+                                        buttons: [
+                                            {
+                                                id: "1",
+                                                render_data: {
+                                                    label: "①",
+                                                    visited_label: "①",
+                                                },
+                                                action: {
+                                                    type: 2,
+                                                    permission: {
+                                                        type: 2,
+                                                    },
+                                                    unsupport_tips: "兼容文本",
+                                                    data: "1",
+                                                    enter: true,
+                                                },
+                                            },
+                                            {
+                                                id: "2",
+                                                render_data: {
+                                                    label: "②",
+                                                    visited_label: "②",
+                                                },
+                                                action: {
+                                                    type: 2,
+                                                    permission: {
+                                                        type: 2,
+                                                    },
+                                                    unsupport_tips: "兼容文本",
+                                                    data: "2",
+                                                    enter: true,
+                                                },
+                                            },
+                                            {
+                                                id: "3",
+                                                render_data: {
+                                                    label: "③",
+                                                    visited_label: "③",
+                                                },
+                                                action: {
+                                                    type: 2,
+                                                    permission: {
+                                                        type: 2,
+                                                    },
+                                                    unsupport_tips: "兼容文本",
+                                                    data: "3",
+                                                    enter: true,
+                                                },
+                                            },
+                                            {
+                                                id: "4",
+                                                render_data: {
+                                                    label: "④",
+                                                    visited_label: "④",
+                                                },
+                                                action: {
+                                                    type: 2,
+                                                    permission: {
+                                                        type: 2,
+                                                    },
+                                                    unsupport_tips: "兼容文本",
+                                                    data: "4",
+                                                    enter: true,
+                                                },
+                                            },
+                                            {
+                                                id: "5",
+                                                render_data: {
+                                                    label: "⑤",
+                                                    visited_label: "⑤",
+                                                },
+                                                action: {
+                                                    type: 2,
+                                                    permission: {
+                                                        type: 2,
+                                                    },
+                                                    unsupport_tips: "兼容文本",
+                                                    data: "5",
+                                                    enter: true,
+                                                },
+                                            },
+                                        ],
+                                    },
+                                ],
+                            },
+                        },
+                        msg_id: session.messageId,
+                        timestamp: session.timestamp,
+                        msg_seq: Math.floor(Math.random() * 500),
+                    });
+                    const awlist = [1, 2, 3, 4, 5];
+                    const awser = +(await session.prompt(50 * 1000))
+                        ?.replace(/\s+/g, "")
+                        ?.slice(-1) || NaN;
+                    if (awlist.includes(awser)) {
+                        return lib.screenShot(res_url[awser - 1], ctx, itemName, config);
+                    }
+                    else if (Number.isNaN(awser)) {
+                        return `您输入的选项有误，已完结本轮查询。如需，如有需要，请重新发起查询.`;
+                    }
+                }
+            }
+        }
+    });
+}
+exports.apply = apply;

package/lib/lib.d.ts

@@ -0,0 +1,34 @@
+import { Context } from "koishi";
+import { Config } from ".";
+/**
+ * 检查文件是否存在
+ * @param filePath 文件路径
+ * @returns true:存在 false:不存在
+ */
+export declare function checkFileExists(filePath: string): boolean;
+/**
+ * @desc 获取文件修改时间
+ * @param filePath 文件路径
+ * @returns string 文件修改时间
+ */
+export declare function getFileModifyTime(filePath: string): string;
+/**
+ * @desc 截屏
+ * @param url 截屏地址
+ * @param ctx 上下文
+ * @param config 配置
+ * @param itemName 物品名称
+ * @returns string
+ */
+export declare function screenShot(url: string, ctx: Context, itemName: string, config: Config): Promise<string>;
+/**
+ *
+ * @param ms 延迟毫秒数
+ * @returns void
+ */
+export declare function delay(ms: number): Promise<unknown>;
+/**
+ * @desc 从fandom获取数据
+ * @returns []
+ */
+export declare function getFromFandom(url: string, ctx: Context, itemName: string): Promise<any[]>;

package/lib/lib.js

@@ -0,0 +1,126 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getFromFandom = exports.delay = exports.screenShot = exports.getFileModifyTime = exports.checkFileExists = void 0;
+const fs_1 = __importDefault(require("fs"));
+/**
+ * 检查文件是否存在
+ * @param filePath 文件路径
+ * @returns true:存在 false:不存在
+ */
+function checkFileExists(filePath) {
+    try {
+        fs_1.default.accessSync(filePath, fs_1.default.constants.F_OK);
+        return true;
+    }
+    catch (err) {
+        return false;
+    }
+}
+exports.checkFileExists = checkFileExists;
+/**
+ * @desc 获取文件修改时间
+ * @param filePath 文件路径
+ * @returns string 文件修改时间
+ */
+function getFileModifyTime(filePath) {
+    const stats = fs_1.default.statSync(filePath);
+    const fileModifiedTime = stats.mtime.getTime();
+    return new Date(fileModifiedTime).toLocaleDateString("zh-CN", {
+        year: "numeric",
+        month: "long",
+        day: "numeric",
+        hour: "numeric",
+        minute: "numeric",
+        second: "numeric",
+    });
+}
+exports.getFileModifyTime = getFileModifyTime;
+/**
+ * @desc 截屏
+ * @param url 截屏地址
+ * @param ctx 上下文
+ * @param config 配置
+ * @param itemName 物品名称
+ * @returns string
+ */
+async function screenShot(url, ctx, itemName, config) {
+    if (!url) {
+        return `你游还没有这些东西...`;
+    }
+    else {
+        const page = await ctx.puppeteer.page();
+        await page.goto(url, {
+            timeout: 0,
+        });
+        await delay(5000);
+        await page.addStyleTag({
+            // 添加详情页边框
+            content: "#mw-content-text{padding: 40px}",
+        });
+        await delay(3000);
+        const selector = await page.$("#mw-content-text");
+        await delay(2000);
+        return await selector
+            .screenshot({
+            type: "jpeg",
+            quality: 80,
+            path: `${config.imgPath}${itemName
+                .replace(/\//g, "-")
+                .replace(/:/g, "-")
+                .replace(/'/g, "-")}.jpeg`,
+        })
+            .then(async () => {
+            // console.info(`截图成功...`);
+            return `截图已保存到以下网址，请自行点击查看:\n ${config.publicUrl}${encodeURI(itemName.replace(/\//g, "-").replace(/:/g, "-").replace(/'/g, "-"))}.jpeg`;
+        })
+            .catch((err) => {
+            console.error(err);
+            return `截图失败...`;
+        })
+            .finally(async () => {
+            await page.close();
+        });
+    }
+}
+exports.screenShot = screenShot;
+/**
+ *
+ * @param ms 延迟毫秒数
+ * @returns void
+ */
+async function delay(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+exports.delay = delay;
+/**
+ * @desc 从fandom获取数据
+ * @returns []
+ */
+async function getFromFandom(url, ctx, itemName) {
+    return await ctx.http
+        .get(url, {
+        headers: {
+            "Content-Type": "application/json",
+            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36",
+        },
+        params: {
+            action: `opensearch`,
+            search: itemName,
+            limit: 5,
+            redirects: "return",
+            format: "json",
+        },
+    })
+        .then(async (res) => {
+        console.log(res);
+        return [res[1], res[3]];
+    })
+        .catch((err) => {
+        console.error(err);
+        return [];
+    });
+}
+exports.getFromFandom = getFromFandom;

package/package.json

@@ -0,0 +1,20 @@
+{
+  "name": "koishi-plugin-oni-wiki-qq",
+  "description": "缺氧Wiki查询。QQ群版。需要官方群机器人md模板和群机器人权限。自用",
+  "version": "0.0.2",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "dist"
+  ],
+  "license": "MIT",
+  "keywords": [
+    "chatbot",
+    "koishi",
+    "plugin"
+  ],
+  "peerDependencies": {
+    "koishi": "^4.17.1"
+  }
+}

package/readme.md

@@ -0,0 +1,5 @@
+# koishi-plugin-oni-wiki-qq
+
+[![npm](https://img.shields.io/npm/v/koishi-plugin-oni-wiki-qq?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-oni-wiki-qq)
+
+缺氧Wiki查询。QQ群版。需要官方群机器人md模板和群机器人权限。