koishi-plugin-oni-wiki-qq 0.5.1 → 0.6.0

package/lib/index.d.ts

@@ -1,6 +1,6 @@
 import { Context, Schema } from "koishi";
 export declare const name = "oni-wiki-qq";
-export declare const usage = "\n  - 0.5.1 \u539F\u7AD9\u70B9\u91CD\u5B9A\u5411\u81F3\u7B80\u4F53\u4E2D\u6587\u7AD9\u70B9\n  - 0.5.0 \u79FB\u9664\u91CD\u5B9A\u5411\u529F\u80FD GG\u7AD9\u70B9\u5DF2\u4FEE\u590D\uFF0C\u4FDD\u9669\u8D77\u89C1\u4FDD\u7559bwiki\u66F4\u65B0\u529F\u80FD\n  - 0.4.9 \u6DFB\u52A0\u91CD\u5B9A\u5411\u529F\u80FD\n  - 0.4.8 \u91CD\u542Fbwiki\u66F4\u65B0\n  - 0.4.6 \u79FB\u9664\u6CA1\u5FC5\u8981\u7684\u529F\u80FD\n  - 0.4.5 \u68C0\u6D4B\u6559\u7A0B\u9875\u9762\n";
+export declare const usage = "\n  - 0.6.1 \u2705 \u6A21\u7CCA\u5339\u914D\u8FD4\u56DE\u6700\u591A5\u6761\u7ED3\u679C+\u5E8F\u53F7\u7B49\u5F85\u4EA4\u4E92\uFF0C\u8D85\u65F6\u65E0\u8F93\u5165\u5219\u9759\u9ED8\u7ED3\u675F\n  - 0.6.0 \u96C6\u6210pinyin-pro\u62FC\u97F3\u6A21\u7CCA\u5339\u914D\uFF0C\u7CBE\u51C6\u5339\u914D\u4F18\u5148\n  - 0.5.0 \u79FB\u9664\u91CD\u5B9A\u5411\u529F\u80FD GG\u7AD9\u70B9\u5DF2\u4FEE\u590D\uFF0C\u4FDD\u7559bwiki\u66F4\u65B0\u529F\u80FD\n  - 0.4.9 \u6DFB\u52A0\u91CD\u5B9A\u5411\u529F\u80FD\n  - 0.4.8 \u91CD\u542Fbwiki\u66F4\u65B0\n  - 0.4.6 \u79FB\u9664\u6CA1\u5FC5\u8981\u7684\u529F\u80FD\n  - 0.4.5 \u68C0\u6D4B\u6559\u7A0B\u9875\u9762\n";
 export declare const inject: string[];
 declare module "koishi" {
     interface Tables {

package/lib/index.js

@@ -29,10 +29,12 @@ __export(src_exports, {
 module.exports = __toCommonJS(src_exports);
 var import_koishi = require("koishi");
 var import_mwn = require("mwn");
+var import_pinyin_pro = require("pinyin-pro");
 var name = "oni-wiki-qq";
 var usage = `
-  - 0.5.1 原站点重定向至简体中文站点
-  - 0.5.0 移除重定向功能 GG站点已修复，保险起见保留bwiki更新功能
+  - 0.6.1 ✅ 模糊匹配返回最多5条结果+序号等待交互，超时无输入则静默结束
+  - 0.6.0 集成pinyin-pro拼音模糊匹配，精准匹配优先
+  - 0.5.0 移除重定向功能 GG站点已修复，保留bwiki更新功能
   - 0.4.9 添加重定向功能
   - 0.4.8 重启bwiki更新
   - 0.4.6 移除没必要的功能
@@ -42,10 +44,10 @@ var inject = ["database"];
 var Config = import_koishi.Schema.object({
   bot_username: import_koishi.Schema.string().description("机器人用户名"),
   bot_password: import_koishi.Schema.string().description("机器人密码"),
-  bwiki_session: import_koishi.Schema.string().description("bwiki的session,无法连接到gg时使用")
+  bwiki_session: import_koishi.Schema.string().description("bwiki的session，无法连接到gg时使用")
 });
 function apply(ctx, config) {
-  const logger = ctx.logger;
+  const logger = ctx.logger(name);
   let wikibot;
   ctx.model.extend("wikipages", {
     id: "integer",
@@ -58,70 +60,87 @@ function apply(ctx, config) {
       username: config.bot_username,
       password: config.bot_password,
       userAgent: "Charles`Bot/2.1",
-      defaultParams: {
-        assert: "user"
-      }
-    });
-    wikibot.login().then(() => {
-      logger.info("Wiki机器人登录成功");
-    }).catch((err) => {
-      logger.error("Wiki机器人登录失败", err);
+      defaultParams: { assert: "user" }
     });
+    wikibot.login().then(() => logger.info("Wiki机器人登录成功")).catch((err) => logger.error("Wiki机器人登录失败", err));
   });
-  ctx.command("x <itemName>", "查询缺氧中文wiki").alias("/查wiki").action(async ({ session }, itemName = "电解器") => {
+  ctx.command("x <itemName>", "查询缺氧中文wiki，精准匹配+拼音模糊匹配+序号选择").alias("/查wiki").action(async ({ session }, itemName = "电解器") => {
     if (/教程/.test(itemName)) {
       return `请点击链接前往站点查看:
-原站点:  http://oni.wiki/${encodeURI(
-        `教程`
-      )}?variant=zh
+原站点:  http://oni.wiki/${encodeURI(`教程`)}?variant=zh
 镜像站:  http://klei.vip/oni/usiz6d/${encodeURI(`教程`)}`;
     }
-    const res = await ctx.database.get("wikipages", {
-      $or: [
-        {
-          title: itemName
-        },
-        {
-          redirect: itemName
-        }
-      ]
+    const queryKey = itemName.trim();
+    const preciseRes = await ctx.database.get("wikipages", {
+      $or: [{ title: queryKey }, { redirect: queryKey }]
     });
-    if (res.length == 0) {
-      return `在Wiki里没找到,如有需要,请按照游戏内名称重新发起查询....`;
-    }
-    const pageName = res[0].title;
-    return `请点击链接前往站点查看:
-原站点:  http://oni.wiki/${encodeURI(
-      pageName
-    )}?variant=zh
+    if (preciseRes.length > 0) {
+      const pageName = preciseRes[0].title;
+      return `✅ 精准匹配成功
+原站点:  http://oni.wiki/${encodeURI(pageName)}?variant=zh
 镜像站:  http://klei.vip/oni/usiz6d/${encodeURI(pageName)}`;
+    }
+    const allPages = await ctx.database.get("wikipages", {});
+    if (allPages.length === 0) {
+      return `❌ 本地缓存为空，请联系管理员执行【update】指令更新缓存！`;
+    }
+    const userPinyin = (0, import_pinyin_pro.pinyin)(queryKey, { toneType: "none", type: "string", separator: "" });
+    const userFirstLetter = (0, import_pinyin_pro.pinyin)(queryKey, { type: "string", separator: "" }).toLowerCase();
+    const matchResult = [];
+    allPages.forEach((page) => {
+      const targetTitle = page.title || "";
+      if (!targetTitle) return;
+      const titlePinyin = (0, import_pinyin_pro.pinyin)(targetTitle, { toneType: "none", type: "string", separator: "" });
+      const titleFirstLetter = (0, import_pinyin_pro.pinyin)(targetTitle, { type: "string", separator: "" }).toLowerCase();
+      let score = 0;
+      if (titlePinyin.includes(userPinyin) || userPinyin.includes(titlePinyin)) score += 5;
+      if (targetTitle.includes(queryKey)) score += 4;
+      if (titleFirstLetter.includes(userFirstLetter) || userFirstLetter.includes(titleFirstLetter)) score += 3;
+      if (score > 0) matchResult.push({ title: targetTitle, score });
+    });
+    if (matchResult.length === 0) {
+      return `❌ 未找到【${queryKey}】相关内容，请按游戏内标准名称重新查询！`;
+    }
+    const sortedResult = matchResult.sort((a, b) => b.score - a.score);
+    const uniqueResult = Array.from(new Map(sortedResult.map((item) => [item.title, item])).values()).slice(0, 5);
+    const resultCount = uniqueResult.length;
+    let replyMsg = `🔍 未找到精准匹配，为你找到【${resultCount}】个相似结果，请输入序号选择（10秒内有效）：
+`;
+    uniqueResult.forEach((item, index) => {
+      replyMsg += `${index + 1}. ${item.title}
+`;
+    });
+    replyMsg += `
+❗️ 提示：超时将静默结束，无任何回应，没有有待选结果请艾特机器人任意内容结束本轮查询或等待超时后重新发起查询`;
+    await session.send(replyMsg);
+    const userInput = await session.prompt(1e4);
+    if (!userInput) return;
+    const selectNum = parseInt(userInput.trim());
+    if (isNaN(selectNum) || selectNum < 1 || selectNum > resultCount) {
+      return `❌ 输入无效！请输入 1-${resultCount} 之间的数字序号`;
+    }
+    const targetPage = uniqueResult[selectNum - 1].title;
+    return `✅ 选择成功
+原站点:  http://oni.wiki/${encodeURI(targetPage)}?variant=zh
+镜像站:  http://klei.vip/oni/usiz6d/${encodeURI(targetPage)}`;
   });
   ctx.command("update", "更新本地页面缓存", { authority: 2 }).action(async ({ session }) => {
-    wikibot.request({
-      action: "query",
-      list: "allpages",
-      format: "json",
-      aplimit: "max"
-    }).then((res) => {
+    wikibot.request({ action: "query", list: "allpages", format: "json", aplimit: "max" }).then((res) => {
       logger.info("查询成功");
       const pages = res.query.allpages;
       pages.forEach((page) => {
-        ctx.database.upsert("wikipages", () => [
-          { id: page.pageid, title: page.title }
-        ]);
+        ctx.database.upsert("wikipages", () => [{ id: page.pageid, title: page.title }]);
       });
       session.send(`检索到 ${pages.length} 个页面，已尝试更新至数据库`);
       logger.info(`检索到 ${pages.length} 个页面，已尝试更新至数据库`);
-    }).catch((err) => {
-      logger.error("查询失败", err);
-    });
+    }).catch((err) => logger.error("查询失败", err));
   });
-  ctx.command("update.delete", " 删除本地页面缓存", { authority: 4 }).action(async ({ session }) => {
+  ctx.command("update.delete", "删除本地页面缓存", { authority: 4 }).action(async ({ session }) => {
     const count = await ctx.database.remove("wikipages", {});
     session.send(`已删除 ${count.removed} 条本地缓存`);
     logger.info(`已删除 ${count.removed} 条本地缓存`);
   });
-  ctx.command("update.bwiki", "使用bwiki的session更新本地页面缓存", { authority: 2 }).action(async ({ session }) => {
+  ctx.command("update.bwiki", "使用bwiki的session更新缓存", { authority: 2 }).action(async ({ session }) => {
     const headers = {
       "Content-Type": "application/json",
       "user-agent": "Charles'queryBot",
@@ -130,9 +149,7 @@ function apply(ctx, config) {
     const url = `https://wiki.biligame.com/oni/api.php?action=query&list=allpages&apnamespace=0&aplimit=5000&format=json`;
     ctx.http.get(url, { headers }).then((res) => {
       res["query"]["allpages"].forEach((page) => {
-        ctx.database.upsert("wikipages", () => [
-          { id: page.pageid, title: page.title }
-        ]);
+        ctx.database.upsert("wikipages", () => [{ id: page.pageid, title: page.title }]);
       });
       session.send(`检索到 ${res["query"]["allpages"].length} 个页面，已尝试更新至数据库`);
       logger.info(`检索到 ${res["query"]["allpages"].length} 个页面，已尝试更新至数据库`);
@@ -141,7 +158,7 @@ function apply(ctx, config) {
       logger.error("更新失败", err);
     });
   });
-  ctx.command("update.status", "查询本地页面缓存数量", { authority: 1 }).action(async ({ session }) => {
+  ctx.command("update.status", "查询本地缓存数量", { authority: 1 }).action(async ({ session }) => {
     const count = await ctx.database.get("wikipages", {});
     session.send(`数据库中缓存了 ${count.length} 条页面`);
     logger.info(`数据库中缓存了 ${count.length} 条页面`);

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-oni-wiki-qq",
   "description": "缺氧wiki查询，自用",
-  "version": "0.5.1",
+  "version": "0.6.0",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [
@@ -15,9 +15,10 @@
     "plugin"
   ],
   "peerDependencies": {
-    "koishi": "^4.18.9"
+    "koishi": "^4.18.10"
   },
   "dependencies": {
-    "mwn": "^3.0.1"
+    "mwn": "^3.0.1",
+    "pinyin-pro": "^3.27.0"
   }
-}
+}